Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-08 07:10:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Sort order

Browse Source
This commit is contained in:
Nuno Campos 2023-10-18 15:42:13 +01:00
parent b753bf3323
commit e4ae690244
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/langchain/langchain/tools/sql_database/tool.py
View File

@ -97,11 +97,11 @@ class QuerySQLCheckerTool(BaseSQLDatabaseTool, BaseTool):
values["llm_chain"] = LLMChain( values["llm_chain"] = LLMChain(
llm=values.get("llm"), llm=values.get("llm"),
prompt=PromptTemplate( 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( raise ValueError(
"LLM chain for QueryCheckerTool must have input variables ['query', 'dialect']" "LLM chain for QueryCheckerTool must have input variables ['query', 'dialect']"
) )