mirror of
https://github.com/wassname/openrouter-python-sdk-retry-errors.git
synced 2026-07-30 12:20:57 +08:00
OpenRouter SDK Bot
5ab44f08f0
Speakeasy regeneration with latest schema changes including type renames and new server tool models.
73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
|
|
|
from __future__ import annotations
|
|
from .chatassistantmessage import ChatAssistantMessage, ChatAssistantMessageTypedDict
|
|
from .chattokenlogprobs import ChatTokenLogprobs, ChatTokenLogprobsTypedDict
|
|
from openrouter.types import (
|
|
BaseModel,
|
|
Nullable,
|
|
OptionalNullable,
|
|
UNSET,
|
|
UNSET_SENTINEL,
|
|
)
|
|
from pydantic import model_serializer
|
|
from typing import Any
|
|
from typing_extensions import NotRequired, TypedDict
|
|
|
|
|
|
class ChatChoiceTypedDict(TypedDict):
|
|
r"""Chat completion choice"""
|
|
|
|
finish_reason: Nullable[Any]
|
|
index: float
|
|
r"""Choice index"""
|
|
message: ChatAssistantMessageTypedDict
|
|
r"""Assistant message for requests and responses"""
|
|
logprobs: NotRequired[Nullable[ChatTokenLogprobsTypedDict]]
|
|
r"""Log probabilities for the completion"""
|
|
|
|
|
|
class ChatChoice(BaseModel):
|
|
r"""Chat completion choice"""
|
|
|
|
finish_reason: Nullable[Any]
|
|
|
|
index: float
|
|
r"""Choice index"""
|
|
|
|
message: ChatAssistantMessage
|
|
r"""Assistant message for requests and responses"""
|
|
|
|
logprobs: OptionalNullable[ChatTokenLogprobs] = UNSET
|
|
r"""Log probabilities for the completion"""
|
|
|
|
@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
|