From f74f1566805faa9bd17274530cd6e7e0e6683e3f Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Wed, 15 May 2024 11:21:17 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E5=AF=BC=E5=85=A5?= =?UTF-8?q?=E7=9A=84=E6=96=87=E6=A1=A3=E5=88=A0=E9=99=A4=E9=83=A8=E5=88=86?= =?UTF-8?q?=E7=9A=84=E5=88=86=E6=AE=B5=E5=90=8E=EF=BC=8C=E5=AD=97=E7=AC=A6?= =?UTF-8?q?=E6=95=B0=E4=B8=8D=E4=BC=9A=E9=87=8D=E6=96=B0=E8=AE=A1=E7=AE=97?= =?UTF-8?q?=20#421?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/dataset/serializers/paragraph_serializers.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/apps/dataset/serializers/paragraph_serializers.py b/apps/dataset/serializers/paragraph_serializers.py index 589710ce..3f5164a4 100644 --- a/apps/dataset/serializers/paragraph_serializers.py +++ b/apps/dataset/serializers/paragraph_serializers.py @@ -15,7 +15,7 @@ from rest_framework import serializers from common.db.search import page_search -from common.event.listener_manage import ListenerManagement, UpdateEmbeddingDocumentIdArgs, UpdateEmbeddingDatasetIdArgs +from common.event.listener_manage import ListenerManagement, UpdateEmbeddingDocumentIdArgs from common.exception.app_exception import AppApiException from common.mixins.api_mixin import ApiMixin from common.util.common import post @@ -284,6 +284,7 @@ def batch_delete(self, instance: Dict, with_valid=True): paragraph_id_list = instance.get("id_list") QuerySet(Paragraph).filter(id__in=paragraph_id_list).delete() QuerySet(ProblemParagraphMapping).filter(paragraph_id__in=paragraph_id_list).delete() + update_document_char_length(self.data.get('document_id')) # 删除向量库 ListenerManagement.delete_embedding_by_paragraph_ids(paragraph_id_list) return True @@ -370,6 +371,7 @@ def migrate(self, with_valid=True): target_document_id, target_dataset_id)) # 修改段落信息 paragraph_list.update(dataset_id=target_dataset_id, document_id=target_document_id) + update_document_char_length(document_id) @staticmethod def update_problem_paragraph_mapping(target_document_id: str, problem_paragraph_mapping): @@ -527,6 +529,7 @@ def delete(self, with_valid=False): paragraph_id = self.data.get('paragraph_id') QuerySet(Paragraph).filter(id=paragraph_id).delete() QuerySet(ProblemParagraphMapping).filter(paragraph_id=paragraph_id).delete() + update_document_char_length(self.data.get('document_id')) ListenerManagement.delete_embedding_by_paragraph_signal.send(paragraph_id) @staticmethod