From f02c7ca68d102ec9e610c6586b9c7f0fcc542368 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Thu, 21 Mar 2024 15:20:53 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E6=96=87=E6=A1=A3=E6=89=B9=E9=87=8F?= =?UTF-8?q?=E5=88=A0=E9=99=A4=E6=8A=A5=E9=94=99?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/dataset/serializers/document_serializers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/apps/dataset/serializers/document_serializers.py b/apps/dataset/serializers/document_serializers.py index 8a0979c8e..62bff658e 100644 --- a/apps/dataset/serializers/document_serializers.py +++ b/apps/dataset/serializers/document_serializers.py @@ -578,6 +578,7 @@ class DocumentSerializers(ApiMixin, serializers.Serializer): instance.get('id_list')) return True + @transaction.atomic def batch_delete(self, instance: Dict, with_valid=True): if with_valid: 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") QuerySet(Document).filter(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) return True