mirror of
https://github.com/wassname/openrouter-python-sdk-retry-errors.git
synced 2026-08-01 12:40:23 +08:00
OpenRouter Team
8fa4d9051c
- bf855082e30820d466f0f86a6d6e87de2ecfb06f - 335b1cbe361214e48b945b987b4e164457aa817b - 648485fdc7f9d8ae86fd857eccedcc5e41fb2e69 - 80c49474a15a2bd549a5f26efad3858aa45b7c7b - 70e8ac317f9c4472db3decd024d1819650df4182 - 9f2a6659ab6dbf111a0c29e4366328c3d2048826 - a3a28b1fc6ec7d2941232a4ebde8c5f470497876 - 4b5dc9193480b7c5c9ddb882ccf3ee2128ca0d51 - bb413e979a46ef67c2f266743f7ebd798af42afb - b60e92d5dcec83f31986164f51c010d1728b2b17 - 45687f423707f5838461585c568a024a19eb2f06 - 3103d3ca68b6e42ccdb6136588ec2b6edc8dcab0 - fd9f991606027cd86116317c71f8eb7b4e2aa3eb - 0f312019161edd76c1e5923d142c788d22f096d5 - 5dbab2094bc744db4a90b3097520d897e5d0d634 - 4c47d1af1735db18a1c1a7f463e801c7f71ad4e5 GitOrigin-RevId: bf855082e30820d466f0f86a6d6e87de2ecfb06f
125 lines
3.8 KiB
Python
125 lines
3.8 KiB
Python
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
|
|
|
from __future__ import annotations
|
|
from .assistantmessageimages import (
|
|
AssistantMessageImages,
|
|
AssistantMessageImagesTypedDict,
|
|
)
|
|
from .chatmessagecontentitem import (
|
|
ChatMessageContentItem,
|
|
ChatMessageContentItemTypedDict,
|
|
)
|
|
from .chatmessagetoolcall import ChatMessageToolCall, ChatMessageToolCallTypedDict
|
|
from .reasoningdetailunion import ReasoningDetailUnion, ReasoningDetailUnionTypedDict
|
|
from openrouter.types import (
|
|
BaseModel,
|
|
Nullable,
|
|
OptionalNullable,
|
|
UNSET,
|
|
UNSET_SENTINEL,
|
|
)
|
|
from pydantic import model_serializer
|
|
from typing import Any, List, Literal, Optional, Union
|
|
from typing_extensions import NotRequired, TypeAliasType, TypedDict
|
|
|
|
|
|
AssistantMessageRole = Literal["assistant",]
|
|
|
|
|
|
AssistantMessageContentTypedDict = TypeAliasType(
|
|
"AssistantMessageContentTypedDict",
|
|
Union[str, List[ChatMessageContentItemTypedDict], Any],
|
|
)
|
|
r"""Assistant message content"""
|
|
|
|
|
|
AssistantMessageContent = TypeAliasType(
|
|
"AssistantMessageContent", Union[str, List[ChatMessageContentItem], Any]
|
|
)
|
|
r"""Assistant message content"""
|
|
|
|
|
|
class AssistantMessageTypedDict(TypedDict):
|
|
r"""Assistant message for requests and responses"""
|
|
|
|
role: AssistantMessageRole
|
|
content: NotRequired[Nullable[AssistantMessageContentTypedDict]]
|
|
r"""Assistant message content"""
|
|
name: NotRequired[str]
|
|
r"""Optional name for the assistant"""
|
|
tool_calls: NotRequired[List[ChatMessageToolCallTypedDict]]
|
|
r"""Tool calls made by the assistant"""
|
|
refusal: NotRequired[Nullable[str]]
|
|
r"""Refusal message if content was refused"""
|
|
reasoning: NotRequired[Nullable[str]]
|
|
r"""Reasoning output"""
|
|
reasoning_details: NotRequired[List[ReasoningDetailUnionTypedDict]]
|
|
r"""Reasoning details for extended thinking models"""
|
|
images: NotRequired[List[AssistantMessageImagesTypedDict]]
|
|
r"""Generated images from image generation models"""
|
|
|
|
|
|
class AssistantMessage(BaseModel):
|
|
r"""Assistant message for requests and responses"""
|
|
|
|
role: AssistantMessageRole
|
|
|
|
content: OptionalNullable[AssistantMessageContent] = UNSET
|
|
r"""Assistant message content"""
|
|
|
|
name: Optional[str] = None
|
|
r"""Optional name for the assistant"""
|
|
|
|
tool_calls: Optional[List[ChatMessageToolCall]] = None
|
|
r"""Tool calls made by the assistant"""
|
|
|
|
refusal: OptionalNullable[str] = UNSET
|
|
r"""Refusal message if content was refused"""
|
|
|
|
reasoning: OptionalNullable[str] = UNSET
|
|
r"""Reasoning output"""
|
|
|
|
reasoning_details: Optional[List[ReasoningDetailUnion]] = None
|
|
r"""Reasoning details for extended thinking models"""
|
|
|
|
images: Optional[List[AssistantMessageImages]] = None
|
|
r"""Generated images from image generation models"""
|
|
|
|
@model_serializer(mode="wrap")
|
|
def serialize_model(self, handler):
|
|
optional_fields = [
|
|
"content",
|
|
"name",
|
|
"tool_calls",
|
|
"refusal",
|
|
"reasoning",
|
|
"reasoning_details",
|
|
"images",
|
|
]
|
|
nullable_fields = ["content", "refusal", "reasoning"]
|
|
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
|