klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Bug fix] Fix issue for using any metadata db apart from sqlite (#1282)

Browse Source
This commit is contained in:
Deshraj Yadav
2024-02-22 09:52:40 -08:00
committed by GitHub
parent f0400e928a
commit b4bb4cf053
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
embedchain/core/db/database.py
View File

@@ -19,7 +19,12 @@ class DatabaseManager:
def setup_engine(self) -> None: def setup_engine(self) -> None:
"""Initializes the database engine and session factory.""" """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)) self._session_factory = scoped_session(sessionmaker(bind=self.engine))
Base.metadata.bind = self.engine Base.metadata.bind = self.engine

2
pyproject.toml
View File

@@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "embedchain" name = "embedchain"
version = "0.1.84" version = "0.1.85"
description = "Simplest open source retrieval(RAG) framework" description = "Simplest open source retrieval(RAG) framework"
authors = [ authors = [
"Taranjeet Singh <taranjeet@embedchain.ai>", "Taranjeet Singh <taranjeet@embedchain.ai>",