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)