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

from __future__ import annotations
from .assistantmessagetoolcallfunction import (
    AssistantMessageToolCallFunction,
    AssistantMessageToolCallFunctionTypedDict,
)
from friendli.core.types import BaseModel
from friendli.core.utils import validate_const
import pydantic
from pydantic.functional_validators import AfterValidator
from typing import Literal
from typing_extensions import Annotated, TypedDict


class AssistantMessageToolCallTypedDict(TypedDict):
    id: str
    "The ID of tool call."
    function: AssistantMessageToolCallFunctionTypedDict
    type: Literal["function"]
    "The type of tool call."


class AssistantMessageToolCall(BaseModel):
    id: str
    "The ID of tool call."
    function: AssistantMessageToolCallFunction
    TYPE: Annotated[
        Annotated[Literal["function"], AfterValidator(validate_const("function"))],
        pydantic.Field(alias="type"),
    ] = "function"
    "The type of tool call."
