jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: SQLDatabase has no attribute set_event_manager (#4192)

Browse source
This commit is contained in:
Marcelo Nunes Alves 2024-11-03 21:20:33 -03:00 • committed by GitHub
commit dd4a9f908c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 15 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

23
src/backend/base/langflow/components/langchain_utilities/sql_database.py
View file

@ -2,26 +2,33 @@ from langchain_community.utilities.sql_database import SQLDatabase
from sqlalchemy import create_engine from sqlalchemy import create_engine
from sqlalchemy.pool import StaticPool from sqlalchemy.pool import StaticPool
from langflow.custom import CustomComponent from langflow.custom import Component
from langflow.io import (
Output,
StrInput,
)
class SQLDatabaseComponent(CustomComponent): class SQLDatabaseComponent(Component):
display_name = "SQLDatabase" display_name = "SQLDatabase"
description = "SQL Database" description = "SQL Database"
name = "SQLDatabase" name = "SQLDatabase"
def build_config(self): inputs = [
return { StrInput(name="uri", display_name="URI", info="URI to the database.", required=True),
"uri": {"display_name": "URI", "info": "URI to the database."}, ]
}
outputs = [
Output(display_name="SQLDatabase", name="SQLDatabase", method="build_sqldatabase"),
]
def clean_up_uri(self, uri: str) -> str: def clean_up_uri(self, uri: str) -> str:
if uri.startswith("postgres://"): if uri.startswith("postgres://"):
uri = uri.replace("postgres://", "postgresql://") uri = uri.replace("postgres://", "postgresql://")
return uri.strip() return uri.strip()
def build(self, uri: str) -> SQLDatabase: def build_sqldatabase(self) -> SQLDatabase:
uri = self.clean_up_uri(uri) uri = self.clean_up_uri(self.uri)
# Create an engine using SQLAlchemy with StaticPool # Create an engine using SQLAlchemy with StaticPool
engine = create_engine(uri, poolclass=StaticPool) engine = create_engine(uri, poolclass=StaticPool)
return SQLDatabase(engine) return SQLDatabase(engine)