"""Structured HTTP response models."""

from __future__ import annotations
from typing import Literal

from pydantic import BaseModel


class ErrorDetail(BaseModel):
    message: str
    type: str = "server_error"


class ErrorResponse(BaseModel):
    error: ErrorDetail


class GeminiUsageMetadata(BaseModel):
    promptTokenCount: int = 0
    candidatesTokenCount: int = 0
    thoughtsTokenCount: int = 0
    totalTokenCount: int = 0


class GeminiFunctionCallPayload(BaseModel):
    name: str
    args: object | None = None


class GeminiFunctionResponsePayload(BaseModel):
    name: str
    response: object | None = None


class GeminiInlineDataResponse(BaseModel):
    mimeType: str
    data: str


class GeminiPartResponse(BaseModel):
    text: str | None = None
    thought: bool | None = None
    inlineData: GeminiInlineDataResponse | None = None
    thoughtSignature: str | None = None
    functionCall: GeminiFunctionCallPayload | None = None
    functionResponse: GeminiFunctionResponsePayload | None = None


class GeminiContentResponse(BaseModel):
    role: Literal["model"] = "model"
    parts: list[GeminiPartResponse]


class GeminiCandidateResponse(BaseModel):
    content: GeminiContentResponse
    finishReason: str | None = None


class GeminiGenerateContentResponse(BaseModel):
    candidates: list[GeminiCandidateResponse]
    usageMetadata: GeminiUsageMetadata | None = None


class HealthResponse(BaseModel):
    status: str
    busy: bool


class ModelStatsResponse(BaseModel):
    requests: int
    success: int
    rate_limited: int
    errors: int
    prompt_tokens: int
    completion_tokens: int
    total_tokens: int
    last_used: str | None = None


class StatsTotalsResponse(BaseModel):
    requests: int
    success: int
    rate_limited: int
    errors: int
    prompt_tokens: int
    completion_tokens: int
    total_tokens: int


class StatsResponse(BaseModel):
    models: dict[str, ModelStatsResponse]
    totals: StatsTotalsResponse


class GeminiModelResponse(BaseModel):
    name: str
    version: str | None = "001"
    displayName: str | None = None
    description: str | None = None
    inputTokenLimit: int | None = None
    outputTokenLimit: int | None = None
    supportedGenerationMethods: list[str] = [
        "generateContent",
        "countTokens",
        "createCachedContent",
    ]
    temperature: float | None = None
    topP: float | None = None
    topK: int | None = None


class GeminiModelListResponse(BaseModel):
    models: list[GeminiModelResponse]