小鱼开发
106 lines
3.7 KiB
Python
106 lines
3.7 KiB
Python
"""
|
|
脚本相关 Schema
|
|
===============
|
|
"""
|
|
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from app.schemas.segment import Segment
|
|
|
|
ScriptShot = Segment
|
|
|
|
|
|
class SubcategoryItem(BaseModel):
|
|
"""小类项"""
|
|
|
|
code: str = Field(..., description="小类代码")
|
|
name: str = Field(..., description="小类名称")
|
|
count: int = Field(..., description="提示词文件数量")
|
|
|
|
|
|
class CategoryItem(BaseModel):
|
|
"""大类项"""
|
|
|
|
code: str = Field(..., description="大类代码")
|
|
name: str = Field(..., description="大类名称")
|
|
subcategories: list[SubcategoryItem] = Field(..., description="小类列表")
|
|
|
|
|
|
class GenerateScriptRequest(BaseModel):
|
|
"""生成脚本请求"""
|
|
|
|
category: str = Field(..., description="大类代码，如 bk")
|
|
subcategory: str = Field(..., description="小类代码，如 ht")
|
|
duration: int = Field(default=45, ge=30, le=180, description="视频时长（秒）")
|
|
script_type: str = Field(default="干货型", description="脚本类型")
|
|
model: str | None = Field(None, description="指定模型（可选）")
|
|
|
|
|
|
class PolishRequest(BaseModel):
|
|
"""润色请求"""
|
|
|
|
content: str = Field(..., description="待润色内容", min_length=1)
|
|
polish_type: str = Field(default="voiceover", description="润色类型：scene / voiceover")
|
|
shot_type: str | None = Field(
|
|
default="segment",
|
|
description="镜头类型：segment(分镜) / empty_shot(空镜)，用于画面润色时区分",
|
|
)
|
|
|
|
|
|
message: str = Field(..., description="状态描述")
|
|
result: list[Any] | None = Field(None, description="生成的分镜结果（complete 时）")
|
|
|
|
|
|
class ModelHealthInfo(BaseModel):
|
|
"""模型健康信息"""
|
|
|
|
id: str = Field(..., description="模型 ID")
|
|
name: str = Field(..., description="模型名称")
|
|
is_available: bool = Field(..., description="是否可用")
|
|
response_time: float = Field(..., description="响应时间（毫秒）")
|
|
last_error: str | None = Field(None, description="上次错误信息")
|
|
|
|
|
|
class ModelHealthResponse(BaseModel):
|
|
"""模型健康检查响应"""
|
|
|
|
status: str = Field(..., description="整体状态：healthy / unhealthy / error")
|
|
models: list[ModelHealthInfo] = Field(..., description="各模型状态")
|
|
recommended_model: ModelHealthInfo | None = Field(None, description="推荐的模型")
|
|
total_models: int = Field(..., description="模型总数")
|
|
available_models: int = Field(..., description="可用模型数")
|
|
error: str | None = Field(None, description="错误信息")
|
|
|
|
|
|
class TestModelRequest(BaseModel):
|
|
"""测试模型请求"""
|
|
|
|
model_id: str | None = Field(None, description="要测试的模型 ID")
|
|
|
|
|
|
class TestModelResponse(BaseModel):
|
|
"""测试模型响应"""
|
|
|
|
success: bool = Field(..., description="是否成功")
|
|
model: str = Field(..., description="模型名称")
|
|
response_time: float | None = Field(None, description="响应时间（毫秒）")
|
|
error: str | None = Field(None, description="错误信息")
|
|
checked_at: str | None = Field(None, description="检查时间 ISO 格式")
|
|
|
|
|
|
class GenerateTitleRequest(BaseModel):
|
|
"""生成标题请求"""
|
|
|
|
script_content: str = Field(..., description="脚本内容（utterances 文本拼接）", min_length=1)
|
|
title_type: str = Field(..., description="标题类型：main(大标题) / sub(小标题)")
|
|
max_length: int = Field(default=8, ge=1, le=100, description="最大字数限制")
|
|
usage: str = Field(default="video", description="使用场景：video(视频画面标题) / cover(封面标题)")
|
|
|
|
|
|
class GenerateTitleResponse(BaseModel):
|
|
"""生成标题响应"""
|
|
|
|
title: str = Field(..., description="生成的标题")
|