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

openrouter/components/chatresponsechoice.py

@@ -1,32 +1,46 @@
 """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 (
     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 List, Optional
+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
     message: AssistantMessageTypedDict
-    reasoning_details: NotRequired[List[Schema3TypedDict]]
     logprobs: NotRequired[Nullable[ChatMessageTokenLogprobsTypedDict]]
 
 
@@ -39,13 +53,60 @@ 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"]
+        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
+
+
+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 = ["reasoning_details", "logprobs"]
+        optional_fields = ["logprobs"]
         nullable_fields = ["finish_reason", "logprobs"]
         null_default_fields = []
 

openrouter/components/chatstreamingmessagechunk.py

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

openrouter/components/chatstreamingresponsechunk.py

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

openrouter/components/completionchoice.py

@@ -2,19 +2,12 @@
 
 from __future__ import annotations
 from .completionlogprobs import CompletionLogprobs, CompletionLogprobsTypedDict
-from openrouter.types import (
-    BaseModel,
-    Nullable,
-    OptionalNullable,
-    UNSET,
-    UNSET_SENTINEL,
-    UnrecognizedStr,
-)
+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
+from typing import Literal, Union
+from typing_extensions import Annotated, TypedDict
 
 
 CompletionFinishReason = Union[
@@ -32,8 +25,6 @@ class CompletionChoiceTypedDict(TypedDict):
     index: float
     logprobs: Nullable[CompletionLogprobsTypedDict]
     finish_reason: Nullable[CompletionFinishReason]
-    native_finish_reason: NotRequired[str]
-    reasoning: NotRequired[Nullable[str]]
 
 
 class CompletionChoice(BaseModel):
@@ -47,14 +38,10 @@ 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 = ["native_finish_reason", "reasoning"]
-        nullable_fields = ["logprobs", "finish_reason", "reasoning"]
+        optional_fields = []
+        nullable_fields = ["logprobs", "finish_reason"]
         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 get_discriminator, validate_const
+from openrouter.utils import validate_const
 import pydantic
-from pydantic import Discriminator, Tag, model_serializer
+from pydantic import 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 = Annotated[
+CompletionCreateParamsResponseFormatUnion = TypeAliasType(
+    "CompletionCreateParamsResponseFormatUnion",
     Union[
-        Annotated[CompletionCreateParamsResponseFormatText, Tag("text")],
-        Annotated[CompletionCreateParamsResponseFormatJSONObject, Tag("json_object")],
-        Annotated[ResponseFormatJSONSchema, Tag("json_schema")],
-        Annotated[ResponseFormatTextGrammar, Tag("grammar")],
-        Annotated[CompletionCreateParamsResponseFormatPython, Tag("python")],
+        CompletionCreateParamsResponseFormatText,
+        CompletionCreateParamsResponseFormatJSONObject,
+        CompletionCreateParamsResponseFormatPython,
+        ResponseFormatJSONSchema,
+        ResponseFormatTextGrammar,
     ],
-    Discriminator(lambda m: get_discriminator(m, "type", "type")),
-]
+)
 
 
 class CompletionCreateParamsTypedDict(TypedDict):

openrouter/components/completionresponse.py

@@ -17,7 +17,6 @@ class CompletionResponseTypedDict(TypedDict):
     model: str
     choices: List[CompletionChoiceTypedDict]
     object: Literal["text_completion"]
-    provider: NotRequired[str]
     system_fingerprint: NotRequired[str]
     usage: NotRequired[CompletionUsageTypedDict]
 
@@ -39,8 +38,6 @@ 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,9 +10,8 @@ from .systemmessage import SystemMessage, SystemMessageTypedDict
 from .toolresponsemessage import ToolResponseMessage, ToolResponseMessageTypedDict
 from .usermessage import UserMessage, UserMessageTypedDict
 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 List, Literal, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
@@ -57,13 +56,13 @@ MessageTypedDict = TypeAliasType(
 )
 
 
-Message = Annotated[
+Message = TypeAliasType(
+    "Message",
     Union[
-        Annotated[SystemMessage, Tag("system")],
-        Annotated[UserMessage, Tag("user")],
-        Annotated[MessageDeveloper, Tag("developer")],
-        Annotated[AssistantMessage, Tag("assistant")],
-        Annotated[ToolResponseMessage, Tag("tool")],
+        SystemMessage,
+        UserMessage,
+        MessageDeveloper,
+        ToolResponseMessage,
+        AssistantMessage,
     ],
-    Discriminator(lambda m: get_discriminator(m, "role", "role")),
-]
+)

openrouter/components/openairesponsesannotation.py

@@ -4,10 +4,8 @@ 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 Annotated, TypeAliasType
+from typing_extensions import TypeAliasType
 
 
 OpenAIResponsesAnnotationTypedDict = TypeAliasType(
@@ -16,11 +14,6 @@ OpenAIResponsesAnnotationTypedDict = TypeAliasType(
 )
 
 
-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")),
-]
+OpenAIResponsesAnnotation = TypeAliasType(
+    "OpenAIResponsesAnnotation", Union[FilePath, FileCitation, URLCitation]
+)

openrouter/components/openairesponsesreasoningeffort.py

@@ -7,7 +7,6 @@ from typing import Literal, Union
 
 OpenAIResponsesReasoningEffort = Union[
     Literal[
-        "xhigh",
         "high",
         "medium",
         "low",

openrouter/components/openresponsesnonstreamingresponse.py

@@ -48,8 +48,8 @@ from openrouter.types import (
     UNSET,
     UNSET_SENTINEL,
 )
-from openrouter.utils import get_discriminator, validate_open_enum
-from pydantic import Discriminator, Tag, model_serializer
+from openrouter.utils import validate_open_enum
+from pydantic import 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,19 +127,16 @@ OpenResponsesNonStreamingResponseToolUnionTypedDict = TypeAliasType(
 )
 
 
-OpenResponsesNonStreamingResponseToolUnion = Annotated[
+OpenResponsesNonStreamingResponseToolUnion = TypeAliasType(
+    "OpenResponsesNonStreamingResponseToolUnion",
     Union[
-        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")],
+        OpenResponsesWebSearchPreviewTool,
+        OpenResponsesWebSearchPreview20250311Tool,
+        OpenResponsesWebSearchTool,
+        OpenResponsesWebSearch20250826Tool,
+        OpenResponsesNonStreamingResponseToolFunction,
     ],
-    Discriminator(lambda m: get_discriminator(m, "type", "type")),
-]
+)
 
 
 class OpenResponsesNonStreamingResponseTypedDict(TypedDict):