From fc51b9fcc9c2bbdc09a9ac91deea7715872c3f44 Mon Sep 17 00:00:00 2001
From: Rana Banerjee <rana1224@gmail.com>
Date: Wed, 14 Feb 2024 21:09:02 +0530
Subject: [PATCH] Fix: The _indices of KnowledgeGraphIndex method does not
update storage context index (#10687)
---
.../llama_index/core/indices/knowledge_graph/base.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/llama-index-core/llama_index/core/indices/knowledge_graph/base.py b/llama-index-core/llama_index/core/indices/knowledge_graph/base.py
index 472c8d53da..8f6ba10162 100644
--- a/llama-index-core/llama_index/core/indices/knowledge_graph/base.py
+++ b/llama-index-core/llama_index/core/indices/knowledge_graph/base.py
@@ -240,6 +240,9 @@ class KnowledgeGraphIndex(BaseIndex[KG]):
rel_embedding = self._embed_model.get_text_embedding(triplet_str)
self._index_struct.add_to_embedding_dict(triplet_str, rel_embedding)
+ # Update the storage context's index_store
+ self._storage_context.index_store.add_index_struct(self._index_struct)
+
def upsert_triplet(self, triplet: Tuple[str, str, str]) -> None:
"""Insert triplets.
--
GitLab