Added missing args to get_structured_schema call

langchain-ai · Feb 25, 2025 · 13ad624 · 13ad624
1 parent f33ad44
commit 13ad624
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/libs/neo4j/langchain_neo4j/graphs/neo4j_graph.py b/libs/neo4j/langchain_neo4j/graphs/neo4j_graph.py
@@ -284,6 +284,9 @@ def refresh_schema(self) -> None:
         self.structured_schema = get_structured_schema(
             driver=self._driver,
             is_enhanced=self._enhanced_schema,
+            database=self._database,
+            timeout=self.timeout,
+            sanitize=self.sanitize,
         )
         self.schema = format_schema(
             schema=self.structured_schema, is_enhanced=self._enhanced_schema