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.enums.generation_task import (
    ChatGenerationTaskStatus,
    GenerationType,
 )
+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.celery_download_recovery_service import (
@@ -547,7 +548,11 @@ async def _run(task_id: str):
                remote_url=task.remote_result_url,
                generated_at=task.generated_at,
            )
-            await sync_chat_generation_task_media_token_snapshot(db, task)
+            await sync_chat_generation_task_media_token_snapshot(
+                db,
+                task,
+                stage=PricingSnapshotStage.RESOURCE_DOWNLOAD_COMPLETED.value,
+            )

            from app.services.generation_module_hook_service import notify_chat_generation_task_finished