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

openrouter/components/chatresponsechoice.py

@@ -1,6 +1,7 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from ._schema3 import Schema3, Schema3TypedDict
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
 from .chatcompletionfinishreason import ChatCompletionFinishReason
 from .chatmessagetokenlogprobs import (
@@ -17,6 +18,7 @@ from openrouter.types import (
 from openrouter.utils import validate_open_enum
 from pydantic import model_serializer
 from pydantic.functional_validators import PlainValidator
+from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -24,6 +26,7 @@ class ChatResponseChoiceTypedDict(TypedDict):
     finish_reason: Nullable[ChatCompletionFinishReason]
     index: float
     message: AssistantMessageTypedDict
+    reasoning_details: NotRequired[List[Schema3TypedDict]]
     logprobs: NotRequired[Nullable[ChatMessageTokenLogprobsTypedDict]]
 
 
@@ -36,11 +39,13 @@ class ChatResponseChoice(BaseModel):
 
     message: AssistantMessage
 
+    reasoning_details: Optional[List[Schema3]] = None
+
     logprobs: OptionalNullable[ChatMessageTokenLogprobs] = UNSET
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["logprobs"]
+        optional_fields = ["reasoning_details", "logprobs"]
         nullable_fields = ["finish_reason", "logprobs"]
         null_default_fields = []
 

openrouter/components/chatstreamingmessagechunk.py

@@ -1,6 +1,7 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from ._schema3 import Schema3, Schema3TypedDict
 from .chatstreamingmessagetoolcall import (
     ChatStreamingMessageToolCall,
     ChatStreamingMessageToolCallTypedDict,
@@ -26,6 +27,7 @@ class ChatStreamingMessageChunkTypedDict(TypedDict):
     reasoning: NotRequired[Nullable[str]]
     refusal: NotRequired[Nullable[str]]
     tool_calls: NotRequired[List[ChatStreamingMessageToolCallTypedDict]]
+    reasoning_details: NotRequired[List[Schema3TypedDict]]
 
 
 class ChatStreamingMessageChunk(BaseModel):
@@ -39,9 +41,18 @@ class ChatStreamingMessageChunk(BaseModel):
 
     tool_calls: Optional[List[ChatStreamingMessageToolCall]] = None
 
+    reasoning_details: Optional[List[Schema3]] = None
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["role", "content", "reasoning", "refusal", "tool_calls"]
+        optional_fields = [
+            "role",
+            "content",
+            "reasoning",
+            "refusal",
+            "tool_calls",
+            "reasoning_details",
+        ]
         nullable_fields = ["content", "reasoning", "refusal"]
         null_default_fields = []
 

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/datacollection.py

@@ -0,0 +1,19 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from openrouter.types import UnrecognizedStr
+from typing import Literal, Union
+
+
+DataCollection = Union[
+    Literal[
+        "deny",
+        "allow",
+    ],
+    UnrecognizedStr,
+]
+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.
+"""

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/openairesponsesreasoningeffort.py

@@ -7,10 +7,12 @@ from typing import Literal, Union
 
 OpenAIResponsesReasoningEffort = Union[
     Literal[
+        "xhigh",
         "high",
         "medium",
         "low",
         "minimal",
+        "none",
     ],
     UnrecognizedStr,
 ]

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/openresponsesreasoning.py

@@ -57,6 +57,7 @@ OpenResponsesReasoningFormat = Union[
         "openai-responses-v1",
         "xai-responses-v1",
         "anthropic-claude-v1",
+        "google-gemini-v1",
     ],
     UnrecognizedStr,
 ]