"""Pydantic schemas for API request/response validation."""

from app.schemas.common import HealthResponse
from app.schemas.tenant import TenantCreate, TenantResponse
from app.schemas.task import (
    TaskCreate,
    TaskResponse,
    TaskUpdate,
)
from app.schemas.agent import (
    AgentRegisterRequest,
    AgentRegisterResponse,
    AgentHeartbeatResponse,
    AgentResponse,
)
from app.schemas.tasks_extended import (
    FileWritePayload,
    EnvUpdatePayload,
    DockerReloadPayload,
    CompositeSubTask,
    CompositePayload,
)

__all__ = [
    # Common
    "HealthResponse",
    # Tenant
    "TenantCreate",
    "TenantResponse",
    # Task
    "TaskCreate",
    "TaskResponse",
    "TaskUpdate",
    # Task Payloads (for documentation/reference)
    "FileWritePayload",
    "EnvUpdatePayload",
    "DockerReloadPayload",
    "CompositeSubTask",
    "CompositePayload",
    # Agent
    "AgentRegisterRequest",
    "AgentRegisterResponse",
    "AgentHeartbeatResponse",
    "AgentResponse",
]