91 lines
3.8 KiB
Python
91 lines
3.8 KiB
Python
from airshelf.celery import app
|
|
|
|
|
|
@app.task(bind=True, max_retries=3)
|
|
def submit_ai_task(self, task_id: str) -> str:
|
|
return task_id
|
|
|
|
|
|
@app.task(bind=True, max_retries=5)
|
|
def poll_ai_task(self, task_id: str) -> str:
|
|
return task_id
|
|
|
|
|
|
@app.task(bind=True, max_retries=0)
|
|
def generate_standalone_image_task(self, task_id: str) -> str:
|
|
"""单张独立图的慢活(ARK 出图 ~30s)在 worker 内跑,Web 层不被占住。
|
|
幂等且失败自退费(见 run_standalone_image_task),故 max_retries=0,不向上抛重试。"""
|
|
from apps.ai.services import run_standalone_image_task
|
|
|
|
run_standalone_image_task(task_id=task_id)
|
|
return task_id
|
|
|
|
|
|
@app.task(bind=True, max_retries=0)
|
|
def extract_entities_task(self, task_id: str) -> str:
|
|
"""实体提取的慢活(豆包思考模型流式,可达数十秒)在 worker 内跑,Web 层秒回不被占住、不再 502。
|
|
幂等且失败自退费(见 run_extract_entities_task),故 max_retries=0,不向上抛重试。"""
|
|
from apps.ai.services import run_extract_entities_task
|
|
|
|
run_extract_entities_task(task_id=task_id)
|
|
return task_id
|
|
|
|
|
|
@app.task(bind=True, max_retries=0)
|
|
def generate_base_asset_task(self, task_id: str) -> str:
|
|
"""基础资产(商品/人物/场景立绘)的慢出图在 worker 内跑,Web 层不被占住。
|
|
幂等且失败自退费(见 run_base_asset_task),故 max_retries=0,不向上抛重试。"""
|
|
from apps.ai.services import run_base_asset_task
|
|
|
|
run_base_asset_task(task_id=task_id)
|
|
return task_id
|
|
|
|
|
|
@app.task(bind=True, max_retries=0)
|
|
def generate_triview_task(self, task_id: str) -> str:
|
|
"""三视图(image_edit 以立绘为参考,慢)在 worker 内跑,Web 层不被占住。
|
|
幂等且失败自退费(见 run_triview_task),故 max_retries=0,不向上抛重试。"""
|
|
from apps.ai.services import run_triview_task
|
|
|
|
run_triview_task(task_id=task_id)
|
|
return task_id
|
|
|
|
|
|
@app.task(bind=True, max_retries=0)
|
|
def generate_model_triview_task(self, task_id: str) -> str:
|
|
"""团队模特库三视图：不绑定项目，成功更新 Model.triview_asset，失败释放预留积分。"""
|
|
from apps.ai.services import run_model_triview_task
|
|
|
|
run_model_triview_task(task_id=task_id)
|
|
return task_id
|
|
|
|
|
|
@app.task(bind=True, max_retries=0)
|
|
def poll_free_video_task(self, task_id: str, attempt: int = 0) -> str:
|
|
"""自由创作视频·worker 兜底轮询:每 30s 一次自重排(不依赖 celery beat),
|
|
总等待上限由统一视频 generation_timeout 配置控制。finalize 幂等(POSTPROCESSING 认领),
|
|
与前端主动 poll 并存不双扣。轮询本身出错不重试(max_retries=0),下一次自重排继续。"""
|
|
from apps.ai.free_video import finalize_free_video
|
|
from apps.ai.models import AITask
|
|
|
|
task = AITask.objects.select_related("model_config", "model_config__provider", "team").filter(id=task_id).first()
|
|
if task is None:
|
|
return task_id
|
|
try:
|
|
task = finalize_free_video(task=task)
|
|
except Exception: # noqa: BLE001 — 单次轮询失败(网络抖动等)不终结任务,等下一轮
|
|
import logging
|
|
|
|
logging.getLogger(__name__).warning("poll_free_video_task %s attempt %s failed", task_id, attempt, exc_info=True)
|
|
# eager(本地联调/单测)下 apply_async 会内联立即执行,自重排=同步死循环 → 跳过,收尾交给前端主动 poll
|
|
from django.conf import settings as dj_settings
|
|
|
|
if getattr(dj_settings, "CELERY_TASK_ALWAYS_EAGER", False):
|
|
return task_id
|
|
# 是否终止由 finalize_free_video 读取统一 generation_timeout 决定；这里不再维护一套
|
|
# “60 次轮询”硬编码，避免修改成片等待时限时 Worker 与 Web 轮询口径不一致。
|
|
if task.status in (AITask.Status.SUBMITTED, AITask.Status.POLLING):
|
|
poll_free_video_task.apply_async(args=[task_id, attempt + 1], countdown=30)
|
|
return task_id
|
|
|