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,