From 72994712541f0631dd5c6e08080b9a57b20d78e7 Mon Sep 17 00:00:00 2001 From: Eric-Shang Date: Fri, 27 Dec 2024 23:15:29 +0000 Subject: [PATCH] Initilized Property Graph Index with llm parameter --- autogen/agentchat/contrib/graph_rag/neo4j_graph_query_engine.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/autogen/agentchat/contrib/graph_rag/neo4j_graph_query_engine.py b/autogen/agentchat/contrib/graph_rag/neo4j_graph_query_engine.py index 462371930..a7c33348b 100644 --- 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, )