完成极速成品和脚本优化

core/backend/apps/ai/services.py

@@ -3225,8 +3225,31 @@ def collect_video_review_blockers(project, only_segment: "VideoSegment | None" =
     return blockers
 
 
-def submit_video_segment(*, video_segment: VideoSegment, user, prompt: str) -> VideoSegmentVersion | None:
-    model_config = get_default_model(ModelConfig.Capability.VIDEO)
+def submit_video_segment(
+    *,
+    video_segment: VideoSegment,
+    user,
+    prompt: str,
+    model_config_id=None,
+    aspect_ratio: str = "9:16",
+    resolution: str = "720p",
+) -> VideoSegmentVersion | None:
+    model_config = None
+    if model_config_id:
+        model_config = (
+            ModelConfig.objects.select_related("provider")
+            .filter(
+                id=model_config_id,
+                capability=ModelConfig.Capability.VIDEO,
+                status=ModelConfig.Status.ACTIVE,
+                provider__status="active",
+            )
+            .first()
+        )
+        if model_config is None:
+            raise ValueError("selected video model is unavailable")
+    else:
+        model_config = get_default_model(ModelConfig.Capability.VIDEO)
     if model_config is None:
         raise ValueError("no active video model configured")
     project = video_segment.project
@@ -3244,15 +3267,15 @@ def submit_video_segment(*, video_segment: VideoSegment, user, prompt: str) -> V
     reference_images = [r["url"] for r in refs]
     final_prompt = build_video_segment_prompt(project, video_segment, scene, refs, prompt)
 
-    # 视频段 token 计量计价(与自由创作同一成本表+同一毛利):按 9:16/720p/目标时长预估,
+    # 视频段 token 计量计价(与自由创作同一成本表+同一毛利):按用户选定的比例/清晰度/目标时长预估,
     # 预留=积分×buffer,终态按火山真实 usage.total_tokens 结算(poll_video_segment true-up)。
     # 这里终结了「视频 ¥1/段、成本 ¥15」的倒贴定价。
     from apps.billing.pricing import quote_video_estimate, video_reserve_amount
 
     est_tokens, quote = quote_video_estimate(
         model_config,
-        aspect_ratio="9:16",
-        resolution="720p",
+        aspect_ratio=aspect_ratio,
+        resolution=resolution,
         duration=video_segment.target_duration_seconds,
         references=[],
         team=project.team,
@@ -3269,8 +3292,8 @@ def submit_video_segment(*, video_segment: VideoSegment, user, prompt: str) -> V
             "endpoint": model_config.endpoint,
             "prompt": final_prompt,
             "duration": video_segment.target_duration_seconds,
-            "ratio": "9:16",
-            "resolution": "720p",
+            "ratio": aspect_ratio,
+            "resolution": resolution,
             "estimated_tokens": est_tokens,
             # 团队价格系数快照:按实结算用它,中途改价不影响在途任务(jimeng 同款纪律)
             "price_multiplier": quote.meta.get("price_multiplier", "1"),
@@ -3288,8 +3311,8 @@ def submit_video_segment(*, video_segment: VideoSegment, user, prompt: str) -> V
             primary_model=model_config,
             prompt=final_prompt,
             duration=video_segment.target_duration_seconds,
-            ratio="9:16",
-            resolution="720p",
+            ratio=aspect_ratio,
+            resolution=resolution,
             reference_images=reference_images,
             request_summary={"video_segment_id": str(video_segment.id)},
         )