From 08083ec90b418aee06c5b274c9f85ae627f1ac0f Mon Sep 17 00:00:00 2001 From: Marcelo Nunes Alves Date: Thu, 17 Oct 2024 16:40:32 -0300 Subject: [PATCH 1/2] fix: SQLDatabase has no attribute set_event_manager --- .../langchain_utilities/sql_database.py | 24 ++++++++++++------- 1 file changed, 15 insertions(+), 9 deletions(-) diff --git a/src/backend/base/langflow/components/langchain_utilities/sql_database.py b/src/backend/base/langflow/components/langchain_utilities/sql_database.py index 057bb1a7be00..18b835af3077 100644 --- a/src/backend/base/langflow/components/langchain_utilities/sql_database.py +++ b/src/backend/base/langflow/components/langchain_utilities/sql_database.py @@ -1,27 +1,33 @@ from langchain_community.utilities.sql_database import SQLDatabase from sqlalchemy import create_engine from sqlalchemy.pool import StaticPool +from langflow.io import ( + StrInput, + Output, +) +from langflow.custom import Component -from langflow.custom import CustomComponent - -class SQLDatabaseComponent(CustomComponent): +class SQLDatabaseComponent(Component): display_name = "SQLDatabase" description = "SQL Database" name = "SQLDatabase" - def build_config(self): - return { - "uri": {"display_name": "URI", "info": "URI to the database."}, - } + inputs = [ + StrInput(name="uri", display_name="URI", info="URI to the database.", required=True), + ] + + outputs = [ + Output(display_name="SQLDatabase", name="SQLDatabase", method="build_sqldatabase"), + ] def clean_up_uri(self, uri: str) -> str: if uri.startswith("postgres://"): uri = uri.replace("postgres://", "postgresql://") return uri.strip() - def build(self, uri: str) -> SQLDatabase: - uri = self.clean_up_uri(uri) + def build_sqldatabase(self) -> SQLDatabase: + uri = self.clean_up_uri(self.uri) # Create an engine using SQLAlchemy with StaticPool engine = create_engine(uri, poolclass=StaticPool) return SQLDatabase(engine) From c35d4dabde927ca52231bb86bf33d52b6274cd35 Mon Sep 17 00:00:00 2001 From: "autofix-ci[bot]" <114827586+autofix-ci[bot]@users.noreply.github.com> Date: Thu, 17 Oct 2024 19:45:59 +0000 Subject: [PATCH 2/2] [autofix.ci] apply automated fixes --- .../langflow/components/langchain_utilities/sql_database.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/components/langchain_utilities/sql_database.py b/src/backend/base/langflow/components/langchain_utilities/sql_database.py index 18b835af3077..fd86af2b7bf5 100644 --- a/src/backend/base/langflow/components/langchain_utilities/sql_database.py +++ b/src/backend/base/langflow/components/langchain_utilities/sql_database.py @@ -1,11 +1,12 @@ from langchain_community.utilities.sql_database import SQLDatabase from sqlalchemy import create_engine from sqlalchemy.pool import StaticPool + +from langflow.custom import Component from langflow.io import ( - StrInput, Output, + StrInput, ) -from langflow.custom import Component class SQLDatabaseComponent(Component):