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

Add support for supplying custom db params (#1276)

Browse Source
This commit is contained in:
Deshraj Yadav
2024-02-21 16:15:57 -08:00
committed by GitHub
parent f8f69eab03
commit aa5ad625af
12 changed files with 36 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -11,8 +11,8 @@ from .models import Base
class DatabaseManager:
def __init__(self, database_uri: str = "sqlite:///embedchain.db", echo: bool = False):
self.database_uri = database_uri
def __init__(self, echo: bool = False):
self.database_uri = os.environ.get("EMBEDCHAIN_DB_URI")
self.echo = echo
self.engine: Engine = None
self._session_factory = None
@@ -58,7 +58,7 @@ database_manager = DatabaseManager()
# Convenience functions for backward compatibility and ease of use
def setup_engine(database_uri: str = "sqlite:///embedchain.db", echo: bool = False) -> None:
def setup_engine(database_uri: str, echo: bool = False) -> None:
database_manager.database_uri = database_uri
database_manager.echo = echo
database_manager.setup_engine()