mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-04 10:34:30 +00:00
Co-authored-by: WangTing <wangting@192.168.1.10> Co-authored-by: aries_ckt <916701291@qq.com>
This commit is contained in:
@@ -219,6 +219,7 @@ class KnowledgeDocumentDao(BaseDao):
|
||||
)
|
||||
|
||||
results = counts_query.all()
|
||||
session.close()
|
||||
docs_count = {result.space: result.document_count for result in results}
|
||||
return docs_count
|
||||
|
||||
@@ -253,7 +254,9 @@ class KnowledgeDocumentDao(BaseDao):
|
||||
session = self.get_raw_session()
|
||||
updated_space = session.merge(document)
|
||||
session.commit()
|
||||
return updated_space.id
|
||||
update_space_id = updated_space.id
|
||||
session.close()
|
||||
return update_space_id
|
||||
|
||||
#
|
||||
def raw_delete(self, query: KnowledgeDocumentEntity):
|
||||
|
Reference in New Issue
Block a user