github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-07 12:00:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(core): Support multi round conversation operator (#986)

Browse Source
This commit is contained in:
Fangyin Cheng
2023-12-27 23:26:28 +08:00
committed by GitHub
parent 9aec636b02
commit b13d3f6d92
63 changed files with 2011 additions and 314 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
dbgpt/storage/metadata/db_manager.py
View File

@@ -183,6 +183,11 @@ class DatabaseManager:
"""Get the engine.""" ""
return self._engine
@property
def is_initialized(self) -> bool:
"""Whether the database manager is initialized.""" ""
return self._engine is not None and self._session is not None
@contextmanager
def session(self) -> Session:
"""Get the session with context manager.
@@ -200,7 +205,7 @@ class DatabaseManager:
RuntimeError: The database manager is not initialized.
Exception: Any exception.
"""
if not self._session:
if not self.is_initialized:
raise RuntimeError("The database manager is not initialized.")
session = self._session()
try: