From 10db240a6b42f373154ab64fafeb5e0a2e0dfc8d Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 21 Jun 2023 16:08:37 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(custom.py):=20rename=20Query?= =?UTF-8?q?CheckerTool=20to=20QuerySQLCheckerTool=20to=20match=20the=20cor?= =?UTF-8?q?rect=20class=20name=20The=20QueryCheckerTool=20class=20was=20re?= =?UTF-8?q?named=20to=20QuerySQLCheckerTool=20to=20match=20the=20correct?= =?UTF-8?q?=20class=20name.=20This=20change=20ensures=20that=20the=20corre?= =?UTF-8?q?ct=20class=20is=20being=20used=20and=20avoids=20any=20potential?= =?UTF-8?q?=20errors=20that=20may=20arise=20from=20using=20the=20wrong=20c?= =?UTF-8?q?lass=20name.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/agents/custom.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/interface/agents/custom.py b/src/backend/langflow/interface/agents/custom.py index b0c794c2d..a4191a8b7 100644 --- a/src/backend/langflow/interface/agents/custom.py +++ b/src/backend/langflow/interface/agents/custom.py @@ -192,7 +192,7 @@ class SQLAgent(CustomAgentExecutor): from langchain.tools.sql_database.tool import ( InfoSQLDatabaseTool, ListSQLDatabaseTool, - QueryCheckerTool, + QuerySQLCheckerTool, QuerySQLDataBaseTool, ) @@ -207,7 +207,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, llm=llm), # type: ignore + QuerySQLCheckerTool(db=db, llm_chain=llmchain, llm=llm), # type: ignore ] prefix = SQL_PREFIX.format(dialect=toolkit.dialect, top_k=10)