火山引擎SMS API|celery容灾优化|生成模型引擎积分列表API

2026-06-04 16:28:59 +08:00
parent 450e509b1a
commit 8d43d5871c
25 changed files with 1936 additions and 302 deletions
@@ -172,9 +172,9 @@ async def _run(task_id: str):
                task.pipeline_stage = "result_ready"
                await db.commit()

-                from app.tasks.generation_download_tasks import download_generation_result_task
+                from app.tasks.generation_download_tasks import enqueue_download_task

-                download_generation_result_task.delay(task.id)
+                await enqueue_download_task(db, task, reason="create_remote_result_ready")
                return

            else:
@@ -223,9 +223,9 @@ async def _run(task_id: str):
                )

            if task.pipeline_stage == "result_ready":
-                from app.tasks.generation_download_tasks import download_generation_result_task
+                from app.tasks.generation_download_tasks import enqueue_download_task

-                download_generation_result_task.delay(task.id)
+                await enqueue_download_task(db, task, reason="create_result_ready")
            else:
                from app.tasks.generation_poll_tasks import poll_generation_task