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.
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 .chatstreamdelta import ChatStreamDelta, ChatStreamDeltaTypedDict
|
|
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 ChatStreamChoiceTypedDict(TypedDict):
|
|
r"""Streaming completion choice chunk"""
|
|
|
|
delta: ChatStreamDeltaTypedDict
|
|
r"""Delta changes in streaming response"""
|
|
finish_reason: Nullable[Any]
|
|
index: float
|
|
r"""Choice index"""
|
|
logprobs: NotRequired[Nullable[ChatTokenLogprobsTypedDict]]
|
|
r"""Log probabilities for the completion"""
|
|
|
|
|
|
class ChatStreamChoice(BaseModel):
|
|
r"""Streaming completion choice chunk"""
|
|
|
|
delta: ChatStreamDelta
|
|
r"""Delta changes in streaming response"""
|
|
|
|
finish_reason: Nullable[Any]
|
|
|
|
index: float
|
|
r"""Choice index"""
|
|
|
|
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
|