From f00420ecbe4d330a937bab813b8e76527bed232a Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Thu, 27 Apr 2023 12:25:13 -0300 Subject: [PATCH] feat(custom.py): pass llm parameter to SQLDatabaseToolkit constructor --- 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 27159d22a..5942e3811 100644 --- a/src/backend/langflow/interface/agents/custom.py +++ b/src/backend/langflow/interface/agents/custom.py @@ -174,7 +174,7 @@ class SQLAgent(AgentExecutor): def from_toolkit_and_llm(cls, llm: BaseLLM, database_uri: str, **kwargs: Any): """Construct a sql agent from an LLM and tools.""" db = SQLDatabase.from_uri(database_uri) - toolkit = SQLDatabaseToolkit(db=db) + toolkit = SQLDatabaseToolkit(db=db, llm=llm) # The right code should be this, but there is a problem with tools = toolkit.get_tools() # related to `OPENAI_API_KEY`