超分功能完成

video-gen-api/app/services/generation/ai/task_create_service.py

@@ -40,6 +40,7 @@ from app.services.generation.billing_service import OWNER_CHAT_GENERATION_TASK,
 from app.services.operation_log_service import log_operation_event
 from app.services.private_portrait.reference_resolver import resolve_private_portrait_references
 from app.services.resource_capacity_service import assert_user_resource_capacity_available
+from app.services.video_upscale.snapshot_service import build_video_upscale_snapshot
 from app.utils.id_gen import generate_id
 
 
@@ -331,7 +332,15 @@ async def create_generation_task_group(
             raise HTTPException(status_code=400, detail=f"视频时长不能超过 {engine.max_duration} 秒")
 
         input_video_duration = _validate_video_references(refs, max_audio_count=engine.max_audio_count)
+        provider_generation_resolution, upscale_enabled_snapshot, upscale_snapshot_json = await build_video_upscale_snapshot(
+            db,
+            target_resolution=resolution,
+            aspect_ratio=ratio,
+            supported_provider_resolutions=resolutions,
+        )
         video_snapshot = build_video_snapshot(engine, ratio, resolution, duration)
+        video_snapshot["provider_generation_resolution"] = provider_generation_resolution
+        video_snapshot["video_upscale_enabled_snapshot"] = upscale_enabled_snapshot
         video_snapshot["generation_count"] = generation_count
         snapshot_json = _json(video_snapshot) or "{}"
         deadline_at = now + timedelta(hours=settings.CHATAPI_ASYNC_VIDEO_FINAL_DEADLINE_HOURS)
@@ -370,6 +379,9 @@ async def create_generation_task_group(
                 duration=duration,
                 aspect_ratio=ratio,
                 resolution=resolution,
+                provider_generation_resolution=provider_generation_resolution,
+                video_upscale_enabled_snapshot=upscale_enabled_snapshot,
+                video_upscale_snapshot_json=upscale_snapshot_json,
                 image_size=req.image_size or IMAGE_DEFAULT_SIZE,
                 image_proportion=req.image_proportion or IMAGE_DEFAULT_PROPORTION,
                 image_px=normalize_px(req.image_px) or IMAGE_DEFAULT_PX,
@@ -394,6 +406,9 @@ async def create_generation_task_group(
                 duration=duration,
                 aspect_ratio=ratio,
                 resolution=resolution,
+                provider_generation_resolution=provider_generation_resolution,
+                video_upscale_enabled_snapshot=upscale_enabled_snapshot,
+                video_upscale_snapshot_json=upscale_snapshot_json,
                 image_size=req.image_size or IMAGE_DEFAULT_SIZE,
                 image_proportion=req.image_proportion or IMAGE_DEFAULT_PROPORTION,
                 image_px=normalize_px(req.image_px) or IMAGE_DEFAULT_PX,
@@ -438,6 +453,9 @@ async def create_generation_task_group(
                     duration=duration,
                     aspect_ratio=ratio,
                     resolution=resolution,
+                    provider_generation_resolution=provider_generation_resolution,
+                    video_upscale_enabled_snapshot=upscale_enabled_snapshot,
+                    video_upscale_snapshot_json=upscale_snapshot_json,
                     image_size=req.image_size or IMAGE_DEFAULT_SIZE,
                     image_proportion=req.image_proportion or IMAGE_DEFAULT_PROPORTION,
                     image_px=normalize_px(req.image_px) or IMAGE_DEFAULT_PX,
@@ -476,6 +494,8 @@ async def create_generation_task_group(
             "generation_count": generation_count,
             "child_task_ids": child_ids,
             "enqueue_task_ids": enqueue_ids,
+            "video_upscale_enabled_snapshot": bool(locals().get("upscale_enabled_snapshot", False)),
+            "provider_generation_resolution": locals().get("provider_generation_resolution"),
         },
     )
     return GenerationTaskCreateResult(