fix: 修复迁移文档分段数没有重置

Author: shaohuzhang1

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)