Add local model asset catalog

2026-07-06 20:59:03 +02:00
parent 9e20cc82ae
commit 6e2a8cbdc4
33 changed files with 660 additions and 7 deletions
@@ -22,10 +22,33 @@ class DetectionModelsResponse(BaseModel):
    models: list[DetectionModelCapability]


+class ModelAssetRead(BaseModel):
+    model_asset_id: str
+    filename: str
+    display_name: str
+    model_path: str
+    suffix: str
+    framework: str
+    task_type: str
+    size_bytes: int
+    sha256: str
+    active: bool
+    status: str
+    limitation_message: str
+    will_download_models: bool = False
+
+
+class ModelAssetListResponse(BaseModel):
+    items: list[ModelAssetRead]
+    total: int
+    model_directory: str
+
+
 class DetectionRunRequest(BaseModel):
    project_id: UUID
    dataset_id: UUID
    model_id: str
+    model_asset_id: str | None = None
    confidence_threshold: float = Field(default=0.5, ge=0.0, le=1.0)
    class_filter: list[str] | None = None
    tile_manifest_path: str | None = None