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

from __future__ import annotations
from mistralai.types import BaseModel
import pydantic
from typing import List, Optional, Union
from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict


InputsTypedDict = TypeAliasType("InputsTypedDict", Union[str, List[str]])
r"""Text to embed."""


Inputs = TypeAliasType("Inputs", Union[str, List[str]])
r"""Text to embed."""


class EmbeddingRequestTypedDict(TypedDict):
    inputs: InputsTypedDict
    r"""Text to embed."""
    model: NotRequired[str]
    r"""ID of the model to use."""


class EmbeddingRequest(BaseModel):
    inputs: Annotated[Inputs, pydantic.Field(alias="input")]
    r"""Text to embed."""

    model: Optional[str] = "mistral-embed"
    r"""ID of the model to use."""