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

fix(core): Delete corresponding Milvus data when using with TuGraph (#2858)

Browse Source
This commit is contained in:
stevenlx96
2025-09-08 09:55:29 +08:00
committed by GitHub
parent ea6cee49ab
commit 6e304c230b
9 changed files with 74 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
packages/dbgpt-app/src/dbgpt_app/knowledge/service.py
View File

@@ -464,6 +464,23 @@ class KnowledgeService:
)
# delete vector by ids
storage_connector.delete_by_ids(vector_ids)
# we next delete the corresponding CHUNK HISTORY data in Milvus
if (
space.vector_type == "KnowledgeGraph"
and storage_connector._vector_store_config.__type__ == "milvus"
):
# this gives the vector store type
# in case this will support chroma in the future
embedding_vector_type = storage_connector._vector_store_config.__type__
# get the collection name
space_name = space_name + "_CHUNK_HISTORY"
storage_connector = self.storage_manager.get_storage_connector(
index_name=space_name, storage_type=embedding_vector_type
)
# documents[0].id is the id we use to find the corresponding document
storage_connector.delete_by_file_id(documents[0].id)
# delete chunks
document_chunk_dao.raw_delete(documents[0].id)
# delete document