小鱼开发
76 lines
2.1 KiB
Python
76 lines
2.1 KiB
Python
"""
|
|
空镜素材 Schema
|
|
==============
|
|
"""
|
|
|
|
from pydantic import BaseModel, Field, field_validator
|
|
|
|
|
|
class MaterialInfo(BaseModel):
|
|
"""素材条目"""
|
|
|
|
url: str = Field(description="素材 URL（远程或本地路径）")
|
|
duration: float = Field(description="素材时长（秒）")
|
|
|
|
|
|
class MatchMaterialRequest(BaseModel):
|
|
"""匹配素材请求"""
|
|
|
|
scene: str = Field(description="分镜场景描述")
|
|
duration: float = Field(description="所需时长（秒）", gt=0)
|
|
project_id: str | None = Field(default=None, description="项目ID，用于跨分镜去重")
|
|
|
|
@field_validator("scene")
|
|
@classmethod
|
|
def _validate_scene(cls, v: str) -> str:
|
|
v = v.strip()
|
|
if not v:
|
|
raise ValueError("场景描述不能为空")
|
|
return v
|
|
|
|
@field_validator("project_id")
|
|
@classmethod
|
|
def _validate_project_id(cls, v: str | None) -> str | None:
|
|
if v is not None and not v.strip():
|
|
return None
|
|
return v
|
|
|
|
|
|
class BatchMatchSceneItem(BaseModel):
|
|
"""批量匹配场景项"""
|
|
|
|
scene: str = Field(description="分镜场景描述")
|
|
duration: float = Field(description="所需时长（秒）", gt=0)
|
|
|
|
@field_validator("scene")
|
|
@classmethod
|
|
def _validate_scene(cls, v: str) -> str:
|
|
v = v.strip()
|
|
if not v:
|
|
raise ValueError("场景描述不能为空")
|
|
return v
|
|
|
|
|
|
class BatchMatchMaterialRequest(BaseModel):
|
|
"""批量匹配素材请求"""
|
|
|
|
project_id: str | None = Field(default=None, description="项目ID，用于跨分镜去重")
|
|
scenes: list[BatchMatchSceneItem] = Field(
|
|
description="分镜场景列表",
|
|
min_length=1,
|
|
)
|
|
|
|
@field_validator("project_id")
|
|
@classmethod
|
|
def _validate_project_id(cls, v: str | None) -> str | None:
|
|
if v is not None and not v.strip():
|
|
return None
|
|
return v
|
|
|
|
|
|
class BatchMatchMaterialResponse(BaseModel):
|
|
"""批量匹配素材响应"""
|
|
|
|
project_id: str | None = Field(description="项目ID")
|
|
results: list[MaterialInfo | None] = Field(description="匹配结果列表，与 scenes 一一对应")
|