15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-28 23:32:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: 文档批量删除报错

This commit is contained in:
shaohuzhang1 2024-03-21 15:20:53 +08:00
parent df896501b6
commit f02c7ca68d
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'))
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