1

2026-07-11 13:00:06 +08:00
parent 6cc1655c69
commit 344fb789a4
82 changed files with 2594 additions and 7930 deletions
@@ -16,7 +16,6 @@ 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
@@ -369,12 +368,7 @@ 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,
-                    stage=PricingSnapshotStage.PROVIDER_ASYNC_COMPLETED.value,
-                )
+                await sync_chat_generation_task_media_token_snapshot(db, task, provider_response=response_data)

                if not task.remote_result_url:
                    await _mark_failed(db, task, message="供应商任务成功但未返回结果URL", detail=poll_result)