1

2026-07-10 17:09:14 +08:00
parent ad2e40bc67
commit d07cd508ee
54 changed files with 7111 additions and 619 deletions
@@ -19,6 +19,7 @@ from app.models.token_usage import TokenUsage
 from app.services.upload_video_asset_service import resolve_upload_video_path
 from app.services.resource_signed_url_service import build_resource_signed_url
 from app.utils.id_gen import generate_id
+from app.services.model_pricing.usage_normalizer import normalize_text_pricing_usage
 from app.enums.common import LogEventStatusEnum, LogSourceEnum
 from app.enums.shot_replicate import ModuleCodeEnum, ShotReplicateLogEventEnum, ShotReplicateRemoteActionEnum
 from app.services.operation_log_service import log_ai_model_event
@@ -692,7 +693,7 @@ async def analyze_video_for_shot_split(
    result = filter_and_normalize_breakdown(result, mode=mode)

    usage = raw.get("usage") or {}
-    token_usage = {
+    token_usage = normalize_text_pricing_usage(usage, base={
        "input_tokens": _int_usage(usage.get("prompt_tokens") or usage.get("input_tokens")),
        "output_tokens": _int_usage(usage.get("completion_tokens") or usage.get("output_tokens")),
        "total_tokens": _int_usage(usage.get("total_tokens")),
@@ -707,7 +708,7 @@ async def analyze_video_for_shot_split(
        "analysis_mode": mode,
        "trace_id": trace_id,
        "log_request": log_request_data,
-    }
+    })
    if not token_usage["total_tokens"]:
        token_usage["total_tokens"] = token_usage["input_tokens"] + token_usage["output_tokens"]