From e4ae6902442656a0931a8a847af4e2661f20469f Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Wed, 18 Oct 2023 15:42:13 +0100 Subject: [PATCH] Sort order --- libs/langchain/langchain/tools/sql_database/tool.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/langchain/langchain/tools/sql_database/tool.py b/libs/langchain/langchain/tools/sql_database/tool.py index 99289e4a97c..5dfe8f680fd 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']" )