拆镜复刻、爆款开头复刻容灾优化

video-gen-api/app/tasks/generation_create_tasks.py

@@ -1,10 +1,11 @@
 from app.tasks.async_runner import run_async
 import json
-from datetime import datetime, timezone
+from datetime import datetime, timedelta, timezone
 from typing import Any, Optional
 
 from sqlalchemy import select
 
+from app.config import settings
 from app.models.base import async_session
 from app.models.chat_generation_task import ChatGenerationTask
 from app.services.error_codes import extract_error_message
@@ -16,6 +17,8 @@ from app.tasks.celery_app import celery_app
 
 ALLOWED_GENERATION_MODES = {"chatapi_async", "hot_opening_replicate", "shot_replicate"}
 
+def _now() -> datetime:
+    return datetime.now(timezone.utc)
 
 def _get_first_value(obj: Any, *field_names: str) -> Optional[Any]:
     """
@@ -242,8 +245,13 @@ async def _run(task_id: str):
 
                 await enqueue_download_task(db, task, reason="create_result_ready")
             else:
-                from app.tasks.generation_poll_tasks import poll_generation_task
+                from app.tasks.generation_poll_tasks import poll_generation_task, register_poll_active
 
+                await register_poll_active(
+                    task,
+                    reason="create_provider_success",
+                    check_at=_now() + timedelta(seconds=int(settings.POLL_TASK_LEASE_SECONDS or 300)),
+                )
                 poll_generation_task.apply_async(
                     args=[task.id],
                     queue="gen_provider_poll",