38 lines
1.6 KiB
Python
38 lines
1.6 KiB
Python
|
from datetime import datetime
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
from typing import List, Optional
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 请求模型
|
|||
|
|
class ModelCreateRequest(BaseModel):
|
|||
|
|
name: str = Field(..., max_length=255, description="模型名称（必填，如：yolo-v8s-car）")
|
|||
|
|
description: Optional[str] = Field(None, description="模型描述（可选）")
|
|||
|
|
is_default: Optional[bool] = Field(False, description="是否设为默认模型")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ModelUpdateRequest(BaseModel):
|
|||
|
|
name: Optional[str] = Field(None, max_length=255, description="模型名称（可选修改）")
|
|||
|
|
description: Optional[str] = Field(None, description="模型描述（可选修改）")
|
|||
|
|
is_default: Optional[bool] = Field(None, description="是否设为默认模型（可选切换）")
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 响应模型
|
|||
|
|
class ModelResponse(BaseModel):
|
|||
|
|
id: int = Field(..., description="模型ID")
|
|||
|
|
name: str = Field(..., description="模型名称")
|
|||
|
|
path: str = Field(..., description="模型文件相对路径")
|
|||
|
|
is_default: bool = Field(..., description="是否默认模型")
|
|||
|
|
description: Optional[str] = Field(None, description="模型描述")
|
|||
|
|
file_size: Optional[int] = Field(None, description="文件大小（字节）")
|
|||
|
|
created_at: datetime = Field(..., description="创建时间")
|
|||
|
|
updated_at: datetime = Field(..., description="更新时间")
|
|||
|
|
|
|||
|
|
model_config = {"from_attributes": True}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ModelListResponse(BaseModel):
|
|||
|
|
total: int = Field(..., description="总记录数")
|
|||
|
|
models: List[ModelResponse] = Field(..., description="当前页模型列表")
|
|||
|
|
|
|||
|
|
model_config = {"from_attributes": True}
|