zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/main'

Browse Source
This commit is contained in:
liqiang-fit2cloud 2024-03-21 15:28:47 +08:00
commit 41151cbbad
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/dataset/serializers/document_serializers.py
View File

@ -578,6 +578,7 @@ class DocumentSerializers(ApiMixin, serializers.Serializer):
instance.get('id_list')) instance.get('id_list'))
return True return True
@transaction.atomic
def batch_delete(self, instance: Dict, with_valid=True): def batch_delete(self, instance: Dict, with_valid=True):
if with_valid: if with_valid:
BatchSerializer(data=instance).is_valid(model=Document, raise_exception=True) BatchSerializer(data=instance).is_valid(model=Document, raise_exception=True)
@ -585,7 +586,7 @@ class DocumentSerializers(ApiMixin, serializers.Serializer):
document_id_list = instance.get("id_list") document_id_list = instance.get("id_list")
QuerySet(Document).filter(id__in=document_id_list).delete() QuerySet(Document).filter(id__in=document_id_list).delete()
QuerySet(Paragraph).filter(document_id__in=document_id_list).delete() QuerySet(Paragraph).filter(document_id__in=document_id_list).delete()
QuerySet(Problem).filter(document_id__in=document_id_list).delete() QuerySet(ProblemParagraphMapping).filter(document_id__in=document_id_list).delete()
# 删除向量库 # 删除向量库
ListenerManagement.delete_embedding_by_document_list_signal.send(document_id_list) ListenerManagement.delete_embedding_by_document_list_signal.send(document_id_list)
return True return True