Removing new fields from SQL query checker

langchain-ai · Jan 19, 2025 · a1e811a · a1e811a
1 parent c28e1b1
commit a1e811a
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 11 deletions.
diff --git a/libs/community/file::memory:?cache=shared b/libs/community/file::memory:?cache=shared
diff --git a/libs/community/langchain_community/tools/sql_database/prompt.py b/libs/community/langchain_community/tools/sql_database/prompt.py
@@ -1,10 +1,6 @@
 # flake8: noqa
 QUERY_CHECKER = """
-Table Schema:
-{table_info}
-
-Foreign Key Relationships:
-{foreign_keys}
+{query}
 
 Double check the {dialect} query above for common mistakes, including:
 - Using NOT IN with NULL values

diff --git a/libs/community/langchain_community/tools/sql_database/tool.py b/libs/community/langchain_community/tools/sql_database/tool.py
@@ -151,12 +151,7 @@ def initialize_llm_chain(cls, values: Dict[str, Any]) -> Any:
                 llm=values.get("llm"),  # type: ignore[arg-type]
                 prompt=PromptTemplate(
                     template=QUERY_CHECKER,
-                    input_variables=[
-                        "dialect",
-                        "query",
-                        "table_info_str",
-                        "foreign_key_info_str",
-                    ],
+                    input_variables=["dialect", "query"],
                 ),
             )