From 73225f4d819128a5de796309018af2a38d248467 Mon Sep 17 00:00:00 2001 From: Cristhian Zanforlin Lousa Date: Fri, 1 Aug 2025 11:23:59 -0300 Subject: [PATCH] fix: Add outputs support to SQLAgentComponent (#9173) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * ✨ (sql.py): add new output definitions for the SQLAgentComponent to include 'Agent' and 'Response' outputs for better functionality and flexibility * 📝 (sql.py): update import statement for Output class to match new file structure ♻️ (sql.py): refactor Output class usage to remove unnecessary attributes and improve code readability --------- Co-authored-by: Edwin Jose --- .../base/langflow/components/langchain_utilities/sql.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/backend/base/langflow/components/langchain_utilities/sql.py b/src/backend/base/langflow/components/langchain_utilities/sql.py index 5edfd77d6..f830594cd 100644 --- a/src/backend/base/langflow/components/langchain_utilities/sql.py +++ b/src/backend/base/langflow/components/langchain_utilities/sql.py @@ -5,6 +5,7 @@ from langchain_community.utilities import SQLDatabase from langflow.base.agents.agent import LCAgentComponent from langflow.inputs.inputs import HandleInput, MessageTextInput +from langflow.template.field.base import Output class SQLAgentComponent(LCAgentComponent): @@ -25,6 +26,11 @@ class SQLAgentComponent(LCAgentComponent): ), ] + outputs = [ + Output(display_name="Response", name="response", method="message_response"), + Output(display_name="Agent", name="agent", method="build_agent", tool_mode=False), + ] + def build_agent(self) -> AgentExecutor: db = SQLDatabase.from_uri(self.database_uri) toolkit = SQLDatabaseToolkit(db=db, llm=self.llm)