mirror of
https://github.com/wassname/openrouter-python-sdk-retry-errors.git
synced 2026-07-29 11:23:49 +08:00
OpenRouter SDK Bot
5ab44f08f0
Speakeasy regeneration with latest schema changes including type renames and new server tool models.
120 lines
3.1 KiB
Python
120 lines
3.1 KiB
Python
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
|
|
|
from __future__ import annotations
|
|
from .chatstreamchoice import ChatStreamChoice, ChatStreamChoiceTypedDict
|
|
from .chatusage import ChatUsage, ChatUsageTypedDict
|
|
from openrouter.types import (
|
|
BaseModel,
|
|
Nullable,
|
|
OptionalNullable,
|
|
UNSET,
|
|
UNSET_SENTINEL,
|
|
)
|
|
from pydantic import model_serializer
|
|
from typing import List, Literal, Optional
|
|
from typing_extensions import NotRequired, TypedDict
|
|
|
|
|
|
ChatStreamChunkObject = Literal["chat.completion.chunk",]
|
|
|
|
|
|
class ErrorTypedDict(TypedDict):
|
|
r"""Error information"""
|
|
|
|
message: str
|
|
r"""Error message"""
|
|
code: float
|
|
r"""Error code"""
|
|
|
|
|
|
class Error(BaseModel):
|
|
r"""Error information"""
|
|
|
|
message: str
|
|
r"""Error message"""
|
|
|
|
code: float
|
|
r"""Error code"""
|
|
|
|
|
|
class ChatStreamChunkTypedDict(TypedDict):
|
|
r"""Streaming chat completion chunk"""
|
|
|
|
id: str
|
|
r"""Unique chunk identifier"""
|
|
choices: List[ChatStreamChoiceTypedDict]
|
|
r"""List of streaming chunk choices"""
|
|
created: float
|
|
r"""Unix timestamp of creation"""
|
|
model: str
|
|
r"""Model used for completion"""
|
|
object: ChatStreamChunkObject
|
|
system_fingerprint: NotRequired[str]
|
|
r"""System fingerprint"""
|
|
service_tier: NotRequired[Nullable[str]]
|
|
r"""The service tier used by the upstream provider for this request"""
|
|
error: NotRequired[ErrorTypedDict]
|
|
r"""Error information"""
|
|
usage: NotRequired[ChatUsageTypedDict]
|
|
r"""Token usage statistics"""
|
|
|
|
|
|
class ChatStreamChunk(BaseModel):
|
|
r"""Streaming chat completion chunk"""
|
|
|
|
id: str
|
|
r"""Unique chunk identifier"""
|
|
|
|
choices: List[ChatStreamChoice]
|
|
r"""List of streaming chunk choices"""
|
|
|
|
created: float
|
|
r"""Unix timestamp of creation"""
|
|
|
|
model: str
|
|
r"""Model used for completion"""
|
|
|
|
object: ChatStreamChunkObject
|
|
|
|
system_fingerprint: Optional[str] = None
|
|
r"""System fingerprint"""
|
|
|
|
service_tier: OptionalNullable[str] = UNSET
|
|
r"""The service tier used by the upstream provider for this request"""
|
|
|
|
error: Optional[Error] = None
|
|
r"""Error information"""
|
|
|
|
usage: Optional[ChatUsage] = None
|
|
r"""Token usage statistics"""
|
|
|
|
@model_serializer(mode="wrap")
|
|
def serialize_model(self, handler):
|
|
optional_fields = ["system_fingerprint", "service_tier", "error", "usage"]
|
|
nullable_fields = ["service_tier"]
|
|
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
|