fix: Fix bug for issue #1669 (#1672)

Co-authored-by: WangTing <wangting@192.168.1.10>
Co-authored-by: aries_ckt <916701291@qq.com>
This commit is contained in:
Armandwt
2024-06-28 12:44:06 +08:00
committed by GitHub
parent 2eeef2fa75
commit cd2163e444
3 changed files with 14 additions and 6 deletions

View File

@@ -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):