修复生辰任务恢复机制BUG|修复生成视频/图片token快照不回落交易流水BUG

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

@@ -10,6 +10,7 @@ from app.models.base import async_session
 from app.models.generation_record import GenerationRecord
 from app.services.video_gen import get_active_engine, poll_task_status, download_video, _log_video_response
 from app.services.image_gen import get_active_image_engine, download_image
+from app.services.media_token_usage_snapshot_service import sync_generation_record_media_token_snapshot
 from app.services.resource_accounting_service import (
     record_generation_record_generated_resource,
     safe_file_size,
@@ -157,6 +158,7 @@ class TaskQueue:
             else:
                 record.video_url = file_url
             record.video_tokens_used = poll_result.get("video_tokens", 0)
+            await sync_generation_record_media_token_snapshot(db, record, provider_response=resp_data)
             record.status = "completed"
             record.generated_at = datetime.now()
             if record.video_url:
@@ -235,6 +237,7 @@ class TaskQueue:
                 else:
                     record.image_url = remote_url
                 record.image_tokens_used = poll_result.get("image_tokens", 0)
+                await sync_generation_record_media_token_snapshot(db, record, provider_response=poll_result)
                 record.status = "completed"
                 record.generated_at = datetime.now()
                 if record.image_url: