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`