mistralai 1.2.2__py3-none-any.whl → 1.2.4__py3-none-any.whl

mistralai/models/chatclassificationrequest.py

@@ -10,15 +10,18 @@ from mistralai.utils import get_discriminator
 import pydantic
 from pydantic import Discriminator, Tag, model_serializer
 from typing import List, Union
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import Annotated, TypeAliasType, TypedDict
 
 
-TwoTypedDict = Union[
-    SystemMessageTypedDict,
-    UserMessageTypedDict,
-    AssistantMessageTypedDict,
-    ToolMessageTypedDict,
-]
+TwoTypedDict = TypeAliasType(
+    "TwoTypedDict",
+    Union[
+        SystemMessageTypedDict,
+        UserMessageTypedDict,
+        AssistantMessageTypedDict,
+        ToolMessageTypedDict,
+    ],
+)
 
 
 Two = Annotated[
@@ -32,12 +35,15 @@ Two = Annotated[
 ]
 
 
-OneTypedDict = Union[
-    SystemMessageTypedDict,
-    UserMessageTypedDict,
-    AssistantMessageTypedDict,
-    ToolMessageTypedDict,
-]
+OneTypedDict = TypeAliasType(
+    "OneTypedDict",
+    Union[
+        SystemMessageTypedDict,
+        UserMessageTypedDict,
+        AssistantMessageTypedDict,
+        ToolMessageTypedDict,
+    ],
+)
 
 
 One = Annotated[
@@ -51,13 +57,16 @@ One = Annotated[
 ]
 
 
-ChatClassificationRequestInputsTypedDict = Union[
-    List[OneTypedDict], List[List[TwoTypedDict]]
-]
+ChatClassificationRequestInputsTypedDict = TypeAliasType(
+    "ChatClassificationRequestInputsTypedDict",
+    Union[List[OneTypedDict], List[List[TwoTypedDict]]],
+)
 r"""Chat to classify"""
 
 
-ChatClassificationRequestInputs = Union[List[One], List[List[Two]]]
+ChatClassificationRequestInputs = TypeAliasType(
+    "ChatClassificationRequestInputs", Union[List[One], List[List[Two]]]
+)
 r"""Chat to classify"""
 
 

mistralai/models/chatcompletionrequest.py

@@ -13,23 +13,26 @@ from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_
 from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
 from typing import List, Optional, Union
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-StopTypedDict = Union[str, List[str]]
+StopTypedDict = TypeAliasType("StopTypedDict", Union[str, List[str]])
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-Stop = Union[str, List[str]]
+Stop = TypeAliasType("Stop", Union[str, List[str]])
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-MessagesTypedDict = Union[
-    SystemMessageTypedDict,
-    UserMessageTypedDict,
-    AssistantMessageTypedDict,
-    ToolMessageTypedDict,
-]
+MessagesTypedDict = TypeAliasType(
+    "MessagesTypedDict",
+    Union[
+        SystemMessageTypedDict,
+        UserMessageTypedDict,
+        AssistantMessageTypedDict,
+        ToolMessageTypedDict,
+    ],
+)
 
 
 Messages = Annotated[
@@ -43,10 +46,15 @@ Messages = Annotated[
 ]
 
 
-ChatCompletionRequestToolChoiceTypedDict = Union[ToolChoiceTypedDict, ToolChoiceEnum]
+ChatCompletionRequestToolChoiceTypedDict = TypeAliasType(
+    "ChatCompletionRequestToolChoiceTypedDict",
+    Union[ToolChoiceTypedDict, ToolChoiceEnum],
+)
 
 
-ChatCompletionRequestToolChoice = Union[ToolChoice, ToolChoiceEnum]
+ChatCompletionRequestToolChoice = TypeAliasType(
+    "ChatCompletionRequestToolChoice", Union[ToolChoice, ToolChoiceEnum]
+)
 
 
 class ChatCompletionRequestTypedDict(TypedDict):

mistralai/models/chatcompletionstreamrequest.py

@@ -13,23 +13,30 @@ from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_
 from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
 from typing import List, Optional, Union
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-ChatCompletionStreamRequestStopTypedDict = Union[str, List[str]]
+ChatCompletionStreamRequestStopTypedDict = TypeAliasType(
+    "ChatCompletionStreamRequestStopTypedDict", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-ChatCompletionStreamRequestStop = Union[str, List[str]]
+ChatCompletionStreamRequestStop = TypeAliasType(
+    "ChatCompletionStreamRequestStop", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-ChatCompletionStreamRequestMessagesTypedDict = Union[
-    SystemMessageTypedDict,
-    UserMessageTypedDict,
-    AssistantMessageTypedDict,
-    ToolMessageTypedDict,
-]
+ChatCompletionStreamRequestMessagesTypedDict = TypeAliasType(
+    "ChatCompletionStreamRequestMessagesTypedDict",
+    Union[
+        SystemMessageTypedDict,
+        UserMessageTypedDict,
+        AssistantMessageTypedDict,
+        ToolMessageTypedDict,
+    ],
+)
 
 
 ChatCompletionStreamRequestMessages = Annotated[
@@ -43,12 +50,15 @@ ChatCompletionStreamRequestMessages = Annotated[
 ]
 
 
-ChatCompletionStreamRequestToolChoiceTypedDict = Union[
-    ToolChoiceTypedDict, ToolChoiceEnum
-]
+ChatCompletionStreamRequestToolChoiceTypedDict = TypeAliasType(
+    "ChatCompletionStreamRequestToolChoiceTypedDict",
+    Union[ToolChoiceTypedDict, ToolChoiceEnum],
+)
 
 
-ChatCompletionStreamRequestToolChoice = Union[ToolChoice, ToolChoiceEnum]
+ChatCompletionStreamRequestToolChoice = TypeAliasType(
+    "ChatCompletionStreamRequestToolChoice", Union[ToolChoice, ToolChoiceEnum]
+)
 
 
 class ChatCompletionStreamRequestTypedDict(TypedDict):

mistralai/models/classificationrequest.py

@@ -5,14 +5,18 @@ from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_
 import pydantic
 from pydantic import model_serializer
 from typing import List, Union
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-ClassificationRequestInputsTypedDict = Union[str, List[str]]
+ClassificationRequestInputsTypedDict = TypeAliasType(
+    "ClassificationRequestInputsTypedDict", Union[str, List[str]]
+)
 r"""Text to classify."""
 
 
-ClassificationRequestInputs = Union[str, List[str]]
+ClassificationRequestInputs = TypeAliasType(
+    "ClassificationRequestInputs", Union[str, List[str]]
+)
 r"""Text to classify."""
 
 

mistralai/models/contentchunk.py

@@ -2,19 +2,25 @@
 
 from __future__ import annotations
 from .imageurlchunk import ImageURLChunk, ImageURLChunkTypedDict
+from .referencechunk import ReferenceChunk, ReferenceChunkTypedDict
 from .textchunk import TextChunk, TextChunkTypedDict
 from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag
 from typing import Union
-from typing_extensions import Annotated
+from typing_extensions import Annotated, TypeAliasType
 
 
-ContentChunkTypedDict = Union[TextChunkTypedDict, ImageURLChunkTypedDict]
+ContentChunkTypedDict = TypeAliasType(
+    "ContentChunkTypedDict",
+    Union[TextChunkTypedDict, ImageURLChunkTypedDict, ReferenceChunkTypedDict],
+)
 
 
 ContentChunk = Annotated[
     Union[
-        Annotated[ImageURLChunk, Tag("image_url")], Annotated[TextChunk, Tag("text")]
+        Annotated[ImageURLChunk, Tag("image_url")],
+        Annotated[TextChunk, Tag("text")],
+        Annotated[ReferenceChunk, Tag("reference")],
     ],
     Discriminator(lambda m: get_discriminator(m, "type", "type")),
 ]

mistralai/models/deltamessage.py

@@ -6,13 +6,15 @@ from .toolcall import ToolCall, ToolCallTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
 from typing import List, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ContentTypedDict = Union[str, List[ContentChunkTypedDict]]
+ContentTypedDict = TypeAliasType(
+    "ContentTypedDict", Union[str, List[ContentChunkTypedDict]]
+)
 
 
-Content = Union[str, List[ContentChunk]]
+Content = TypeAliasType("Content", Union[str, List[ContentChunk]])
 
 
 class DeltaMessageTypedDict(TypedDict):

mistralai/models/detailedjobout.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 from .checkpointout import CheckpointOut, CheckpointOutTypedDict
 from .eventout import EventOut, EventOutTypedDict
-from .finetuneablemodel import FineTuneableModel
 from .githubrepositoryout import GithubRepositoryOut, GithubRepositoryOutTypedDict
 from .jobmetadataout import JobMetadataOut, JobMetadataOutTypedDict
 from .trainingparameters import TrainingParameters, TrainingParametersTypedDict
@@ -48,7 +47,7 @@ class DetailedJobOutTypedDict(TypedDict):
     id: str
     auto_start: bool
     hyperparameters: TrainingParametersTypedDict
-    model: FineTuneableModel
+    model: str
     r"""The name of the model to fine-tune."""
     status: DetailedJobOutStatus
     job_type: str
@@ -75,7 +74,7 @@ class DetailedJobOut(BaseModel):
 
     hyperparameters: TrainingParameters
 
-    model: FineTuneableModel
+    model: str
     r"""The name of the model to fine-tune."""
 
     status: DetailedJobOutStatus

mistralai/models/embeddingrequest.py

@@ -5,14 +5,14 @@ from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_
 import pydantic
 from pydantic import model_serializer
 from typing import List, Optional, Union
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-InputsTypedDict = Union[str, List[str]]
+InputsTypedDict = TypeAliasType("InputsTypedDict", Union[str, List[str]])
 r"""Text to embed."""
 
 
-Inputs = Union[str, List[str]]
+Inputs = TypeAliasType("Inputs", Union[str, List[str]])
 r"""Text to embed."""
 
 

mistralai/models/files_api_routes_get_signed_urlop.py

@@ -0,0 +1,25 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata, QueryParamMetadata
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class FilesAPIRoutesGetSignedURLRequestTypedDict(TypedDict):
+    file_id: str
+    expiry: NotRequired[int]
+    r"""Number of hours before the url becomes invalid. Defaults to 24h"""
+
+
+class FilesAPIRoutesGetSignedURLRequest(BaseModel):
+    file_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    expiry: Annotated[
+        Optional[int],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = 24
+    r"""Number of hours before the url becomes invalid. Defaults to 24h"""

mistralai/models/filesignedurl.py

@@ -0,0 +1,13 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from typing_extensions import TypedDict
+
+
+class FileSignedURLTypedDict(TypedDict):
+    url: str
+
+
+class FileSignedURL(BaseModel):
+    url: str

mistralai/models/fimcompletionrequest.py

@@ -4,14 +4,18 @@ from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
 from typing import List, Optional, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-FIMCompletionRequestStopTypedDict = Union[str, List[str]]
+FIMCompletionRequestStopTypedDict = TypeAliasType(
+    "FIMCompletionRequestStopTypedDict", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-FIMCompletionRequestStop = Union[str, List[str]]
+FIMCompletionRequestStop = TypeAliasType(
+    "FIMCompletionRequestStop", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 

mistralai/models/fimcompletionstreamrequest.py

@@ -4,14 +4,18 @@ from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
 from typing import List, Optional, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-FIMCompletionStreamRequestStopTypedDict = Union[str, List[str]]
+FIMCompletionStreamRequestStopTypedDict = TypeAliasType(
+    "FIMCompletionStreamRequestStopTypedDict", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-FIMCompletionStreamRequestStop = Union[str, List[str]]
+FIMCompletionStreamRequestStop = TypeAliasType(
+    "FIMCompletionStreamRequestStop", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 

mistralai/models/functioncall.py

@@ -3,13 +3,13 @@
 from __future__ import annotations
 from mistralai.types import BaseModel
 from typing import Any, Dict, Union
-from typing_extensions import TypedDict
+from typing_extensions import TypeAliasType, TypedDict
 
 
-ArgumentsTypedDict = Union[Dict[str, Any], str]
+ArgumentsTypedDict = TypeAliasType("ArgumentsTypedDict", Union[Dict[str, Any], str])
 
 
-Arguments = Union[Dict[str, Any], str]
+Arguments = TypeAliasType("Arguments", Union[Dict[str, Any], str])
 
 
 class FunctionCallTypedDict(TypedDict):

mistralai/models/imageurlchunk.py

@@ -3,26 +3,26 @@
 from __future__ import annotations
 from .imageurl import ImageURL, ImageURLTypedDict
 from mistralai.types import BaseModel
-from mistralai.utils import validate_const
-import pydantic
-from pydantic.functional_validators import AfterValidator
 from typing import Literal, Optional, Union
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ImageURLChunkType = Literal["image_url"]
+ImageURLChunkImageURLTypedDict = TypeAliasType(
+    "ImageURLChunkImageURLTypedDict", Union[ImageURLTypedDict, str]
+)
+
 
-ImageURLChunkImageURLTypedDict = Union[ImageURLTypedDict, str]
+ImageURLChunkImageURL = TypeAliasType("ImageURLChunkImageURL", Union[ImageURL, str])
 
 
-ImageURLChunkImageURL = Union[ImageURL, str]
+ImageURLChunkType = Literal["image_url"]
 
 
 class ImageURLChunkTypedDict(TypedDict):
     r"""{\"type\":\"image_url\",\"image_url\":{\"url\":\"data:image/png;base64,iVBORw0"""
 
     image_url: ImageURLChunkImageURLTypedDict
-    type: ImageURLChunkType
+    type: NotRequired[ImageURLChunkType]
 
 
 class ImageURLChunk(BaseModel):
@@ -30,9 +30,4 @@ class ImageURLChunk(BaseModel):
 
     image_url: ImageURLChunkImageURL
 
-    TYPE: Annotated[
-        Annotated[
-            Optional[ImageURLChunkType], AfterValidator(validate_const("image_url"))
-        ],
-        pydantic.Field(alias="type"),
-    ] = "image_url"
+    type: Optional[ImageURLChunkType] = "image_url"

mistralai/models/jobin.py

@@ -1,7 +1,6 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .finetuneablemodel import FineTuneableModel
 from .githubrepositoryin import GithubRepositoryIn, GithubRepositoryInTypedDict
 from .trainingfile import TrainingFile, TrainingFileTypedDict
 from .trainingparametersin import TrainingParametersIn, TrainingParametersInTypedDict
@@ -25,7 +24,7 @@ JobInRepositories = GithubRepositoryIn
 
 
 class JobInTypedDict(TypedDict):
-    model: FineTuneableModel
+    model: str
     r"""The name of the model to fine-tune."""
     hyperparameters: TrainingParametersInTypedDict
     r"""The fine-tuning hyperparameter settings used in a fine-tune job."""
@@ -42,7 +41,7 @@ class JobInTypedDict(TypedDict):
 
 
 class JobIn(BaseModel):
-    model: FineTuneableModel
+    model: str
     r"""The name of the model to fine-tune."""
 
     hyperparameters: TrainingParametersIn

mistralai/models/jobout.py

@@ -1,7 +1,6 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .finetuneablemodel import FineTuneableModel
 from .githubrepositoryout import GithubRepositoryOut, GithubRepositoryOutTypedDict
 from .jobmetadataout import JobMetadataOut, JobMetadataOutTypedDict
 from .trainingparameters import TrainingParameters, TrainingParametersTypedDict
@@ -49,7 +48,7 @@ class JobOutTypedDict(TypedDict):
     r"""The ID of the job."""
     auto_start: bool
     hyperparameters: TrainingParametersTypedDict
-    model: FineTuneableModel
+    model: str
     r"""The name of the model to fine-tune."""
     status: Status
     r"""The current status of the fine-tuning job."""
@@ -85,7 +84,7 @@ class JobOut(BaseModel):
 
     hyperparameters: TrainingParameters
 
-    model: FineTuneableModel
+    model: str
     r"""The name of the model to fine-tune."""
 
     status: Status

mistralai/models/jobs_api_routes_fine_tuning_create_fine_tuning_jobop.py

@@ -4,13 +4,18 @@ from __future__ import annotations
 from .jobout import JobOut, JobOutTypedDict
 from .legacyjobmetadataout import LegacyJobMetadataOut, LegacyJobMetadataOutTypedDict
 from typing import Union
+from typing_extensions import TypeAliasType
 
 
-JobsAPIRoutesFineTuningCreateFineTuningJobResponseTypedDict = Union[
-    LegacyJobMetadataOutTypedDict, JobOutTypedDict
-]
+JobsAPIRoutesFineTuningCreateFineTuningJobResponseTypedDict = TypeAliasType(
+    "JobsAPIRoutesFineTuningCreateFineTuningJobResponseTypedDict",
+    Union[LegacyJobMetadataOutTypedDict, JobOutTypedDict],
+)
 r"""OK"""
 
 
-JobsAPIRoutesFineTuningCreateFineTuningJobResponse = Union[LegacyJobMetadataOut, JobOut]
+JobsAPIRoutesFineTuningCreateFineTuningJobResponse = TypeAliasType(
+    "JobsAPIRoutesFineTuningCreateFineTuningJobResponse",
+    Union[LegacyJobMetadataOut, JobOut],
+)
 r"""OK"""

mistralai/models/modellist.py

@@ -7,10 +7,12 @@ from mistralai.types import BaseModel
 from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag
 from typing import List, Optional, Union
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-DataTypedDict = Union[BaseModelCardTypedDict, FTModelCardTypedDict]
+DataTypedDict = TypeAliasType(
+    "DataTypedDict", Union[BaseModelCardTypedDict, FTModelCardTypedDict]
+)
 
 
 Data = Annotated[

mistralai/models/referencechunk.py

@@ -0,0 +1,20 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from typing import List, Literal, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+ReferenceChunkType = Literal["reference"]
+
+
+class ReferenceChunkTypedDict(TypedDict):
+    reference_ids: List[int]
+    type: NotRequired[ReferenceChunkType]
+
+
+class ReferenceChunk(BaseModel):
+    reference_ids: List[int]
+
+    type: Optional[ReferenceChunkType] = "reference"

mistralai/models/retrieve_model_v1_models_model_id_getop.py

@@ -7,7 +7,7 @@ from mistralai.types import BaseModel
 from mistralai.utils import FieldMetadata, PathParamMetadata, get_discriminator
 from pydantic import Discriminator, Tag
 from typing import Union
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import Annotated, TypeAliasType, TypedDict
 
 
 class RetrieveModelV1ModelsModelIDGetRequestTypedDict(TypedDict):
@@ -22,9 +22,10 @@ class RetrieveModelV1ModelsModelIDGetRequest(BaseModel):
     r"""The ID of the model to retrieve."""
 
 
-RetrieveModelV1ModelsModelIDGetResponseRetrieveModelV1ModelsModelIDGetTypedDict = Union[
-    BaseModelCardTypedDict, FTModelCardTypedDict
-]
+RetrieveModelV1ModelsModelIDGetResponseRetrieveModelV1ModelsModelIDGetTypedDict = TypeAliasType(
+    "RetrieveModelV1ModelsModelIDGetResponseRetrieveModelV1ModelsModelIDGetTypedDict",
+    Union[BaseModelCardTypedDict, FTModelCardTypedDict],
+)
 r"""Successful Response"""
 
 

mistralai/models/systemmessage.py

@@ -4,13 +4,17 @@ from __future__ import annotations
 from .textchunk import TextChunk, TextChunkTypedDict
 from mistralai.types import BaseModel
 from typing import List, Literal, Optional, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-SystemMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
+SystemMessageContentTypedDict = TypeAliasType(
+    "SystemMessageContentTypedDict", Union[str, List[TextChunkTypedDict]]
+)
 
 
-SystemMessageContent = Union[str, List[TextChunk]]
+SystemMessageContent = TypeAliasType(
+    "SystemMessageContent", Union[str, List[TextChunk]]
+)
 
 
 Role = Literal["system"]

mistralai/models/textchunk.py

@@ -2,11 +2,8 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel
-from mistralai.utils import validate_const
-import pydantic
-from pydantic.functional_validators import AfterValidator
 from typing import Literal, Optional
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 TextChunkType = Literal["text"]
@@ -14,13 +11,10 @@ TextChunkType = Literal["text"]
 
 class TextChunkTypedDict(TypedDict):
     text: str
-    type: TextChunkType
+    type: NotRequired[TextChunkType]
 
 
 class TextChunk(BaseModel):
     text: str
 
-    TYPE: Annotated[
-        Annotated[Optional[TextChunkType], AfterValidator(validate_const("text"))],
-        pydantic.Field(alias="type"),
-    ] = "text"
+    type: Optional[TextChunkType] = "text"