From b4bb4cf053fc079cb81135d1696a2734433a181c Mon Sep 17 00:00:00 2001 From: Deshraj Yadav Date: Thu, 22 Feb 2024 09:52:40 -0800 Subject: [PATCH] [Bug fix] Fix issue for using any metadata db apart from sqlite (#1282) --- embedchain/core/db/database.py | 7 ++++++- pyproject.toml | 2 +- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/embedchain/core/db/database.py b/embedchain/core/db/database.py index f460c5a4..0965ca8f 100644 --- a/embedchain/core/db/database.py +++ b/embedchain/core/db/database.py @@ -19,7 +19,12 @@ class DatabaseManager: def setup_engine(self) -> None: """Initializes the database engine and session factory.""" - self.engine = create_engine(self.database_uri, echo=self.echo, connect_args={"check_same_thread": False}) + if not self.database_uri: + raise RuntimeError("Database URI is not set. Set the EMBEDCHAIN_DB_URI environment variable.") + connect_args = {} + if self.database_uri.startswith("sqlite"): + connect_args["check_same_thread"] = False + self.engine = create_engine(self.database_uri, echo=self.echo, connect_args=connect_args) self._session_factory = scoped_session(sessionmaker(bind=self.engine)) Base.metadata.bind = self.engine diff --git a/pyproject.toml b/pyproject.toml index da49da99..ffbba155 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "embedchain" -version = "0.1.84" +version = "0.1.85" description = "Simplest open source retrieval(RAG) framework" authors = [ "Taranjeet Singh ",