chore: 重复提交的生成任务弹框提示

Author: liuruibin

apps/dataset/serializers/document_serializers.py

@@ -977,7 +977,10 @@ def generate_related(self, model_id, prompt, with_valid=True):
                 self.is_valid(raise_exception=True)
             document_id = self.data.get('document_id')
             QuerySet(Document).filter(id=document_id).update(status=Status.queue_up)
-            generate_related_by_document_id.delay(document_id, model_id, prompt)
+            try:
+                generate_related_by_document_id.delay(document_id, model_id, prompt)
+            except AlreadyQueued as e:
+                raise AppApiException(500, "任务正在执行中,请勿重复下发")
 
 
 

apps/dataset/serializers/paragraph_serializers.py

@@ -9,6 +9,7 @@
 import uuid
 from typing import Dict
 
+from celery_once import AlreadyQueued
 from django.db import transaction
 from django.db.models import QuerySet
 from drf_yasg import openapi
@@ -733,7 +734,10 @@ def batch_generate_related(self, instance: Dict, with_valid=True):
             paragraph_id_list = instance.get("paragraph_id_list")
             model_id = instance.get("model_id")
             prompt = instance.get("prompt")
-            generate_related_by_paragraph_id_list.delay(paragraph_id_list, model_id, prompt)
+            try:
+                generate_related_by_paragraph_id_list.delay(paragraph_id_list, model_id, prompt)
+            except AlreadyQueued as e:
+                raise AppApiException(500, "任务正在执行中,请勿重复下发")