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

openrouter/components/chatmessagecontentitemcachecontrol.py

@@ -0,0 +1,32 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from openrouter.types import BaseModel, UnrecognizedStr
+from openrouter.utils import validate_const, validate_open_enum
+import pydantic
+from pydantic.functional_validators import AfterValidator, PlainValidator
+from typing import Literal, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+TTL = Union[
+    Literal[
+        "5m",
+        "1h",
+    ],
+    UnrecognizedStr,
+]
+
+
+class ChatMessageContentItemCacheControlTypedDict(TypedDict):
+    type: Literal["ephemeral"]
+    ttl: NotRequired[TTL]
+
+
+class ChatMessageContentItemCacheControl(BaseModel):
+    TYPE: Annotated[
+        Annotated[Literal["ephemeral"], AfterValidator(validate_const("ephemeral"))],
+        pydantic.Field(alias="type"),
+    ] = "ephemeral"
+
+    ttl: Annotated[Optional[TTL], PlainValidator(validate_open_enum(False))] = None

openrouter/components/chatmessagecontentitemtext.py

@@ -1,17 +1,22 @@
 """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
-from typing_extensions import Annotated, TypedDict
+from typing import Literal, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class ChatMessageContentItemTextTypedDict(TypedDict):
     text: str
     type: Literal["text"]
+    cache_control: NotRequired[ChatMessageContentItemCacheControlTypedDict]
 
 
 class ChatMessageContentItemText(BaseModel):
@@ -21,3 +26,5 @@ 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,8 +2,9 @@
 
 from __future__ import annotations
 from openrouter.types import BaseModel
-from openrouter.utils import validate_const
+from openrouter.utils import get_discriminator, 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
@@ -64,7 +65,10 @@ ChatMessageContentItemVideoTypedDict = TypeAliasType(
 )
 
 
-ChatMessageContentItemVideo = TypeAliasType(
-    "ChatMessageContentItemVideo",
-    Union[ChatMessageContentItemVideoInputVideo, ChatMessageContentItemVideoVideoURL],
-)
+ChatMessageContentItemVideo = Annotated[
+    Union[
+        Annotated[ChatMessageContentItemVideoInputVideo, Tag("input_video")],
+        Annotated[ChatMessageContentItemVideoVideoURL, Tag("video_url")],
+    ],
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]

openrouter/components/chatresponsechoice.py

@@ -2,41 +2,24 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .chatcompletionfinishreason import ChatCompletionFinishReason
 from .chatmessagetokenlogprobs import (
     ChatMessageTokenLogprobs,
     ChatMessageTokenLogprobsTypedDict,
 )
-from .chatstreamingmessagechunk import (
-    ChatStreamingMessageChunk,
-    ChatStreamingMessageChunkTypedDict,
-)
 from openrouter.types import (
     BaseModel,
     Nullable,
     OptionalNullable,
     UNSET,
     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, Union
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-ChatCompletionFinishReason = Union[
-    Literal[
-        "tool_calls",
-        "stop",
-        "length",
-        "content_filter",
-        "error",
-    ],
-    UnrecognizedStr,
-]
-
-
 class ChatResponseChoiceTypedDict(TypedDict):
     finish_reason: Nullable[ChatCompletionFinishReason]
     index: float
@@ -84,52 +67,3 @@ class ChatResponseChoice(BaseModel):
                 m[k] = val
 
         return m
-
-
-class ChatStreamingChoiceTypedDict(TypedDict):
-    delta: ChatStreamingMessageChunkTypedDict
-    finish_reason: Nullable[ChatCompletionFinishReason]
-    index: float
-    logprobs: NotRequired[Nullable[ChatMessageTokenLogprobsTypedDict]]
-
-
-class ChatStreamingChoice(BaseModel):
-    delta: ChatStreamingMessageChunk
-
-    finish_reason: Annotated[
-        Nullable[ChatCompletionFinishReason], PlainValidator(validate_open_enum(False))
-    ]
-
-    index: float
-
-    logprobs: OptionalNullable[ChatMessageTokenLogprobs] = UNSET
-
-    @model_serializer(mode="wrap")
-    def serialize_model(self, handler):
-        optional_fields = ["logprobs"]
-        nullable_fields = ["finish_reason", "logprobs"]
-        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

openrouter/components/chatstreamingchoice.py

@@ -0,0 +1,72 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .chatcompletionfinishreason import ChatCompletionFinishReason
+from .chatmessagetokenlogprobs import (
+    ChatMessageTokenLogprobs,
+    ChatMessageTokenLogprobsTypedDict,
+)
+from .chatstreamingmessagechunk import (
+    ChatStreamingMessageChunk,
+    ChatStreamingMessageChunkTypedDict,
+)
+from openrouter.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
+from openrouter.utils import validate_open_enum
+from pydantic import model_serializer
+from pydantic.functional_validators import PlainValidator
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class ChatStreamingChoiceTypedDict(TypedDict):
+    delta: ChatStreamingMessageChunkTypedDict
+    finish_reason: Nullable[ChatCompletionFinishReason]
+    index: float
+    logprobs: NotRequired[Nullable[ChatMessageTokenLogprobsTypedDict]]
+
+
+class ChatStreamingChoice(BaseModel):
+    delta: ChatStreamingMessageChunk
+
+    finish_reason: Annotated[
+        Nullable[ChatCompletionFinishReason], PlainValidator(validate_open_enum(False))
+    ]
+
+    index: float
+
+    logprobs: OptionalNullable[ChatMessageTokenLogprobs] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["logprobs"]
+        nullable_fields = ["finish_reason", "logprobs"]
+        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

openrouter/components/chatstreamingresponsechunk.py

@@ -5,7 +5,7 @@ from .chatgenerationtokenusage import (
     ChatGenerationTokenUsage,
     ChatGenerationTokenUsageTypedDict,
 )
-from .chatresponsechoice import ChatStreamingChoice, ChatStreamingChoiceTypedDict
+from .chatstreamingchoice import ChatStreamingChoice, ChatStreamingChoiceTypedDict
 from openrouter.types import (
     BaseModel,
     Nullable,

openrouter/components/completionchoice.py

@@ -2,12 +2,19 @@
 
 from __future__ import annotations
 from .completionlogprobs import CompletionLogprobs, CompletionLogprobsTypedDict
-from openrouter.types import BaseModel, Nullable, UNSET_SENTINEL, UnrecognizedStr
+from openrouter.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    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, Union
-from typing_extensions import Annotated, TypedDict
+from typing import Literal, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 CompletionFinishReason = Union[
@@ -25,6 +32,8 @@ class CompletionChoiceTypedDict(TypedDict):
     index: float
     logprobs: Nullable[CompletionLogprobsTypedDict]
     finish_reason: Nullable[CompletionFinishReason]
+    native_finish_reason: NotRequired[str]
+    reasoning: NotRequired[Nullable[str]]
 
 
 class CompletionChoice(BaseModel):
@@ -38,10 +47,14 @@ class CompletionChoice(BaseModel):
         Nullable[CompletionFinishReason], PlainValidator(validate_open_enum(False))
     ]
 
+    native_finish_reason: Optional[str] = None
+
+    reasoning: OptionalNullable[str] = UNSET
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = []
-        nullable_fields = ["logprobs", "finish_reason"]
+        optional_fields = ["native_finish_reason", "reasoning"]
+        nullable_fields = ["logprobs", "finish_reason", "reasoning"]
         null_default_fields = []
 
         serialized = handler(self)

openrouter/components/completioncreateparams.py

@@ -16,9 +16,9 @@ from openrouter.types import (
     UNSET,
     UNSET_SENTINEL,
 )
-from openrouter.utils import validate_const
+from openrouter.utils import get_discriminator, validate_const
 import pydantic
-from pydantic import model_serializer
+from pydantic import Discriminator, Tag, model_serializer
 from pydantic.functional_validators import AfterValidator
 from typing import Dict, List, Literal, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
@@ -127,16 +127,16 @@ CompletionCreateParamsResponseFormatUnionTypedDict = TypeAliasType(
 )
 
 
-CompletionCreateParamsResponseFormatUnion = TypeAliasType(
-    "CompletionCreateParamsResponseFormatUnion",
+CompletionCreateParamsResponseFormatUnion = Annotated[
     Union[
-        CompletionCreateParamsResponseFormatText,
-        CompletionCreateParamsResponseFormatJSONObject,
-        CompletionCreateParamsResponseFormatPython,
-        ResponseFormatJSONSchema,
-        ResponseFormatTextGrammar,
+        Annotated[CompletionCreateParamsResponseFormatText, Tag("text")],
+        Annotated[CompletionCreateParamsResponseFormatJSONObject, Tag("json_object")],
+        Annotated[ResponseFormatJSONSchema, Tag("json_schema")],
+        Annotated[ResponseFormatTextGrammar, Tag("grammar")],
+        Annotated[CompletionCreateParamsResponseFormatPython, Tag("python")],
     ],
-)
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]
 
 
 class CompletionCreateParamsTypedDict(TypedDict):

openrouter/components/completionresponse.py

@@ -17,6 +17,7 @@ class CompletionResponseTypedDict(TypedDict):
     model: str
     choices: List[CompletionChoiceTypedDict]
     object: Literal["text_completion"]
+    provider: NotRequired[str]
     system_fingerprint: NotRequired[str]
     usage: NotRequired[CompletionUsageTypedDict]
 
@@ -38,6 +39,8 @@ class CompletionResponse(BaseModel):
         pydantic.Field(alias="object"),
     ] = "text_completion"
 
+    provider: Optional[str] = None
+
     system_fingerprint: Optional[str] = None
 
     usage: Optional[CompletionUsage] = None

openrouter/components/message.py

@@ -10,8 +10,9 @@ from .systemmessage import SystemMessage, SystemMessageTypedDict
 from .toolresponsemessage import ToolResponseMessage, ToolResponseMessageTypedDict
 from .usermessage import UserMessage, UserMessageTypedDict
 from openrouter.types import BaseModel
-from openrouter.utils import validate_const
+from openrouter.utils import get_discriminator, validate_const
 import pydantic
+from pydantic import Discriminator, Tag
 from pydantic.functional_validators import AfterValidator
 from typing import List, Literal, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
@@ -56,13 +57,13 @@ MessageTypedDict = TypeAliasType(
 )
 
 
-Message = TypeAliasType(
-    "Message",
+Message = Annotated[
     Union[
-        SystemMessage,
-        UserMessage,
-        MessageDeveloper,
-        ToolResponseMessage,
-        AssistantMessage,
+        Annotated[SystemMessage, Tag("system")],
+        Annotated[UserMessage, Tag("user")],
+        Annotated[MessageDeveloper, Tag("developer")],
+        Annotated[AssistantMessage, Tag("assistant")],
+        Annotated[ToolResponseMessage, Tag("tool")],
     ],
-)
+    Discriminator(lambda m: get_discriminator(m, "role", "role")),
+]

openrouter/components/openairesponsesannotation.py

@@ -4,8 +4,10 @@ from __future__ import annotations
 from .filecitation import FileCitation, FileCitationTypedDict
 from .filepath import FilePath, FilePathTypedDict
 from .urlcitation import URLCitation, URLCitationTypedDict
+from openrouter.utils import get_discriminator
+from pydantic import Discriminator, Tag
 from typing import Union
-from typing_extensions import TypeAliasType
+from typing_extensions import Annotated, TypeAliasType
 
 
 OpenAIResponsesAnnotationTypedDict = TypeAliasType(
@@ -14,6 +16,11 @@ OpenAIResponsesAnnotationTypedDict = TypeAliasType(
 )
 
 
-OpenAIResponsesAnnotation = TypeAliasType(
-    "OpenAIResponsesAnnotation", Union[FilePath, FileCitation, URLCitation]
-)
+OpenAIResponsesAnnotation = Annotated[
+    Union[
+        Annotated[FileCitation, Tag("file_citation")],
+        Annotated[URLCitation, Tag("url_citation")],
+        Annotated[FilePath, Tag("file_path")],
+    ],
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]

openrouter/components/openresponsesnonstreamingresponse.py

@@ -48,8 +48,8 @@ from openrouter.types import (
     UNSET,
     UNSET_SENTINEL,
 )
-from openrouter.utils import validate_open_enum
-from pydantic import model_serializer
+from openrouter.utils import get_discriminator, validate_open_enum
+from pydantic import Discriminator, Tag, model_serializer
 from pydantic.functional_validators import PlainValidator
 from typing import Any, Dict, List, Literal, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
@@ -127,16 +127,19 @@ OpenResponsesNonStreamingResponseToolUnionTypedDict = TypeAliasType(
 )
 
 
-OpenResponsesNonStreamingResponseToolUnion = TypeAliasType(
-    "OpenResponsesNonStreamingResponseToolUnion",
+OpenResponsesNonStreamingResponseToolUnion = Annotated[
     Union[
-        OpenResponsesWebSearchPreviewTool,
-        OpenResponsesWebSearchPreview20250311Tool,
-        OpenResponsesWebSearchTool,
-        OpenResponsesWebSearch20250826Tool,
-        OpenResponsesNonStreamingResponseToolFunction,
+        Annotated[OpenResponsesNonStreamingResponseToolFunction, Tag("function")],
+        Annotated[OpenResponsesWebSearchPreviewTool, Tag("web_search_preview")],
+        Annotated[
+            OpenResponsesWebSearchPreview20250311Tool,
+            Tag("web_search_preview_2025_03_11"),
+        ],
+        Annotated[OpenResponsesWebSearchTool, Tag("web_search")],
+        Annotated[OpenResponsesWebSearch20250826Tool, Tag("web_search_2025_08_26")],
     ],
-)
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]
 
 
 class OpenResponsesNonStreamingResponseTypedDict(TypedDict):

openrouter/components/openresponsesrequest.py

@@ -44,9 +44,10 @@ from openrouter.types import (
     UNSET_SENTINEL,
     UnrecognizedStr,
 )
-from openrouter.utils import validate_open_enum
-from pydantic import model_serializer
-from pydantic.functional_validators import PlainValidator
+from openrouter.utils import get_discriminator, validate_const, validate_open_enum
+import pydantic
+from pydantic import Discriminator, Tag, 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
 
@@ -120,28 +121,22 @@ OpenResponsesRequestToolUnionTypedDict = TypeAliasType(
 )
 
 
-OpenResponsesRequestToolUnion = TypeAliasType(
-    "OpenResponsesRequestToolUnion",
+OpenResponsesRequestToolUnion = Annotated[
     Union[
-        OpenResponsesWebSearchPreviewTool,
-        OpenResponsesWebSearchPreview20250311Tool,
-        OpenResponsesWebSearchTool,
-        OpenResponsesWebSearch20250826Tool,
-        OpenResponsesRequestToolFunction,
+        Annotated[OpenResponsesRequestToolFunction, Tag("function")],
+        Annotated[OpenResponsesWebSearchPreviewTool, Tag("web_search_preview")],
+        Annotated[
+            OpenResponsesWebSearchPreview20250311Tool,
+            Tag("web_search_preview_2025_03_11"),
+        ],
+        Annotated[OpenResponsesWebSearchTool, Tag("web_search")],
+        Annotated[OpenResponsesWebSearch20250826Tool, Tag("web_search_2025_08_26")],
     ],
-)
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]
 
 
-ServiceTier = Union[
-    Literal[
-        "auto",
-        "default",
-        "flex",
-        "priority",
-        "scale",
-    ],
-    UnrecognizedStr,
-]
+ServiceTier = Literal["auto",]
 
 
 Truncation = Union[
@@ -353,6 +348,17 @@ class Provider(BaseModel):
         return m
 
 
+IDResponseHealing = Literal["response-healing",]
+
+
+class PluginResponseHealingTypedDict(TypedDict):
+    id: IDResponseHealing
+
+
+class PluginResponseHealing(BaseModel):
+    id: IDResponseHealing
+
+
 IDFileParser = Literal["file-parser",]
 
 
@@ -434,11 +440,24 @@ class PluginModeration(BaseModel):
 
 PluginTypedDict = TypeAliasType(
     "PluginTypedDict",
-    Union[PluginModerationTypedDict, PluginFileParserTypedDict, PluginWebTypedDict],
+    Union[
+        PluginModerationTypedDict,
+        PluginResponseHealingTypedDict,
+        PluginFileParserTypedDict,
+        PluginWebTypedDict,
+    ],
 )
 
 
-Plugin = TypeAliasType("Plugin", Union[PluginModeration, PluginFileParser, PluginWeb])
+Plugin = Annotated[
+    Union[
+        Annotated[PluginModeration, Tag("moderation")],
+        Annotated[PluginWeb, Tag("web")],
+        Annotated[PluginFileParser, Tag("file-parser")],
+        Annotated[PluginResponseHealing, Tag("response-healing")],
+    ],
+    Discriminator(lambda m: get_discriminator(m, "id", "id")),
+]
 
 
 class OpenResponsesRequestTypedDict(TypedDict):
@@ -468,8 +487,8 @@ class OpenResponsesRequestTypedDict(TypedDict):
     include: NotRequired[Nullable[List[OpenAIResponsesIncludable]]]
     background: NotRequired[Nullable[bool]]
     safety_identifier: NotRequired[Nullable[str]]
-    store: NotRequired[Nullable[bool]]
-    service_tier: NotRequired[Nullable[ServiceTier]]
+    store: Literal[False]
+    service_tier: NotRequired[ServiceTier]
     truncation: NotRequired[Nullable[Truncation]]
     stream: NotRequired[bool]
     provider: NotRequired[Nullable[ProviderTypedDict]]
@@ -478,6 +497,8 @@ class OpenResponsesRequestTypedDict(TypedDict):
     r"""Plugins you want to enable for this request, including their settings."""
     user: NotRequired[str]
     r"""A unique identifier representing your end-user, which helps distinguish between different users of your app. This allows your app to identify specific users in case of abuse reports, preventing your entire app from being affected by the actions of individual users. Maximum of 128 characters."""
+    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."""
 
 
 class OpenResponsesRequest(BaseModel):
@@ -533,11 +554,12 @@ class OpenResponsesRequest(BaseModel):
 
     safety_identifier: OptionalNullable[str] = UNSET
 
-    store: OptionalNullable[bool] = UNSET
+    STORE: Annotated[
+        Annotated[Optional[Literal[False]], AfterValidator(validate_const(False))],
+        pydantic.Field(alias="store"),
+    ] = False
 
-    service_tier: Annotated[
-        OptionalNullable[ServiceTier], PlainValidator(validate_open_enum(False))
-    ] = UNSET
+    service_tier: Optional[ServiceTier] = "auto"
 
     truncation: Annotated[
         OptionalNullable[Truncation], PlainValidator(validate_open_enum(False))
@@ -554,6 +576,9 @@ class OpenResponsesRequest(BaseModel):
     user: Optional[str] = None
     r"""A unique identifier representing your end-user, which helps distinguish between different users of your app. This allows your app to identify specific users in case of abuse reports, preventing your entire app from being affected by the actions of individual users. Maximum of 128 characters."""
 
+    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_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = [
@@ -584,6 +609,7 @@ class OpenResponsesRequest(BaseModel):
             "provider",
             "plugins",
             "user",
+            "session_id",
         ]
         nullable_fields = [
             "instructions",
@@ -599,8 +625,6 @@ class OpenResponsesRequest(BaseModel):
             "include",
             "background",
             "safety_identifier",
-            "store",
-            "service_tier",
             "truncation",
             "provider",
         ]