From 948d0c4f3cf6385c00551241fc9e8883bd9e2adc Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Mon, 19 Feb 2024 10:55:13 +0800 Subject: [PATCH] =?UTF-8?q?fix:=E3=80=90=E5=AF=B9=E8=AF=9D=E6=97=A5?= =?UTF-8?q?=E5=BF=97=E3=80=91=E4=BF=AE=E6=94=B9=E6=97=A5=E5=BF=97=E7=AD=94?= =?UTF-8?q?=E6=A1=88=E4=BF=9D=E5=AD=98=E5=88=B0=E6=AE=B5=E8=90=BD=E5=90=8E?= =?UTF-8?q?=EF=BC=8C=E5=85=B3=E8=81=94=E9=97=AE=E9=A2=98=E6=9C=AA=E5=81=9A?= =?UTF-8?q?=E5=90=91=E9=87=8F=E5=8C=96=E5=A4=84=E7=90=86=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/application/serializers/chat_serializers.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/apps/application/serializers/chat_serializers.py b/apps/application/serializers/chat_serializers.py index 5eb0d14dd..0c60ab662 100644 --- a/apps/application/serializers/chat_serializers.py +++ b/apps/application/serializers/chat_serializers.py @@ -27,6 +27,7 @@ from application.serializers.chat_message_serializers import ChatInfo from common.db.search import native_search, native_page_search, page_search, get_dynamics_model from common.event import ListenerManagement from common.exception.app_exception import AppApiException +from common.util.common import post from common.util.file_util import get_file_content from common.util.lock import try_lock, un_lock from common.util.rsa_util import decrypt @@ -360,6 +361,13 @@ class ChatRecordSerializer(serializers.Serializer): dataset_id=self.data.get('dataset_id')).exists(): raise AppApiException(500, "文档id不正确") + @staticmethod + def post_embedding_paragraph(chat_record, paragraph_id): + # 发送向量化事件 + ListenerManagement.embedding_by_paragraph_signal.send(paragraph_id) + return chat_record + + @post(post_function=post_embedding_paragraph) @transaction.atomic def improve(self, instance: Dict, with_valid=True): if with_valid: @@ -388,5 +396,4 @@ class ChatRecordSerializer(serializers.Serializer): chat_record.improve_paragraph_id_list.append(paragraph.id) # 添加标注 chat_record.save() - ListenerManagement.embedding_by_paragraph_signal.send(paragraph.id) - return ChatRecordSerializerModel(chat_record).data + return ChatRecordSerializerModel(chat_record).data, paragraph.id