114 lines
4.1 KiB
Python
114 lines
4.1 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field, field_validator
|
|
|
|
|
|
class ModuleGenerationVideoConfigV2(BaseModel):
|
|
model_config = ConfigDict(extra="ignore")
|
|
|
|
engine_id: str = Field(..., min_length=1, max_length=32, description="视频引擎ID")
|
|
duration: int = Field(..., ge=1, le=120, description="视频时长，单位秒")
|
|
aspect_ratio: str = Field(..., min_length=1, max_length=16, description="视频比例")
|
|
resolution: str = Field(..., min_length=1, max_length=16, description="目标分辨率")
|
|
|
|
@field_validator("engine_id", "aspect_ratio", "resolution", mode="before")
|
|
@classmethod
|
|
def _strip_required(cls, value: str) -> str:
|
|
value = str(value or "").strip()
|
|
if not value:
|
|
raise ValueError("字段不能为空")
|
|
return value
|
|
|
|
|
|
class HotOpeningTaskCreateV2(BaseModel):
|
|
model_config = ConfigDict(extra="ignore")
|
|
|
|
material_video_url: str = Field(..., min_length=1, description="爆款参考视频，步骤2分析使用，步骤3绝不携带")
|
|
material_video_resource_id: str | None = Field(None, max_length=32)
|
|
material_video_duration_seconds: float | None = Field(None, gt=0)
|
|
material_image_url: str | None = Field(None, description="可选参考图片；仅支持本地上传/历史素材")
|
|
material_image_resource_id: str | None = Field(None, max_length=32)
|
|
source_project_name: str | None = Field(None, max_length=40)
|
|
target_project_name: str | None = Field(None, max_length=40)
|
|
core_content_point: str | None = Field(None, max_length=200)
|
|
project_description: str | None = Field(None, max_length=1000)
|
|
target_platform: str | None = Field(None, max_length=64)
|
|
video_config: ModuleGenerationVideoConfigV2
|
|
idempotency_key: str | None = Field(None, max_length=64)
|
|
|
|
@field_validator(
|
|
"material_video_url",
|
|
"material_image_url",
|
|
"source_project_name",
|
|
"target_project_name",
|
|
"core_content_point",
|
|
"project_description",
|
|
"target_platform",
|
|
"idempotency_key",
|
|
mode="before",
|
|
)
|
|
@classmethod
|
|
def _strip_text(cls, value: str | None) -> str | None:
|
|
if value is None:
|
|
return None
|
|
value = str(value).strip()
|
|
return value or None
|
|
|
|
|
|
class ShotReplicateProjectCreateV2(BaseModel):
|
|
model_config = ConfigDict(extra="ignore")
|
|
|
|
material_image_url: str | None = Field(None, description="可选参考图片；仅支持本地上传/历史素材")
|
|
material_image_resource_id: str | None = Field(None, max_length=32)
|
|
target_project_name: str | None = Field(None, max_length=40)
|
|
core_content_point: str | None = Field(None, max_length=200)
|
|
project_description: str | None = Field(None, max_length=1000)
|
|
target_platform: str | None = Field(None, max_length=64)
|
|
video_config: ModuleGenerationVideoConfigV2
|
|
idempotency_key: str | None = Field(None, max_length=64)
|
|
|
|
@field_validator(
|
|
"material_image_url",
|
|
"target_project_name",
|
|
"core_content_point",
|
|
"project_description",
|
|
"target_platform",
|
|
"idempotency_key",
|
|
mode="before",
|
|
)
|
|
@classmethod
|
|
def _strip_text(cls, value: str | None) -> str | None:
|
|
if value is None:
|
|
return None
|
|
value = str(value).strip()
|
|
return value or None
|
|
|
|
|
|
class ModuleVideoPromptRetryV2(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
video_config: ModuleGenerationVideoConfigV2
|
|
|
|
|
|
class ModuleVideoPromptSchemaUpdateV2(BaseModel):
|
|
model_config = ConfigDict(extra="ignore")
|
|
|
|
prompt_schema: dict[str, Any] = Field(..., description="允许编辑字段的 JSON patch")
|
|
|
|
@field_validator("prompt_schema")
|
|
@classmethod
|
|
def _non_empty_schema(cls, value: dict[str, Any]) -> dict[str, Any]:
|
|
if not isinstance(value, dict) or not value:
|
|
raise ValueError("prompt_schema 必须为非空对象")
|
|
return value
|
|
|
|
|
|
class ModuleGenerationV2ActionOut(BaseModel):
|
|
message: str
|
|
project_id: str
|
|
step_id: str | None = None
|
|
flow_version: str = "v2"
|
|
detail: Any | None = None
|