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

refactor: RAG Refactor (#985)

Browse Source 
Co-authored-by: Aralhi <xiaoping0501@gmail.com>
Co-authored-by: csunny <cfqsunny@163.com>
This commit is contained in:
Aries-ckt
2024-01-03 09:45:26 +08:00
committed by GitHub
parent 90775aad50
commit 9ad70a2961
206 changed files with 5766 additions and 2419 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
dbgpt/storage/metadata/_base_dao.py
View File

@@ -139,8 +139,9 @@ class BaseDao(Generic[T, REQ, RES]):
RES: The response schema object.
"""
entry = self.from_request(request)
with self.session() as session:
with self.session(commit=False) as session:
session.add(entry)
session.commit()
return self.get_one(self.to_request(entry))
def update(self, query_request: QUERY_SPEC, update_request: REQ) -> RES:

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

@@ -299,7 +299,8 @@ class DatabaseManager:
session_options.setdefault("class_", Session)
session_options.setdefault("query_cls", self.Query)
session_factory = sessionmaker(bind=self._engine, **session_options)
self._session = scoped_session(session_factory)
# self._session = scoped_session(session_factory)
self._session = session_factory
self._base.metadata.bind = self._engine
def init_default_db(