agent-platform/services/api-gateway/app/schemas/gateway.py

from pydantic import BaseModel
from datetime import datetime

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from app.db.models import GatewayRequestAudit


class DownstreamServiceHealth(BaseModel):
    service: str
    status: str
    url: str
    status_code: int | None = None
    error_message: str | None = None


class GatewayServicesHealthResponse(BaseModel):
    service: str = "api-gateway"
    status: str
    downstream_services: list[DownstreamServiceHealth]


class GatewayRequestAuditResponse(BaseModel):
    id: str
    tenant_id: str
    request_id: str
    method: str
    path: str
    query_string: str | None = None
    target_service: str | None = None
    target_url: str | None = None
    status_code: int | None = None
    duration_ms: int
    client_host: str | None = None
    user_agent: str | None = None
    error_message: str | None = None
    created_time: datetime

    @classmethod
    def from_entity(cls, entity: "GatewayRequestAudit") -> "GatewayRequestAuditResponse":
        return cls.model_validate(entity, from_attributes=True)