爆款开头API开发完成文件追加

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

@@ -20,6 +20,7 @@ CHARGE_MEDIA = "media"
 
 OWNER_GENERATION_RECORD = "generation_record"
 OWNER_CHAT_GENERATION_TASK = "chat_generation_task"
+OWNER_MODULE_GENERATION_STEP = "module_generation_step"
 
 _BIZ_KEY_PATTERN = re.compile(
     r"^(?P<owner_type>[^:]+):(?P<owner_id>[^:]+):attempt:(?P<attempt_no>\d+):(?P<charge_kind>[^:]+):(?P<action>charge|refund)$"
@@ -287,6 +288,45 @@ async def charge_chatapi_prompt_usage(
     return BillingSummary(record_id=record.id, user_id=record.user_id, items=items)
 
 
+async def charge_module_prompt_usage(
+    db: AsyncSession,
+    *,
+    user_id: str,
+    step_id: str,
+    usage: Mapping[str, Any],
+    description: str,
+    attempt_no: int = 1,
+) -> BillingSummary:
+    """爆款开头复刻模块图片/视频 AI 提词扣文本积分。
+
+    文本提词属于已经发生的 LLM 消费：
+    - 调用成功后按 input_tokens + output_tokens 扣费。
+    - 不参与后续图片/视频媒体生成失败退款。
+    - 通过 module_generation_step:{step_id}:attempt:1:text_prompt:charge 幂等。
+    """
+    input_tokens = _safe_int(usage.get("input_tokens"))
+    output_tokens = _safe_int(usage.get("output_tokens"))
+    text_credits = await calc_text_credits(db, input_tokens, output_tokens)
+    biz_key = build_credit_biz_key(
+        owner_type=OWNER_MODULE_GENERATION_STEP,
+        owner_id=step_id,
+        attempt_no=attempt_no,
+        charge_kind=CHARGE_TEXT_PROMPT,
+        action="charge",
+    )
+    item = await deduct_credits_locked_once(
+        db,
+        user_id=user_id,
+        amount=text_credits,
+        description=description,
+        related_id=step_id,
+        charge_key=CHARGE_TEXT_PROMPT,
+        biz_key=biz_key,
+        attempt_no=attempt_no,
+    )
+    return BillingSummary(record_id=step_id, user_id=user_id, items=[item])
+
+
 async def charge_generation_media_by_params(
     db: AsyncSession,
     *,

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

@@ -17,10 +17,13 @@ from app.services.celery_download_recovery_service import (
     remove_download_active,
 )
 from app.services.generation_log_service import log_task_event
+from app.services.generation_module_hook_service import notify_chat_generation_task_finished
 from app.services.generation_refund_service import mark_chat_generation_task_failed_and_refund_once
 
 logger = logging.getLogger("video_gen")
 
+ALLOWED_GENERATION_MODES = {"chatapi_async", "hot_opening_replicate"}
+
 
 def _now() -> datetime:
     return datetime.now(timezone.utc)
@@ -73,7 +76,7 @@ async def recover_one_download_task(
 
     if not task:
         return "skip_missing_task"
-    if task.generation_mode != "chatapi_async":
+    if task.generation_mode not in ALLOWED_GENERATION_MODES:
         await remove_download_active(task.id)
         return "clean_invalid_mode"
     if _is_final_task_state(task):
@@ -218,7 +221,7 @@ async def recover_download_tasks_once(db: AsyncSession) -> dict[str, Any]:
         select(ChatGenerationTask)
         .where(
             ChatGenerationTask.deleted_at.is_(None),
-            ChatGenerationTask.generation_mode == "chatapi_async",
+            ChatGenerationTask.generation_mode.in_(["chatapi_async", "hot_opening_replicate"]),
             ChatGenerationTask.status == "generating",
             ChatGenerationTask.remote_result_url.is_not(None),
             ChatGenerationTask.pipeline_stage.in_(
@@ -263,7 +266,7 @@ async def recover_generation_tasks_once(db: AsyncSession) -> dict[str, Any]:
         select(ChatGenerationTask)
         .where(
             ChatGenerationTask.deleted_at.is_(None),
-            ChatGenerationTask.generation_mode == "chatapi_async",
+            ChatGenerationTask.generation_mode.in_(["chatapi_async", "hot_opening_replicate"]),
             ChatGenerationTask.status == "generating",
             ChatGenerationTask.pipeline_stage.in_(
                 [
@@ -290,6 +293,7 @@ async def recover_generation_tasks_once(db: AsyncSession) -> dict[str, Any]:
                 error_message="任务超时",
                 pipeline_stage="timeout",
             )
+            await notify_chat_generation_task_finished(db, task)
             await db.commit()
             await log_task_event(
                 task,

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

@@ -180,7 +180,7 @@ async def mark_chat_generation_task_failed_and_refund_once(
             select(ChatGenerationTask)
             .where(
                 ChatGenerationTask.id == task_id,
-                ChatGenerationTask.generation_mode == "chatapi_async",
+                ChatGenerationTask.generation_mode.in_(["chatapi_async", "hot_opening_replicate"]),
                 ChatGenerationTask.deleted_at.is_(None),
             )
             .with_for_update()