from pydantic import BaseModel

from app.dsp.enums import DSPClientStatus, DSPId
from app.dsp.stats import RequestStats


class DSPClient(BaseModel):
    id: int
    name: str
    display_name: str
    client_id: str
    status: DSPClientStatus
    configured: bool
    stats: RequestStats | None
    nominal_rps: int | None


class DSP(BaseModel):
    id: DSPId
    clients: list[DSPClient]