AiDA_Python/app/schemas/design.py

from typing import List, Optional

from pydantic import BaseModel, Field


class SAMRequestModel(BaseModel):
    user_id: int = Field(..., description="用户id, 必填字段")
    image_path: str = Field(..., description="图片路径，必填字段")
    type: str = Field(..., description="推理类型，必填字段")
    points: Optional[List[List[float]]] | None = None
    labels: Optional[List[int]] | None = None
    box: Optional[List[int]] | None = 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