Skip to content
Snippets Groups Projects
Unverified Commit 7acf7f0c authored by Kaan Ozdogan's avatar Kaan Ozdogan Committed by GitHub
Browse files

fixed storage context update & service context issue (#11475)

parent d3724dbd
No related branches found
No related tags found
No related merge requests found
......@@ -259,10 +259,9 @@ class KnowledgeGraphIndex(BaseIndex[KG]):
self._graph_store.upsert_triplet(*triplet)
triplet_str = str(triplet)
if include_embeddings:
set_embedding = self._service_context.embed_model.get_text_embedding(
triplet_str
)
set_embedding = self._embed_model.get_text_embedding(triplet_str)
self._index_struct.add_to_embedding_dict(str(triplet), set_embedding)
self._storage_context.index_store.add_index_struct(self._index_struct)
def add_node(self, keywords: List[str], node: BaseNode) -> None:
"""Add node.
......@@ -300,10 +299,9 @@ class KnowledgeGraphIndex(BaseIndex[KG]):
self.add_node([subj, obj], node)
triplet_str = str(triplet)
if include_embeddings:
set_embedding = self._service_context.embed_model.get_text_embedding(
triplet_str
)
set_embedding = self._embed_model.get_text_embedding(triplet_str)
self._index_struct.add_to_embedding_dict(str(triplet), set_embedding)
self._storage_context.index_store.add_index_struct(self._index_struct)
def _delete_node(self, node_id: str, **delete_kwargs: Any) -> None:
"""Delete a node."""
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment