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

refactor: Refactor storage and new serve template (#947)

Browse Source
This commit is contained in:
Fangyin Cheng
2023-12-18 19:30:40 +08:00
committed by GitHub
parent 22d95b444b
commit 511a43b849
63 changed files with 1891 additions and 229 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
dbgpt/app/knowledge/service.py
View File

@@ -367,9 +367,9 @@ class KnowledgeService:
# delete chunks
documents = knowledge_document_dao.get_documents(document_query)
for document in documents:
document_chunk_dao.delete(document.id)
document_chunk_dao.raw_delete(document.id)
# delete documents
knowledge_document_dao.delete(document_query)
knowledge_document_dao.raw_delete(document_query)
# delete space
return knowledge_space_dao.delete_knowledge_space(space)
@@ -395,9 +395,9 @@ class KnowledgeService:
# delete vector by ids
vector_client.delete_by_ids(vector_ids)
# delete chunks
document_chunk_dao.delete(documents[0].id)
document_chunk_dao.raw_delete(documents[0].id)
# delete document
return knowledge_document_dao.delete(document_query)
return knowledge_document_dao.raw_delete(document_query)
def get_document_chunks(self, request: ChunkQueryRequest):
"""get document chunks