fix: add overlay to remove nullable from pagination offset params (#121)

2026-07-29 11:23:49 +08:00 · 2026-04-14 12:48:15 -04:00
parent 2bba049182
commit b2386114cd
440 changed files with 36150 additions and 32168 deletions
@@ -0,0 +1,207 @@
+"""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 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",
+        "Amazon Bedrock",
+        "Amazon Nova",
+        "Anthropic",
+        "Arcee AI",
+        "AtlasCloud",
+        "Avian",
+        "Azure",
+        "BaseTen",
+        "BytePlus",
+        "Black Forest Labs",
+        "Cerebras",
+        "Chutes",
+        "Cirrascale",
+        "Clarifai",
+        "Cloudflare",
+        "Cohere",
+        "Crusoe",
+        "DeepInfra",
+        "DeepSeek",
+        "DekaLLM",
+        "Featherless",
+        "Fireworks",
+        "Friendli",
+        "GMICloud",
+        "Google",
+        "Google AI Studio",
+        "Groq",
+        "Hyperbolic",
+        "Inception",
+        "Inceptron",
+        "InferenceNet",
+        "Ionstream",
+        "Infermatic",
+        "Io Net",
+        "Inflection",
+        "Liquid",
+        "Mara",
+        "Mancer 2",
+        "Minimax",
+        "ModelRun",
+        "Mistral",
+        "Modular",
+        "Moonshot AI",
+        "Morph",
+        "NCompass",
+        "Nebius",
+        "NextBit",
+        "Novita",
+        "Nvidia",
+        "OpenAI",
+        "OpenInference",
+        "Parasail",
+        "Perplexity",
+        "Phala",
+        "Recraft",
+        "Reka",
+        "Relace",
+        "SambaNova",
+        "Seed",
+        "SiliconFlow",
+        "Sourceful",
+        "StepFun",
+        "Stealth",
+        "StreamLake",
+        "Switchpoint",
+        "Together",
+        "Upstage",
+        "Venice",
+        "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[float]
+    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[float]
+    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