69 lines
1.5 KiB
Python
69 lines
1.5 KiB
Python
"""Pydantic request/response models for the platform SDK REST API."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class TaskConstraintsModel(BaseModel):
|
|
driver_type: str | None = None
|
|
capability_tags: list[str] = Field(default_factory=list)
|
|
|
|
|
|
class TaskSubmissionRequest(BaseModel):
|
|
goal: str | None = None
|
|
workflow_definition_id: str | None = None
|
|
constraints: TaskConstraintsModel = Field(default_factory=TaskConstraintsModel)
|
|
|
|
|
|
class TaskSubmissionResponse(BaseModel):
|
|
task_id: str
|
|
|
|
|
|
class TaskStatusResponse(BaseModel):
|
|
id: str
|
|
status: str
|
|
goal: str | None = None
|
|
workflow_definition_id: str | None = None
|
|
assigned_device_id: str | None = None
|
|
assigned_host_id: str | None = None
|
|
attempt_count: int = 0
|
|
lease_expires_at: datetime | None = None
|
|
failure_reason: str | None = None
|
|
|
|
|
|
class DeviceResponse(BaseModel):
|
|
device_id: str
|
|
host_id: str
|
|
driver_type: str
|
|
status: str
|
|
capability_tags: list[str] = Field(default_factory=list)
|
|
|
|
|
|
class HostResponse(BaseModel):
|
|
host_id: str
|
|
address: str | None = None
|
|
last_seen_at: str
|
|
|
|
|
|
class PluginRegistrationRequest(BaseModel):
|
|
name: str
|
|
version: str
|
|
entry_point_kind: Literal["driver", "tool", "skill"]
|
|
target: str
|
|
|
|
|
|
class PluginResponse(BaseModel):
|
|
name: str
|
|
version: str
|
|
entry_point_kind: str
|
|
target: str
|
|
wired: bool
|
|
|
|
|
|
class ErrorResponse(BaseModel):
|
|
detail: str
|