zcr
7e30779aec
All checks were successful
git commit AiDA python develop 分支构建部署 / scheduled_deploy (push) Has been skipped
40 lines
867 B
Python
40 lines
867 B
Python
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
|
|
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
|