"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""

from __future__ import annotations
from openrouter.types import BaseModel, UNSET_SENTINEL
from pydantic import model_serializer
from typing import Literal, Optional
from typing_extensions import NotRequired, TypedDict


class ChatStreamToolCallFunctionTypedDict(TypedDict):
    r"""Function call details"""

    arguments: NotRequired[str]
    r"""Function arguments as JSON string"""
    name: NotRequired[str]
    r"""Function name"""


class ChatStreamToolCallFunction(BaseModel):
    r"""Function call details"""

    arguments: Optional[str] = None
    r"""Function arguments as JSON string"""

    name: Optional[str] = None
    r"""Function name"""

    @model_serializer(mode="wrap")
    def serialize_model(self, handler):
        optional_fields = set(["arguments", "name"])
        serialized = handler(self)
        m = {}

        for n, f in type(self).model_fields.items():
            k = f.alias or n
            val = serialized.get(k, serialized.get(n))

            if val != UNSET_SENTINEL:
                if val is not None or k not in optional_fields:
                    m[k] = val

        return m


ChatStreamToolCallType = Literal["function",]
r"""Tool call type"""


class ChatStreamToolCallTypedDict(TypedDict):
    r"""Tool call delta for streaming responses"""

    index: int
    r"""Tool call index in the array"""
    function: NotRequired[ChatStreamToolCallFunctionTypedDict]
    r"""Function call details"""
    id: NotRequired[str]
    r"""Tool call identifier"""
    type: NotRequired[ChatStreamToolCallType]
    r"""Tool call type"""


class ChatStreamToolCall(BaseModel):
    r"""Tool call delta for streaming responses"""

    index: int
    r"""Tool call index in the array"""

    function: Optional[ChatStreamToolCallFunction] = None
    r"""Function call details"""

    id: Optional[str] = None
    r"""Tool call identifier"""

    type: Optional[ChatStreamToolCallType] = None
    r"""Tool call type"""

    @model_serializer(mode="wrap")
    def serialize_model(self, handler):
        optional_fields = set(["function", "id", "type"])
        serialized = handler(self)
        m = {}

        for n, f in type(self).model_fields.items():
            k = f.alias or n
            val = serialized.get(k, serialized.get(n))

            if val != UNSET_SENTINEL:
                if val is not None or k not in optional_fields:
                    m[k] = val

        return m