openrouter 0.0.17__py3-none-any.whl → 0.0.19__py3-none-any.whl

openrouter/components/chatgenerationparams.py

@@ -1,10 +1,8 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from ._schema0 import Schema0, Schema0TypedDict
 from .chatstreamoptions import ChatStreamOptions, ChatStreamOptionsTypedDict
 from .message import Message, MessageTypedDict
-from .providersortunion import ProviderSortUnion, ProviderSortUnionTypedDict
 from .reasoningsummaryverbosity import ReasoningSummaryVerbosity
 from .responseformatjsonschema import (
     ResponseFormatJSONSchema,
@@ -23,353 +21,21 @@ from openrouter.types import (
     UNSET_SENTINEL,
     UnrecognizedStr,
 )
-from openrouter.utils import get_discriminator, validate_const, validate_open_enum
+from openrouter.utils import validate_const, validate_open_enum
 import pydantic
-from pydantic import Discriminator, Tag, model_serializer
+from pydantic import model_serializer
 from pydantic.functional_validators import AfterValidator, PlainValidator
 from typing import Any, Dict, List, Literal, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-ChatGenerationParamsDataCollection = Union[
-    Literal[
-        "deny",
-        "allow",
-    ],
-    UnrecognizedStr,
-]
-
-
-Quantizations = Union[
-    Literal[
-        "int4",
-        "int8",
-        "fp4",
-        "fp6",
-        "fp8",
-        "fp16",
-        "bf16",
-        "fp32",
-        "unknown",
-    ],
-    UnrecognizedStr,
-]
-
-
-class ChatGenerationParamsMaxPriceTypedDict(TypedDict):
-    r"""The object specifying the maximum price you want to pay for this request. USD price per million tokens, for prompt and completion."""
-
-    prompt: NotRequired[Any]
-    completion: NotRequired[Any]
-    image: NotRequired[Any]
-    audio: NotRequired[Any]
-    request: NotRequired[Any]
-
-
-class ChatGenerationParamsMaxPrice(BaseModel):
-    r"""The object specifying the maximum price you want to pay for this request. USD price per million tokens, for prompt and completion."""
-
-    prompt: Optional[Any] = None
-
-    completion: Optional[Any] = None
-
-    image: Optional[Any] = None
-
-    audio: Optional[Any] = None
-
-    request: Optional[Any] = None
-
-
-class ChatGenerationParamsProviderTypedDict(TypedDict):
-    allow_fallbacks: NotRequired[Nullable[bool]]
-    r"""Whether to allow backup providers to serve requests
-    - true: (default) when the primary provider (or your custom providers in \"order\") is unavailable, use the next best provider.
-    - false: use only the primary/custom provider, and return the upstream error if it's unavailable.
-
-    """
-    require_parameters: NotRequired[Nullable[bool]]
-    r"""Whether to filter providers to only those that support the parameters you've provided. If this setting is omitted or set to false, then providers will receive only the parameters they support, and ignore the rest."""
-    data_collection: NotRequired[Nullable[ChatGenerationParamsDataCollection]]
-    r"""Data collection setting. If no available model provider meets the requirement, your request will return an error.
-    - allow: (default) allow providers which store user data non-transiently and may train on it
-
-    - deny: use only providers which do not collect user data.
-    """
-    zdr: NotRequired[Nullable[bool]]
-    enforce_distillable_text: NotRequired[Nullable[bool]]
-    order: NotRequired[Nullable[List[Schema0TypedDict]]]
-    r"""An ordered list of provider slugs. The router will attempt to use the first provider in the subset of this list that supports your requested model, and fall back to the next if it is unavailable. If no providers are available, the request will fail with an error message."""
-    only: NotRequired[Nullable[List[Schema0TypedDict]]]
-    r"""List of provider slugs to allow. If provided, this list is merged with your account-wide allowed provider settings for this request."""
-    ignore: NotRequired[Nullable[List[Schema0TypedDict]]]
-    r"""List of provider slugs to ignore. If provided, this list is merged with your account-wide ignored provider settings for this request."""
-    quantizations: NotRequired[Nullable[List[Quantizations]]]
-    r"""A list of quantization levels to filter the provider by."""
-    sort: NotRequired[Nullable[ProviderSortUnionTypedDict]]
-    r"""The sorting strategy to use for this request, if \"order\" is not specified. When set, no load balancing is performed."""
-    max_price: NotRequired[ChatGenerationParamsMaxPriceTypedDict]
-    r"""The object specifying the maximum price you want to pay for this request. USD price per million tokens, for prompt and completion."""
-    preferred_min_throughput: NotRequired[Nullable[float]]
-    preferred_max_latency: NotRequired[Nullable[float]]
-    min_throughput: NotRequired[Nullable[float]]
-    max_latency: NotRequired[Nullable[float]]
-
-
-class ChatGenerationParamsProvider(BaseModel):
-    allow_fallbacks: OptionalNullable[bool] = UNSET
-    r"""Whether to allow backup providers to serve requests
-    - true: (default) when the primary provider (or your custom providers in \"order\") is unavailable, use the next best provider.
-    - false: use only the primary/custom provider, and return the upstream error if it's unavailable.
-
-    """
-
-    require_parameters: OptionalNullable[bool] = UNSET
-    r"""Whether to filter providers to only those that support the parameters you've provided. If this setting is omitted or set to false, then providers will receive only the parameters they support, and ignore the rest."""
-
-    data_collection: Annotated[
-        OptionalNullable[ChatGenerationParamsDataCollection],
-        PlainValidator(validate_open_enum(False)),
-    ] = UNSET
-    r"""Data collection setting. If no available model provider meets the requirement, your request will return an error.
-    - allow: (default) allow providers which store user data non-transiently and may train on it
-
-    - deny: use only providers which do not collect user data.
-    """
-
-    zdr: OptionalNullable[bool] = UNSET
-
-    enforce_distillable_text: OptionalNullable[bool] = UNSET
-
-    order: OptionalNullable[List[Schema0]] = UNSET
-    r"""An ordered list of provider slugs. The router will attempt to use the first provider in the subset of this list that supports your requested model, and fall back to the next if it is unavailable. If no providers are available, the request will fail with an error message."""
-
-    only: OptionalNullable[List[Schema0]] = UNSET
-    r"""List of provider slugs to allow. If provided, this list is merged with your account-wide allowed provider settings for this request."""
-
-    ignore: OptionalNullable[List[Schema0]] = UNSET
-    r"""List of provider slugs to ignore. If provided, this list is merged with your account-wide ignored provider settings for this request."""
-
-    quantizations: OptionalNullable[
-        List[Annotated[Quantizations, PlainValidator(validate_open_enum(False))]]
-    ] = UNSET
-    r"""A list of quantization levels to filter the provider by."""
-
-    sort: OptionalNullable[ProviderSortUnion] = UNSET
-    r"""The sorting strategy to use for this request, if \"order\" is not specified. When set, no load balancing is performed."""
-
-    max_price: Optional[ChatGenerationParamsMaxPrice] = None
-    r"""The object specifying the maximum price you want to pay for this request. USD price per million tokens, for prompt and completion."""
-
-    preferred_min_throughput: OptionalNullable[float] = UNSET
-
-    preferred_max_latency: OptionalNullable[float] = UNSET
-
-    min_throughput: OptionalNullable[float] = UNSET
-
-    max_latency: OptionalNullable[float] = UNSET
-
-    @model_serializer(mode="wrap")
-    def serialize_model(self, handler):
-        optional_fields = [
-            "allow_fallbacks",
-            "require_parameters",
-            "data_collection",
-            "zdr",
-            "enforce_distillable_text",
-            "order",
-            "only",
-            "ignore",
-            "quantizations",
-            "sort",
-            "max_price",
-            "preferred_min_throughput",
-            "preferred_max_latency",
-            "min_throughput",
-            "max_latency",
-        ]
-        nullable_fields = [
-            "allow_fallbacks",
-            "require_parameters",
-            "data_collection",
-            "zdr",
-            "enforce_distillable_text",
-            "order",
-            "only",
-            "ignore",
-            "quantizations",
-            "sort",
-            "preferred_min_throughput",
-            "preferred_max_latency",
-            "min_throughput",
-            "max_latency",
-        ]
-        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 ChatGenerationParamsPluginResponseHealingTypedDict(TypedDict):
-    id: Literal["response-healing"]
-    enabled: NotRequired[bool]
-
-
-class ChatGenerationParamsPluginResponseHealing(BaseModel):
-    ID: Annotated[
-        Annotated[
-            Literal["response-healing"],
-            AfterValidator(validate_const("response-healing")),
-        ],
-        pydantic.Field(alias="id"),
-    ] = "response-healing"
-
-    enabled: Optional[bool] = None
-
-
-PdfEngine = Union[
-    Literal[
-        "mistral-ocr",
-        "pdf-text",
-        "native",
-    ],
-    UnrecognizedStr,
-]
-
-
-class PdfTypedDict(TypedDict):
-    engine: NotRequired[PdfEngine]
-
-
-class Pdf(BaseModel):
-    engine: Annotated[
-        Optional[PdfEngine], PlainValidator(validate_open_enum(False))
-    ] = None
-
-
-class ChatGenerationParamsPluginFileParserTypedDict(TypedDict):
-    id: Literal["file-parser"]
-    enabled: NotRequired[bool]
-    pdf: NotRequired[PdfTypedDict]
-
-
-class ChatGenerationParamsPluginFileParser(BaseModel):
-    ID: Annotated[
-        Annotated[
-            Literal["file-parser"], AfterValidator(validate_const("file-parser"))
-        ],
-        pydantic.Field(alias="id"),
-    ] = "file-parser"
-
-    enabled: Optional[bool] = None
-
-    pdf: Optional[Pdf] = None
-
-
-Engine = Union[
-    Literal[
-        "native",
-        "exa",
-    ],
-    UnrecognizedStr,
-]
-
-
-class ChatGenerationParamsPluginWebTypedDict(TypedDict):
-    id: Literal["web"]
-    enabled: NotRequired[bool]
-    max_results: NotRequired[float]
-    search_prompt: NotRequired[str]
-    engine: NotRequired[Engine]
-
-
-class ChatGenerationParamsPluginWeb(BaseModel):
-    ID: Annotated[
-        Annotated[Literal["web"], AfterValidator(validate_const("web"))],
-        pydantic.Field(alias="id"),
-    ] = "web"
-
-    enabled: Optional[bool] = None
-
-    max_results: Optional[float] = None
-
-    search_prompt: Optional[str] = None
-
-    engine: Annotated[Optional[Engine], PlainValidator(validate_open_enum(False))] = (
-        None
-    )
-
-
-class ChatGenerationParamsPluginModerationTypedDict(TypedDict):
-    id: Literal["moderation"]
-
-
-class ChatGenerationParamsPluginModeration(BaseModel):
-    ID: Annotated[
-        Annotated[Literal["moderation"], AfterValidator(validate_const("moderation"))],
-        pydantic.Field(alias="id"),
-    ] = "moderation"
-
-
-ChatGenerationParamsPluginUnionTypedDict = TypeAliasType(
-    "ChatGenerationParamsPluginUnionTypedDict",
-    Union[
-        ChatGenerationParamsPluginModerationTypedDict,
-        ChatGenerationParamsPluginResponseHealingTypedDict,
-        ChatGenerationParamsPluginFileParserTypedDict,
-        ChatGenerationParamsPluginWebTypedDict,
-    ],
-)
-
-
-ChatGenerationParamsPluginUnion = Annotated[
-    Union[
-        Annotated[ChatGenerationParamsPluginModeration, Tag("moderation")],
-        Annotated[ChatGenerationParamsPluginWeb, Tag("web")],
-        Annotated[ChatGenerationParamsPluginFileParser, Tag("file-parser")],
-        Annotated[ChatGenerationParamsPluginResponseHealing, Tag("response-healing")],
-    ],
-    Discriminator(lambda m: get_discriminator(m, "id", "id")),
-]
-
-
-Route = Union[
-    Literal[
-        "fallback",
-        "sort",
-    ],
-    UnrecognizedStr,
-]
-
-
 Effort = Union[
     Literal[
-        "xhigh",
-        "high",
-        "medium",
-        "low",
-        "minimal",
         "none",
+        "minimal",
+        "low",
+        "medium",
+        "high",
     ],
     UnrecognizedStr,
 ]
@@ -468,16 +134,16 @@ ChatGenerationParamsResponseFormatUnionTypedDict = TypeAliasType(
 )
 
 
-ChatGenerationParamsResponseFormatUnion = Annotated[
+ChatGenerationParamsResponseFormatUnion = TypeAliasType(
+    "ChatGenerationParamsResponseFormatUnion",
     Union[
-        Annotated[ChatGenerationParamsResponseFormatText, Tag("text")],
-        Annotated[ChatGenerationParamsResponseFormatJSONObject, Tag("json_object")],
-        Annotated[ResponseFormatJSONSchema, Tag("json_schema")],
-        Annotated[ResponseFormatTextGrammar, Tag("grammar")],
-        Annotated[ChatGenerationParamsResponseFormatPython, Tag("python")],
+        ChatGenerationParamsResponseFormatText,
+        ChatGenerationParamsResponseFormatJSONObject,
+        ChatGenerationParamsResponseFormatPython,
+        ResponseFormatJSONSchema,
+        ResponseFormatTextGrammar,
     ],
-    Discriminator(lambda m: get_discriminator(m, "type", "type")),
-]
+)
 
 
 ChatGenerationParamsStopTypedDict = TypeAliasType(
@@ -490,24 +156,8 @@ ChatGenerationParamsStop = TypeAliasType(
 )
 
 
-class DebugTypedDict(TypedDict):
-    echo_upstream_body: NotRequired[bool]
-
-
-class Debug(BaseModel):
-    echo_upstream_body: Optional[bool] = None
-
-
 class ChatGenerationParamsTypedDict(TypedDict):
     messages: List[MessageTypedDict]
-    provider: NotRequired[Nullable[ChatGenerationParamsProviderTypedDict]]
-    r"""When multiple model providers are available, optionally indicate your routing preference."""
-    plugins: NotRequired[List[ChatGenerationParamsPluginUnionTypedDict]]
-    r"""Plugins you want to enable for this request, including their settings."""
-    route: NotRequired[Nullable[Route]]
-    user: NotRequired[str]
-    session_id: NotRequired[str]
-    r"""A unique identifier for grouping related requests (e.g., a conversation or agent workflow) for observability. If provided in both the request body and the x-session-id header, the body value takes precedence. Maximum of 128 characters."""
     model: NotRequired[str]
     models: NotRequired[List[str]]
     frequency_penalty: NotRequired[Nullable[float]]
@@ -528,27 +178,12 @@ class ChatGenerationParamsTypedDict(TypedDict):
     tool_choice: NotRequired[Any]
     tools: NotRequired[List[ToolDefinitionJSONTypedDict]]
     top_p: NotRequired[Nullable[float]]
-    debug: NotRequired[DebugTypedDict]
+    user: NotRequired[str]
 
 
 class ChatGenerationParams(BaseModel):
     messages: List[Message]
 
-    provider: OptionalNullable[ChatGenerationParamsProvider] = UNSET
-    r"""When multiple model providers are available, optionally indicate your routing preference."""
-
-    plugins: Optional[List[ChatGenerationParamsPluginUnion]] = None
-    r"""Plugins you want to enable for this request, including their settings."""
-
-    route: Annotated[
-        OptionalNullable[Route], PlainValidator(validate_open_enum(False))
-    ] = UNSET
-
-    user: Optional[str] = None
-
-    session_id: Optional[str] = None
-    r"""A unique identifier for grouping related requests (e.g., a conversation or agent workflow) for observability. If provided in both the request body and the x-session-id header, the body value takes precedence. Maximum of 128 characters."""
-
     model: Optional[str] = None
 
     models: Optional[List[str]] = None
@@ -589,16 +224,11 @@ class ChatGenerationParams(BaseModel):
 
     top_p: OptionalNullable[float] = UNSET
 
-    debug: Optional[Debug] = None
+    user: Optional[str] = None
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = [
-            "provider",
-            "plugins",
-            "route",
-            "user",
-            "session_id",
             "model",
             "models",
             "frequency_penalty",
@@ -619,11 +249,9 @@ class ChatGenerationParams(BaseModel):
             "tool_choice",
             "tools",
             "top_p",
-            "debug",
+            "user",
         ]
         nullable_fields = [
-            "provider",
-            "route",
             "frequency_penalty",
             "logit_bias",
             "logprobs",

openrouter/components/chatmessagecontentitem.py

@@ -26,9 +26,9 @@ from typing_extensions import Annotated, TypeAliasType
 ChatMessageContentItemTypedDict = TypeAliasType(
     "ChatMessageContentItemTypedDict",
     Union[
+        ChatMessageContentItemTextTypedDict,
         ChatMessageContentItemImageTypedDict,
         ChatMessageContentItemAudioTypedDict,
-        ChatMessageContentItemTextTypedDict,
         ChatMessageContentItemVideoTypedDict,
     ],
 )

openrouter/components/chatmessagecontentitemaudio.py

@@ -1,23 +1,42 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from openrouter.types import BaseModel
-from openrouter.utils import validate_const
+from openrouter.types import BaseModel, UnrecognizedStr
+from openrouter.utils import validate_const, validate_open_enum
 import pydantic
-from pydantic.functional_validators import AfterValidator
-from typing import Literal
+from pydantic.functional_validators import AfterValidator, PlainValidator
+from typing import Literal, Union
 from typing_extensions import Annotated, TypedDict
 
 
+ChatMessageContentItemAudioFormat = Union[
+    Literal[
+        "wav",
+        "mp3",
+        "flac",
+        "m4a",
+        "ogg",
+        "pcm16",
+        "pcm24",
+    ],
+    UnrecognizedStr,
+]
+
+
 class ChatMessageContentItemAudioInputAudioTypedDict(TypedDict):
     data: str
-    format_: str
+    format_: ChatMessageContentItemAudioFormat
 
 
 class ChatMessageContentItemAudioInputAudio(BaseModel):
     data: str
 
-    format_: Annotated[str, pydantic.Field(alias="format")]
+    format_: Annotated[
+        Annotated[
+            ChatMessageContentItemAudioFormat, PlainValidator(validate_open_enum(False))
+        ],
+        pydantic.Field(alias="format"),
+    ]
 
 
 class ChatMessageContentItemAudioTypedDict(TypedDict):

openrouter/components/chatmessagecontentitemtext.py

@@ -1,22 +1,17 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .chatmessagecontentitemcachecontrol import (
-    ChatMessageContentItemCacheControl,
-    ChatMessageContentItemCacheControlTypedDict,
-)
 from openrouter.types import BaseModel
 from openrouter.utils import validate_const
 import pydantic
 from pydantic.functional_validators import AfterValidator
-from typing import Literal, Optional
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing import Literal
+from typing_extensions import Annotated, TypedDict
 
 
 class ChatMessageContentItemTextTypedDict(TypedDict):
     text: str
     type: Literal["text"]
-    cache_control: NotRequired[ChatMessageContentItemCacheControlTypedDict]
 
 
 class ChatMessageContentItemText(BaseModel):
@@ -26,5 +21,3 @@ class ChatMessageContentItemText(BaseModel):
         Annotated[Literal["text"], AfterValidator(validate_const("text"))],
         pydantic.Field(alias="type"),
     ] = "text"
-
-    cache_control: Optional[ChatMessageContentItemCacheControl] = None

openrouter/components/chatmessagecontentitemvideo.py

@@ -2,9 +2,8 @@
 
 from __future__ import annotations
 from openrouter.types import BaseModel
-from openrouter.utils import get_discriminator, validate_const
+from openrouter.utils import validate_const
 import pydantic
-from pydantic import Discriminator, Tag
 from pydantic.functional_validators import AfterValidator
 from typing import Literal, Union
 from typing_extensions import Annotated, TypeAliasType, TypedDict
@@ -65,10 +64,7 @@ ChatMessageContentItemVideoTypedDict = TypeAliasType(
 )
 
 
-ChatMessageContentItemVideo = Annotated[
-    Union[
-        Annotated[ChatMessageContentItemVideoInputVideo, Tag("input_video")],
-        Annotated[ChatMessageContentItemVideoVideoURL, Tag("video_url")],
-    ],
-    Discriminator(lambda m: get_discriminator(m, "type", "type")),
-]
+ChatMessageContentItemVideo = TypeAliasType(
+    "ChatMessageContentItemVideo",
+    Union[ChatMessageContentItemVideoInputVideo, ChatMessageContentItemVideoVideoURL],
+)