AI计价规则BUG修复

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

@@ -241,6 +241,7 @@ async def charge_chatapi_prompt_usage(
         attempt_no=attempt_no,
         charge_kind=CHARGE_TEXT_PROMPT,
         usage=usage,
+        media_references=record.media_references,
     )
     items.append(
         await deduct_credits_locked_once(
@@ -270,6 +271,7 @@ async def charge_chatapi_prompt_usage(
         attempt_no=attempt_no,
         charge_kind=CHARGE_FILE_PARSE,
         usage={**dict(usage), "total_tokens": _safe_int(file_tokens), "input_tokens": _safe_int(file_tokens), "output_tokens": 0},
+        media_references=record.media_references,
     )
     items.append(
         await deduct_credits_locked_once(
@@ -299,6 +301,7 @@ async def charge_chatapi_prompt_usage(
         attempt_no=attempt_no,
         charge_kind=CHARGE_VISION_INPUT,
         usage={**dict(usage), "total_tokens": _safe_int(vision_tokens), "input_tokens": _safe_int(vision_tokens), "output_tokens": 0},
+        media_references=record.media_references,
     )
     items.append(
         await deduct_credits_locked_once(
@@ -477,6 +480,7 @@ async def charge_generation_media_by_params(
     billing_scene: str | None = None,
     media_references: Any = None,
     requested_output_count: int = 1,
+    provider_uses_media_references: bool | None = None,
 ) -> BillingSummary:
     """图片/视频媒体生成扣费。
 
@@ -522,6 +526,7 @@ async def charge_generation_media_by_params(
         inference_mode=inference_mode,
         input_video_duration=input_video_duration,
         requested_output_count=requested_output_count,
+        provider_uses_media_references=provider_uses_media_references,
     )
 
     if gen_type == "image":
@@ -589,4 +594,5 @@ async def charge_generation_media_for_record(
         attempt_no=attempt_no,
         source_module=CreditRecordSourceModule.GENERATION_RECORD.value,
         media_references=record.media_references,
+        provider_uses_media_references=False,
     )