dunno

libs/langchain/langchain/chains/graph_qa/cypher.py

@@ -37,7 +37,7 @@ def extract_cypher(text: str) -> str:
 
 
 def construct_schema(
-    structured_schema: Dict[str, Union[Any, Dict[str, Any]]],
+    structured_schema: Dict[str, Any],
     include_types: List[str],
     exclude_types: List[str],
 ) -> str:
@@ -168,9 +168,11 @@ def from_llm(
         """Initialize from LLM."""
 
         if not cypher_llm and not llm:
-            raise ValueError("Either `llm` or `cypher_llm` parameters must be provided")
+            raise ValueError(
+                "Either `llm` or `cypher_llm` parameters must be provided")
         if not qa_llm and not llm:
-            raise ValueError("Either `llm` or `qa_llm` parameters must be provided")
+            raise ValueError(
+                "Either `llm` or `qa_llm` parameters must be provided")
         if cypher_llm and qa_llm and llm:
             raise ValueError(
                 "You can specify up to two of 'cypher_llm', 'qa_llm'"
@@ -254,7 +256,8 @@ def _call(
         if self.cypher_query_corrector:
             generated_cypher = self.cypher_query_corrector(generated_cypher)
 
-        _run_manager.on_text("Generated Cypher:", end="\n", verbose=self.verbose)
+        _run_manager.on_text("Generated Cypher:",
+                             end="\n", verbose=self.verbose)
         _run_manager.on_text(
             generated_cypher, color="green", end="\n", verbose=self.verbose
         )
@@ -271,7 +274,8 @@ def _call(
         if self.return_direct:
             final_result = context
         else:
-            _run_manager.on_text("Full Context:", end="\n", verbose=self.verbose)
+            _run_manager.on_text("Full Context:", end="\n",
+                                 verbose=self.verbose)
             _run_manager.on_text(
                 str(context), color="green", end="\n", verbose=self.verbose
             )