拆镜复刻开发完成

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

@@ -0,0 +1,42 @@
+from __future__ import annotations
+
+from app.models.base import async_session
+from app.services.shot_replicate_flow_service import run_image_prompt_optimize, run_video_prompt_optimize
+from app.tasks.async_runner import run_async
+from app.tasks.celery_app import celery_app
+
+
+async def _run_image_prompt(project_id: str, step_id: str | None = None):
+    async with async_session() as db:
+        await run_image_prompt_optimize(db, project_id=project_id, step_id=step_id)
+        await db.commit()
+
+
+async def _run_video_prompt(project_id: str, step_id: str | None = None):
+    async with async_session() as db:
+        await run_video_prompt_optimize(db, project_id=project_id, step_id=step_id)
+        await db.commit()
+
+
+if celery_app:
+
+    @celery_app.task(name="shot_replicate.start_image_prompt_optimize", bind=True, max_retries=3, default_retry_delay=30)
+    def start_image_prompt_optimize(self, project_id: str, step_id: str | None = None):
+        return run_async(_run_image_prompt(project_id, step_id))
+
+
+    @celery_app.task(name="shot_replicate.start_video_prompt_optimize", bind=True, max_retries=3, default_retry_delay=30)
+    def start_video_prompt_optimize(self, project_id: str, step_id: str | None = None):
+        return run_async(_run_video_prompt(project_id, step_id))
+
+else:
+
+    class _DisabledTask:
+        def delay(self, *args, **kwargs):
+            raise RuntimeError("Celery is disabled")
+
+        def apply_async(self, *args, **kwargs):
+            raise RuntimeError("Celery is disabled")
+
+    start_image_prompt_optimize = _DisabledTask()
+    start_video_prompt_optimize = _DisabledTask()