41 lines
951 B
Python
41 lines
951 B
Python
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 = 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
|