修复生辰任务恢复机制BUG|修复生成视频/图片token快照不回落交易流水BUG

video-gen-api/app/services/generation_module_hook_service.py

@@ -2,36 +2,40 @@ from __future__ import annotations
 
 from sqlalchemy.ext.asyncio import AsyncSession
 
+from app.enums.generation_task import ChatGenerationTaskStatus, GenerationMode
 from app.models.chat_generation_task import ChatGenerationTask
 
 
 async def notify_chat_generation_task_finished(db: AsyncSession, task: ChatGenerationTask) -> None:
     """通知业务模块 ChatGenerationTask 已进入终态。
 
-    当前用于爆款开头复刻：
-    - image_generate 完成后自动进入 video_prompt_optimize
-    - video_generate 完成后总任务完成
+    该方法必须幂等：下载恢复任务、重试任务、服务重启补偿都可能重复调用。
+    具体模块服务需要自行判断 step/project 是否已经完成或失败，避免重复推进。
     """
     if not task:
         return
-    if task.generation_mode == "hot_opening_replicate":
+
+    status = getattr(task, "status", None)
+    generation_mode = getattr(task, "generation_mode", None)
+
+    if generation_mode == GenerationMode.HOT_OPENING_REPLICATE.value:
         from app.services.hot_opening_replicate_service import (
             handle_chat_generation_task_completed,
             handle_chat_generation_task_failed,
         )
-        if task.status == "completed":
+        if status == ChatGenerationTaskStatus.COMPLETED.value:
             await handle_chat_generation_task_completed(db, task)
-        elif task.status == "failed":
+        elif status == ChatGenerationTaskStatus.FAILED.value:
             await handle_chat_generation_task_failed(db, task)
         return
 
-    if task.generation_mode == "shot_replicate":
+    if generation_mode == GenerationMode.SHOT_REPLICATE.value:
         from app.services.shot_replicate_flow_service import (
             handle_chat_generation_task_completed,
             handle_chat_generation_task_failed,
         )
-        if task.status == "completed":
+        if status == ChatGenerationTaskStatus.COMPLETED.value:
             await handle_chat_generation_task_completed(db, task)
-        elif task.status == "failed":
+        elif status == ChatGenerationTaskStatus.FAILED.value:
             await handle_chat_generation_task_failed(db, task)
         return