1

2026-07-11 13:00:06 +08:00
parent 6cc1655c69
commit 344fb789a4
82 changed files with 2594 additions and 7930 deletions
@@ -14,7 +14,7 @@ from app.models.chat_generation_task import ChatGenerationTask
 from app.models.image_engine import ImageEngine
 from app.models.video_engine import VideoEngine
 from app.services.generation_log_service import log_provider_call
-from app.services.image_gen import submit_image_task
+from app.services.image_gen import poll_image_task_status, submit_image_task
 from app.services.provider_limit import provider_limit
 from app.services.video_gen import poll_task_status, submit_video_task

@@ -165,7 +165,8 @@ def _try_json(text: Any) -> Any:
 async def poll_provider_task(db: AsyncSession, task: ChatGenerationTask) -> dict:
    engine = await get_runtime_engine(db, task)
    task_id = task.seedance_task_id or task.provider_task_id
-    if task.gen_type != "video":
-        raise ValueError("当前火山图片引擎为同步生成，不允许进入 Provider 轮询链路")
-    async with provider_limit("ark_video_poll", settings.ARK_VIDEO_POLL_MAX_CONCURRENCY):
-        return await poll_task_status(engine, task_id)
+    if task.gen_type == "video":
+        async with provider_limit("ark_video_poll", settings.ARK_VIDEO_POLL_MAX_CONCURRENCY):
+            return await poll_task_status(engine, task_id)
+    async with provider_limit("ark_image_poll", settings.ARK_IMAGE_POLL_MAX_CONCURRENCY):
+        return await poll_image_task_status(engine, task_id)