fix: 修复迁移文档分段数没有重置 (1Panel-dev#1760)

imrobin · Dec 4, 2024 · a71adf1 · a71adf1
1 parent f945141
commit a71adf1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/apps/dataset/serializers/document_serializers.py b/apps/dataset/serializers/document_serializers.py
@@ -301,6 +301,8 @@ def migrate(self, with_valid=True):
                 ListenerManagement.update_status(QuerySet(Paragraph).filter(document_id__in=document_id_list),
                                                  TaskType.EMBEDDING,
                                                  State.PENDING)
+                ListenerManagement.get_aggregation_document_status_by_query_set(
+                    QuerySet(Document).filter(id__in=document_id_list))()
                 embedding_by_document_list.delay(document_id_list, model_id)
             else:
                 update_embedding_dataset_id(pid_list, target_dataset_id)