Skip to content

Commit 3e1e434

Browse files
committed
feat: delete associated DocumentTag entries when deleting documents
1 parent 55dc3a6 commit 3e1e434

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

apps/knowledge/serializers/document.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -681,6 +681,7 @@ def delete(self):
681681
delete_problems_and_mappings([document_id])
682682
# 删除向量库
683683
delete_embedding_by_document(document_id)
684+
QuerySet(model=DocumentTag).filter(document_id=document_id).delete()
684685
QuerySet(model=Document).filter(id=document_id).delete()
685686
return True
686687

@@ -1209,6 +1210,7 @@ def batch_delete(self, instance: Dict, with_valid=True):
12091210
QuerySet(File).filter(id__in=source_file_ids).delete()
12101211
QuerySet(Document).filter(id__in=document_id_list).delete()
12111212
QuerySet(Paragraph).filter(document_id__in=document_id_list).delete()
1213+
QuerySet(DocumentTag).filter(document_id__in=document_id_list).delete()
12121214
delete_problems_and_mappings(document_id_list)
12131215
# 删除向量库
12141216
delete_embedding_by_document_list(document_id_list)

0 commit comments

Comments
 (0)