Merge pull request #305 from ag2ai/neo4j_fix_llm

Initilized Property Graph Index with llm parameter
ag2ai · Dec 28, 2024 · 721ee7a · 721ee7a
2 parents 57f229d + 199d0ed
commit 721ee7a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/autogen/agentchat/contrib/graph_rag/neo4j_graph_query_engine.py b/autogen/agentchat/contrib/graph_rag/neo4j_graph_query_engine.py
@@ -107,6 +107,7 @@ def init_db(self, input_doc: list[Document] | None = None):
 
         self.index = PropertyGraphIndex.from_documents(
             self.documents,
+            llm=self.llm,
             embed_model=self.embedding,
             kg_extractors=self.kg_extractors,
             property_graph_store=self.graph_store,
@@ -129,6 +130,7 @@ def connect_db(self):
         self.index = PropertyGraphIndex.from_existing(
             property_graph_store=self.graph_store,
             kg_extractors=self.kg_extractors,
+            llm=self.llm,
             embed_model=self.embedding,
             show_progress=True,
         )