Sort order

11 months ago · e4ae690244
parent b753bf3323
commit e4ae690244
1 changed files with 2 additions and 2 deletions
--- 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']"
            )