diff --git a/libs/langchain/langchain/tools/sql_database/tool.py b/libs/langchain/langchain/tools/sql_database/tool.py index 99289e4a97..5dfe8f680f 100644 --- a/libs/langchain/langchain/tools/sql_database/tool.py +++ b/libs/langchain/langchain/tools/sql_database/tool.py @@ -97,11 +97,11 @@ class QuerySQLCheckerTool(BaseSQLDatabaseTool, BaseTool): values["llm_chain"] = LLMChain( llm=values.get("llm"), prompt=PromptTemplate( - template=QUERY_CHECKER, input_variables=["query", "dialect"] + template=QUERY_CHECKER, input_variables=["dialect", "query"] ), ) - if values["llm_chain"].prompt.input_variables != ["query", "dialect"]: + if values["llm_chain"].prompt.input_variables != ["dialect", "query"]: raise ValueError( "LLM chain for QueryCheckerTool must have input variables ['query', 'dialect']" )