From 413fa6f0c28ba4303e487796bf083fb687627083 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Wed, 26 Feb 2025 11:00:07 +0800 Subject: [PATCH] fix: When generating a problem, the generated data is incorrect (#2408) --- apps/dataset/serializers/document_serializers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/apps/dataset/serializers/document_serializers.py b/apps/dataset/serializers/document_serializers.py index e30e9f360..ad9199a52 100644 --- a/apps/dataset/serializers/document_serializers.py +++ b/apps/dataset/serializers/document_serializers.py @@ -1225,7 +1225,7 @@ class DocumentSerializers(ApiMixin, serializers.Serializer): 1), ).filter(task_type_status__in=state_list, document_id__in=document_id_list) .values('id'), - TaskType.EMBEDDING, + TaskType.GENERATE_PROBLEM, State.PENDING) ListenerManagement.get_aggregation_document_status_by_query_set( QuerySet(Document).filter(id__in=document_id_list))()