73 lines
4.0 KiB
Python
73 lines
4.0 KiB
Python
"""
|
||
Operator Schemas
|
||
算子 Schema 定义
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
from typing import List, Optional, Dict, Any
|
||
from datetime import datetime
|
||
from pydantic import BaseModel, Field
|
||
|
||
from app.module.shared.schema import BaseResponseModel
|
||
from .release import OperatorReleaseDto
|
||
|
||
|
||
class OperatorDto(BaseResponseModel):
|
||
"""算子 DTO"""
|
||
id: str = Field(..., description="算子ID")
|
||
name: str = Field(..., description="算子名称")
|
||
description: Optional[str] = Field(None, description="算子描述")
|
||
version: str = Field(..., description="算子版本")
|
||
inputs: Optional[str] = Field(None, description="输入定义（JSON）")
|
||
outputs: Optional[str] = Field(None, description="输出定义（JSON）")
|
||
runtime: Optional[str] = Field(None, description="运行时配置（JSON）")
|
||
settings: Optional[str] = Field(None, description="算子设置（JSON）")
|
||
file_name: Optional[str] = Field(None, description="文件名")
|
||
file_size: Optional[int] = Field(None, description="文件大小（字节）")
|
||
metrics: Optional[str] = Field(None, description="算子指标（JSON）")
|
||
usage_count: Optional[int] = Field(None, description="使用次数")
|
||
is_star: Optional[bool] = Field(None, description="是否收藏")
|
||
categories: Optional[List[str]] = Field(None, description="分类ID列表")
|
||
overrides: Optional[Dict[str, Any]] = Field(None, description="设置覆盖值")
|
||
requirements: Optional[List[str]] = Field(None, description="Python 依赖列表")
|
||
readme: Optional[str] = Field(None, description="README 内容")
|
||
releases: Optional[List[OperatorReleaseDto]] = Field(None, description="发布版本列表")
|
||
created_at: Optional[datetime] = Field(None, description="创建时间")
|
||
updated_at: Optional[datetime] = Field(None, description="更新时间")
|
||
|
||
|
||
class OperatorListRequest(BaseResponseModel):
|
||
"""算子列表查询请求"""
|
||
page: int = Field(1, ge=0, description="页码（从0开始）")
|
||
size: int = Field(10, ge=1, description="页大小")
|
||
categories: List[List[str]] = Field(default_factory=list, description="分类ID列表（每个父分类下的id放到一个列表中）")
|
||
keyword: Optional[str] = Field(None, description="搜索关键词")
|
||
label_name: Optional[str] = Field(None, description="标签名称（暂不支持）")
|
||
is_star: Optional[bool] = Field(None, description="是否收藏")
|
||
|
||
|
||
class PreUploadResponse(BaseResponseModel):
|
||
"""预上传响应"""
|
||
req_id: str = Field(..., description="请求ID")
|
||
|
||
|
||
class OperatorUpdateDto(BaseResponseModel):
|
||
"""算子更新 DTO（所有字段可选）"""
|
||
name: Optional[str] = Field(None, description="算子名称")
|
||
description: Optional[str] = Field(None, description="算子描述")
|
||
version: Optional[str] = Field(None, description="算子版本")
|
||
inputs: Optional[str] = Field(None, description="输入定义（JSON）")
|
||
outputs: Optional[str] = Field(None, description="输出定义（JSON）")
|
||
runtime: Optional[str] = Field(None, description="运行时配置（JSON）")
|
||
settings: Optional[str] = Field(None, description="算子设置（JSON）")
|
||
file_name: Optional[str] = Field(None, description="文件名")
|
||
file_size: Optional[int] = Field(None, description="文件大小（字节）")
|
||
metrics: Optional[str] = Field(None, description="算子指标（JSON）")
|
||
usage_count: Optional[int] = Field(None, description="使用次数")
|
||
is_star: Optional[bool] = Field(None, description="是否收藏")
|
||
categories: Optional[List[str]] = Field(None, description="分类ID列表")
|
||
overrides: Optional[Dict[str, Any]] = Field(None, description="设置覆盖值")
|
||
requirements: Optional[List[str]] = Field(None, description="Python 依赖列表")
|
||
readme: Optional[str] = Field(None, description="README 内容")
|
||
releases: Optional[List[OperatorReleaseDto]] = Field(None, description="发布版本列表")
|