mistralai 1.2.3__py3-none-any.whl → 1.2.5__py3-none-any.whl

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

@@ -7,12 +7,13 @@ 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, ReferenceChunkTypedDict
-]
+ContentChunkTypedDict = TypeAliasType(
+    "ContentChunkTypedDict",
+    Union[TextChunkTypedDict, ImageURLChunkTypedDict, ReferenceChunkTypedDict],
+)
 
 
 ContentChunk = Annotated[

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

@@ -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 List, Literal, Optional
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 ReferenceChunkType = Literal["reference"]
@@ -14,15 +11,10 @@ ReferenceChunkType = Literal["reference"]
 
 class ReferenceChunkTypedDict(TypedDict):
     reference_ids: List[int]
-    type: ReferenceChunkType
+    type: NotRequired[ReferenceChunkType]
 
 
 class ReferenceChunk(BaseModel):
     reference_ids: List[int]
 
-    TYPE: Annotated[
-        Annotated[
-            Optional[ReferenceChunkType], AfterValidator(validate_const("reference"))
-        ],
-        pydantic.Field(alias="type"),
-    ] = "reference"
+    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"

mistralai/models/toolmessage.py

@@ -5,13 +5,15 @@ from .contentchunk import ContentChunk, ContentChunkTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
 from typing import List, Literal, Optional, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ToolMessageContentTypedDict = Union[str, List[ContentChunkTypedDict]]
+ToolMessageContentTypedDict = TypeAliasType(
+    "ToolMessageContentTypedDict", Union[str, List[ContentChunkTypedDict]]
+)
 
 
-ToolMessageContent = Union[str, List[ContentChunk]]
+ToolMessageContent = TypeAliasType("ToolMessageContent", Union[str, List[ContentChunk]])
 
 
 ToolMessageRole = Literal["tool"]

mistralai/models/usermessage.py

@@ -5,13 +5,15 @@ from .contentchunk import ContentChunk, ContentChunkTypedDict
 from mistralai.types import BaseModel, Nullable, UNSET_SENTINEL
 from pydantic import model_serializer
 from typing import List, Literal, Optional, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-UserMessageContentTypedDict = Union[str, List[ContentChunkTypedDict]]
+UserMessageContentTypedDict = TypeAliasType(
+    "UserMessageContentTypedDict", Union[str, List[ContentChunkTypedDict]]
+)
 
 
-UserMessageContent = Union[str, List[ContentChunk]]
+UserMessageContent = TypeAliasType("UserMessageContent", Union[str, List[ContentChunk]])
 
 
 UserMessageRole = Literal["user"]

mistralai/models/validationerror.py

@@ -3,13 +3,13 @@
 from __future__ import annotations
 from mistralai.types import BaseModel
 from typing import List, Union
-from typing_extensions import TypedDict
+from typing_extensions import TypeAliasType, TypedDict
 
 
-LocTypedDict = Union[str, int]
+LocTypedDict = TypeAliasType("LocTypedDict", Union[str, int])
 
 
-Loc = Union[str, int]
+Loc = TypeAliasType("Loc", Union[str, int])
 
 
 class ValidationErrorTypedDict(TypedDict):

mistralai/sdk.py

@@ -129,3 +129,17 @@ class Mistral(BaseSDK):
         self.agents = Agents(self.sdk_configuration)
         self.embeddings = Embeddings(self.sdk_configuration)
         self.classifiers = Classifiers(self.sdk_configuration)
+
+    def __enter__(self):
+        return self
+
+    async def __aenter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        if self.sdk_configuration.client is not None:
+            self.sdk_configuration.client.close()
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        if self.sdk_configuration.async_client is not None:
+            await self.sdk_configuration.async_client.aclose()

mistralai/sdkconfiguration.py

@@ -28,9 +28,9 @@ class SDKConfiguration:
     server: Optional[str] = ""
     language: str = "python"
     openapi_doc_version: str = "0.0.2"
-    sdk_version: str = "1.2.3"
-    gen_version: str = "2.460.1"
-    user_agent: str = "speakeasy-sdk/python 1.2.3 2.460.1 0.0.2 mistralai"
+    sdk_version: str = "1.2.5"
+    gen_version: str = "2.470.1"
+    user_agent: str = "speakeasy-sdk/python 1.2.5 2.470.1 0.0.2 mistralai"
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None