Commits included in this export:

- 68922847c2969b773887310eb7aa9a814e199c5a
  - 729ee2338c39df22461526153e7f4ba868fd498e
  - e1074ff86fccf5e48406b7595298873106d3cc5a
  - 123bbdaf841d299162a82ee2573adfbb70a45b8f
  - ac457b8a8dad98fdf5ddcae8dd4711361cea3084
  - ea825199f4c59dd68fec01cf8d18a582963c9f25
  - 4d48e4f6bb936e785787314a965298da6b78fea0
  - 718729a573ee98893a4408d5a52ab4686661fe54
  - b867adcde6b64f3f2a11c4626254b39098e4b0b4
  - 21912c097b4066d109766c6b4492826595cd57c5
  - bc1ed85adc976a4634ca8a41b82e4fc2c39a864f
  - d145c89c941937a312ed1357590d2bfda9a3603d
  - 12f05b5dee397ead47cfcd545a7a2e3c7ff57524
  - dd28f75a19a015809797c7abf3f8758f7c37262e
  - c96e569781e07ab44172d9c69940ec17bafd81f4
  - c9d5a90e3c2b5077a2bdd49773f1a46d1a5a79c7
  - 58b08f87116ceab9076de2fdc0a4c74c919b44c4

GitOrigin-RevId: 68922847c2969b773887310eb7aa9a814e199c5a

src/openrouter/operations/getgeneration.py

@@ -5,8 +5,8 @@ from openrouter.types import BaseModel, Nullable, UNSET_SENTINEL, UnrecognizedSt
 from openrouter.utils import FieldMetadata, QueryParamMetadata, validate_open_enum
 from pydantic import model_serializer
 from pydantic.functional_validators import PlainValidator
-from typing import Literal, Union
-from typing_extensions import Annotated, TypedDict
+from typing import List, Literal, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class GetGenerationRequestTypedDict(TypedDict):
@@ -29,6 +29,178 @@ APIType = Union[
 r"""Type of API used for the generation"""
 
 
+ProviderName = 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",
+        "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",
+        "Featherless",
+        "Fireworks",
+        "Friendli",
+        "GMICloud",
+        "Google",
+        "Google AI Studio",
+        "Groq",
+        "Hyperbolic",
+        "Inception",
+        "Inceptron",
+        "InferenceNet",
+        "Infermatic",
+        "Inflection",
+        "Liquid",
+        "Mara",
+        "Mancer 2",
+        "Minimax",
+        "ModelRun",
+        "Mistral",
+        "Modular",
+        "Moonshot AI",
+        "Morph",
+        "NCompass",
+        "Nebius",
+        "NextBit",
+        "Novita",
+        "Nvidia",
+        "OpenAI",
+        "OpenInference",
+        "Parasail",
+        "Perplexity",
+        "Phala",
+        "Relace",
+        "SambaNova",
+        "Seed",
+        "SiliconFlow",
+        "Sourceful",
+        "StepFun",
+        "Stealth",
+        "StreamLake",
+        "Switchpoint",
+        "Together",
+        "Upstage",
+        "Venice",
+        "WandB",
+        "Xiaomi",
+        "xAI",
+        "Z.AI",
+        "FakeProvider",
+    ],
+    UnrecognizedStr,
+]
+
+
+class ProviderResponseTypedDict(TypedDict):
+    status: Nullable[float]
+    id: NotRequired[str]
+    endpoint_id: NotRequired[str]
+    model_permaslug: NotRequired[str]
+    provider_name: NotRequired[ProviderName]
+    latency: NotRequired[float]
+    is_byok: NotRequired[bool]
+
+
+class ProviderResponse(BaseModel):
+    status: Nullable[float]
+
+    id: Optional[str] = None
+
+    endpoint_id: Optional[str] = None
+
+    model_permaslug: Optional[str] = None
+
+    provider_name: Annotated[
+        Optional[ProviderName], PlainValidator(validate_open_enum(False))
+    ] = None
+
+    latency: Optional[float] = None
+
+    is_byok: Optional[bool] = None
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "id",
+            "endpoint_id",
+            "model_permaslug",
+            "provider_name",
+            "latency",
+            "is_byok",
+        ]
+        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
+
+
 class GetGenerationDataTypedDict(TypedDict):
     r"""Generation data"""
 
@@ -98,6 +270,8 @@ class GetGenerationDataTypedDict(TypedDict):
     r"""Type of API used for the generation"""
     router: Nullable[str]
     r"""Router used for the request (e.g., openrouter/auto)"""
+    provider_responses: Nullable[List[ProviderResponseTypedDict]]
+    r"""List of provider responses for this generation, including fallback attempts"""
 
 
 class GetGenerationData(BaseModel):
@@ -202,6 +376,9 @@ class GetGenerationData(BaseModel):
     router: Nullable[str]
     r"""Router used for the request (e.g., openrouter/auto)"""
 
+    provider_responses: Nullable[List[ProviderResponse]]
+    r"""List of provider responses for this generation, including fallback attempts"""
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = []
@@ -232,6 +409,7 @@ class GetGenerationData(BaseModel):
             "external_user",
             "api_type",
             "router",
+            "provider_responses",
         ]
         null_default_fields = []