from typing import Optional, List

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