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

from __future__ import annotations
from mistralai.types import BaseModel
from typing import Optional
from typing_extensions import NotRequired, TypedDict


class ModelCapabilitiesTypedDict(TypedDict):
    completion_chat: NotRequired[bool]
    function_calling: NotRequired[bool]
    completion_fim: NotRequired[bool]
    fine_tuning: NotRequired[bool]
    vision: NotRequired[bool]
    ocr: NotRequired[bool]
    classification: NotRequired[bool]
    moderation: NotRequired[bool]
    audio: NotRequired[bool]
    audio_transcription: NotRequired[bool]


class ModelCapabilities(BaseModel):
    completion_chat: Optional[bool] = False

    function_calling: Optional[bool] = False

    completion_fim: Optional[bool] = False

    fine_tuning: Optional[bool] = False

    vision: Optional[bool] = False

    ocr: Optional[bool] = False

    classification: Optional[bool] = False

    moderation: Optional[bool] = False

    audio: Optional[bool] = False

    audio_transcription: Optional[bool] = False