from __future__ import annotations

from typing import Any, Dict, Literal

from pydantic import BaseModel


class ServiceHealth(BaseModel):
    status: Literal["healthy", "unhealthy"]
    details: Dict[str, Any]


class InfraHealthResponse(BaseModel):
    status: Literal["healthy", "unhealthy"]
    services: Dict[str, ServiceHealth]
