From 0bcf7e311c3ff91488344340852a0cd576742b76 Mon Sep 17 00:00:00 2001 From: CaptainB Date: Thu, 10 Jul 2025 16:51:14 +0800 Subject: [PATCH] feat: update paragraph processing to include status updates for embedding tasks MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --bug=1058487 --user=刘瑞斌 【知识库】在文档中添加分段，没有执行向量化操作 https://www.tapd.cn/62980211/s/1728041 --- apps/knowledge/serializers/paragraph.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/apps/knowledge/serializers/paragraph.py b/apps/knowledge/serializers/paragraph.py index 518666b63..c02fe93c4 100644 --- a/apps/knowledge/serializers/paragraph.py +++ b/apps/knowledge/serializers/paragraph.py @@ -305,6 +305,10 @@ class ParagraphSerializers(serializers.Serializer): if with_embedding: model_id = get_embedding_model_id_by_knowledge_id(knowledge_id) embedding_by_paragraph(str(paragraph.id), model_id) + ListenerManagement.update_status( + QuerySet(Document).filter(id=document_id), TaskType.EMBEDDING, State.SUCCESS + ) + ListenerManagement.get_aggregation_document_status(document_id)() return ParagraphSerializers.Operate( data={ 'paragraph_id': str(paragraph.id),