From 11341fcecbb92ccdb8ee2edf72c83342ed59f6fd Mon Sep 17 00:00:00 2001 From: Chandan Routray Date: Wed, 17 May 2023 04:28:58 +0530 Subject: [PATCH] Fixed query checker for SQLDatabaseChain (#4780) # Fixed query checker for SQLDatabaseChain When `SQLDatabaseChain`'s llm attribute was deprecated, the query checker stopped working if `SQLDatabaseChain` is initialised via `from_llm` method. With this fix, `SQLDatabaseChain`'s query checker would use the same `llm` as used in the `llm_chain` ## Who can review? @hwchase17 - project lead Co-authored-by: imeckr --- langchain/chains/sql_database/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/langchain/chains/sql_database/base.py b/langchain/chains/sql_database/base.py index a2eb4a90..fe08b82c 100644 --- a/langchain/chains/sql_database/base.py +++ b/langchain/chains/sql_database/base.py @@ -130,7 +130,7 @@ class SQLDatabaseChain(Chain): template=QUERY_CHECKER, input_variables=["query", "dialect"] ) query_checker_chain = LLMChain( - llm=self.llm, prompt=query_checker_prompt + llm=self.llm_chain.llm, prompt=query_checker_prompt ) query_checker_inputs = { "query": sql_cmd,