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

from __future__ import annotations
from .chatcompletebodytoolchoicefunction import (
    ChatCompleteBodyToolChoiceFunction,
    ChatCompleteBodyToolChoiceFunctionTypedDict,
)
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 ChatCompleteBodyToolChoiceTypedDict(TypedDict):
    function: ChatCompleteBodyToolChoiceFunctionTypedDict
    type: Literal["function"]
    "The type of the tool. Currently, only `function` is supported."


class ChatCompleteBodyToolChoice(BaseModel):
    function: ChatCompleteBodyToolChoiceFunction
    TYPE: Annotated[
        Annotated[Literal["function"], AfterValidator(validate_const("function"))],
        pydantic.Field(alias="type"),
    ] = "function"
    "The type of the tool. Currently, only `function` is supported."
