diff --git a/libs/community/langchain_community/utilities/sql_database.py b/libs/community/langchain_community/utilities/sql_database.py index ace89c1d035..fa57ea60ff5 100644 --- a/libs/community/langchain_community/utilities/sql_database.py +++ b/libs/community/langchain_community/utilities/sql_database.py @@ -14,7 +14,7 @@ from sqlalchemy import ( select, text, ) -from sqlalchemy.engine import Engine, Result +from sqlalchemy.engine import URL, Engine, Result from sqlalchemy.exc import ProgrammingError, SQLAlchemyError from sqlalchemy.schema import CreateTable from sqlalchemy.sql.expression import Executable @@ -128,7 +128,10 @@ class SQLDatabase: @classmethod def from_uri( - cls, database_uri: str, engine_args: Optional[dict] = None, **kwargs: Any + cls, + database_uri: Union[str, URL], + engine_args: Optional[dict] = None, + **kwargs: Any, ) -> SQLDatabase: """Construct a SQLAlchemy engine from URI.""" _engine_args = engine_args or {}