from typing import Optional, List

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 = None
    labels: Optional[List[int]] | None = None
    box: Optional[List[int]] | None = None