From f44e2f97f7fb1c053288886e5c051c2461654fe4 Mon Sep 17 00:00:00 2001 From: Gabriel Ferreira Rosalino Date: Sat, 27 May 2023 14:12:18 -0300 Subject: [PATCH] fix: SQLAgent The queryCheckerTool now needs an extra arg with the LLM instance: https://github.com/hwchase17/langchain/blob/master/langchain/tools/sql_database/tool.py#L108 I added it so the SQLAgent can work normally now --- src/backend/langflow/interface/agents/custom.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/langflow/interface/agents/custom.py b/src/backend/langflow/interface/agents/custom.py index 0cbf7baca..355856c74 100644 --- a/src/backend/langflow/interface/agents/custom.py +++ b/src/backend/langflow/interface/agents/custom.py @@ -220,7 +220,7 @@ class SQLAgent(CustomAgentExecutor): QuerySQLDataBaseTool(db=db), # type: ignore InfoSQLDatabaseTool(db=db), # type: ignore ListSQLDatabaseTool(db=db), # type: ignore - QueryCheckerTool(db=db, llm_chain=llmchain), # type: ignore + QueryCheckerTool(db=db, llm_chain=llmchain, llm=llm), # type: ignore ] prefix = SQL_PREFIX.format(dialect=toolkit.dialect, top_k=10)