交易流水对账明细 | AI引擎计价规则

video-gen-api/app/tasks/generation_poll_tasks.py

@@ -16,6 +16,7 @@ from app.enums.generation_task import (
     PROVIDER_FAILED_STATUSES,
     PROVIDER_SUCCESS_STATUSES,
 )
+from app.enums.model_pricing import PricingSnapshotStage
 from app.models.base import async_session
 from app.models.chat_generation_task import ChatGenerationTask
 from app.services.error_codes import extract_error_message
@@ -368,7 +369,12 @@ async def _run(task_id: str, *, force_due: bool = False):
                     task.video_tokens_used = poll_result.get("video_tokens", 0) or 0
 
                 task.provider_response_json = response_data
-                await sync_chat_generation_task_media_token_snapshot(db, task, provider_response=response_data)
+                await sync_chat_generation_task_media_token_snapshot(
+                    db,
+                    task,
+                    provider_response=response_data,
+                    stage=PricingSnapshotStage.PROVIDER_ASYNC_COMPLETED.value,
+                )
 
                 if not task.remote_result_url:
                     await _mark_failed(db, task, message="供应商任务成功但未返回结果URL", detail=poll_result)