爆款开头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,
     *,