fix: chat

arangodb · Jan 3, 2025 · 09fa974 · 09fa974
1 parent 46e6b69
commit 09fa974
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/nx_arangodb/classes/graph.py b/nx_arangodb/classes/graph.py
@@ -621,9 +621,11 @@ def chat(
         if llm is None:
             llm = ChatOpenAI(temperature=0, model_name="gpt-4")
 
+        graph = ArangoGraph(self.db, graph_name=self.name)
+
         chain = ArangoGraphQAChain.from_llm(
             llm=llm,
-            graph=ArangoGraph(self.db),
+            graph=graph,
             verbose=verbose,
         )