拆镜复刻开发完成

2026-06-11 17:54:40 +08:00
parent a6b6e5f822
commit 74266a126c
34 changed files with 7105 additions and 470 deletions
@@ -23,3 +23,15 @@ async def notify_chat_generation_task_finished(db: AsyncSession, task: ChatGener
            await handle_chat_generation_task_completed(db, task)
        elif task.status == "failed":
            await handle_chat_generation_task_failed(db, task)
+        return
+
+    if task.generation_mode == "shot_replicate":
+        from app.services.shot_replicate_flow_service import (
+            handle_chat_generation_task_completed,
+            handle_chat_generation_task_failed,
+        )
+        if task.status == "completed":
+            await handle_chat_generation_task_completed(db, task)
+        elif task.status == "failed":
+            await handle_chat_generation_task_failed(db, task)
+        return