"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""

from __future__ import annotations
from openrouter.types import BaseModel, Nullable, UNSET_SENTINEL, UnrecognizedStr
from openrouter.utils import validate_open_enum
from pydantic import model_serializer
from pydantic.functional_validators import PlainValidator
from typing import Literal, Optional, Union
from typing_extensions import Annotated, NotRequired, TypedDict


ProviderResponseProviderName = Union[
    Literal[
        "AnyScale",
        "Atoma",
        "Cent-ML",
        "CrofAI",
        "Enfer",
        "GoPomelo",
        "HuggingFace",
        "Hyperbolic",
        "Hyperbolic 2",
        "InoCloud",
        "Kluster",
        "Lambda",
        "Lepton",
        "Lynn 2",
        "Lynn",
        "Mancer",
        "Meta",
        "Modal",
        "Nineteen",
        "OctoAI",
        "Recursal",
        "Reflection",
        "Replicate",
        "SambaNova 2",
        "SF Compute",
        "Targon",
        "Together 2",
        "Ubicloud",
        "01.AI",
        "AkashML",
        "AI21",
        "AionLabs",
        "Alibaba",
        "Ambient",
        "Baidu",
        "Amazon Bedrock",
        "Amazon Nova",
        "Anthropic",
        "Arcee AI",
        "AtlasCloud",
        "Avian",
        "Azure",
        "BaseTen",
        "BytePlus",
        "Black Forest Labs",
        "Cerebras",
        "Chutes",
        "Cirrascale",
        "Clarifai",
        "Cloudflare",
        "Cohere",
        "Crucible",
        "Crusoe",
        "Darkbloom",
        "Decart",
        "DeepInfra",
        "DeepSeek",
        "DekaLLM",
        "DigitalOcean",
        "Featherless",
        "Fireworks",
        "Friendli",
        "GMICloud",
        "Google",
        "Google AI Studio",
        "Groq",
        "Inception",
        "Inceptron",
        "InferenceNet",
        "Ionstream",
        "Infermatic",
        "Io Net",
        "Inflection",
        "Liquid",
        "Mara",
        "Mancer 2",
        "Minimax",
        "ModelRun",
        "Mistral",
        "Modular",
        "Moonshot AI",
        "Morph",
        "NCompass",
        "Nebius",
        "Nex AGI",
        "NextBit",
        "Novita",
        "Nvidia",
        "OpenAI",
        "OpenInference",
        "Parasail",
        "Poolside",
        "Perceptron",
        "Perplexity",
        "Phala",
        "Recraft",
        "Reka",
        "Relace",
        "SambaNova",
        "Seed",
        "SiliconFlow",
        "Sourceful",
        "StepFun",
        "Stealth",
        "StreamLake",
        "Switchpoint",
        "Together",
        "Upstage",
        "Venice",
        "Wafer",
        "WandB",
        "Xiaomi",
        "xAI",
        "Z.AI",
        "FakeProvider",
    ],
    UnrecognizedStr,
]
r"""Name of the provider"""


class ProviderResponseTypedDict(TypedDict):
    r"""Details of a provider response for a generation attempt"""

    status: Nullable[int]
    r"""HTTP status code from the provider"""
    endpoint_id: NotRequired[str]
    r"""Internal endpoint identifier"""
    id: NotRequired[str]
    r"""Upstream provider response identifier"""
    is_byok: NotRequired[bool]
    r"""Whether the request used a bring-your-own-key"""
    latency: NotRequired[float]
    r"""Response latency in milliseconds"""
    model_permaslug: NotRequired[str]
    r"""Canonical model slug"""
    provider_name: NotRequired[ProviderResponseProviderName]
    r"""Name of the provider"""


class ProviderResponse(BaseModel):
    r"""Details of a provider response for a generation attempt"""

    status: Nullable[int]
    r"""HTTP status code from the provider"""

    endpoint_id: Optional[str] = None
    r"""Internal endpoint identifier"""

    id: Optional[str] = None
    r"""Upstream provider response identifier"""

    is_byok: Optional[bool] = None
    r"""Whether the request used a bring-your-own-key"""

    latency: Optional[float] = None
    r"""Response latency in milliseconds"""

    model_permaslug: Optional[str] = None
    r"""Canonical model slug"""

    provider_name: Annotated[
        Optional[ProviderResponseProviderName],
        PlainValidator(validate_open_enum(False)),
    ] = None
    r"""Name of the provider"""

    @model_serializer(mode="wrap")
    def serialize_model(self, handler):
        optional_fields = [
            "endpoint_id",
            "id",
            "is_byok",
            "latency",
            "model_permaslug",
            "provider_name",
        ]
        nullable_fields = ["status"]
        null_default_fields = []

        serialized = handler(self)

        m = {}

        for n, f in type(self).model_fields.items():
            k = f.alias or n
            val = serialized.get(k)
            serialized.pop(k, None)

            optional_nullable = k in optional_fields and k in nullable_fields
            is_set = (
                self.__pydantic_fields_set__.intersection({n})
                or k in null_default_fields
            )  # pylint: disable=no-member

            if val is not None and val != UNSET_SENTINEL:
                m[k] = val
            elif val != UNSET_SENTINEL and (
                not k in optional_fields or (optional_nullable and is_set)
            ):
                m[k] = val

        return m
