"""
Data models for the dashboard API.

Includes Pydantic schemas for devices, tasks, and WebSocket messages.
"""

from dashboard.models.device import (
    DeviceType,
    DeviceStatus,
    DeviceSchema,
    DeviceInfo,
)
from dashboard.models.task import (
    TaskStatus,
    TaskRequest,
    TaskSchema,
    TaskCreateRequest,
)
from dashboard.models.ws_messages import (
    WSMessageType,
    WSMessage,
    StepUpdate,
)

__all__ = [
    "DeviceType",
    "DeviceStatus",
    "DeviceSchema",
    "DeviceInfo",
    "TaskStatus",
    "TaskRequest",
    "TaskSchema",
    "TaskCreateRequest",
    "WSMessageType",
    "WSMessage",
    "StepUpdate",
]