zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
CaptainB 2024-10-29 16:10:14 +08:00 committed by 刘瑞斌
parent 0f63db683c
commit 67fd113449
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/dataset/serializers/document_serializers.py
View File

@ -977,7 +977,10 @@ class DocumentSerializers(ApiMixin, serializers.Serializer):
self.is_valid(raise_exception=True) self.is_valid(raise_exception=True)
document_id = self.data.get('document_id') document_id = self.data.get('document_id')
QuerySet(Document).filter(id=document_id).update(status=Status.queue_up) 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, "任务正在执行中,请勿重复下发")

6
apps/dataset/serializers/paragraph_serializers.py
View File

@ -9,6 +9,7 @@
import uuid import uuid
from typing import Dict from typing import Dict
from celery_once import AlreadyQueued
from django.db import transaction from django.db import transaction
from django.db.models import QuerySet from django.db.models import QuerySet
from drf_yasg import openapi from drf_yasg import openapi
@ -733,7 +734,10 @@ class ParagraphSerializers(ApiMixin, serializers.Serializer):
paragraph_id_list = instance.get("paragraph_id_list") paragraph_id_list = instance.get("paragraph_id_list")
model_id = instance.get("model_id") model_id = instance.get("model_id")
prompt = instance.get("prompt") 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, "任务正在执行中,请勿重复下发")