From bcfd071784b091bb810ea1c1c43eed46c0bd2b46 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Wed, 29 Mar 2023 22:29:38 -0700 Subject: [PATCH] Harrison/engine args (#2177) Co-authored-by: Alvaro Sevilla --- langchain/sql_database.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/langchain/sql_database.py b/langchain/sql_database.py index 424f789a..844084b7 100644 --- a/langchain/sql_database.py +++ b/langchain/sql_database.py @@ -69,9 +69,12 @@ class SQLDatabase: self._metadata.reflect(bind=self._engine) @classmethod - def from_uri(cls, database_uri: str, **kwargs: Any) -> SQLDatabase: + def from_uri( + cls, database_uri: str, engine_args: Optional[dict] = None, **kwargs: Any + ) -> SQLDatabase: """Construct a SQLAlchemy engine from URI.""" - return cls(create_engine(database_uri), **kwargs) + _engine_args = engine_args or {} + return cls(create_engine(database_uri, **_engine_args), **kwargs) @property def dialect(self) -> str: