fix: When generating a problem, the generated data is incorrect (#2408)

2025-02-26 11:00:07 +08:00 · 2025-02-26 11:00:07 +08:00 · 413fa6f0c2
parent c6c3799d08
commit 413fa6f0c2
1 changed files with 1 additions and 1 deletions
--- 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))()