AiDA_Python/app/schemas/design.py

from typing import List, Optional

from pydantic import BaseModel, Field


class SAMRequestModel(BaseModel):
    bucket: str = Field(..., description="minio bucket name ")
    object_name: str = Field(..., description="minio object name ")
    image_path: str = Field(..., description="图片路径，必填字段")
    type: str = Field(..., description="推理类型，必填字段")
    points: Optional[List[List[float]]] = None
    labels: Optional[List[int]] = None
    box: Optional[List[int]] = None


class DesignModel(BaseModel):
    objects: list[dict]
    process_id: str


class DesignStreamModel(BaseModel):
    objects: list[dict]
    process_id: str
    requestId: str
    callback_url: str


class DesignProgressModel(BaseModel):
    process_id: str


class ModelProgressModel(BaseModel):
    model_path: str


class DBGConfigModel(BaseModel):
    tasks_id: str
    user_id: str
    file_name: str
    total: int