mistralai 1.10.1__py3-none-any.whl → 1.12.0__py3-none-any.whl

mistralai/models/conversationrestartrequest.py

@@ -5,11 +5,26 @@ from .completionargs import CompletionArgs, CompletionArgsTypedDict
 from .conversationinputs import ConversationInputs, ConversationInputsTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Any, Dict, Literal, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import Any, Dict, Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ConversationRestartRequestHandoffExecution = Literal["client", "server"]
+ConversationRestartRequestHandoffExecution = Literal[
+    "client",
+    "server",
+]
+
+
+ConversationRestartRequestAgentVersionTypedDict = TypeAliasType(
+    "ConversationRestartRequestAgentVersionTypedDict", Union[str, int]
+)
+r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
+
+
+ConversationRestartRequestAgentVersion = TypeAliasType(
+    "ConversationRestartRequestAgentVersion", Union[str, int]
+)
+r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
 
 
 class ConversationRestartRequestTypedDict(TypedDict):
@@ -25,7 +40,9 @@ class ConversationRestartRequestTypedDict(TypedDict):
     r"""White-listed arguments from the completion API"""
     metadata: NotRequired[Nullable[Dict[str, Any]]]
     r"""Custom metadata for the conversation."""
-    agent_version: NotRequired[Nullable[int]]
+    agent_version: NotRequired[
+        Nullable[ConversationRestartRequestAgentVersionTypedDict]
+    ]
     r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
 
 
@@ -49,7 +66,7 @@ class ConversationRestartRequest(BaseModel):
     metadata: OptionalNullable[Dict[str, Any]] = UNSET
     r"""Custom metadata for the conversation."""
 
-    agent_version: OptionalNullable[int] = UNSET
+    agent_version: OptionalNullable[ConversationRestartRequestAgentVersion] = UNSET
     r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
 
     @model_serializer(mode="wrap")

mistralai/models/conversationrestartstreamrequest.py

@@ -5,11 +5,26 @@ from .completionargs import CompletionArgs, CompletionArgsTypedDict
 from .conversationinputs import ConversationInputs, ConversationInputsTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Any, Dict, Literal, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import Any, Dict, Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ConversationRestartStreamRequestHandoffExecution = Literal["client", "server"]
+ConversationRestartStreamRequestHandoffExecution = Literal[
+    "client",
+    "server",
+]
+
+
+ConversationRestartStreamRequestAgentVersionTypedDict = TypeAliasType(
+    "ConversationRestartStreamRequestAgentVersionTypedDict", Union[str, int]
+)
+r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
+
+
+ConversationRestartStreamRequestAgentVersion = TypeAliasType(
+    "ConversationRestartStreamRequestAgentVersion", Union[str, int]
+)
+r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
 
 
 class ConversationRestartStreamRequestTypedDict(TypedDict):
@@ -25,7 +40,9 @@ class ConversationRestartStreamRequestTypedDict(TypedDict):
     r"""White-listed arguments from the completion API"""
     metadata: NotRequired[Nullable[Dict[str, Any]]]
     r"""Custom metadata for the conversation."""
-    agent_version: NotRequired[Nullable[int]]
+    agent_version: NotRequired[
+        Nullable[ConversationRestartStreamRequestAgentVersionTypedDict]
+    ]
     r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
 
 
@@ -51,7 +68,9 @@ class ConversationRestartStreamRequest(BaseModel):
     metadata: OptionalNullable[Dict[str, Any]] = UNSET
     r"""Custom metadata for the conversation."""
 
-    agent_version: OptionalNullable[int] = UNSET
+    agent_version: OptionalNullable[ConversationRestartStreamRequestAgentVersion] = (
+        UNSET
+    )
     r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
 
     @model_serializer(mode="wrap")

mistralai/models/conversationstreamrequest.py

@@ -16,7 +16,11 @@ from typing import Any, Dict, List, Literal, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-ConversationStreamRequestHandoffExecution = Literal["client", "server"]
+ConversationStreamRequestHandoffExecution = Literal[
+    "client",
+    "server",
+]
+
 
 ConversationStreamRequestToolsTypedDict = TypeAliasType(
     "ConversationStreamRequestToolsTypedDict",
@@ -44,6 +48,16 @@ ConversationStreamRequestTools = Annotated[
 ]
 
 
+ConversationStreamRequestAgentVersionTypedDict = TypeAliasType(
+    "ConversationStreamRequestAgentVersionTypedDict", Union[str, int]
+)
+
+
+ConversationStreamRequestAgentVersion = TypeAliasType(
+    "ConversationStreamRequestAgentVersion", Union[str, int]
+)
+
+
 class ConversationStreamRequestTypedDict(TypedDict):
     inputs: ConversationInputsTypedDict
     stream: NotRequired[bool]
@@ -57,7 +71,7 @@ class ConversationStreamRequestTypedDict(TypedDict):
     description: NotRequired[Nullable[str]]
     metadata: NotRequired[Nullable[Dict[str, Any]]]
     agent_id: NotRequired[Nullable[str]]
-    agent_version: NotRequired[Nullable[int]]
+    agent_version: NotRequired[Nullable[ConversationStreamRequestAgentVersionTypedDict]]
     model: NotRequired[Nullable[str]]
 
 
@@ -87,7 +101,7 @@ class ConversationStreamRequest(BaseModel):
 
     agent_id: OptionalNullable[str] = UNSET
 
-    agent_version: OptionalNullable[int] = UNSET
+    agent_version: OptionalNullable[ConversationStreamRequestAgentVersion] = UNSET
 
     model: OptionalNullable[str] = UNSET
 

mistralai/models/documentlibrarytool.py

@@ -6,7 +6,7 @@ from typing import List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-DocumentLibraryToolType = Literal["document_library"]
+DocumentLibraryToolType = Literal["document_library",]
 
 
 class DocumentLibraryToolTypedDict(TypedDict):

mistralai/models/documenturlchunk.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-DocumentURLChunkType = Literal["document_url"]
+DocumentURLChunkType = Literal["document_url",]
 
 
 class DocumentURLChunkTypedDict(TypedDict):

mistralai/models/embeddingdtype.py

@@ -4,4 +4,10 @@ from __future__ import annotations
 from typing import Literal
 
 
-EmbeddingDtype = Literal["float", "int8", "uint8", "binary", "ubinary"]
+EmbeddingDtype = Literal[
+    "float",
+    "int8",
+    "uint8",
+    "binary",
+    "ubinary",
+]

mistralai/models/encodingformat.py

@@ -4,4 +4,7 @@ from __future__ import annotations
 from typing import Literal
 
 
-EncodingFormat = Literal["float", "base64"]
+EncodingFormat = Literal[
+    "float",
+    "base64",
+]

mistralai/models/entitytype.py

@@ -5,5 +5,12 @@ from mistralai.types import UnrecognizedStr
 from typing import Literal, Union
 
 
-EntityType = Union[Literal["User", "Workspace", "Org"], UnrecognizedStr]
+EntityType = Union[
+    Literal[
+        "User",
+        "Workspace",
+        "Org",
+    ],
+    UnrecognizedStr,
+]
 r"""The type of entity, used to share a library."""

mistralai/models/filepurpose.py

@@ -5,4 +5,11 @@ from mistralai.types import UnrecognizedStr
 from typing import Literal, Union
 
 
-FilePurpose = Union[Literal["fine-tune", "batch", "ocr"], UnrecognizedStr]
+FilePurpose = Union[
+    Literal[
+        "fine-tune",
+        "batch",
+        "ocr",
+    ],
+    UnrecognizedStr,
+]

mistralai/models/files_api_routes_list_filesop.py

@@ -5,9 +5,8 @@ from .filepurpose import FilePurpose
 from .sampletype import SampleType
 from .source import Source
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import FieldMetadata, QueryParamMetadata, validate_open_enum
+from mistralai.utils import FieldMetadata, QueryParamMetadata
 from pydantic import model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -20,6 +19,7 @@ class FilesAPIRoutesListFilesRequestTypedDict(TypedDict):
     source: NotRequired[Nullable[List[Source]]]
     search: NotRequired[Nullable[str]]
     purpose: NotRequired[Nullable[FilePurpose]]
+    mimetypes: NotRequired[Nullable[List[str]]]
 
 
 class FilesAPIRoutesListFilesRequest(BaseModel):
@@ -39,16 +39,12 @@ class FilesAPIRoutesListFilesRequest(BaseModel):
     ] = True
 
     sample_type: Annotated[
-        OptionalNullable[
-            List[Annotated[SampleType, PlainValidator(validate_open_enum(False))]]
-        ],
+        OptionalNullable[List[SampleType]],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = UNSET
 
     source: Annotated[
-        OptionalNullable[
-            List[Annotated[Source, PlainValidator(validate_open_enum(False))]]
-        ],
+        OptionalNullable[List[Source]],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = UNSET
 
@@ -58,9 +54,12 @@ class FilesAPIRoutesListFilesRequest(BaseModel):
     ] = UNSET
 
     purpose: Annotated[
-        Annotated[
-            OptionalNullable[FilePurpose], PlainValidator(validate_open_enum(False))
-        ],
+        OptionalNullable[FilePurpose],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = UNSET
+
+    mimetypes: Annotated[
+        OptionalNullable[List[str]],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = UNSET
 
@@ -74,8 +73,9 @@ class FilesAPIRoutesListFilesRequest(BaseModel):
             "source",
             "search",
             "purpose",
+            "mimetypes",
         ]
-        nullable_fields = ["sample_type", "source", "search", "purpose"]
+        nullable_fields = ["sample_type", "source", "search", "purpose", "mimetypes"]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/files_api_routes_upload_fileop.py

@@ -4,8 +4,7 @@ from __future__ import annotations
 from .file import File, FileTypedDict
 from .filepurpose import FilePurpose
 from mistralai.types import BaseModel
-from mistralai.utils import FieldMetadata, MultipartFormMetadata, validate_open_enum
-from pydantic.functional_validators import PlainValidator
+from mistralai.utils import FieldMetadata, MultipartFormMetadata
 from typing import Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -38,7 +37,4 @@ class FilesAPIRoutesUploadFileMultiPartBodyParams(BaseModel):
     ```
     """
 
-    purpose: Annotated[
-        Annotated[Optional[FilePurpose], PlainValidator(validate_open_enum(False))],
-        FieldMetadata(multipart=True),
-    ] = None
+    purpose: Annotated[Optional[FilePurpose], FieldMetadata(multipart=True)] = None

mistralai/models/fileschema.py

@@ -5,10 +5,8 @@ from .filepurpose import FilePurpose
 from .sampletype import SampleType
 from .source import Source
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import validate_open_enum
 import pydantic
 from pydantic import model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -47,11 +45,11 @@ class FileSchema(BaseModel):
     filename: str
     r"""The name of the uploaded file."""
 
-    purpose: Annotated[FilePurpose, PlainValidator(validate_open_enum(False))]
+    purpose: FilePurpose
 
-    sample_type: Annotated[SampleType, PlainValidator(validate_open_enum(False))]
+    sample_type: SampleType
 
-    source: Annotated[Source, PlainValidator(validate_open_enum(False))]
+    source: Source
 
     num_lines: OptionalNullable[int] = UNSET
 

mistralai/models/finetuneablemodeltype.py

@@ -4,4 +4,7 @@ from __future__ import annotations
 from typing import Literal
 
 
-FineTuneableModelType = Literal["completion", "classifier"]
+FineTuneableModelType = Literal[
+    "completion",
+    "classifier",
+]

mistralai/models/ftclassifierlossfunction.py

@@ -4,4 +4,7 @@ from __future__ import annotations
 from typing import Literal
 
 
-FTClassifierLossFunction = Literal["single_class", "multi_class"]
+FTClassifierLossFunction = Literal[
+    "single_class",
+    "multi_class",
+]

mistralai/models/ftmodelcard.py

@@ -12,7 +12,7 @@ from typing import List, Literal, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-FTModelCardType = Literal["fine-tuned"]
+FTModelCardType = Literal["fine-tuned",]
 
 
 class FTModelCardTypedDict(TypedDict):

mistralai/models/functioncallentry.py

@@ -12,9 +12,10 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-FunctionCallEntryObject = Literal["entry"]
+FunctionCallEntryObject = Literal["entry",]
 
-FunctionCallEntryType = Literal["function.call"]
+
+FunctionCallEntryType = Literal["function.call",]
 
 
 class FunctionCallEntryTypedDict(TypedDict):

mistralai/models/functioncallevent.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-FunctionCallEventType = Literal["function.call.delta"]
+FunctionCallEventType = Literal["function.call.delta",]
 
 
 class FunctionCallEventTypedDict(TypedDict):

mistralai/models/functionresultentry.py

@@ -8,9 +8,10 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-FunctionResultEntryObject = Literal["entry"]
+FunctionResultEntryObject = Literal["entry",]
 
-FunctionResultEntryType = Literal["function.result"]
+
+FunctionResultEntryType = Literal["function.result",]
 
 
 class FunctionResultEntryTypedDict(TypedDict):

mistralai/models/functiontool.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-FunctionToolType = Literal["function"]
+FunctionToolType = Literal["function",]
 
 
 class FunctionToolTypedDict(TypedDict):

mistralai/models/githubrepositoryin.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-GithubRepositoryInType = Literal["github"]
+GithubRepositoryInType = Literal["github",]
 
 
 class GithubRepositoryInTypedDict(TypedDict):

mistralai/models/githubrepositoryout.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-GithubRepositoryOutType = Literal["github"]
+GithubRepositoryOutType = Literal["github",]
 
 
 class GithubRepositoryOutTypedDict(TypedDict):

mistralai/models/httpvalidationerror.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .validationerror import ValidationError
+from dataclasses import dataclass, field
 import httpx
 from mistralai.models import MistralError
 from mistralai.types import BaseModel
@@ -12,8 +13,9 @@ class HTTPValidationErrorData(BaseModel):
     detail: Optional[List[ValidationError]] = None
 
 
+@dataclass(unsafe_hash=True)
 class HTTPValidationError(MistralError):
-    data: HTTPValidationErrorData
+    data: HTTPValidationErrorData = field(hash=False)
 
     def __init__(
         self,
@@ -23,4 +25,4 @@ class HTTPValidationError(MistralError):
     ):
         message = body or raw_response.text
         super().__init__(message, raw_response, body)
-        self.data = data
+        object.__setattr__(self, "data", data)

mistralai/models/imagegenerationtool.py

@@ -6,7 +6,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ImageGenerationToolType = Literal["image_generation"]
+ImageGenerationToolType = Literal["image_generation",]
 
 
 class ImageGenerationToolTypedDict(TypedDict):

mistralai/models/imageurlchunk.py

@@ -15,7 +15,7 @@ ImageURLChunkImageURLTypedDict = TypeAliasType(
 ImageURLChunkImageURL = TypeAliasType("ImageURLChunkImageURL", Union[ImageURL, str])
 
 
-ImageURLChunkType = Literal["image_url"]
+ImageURLChunkType = Literal["image_url",]
 
 
 class ImageURLChunkTypedDict(TypedDict):

mistralai/models/jobsout.py

@@ -24,7 +24,7 @@ JobsOutData = Annotated[
 ]
 
 
-JobsOutObject = Literal["list"]
+JobsOutObject = Literal["list",]
 
 
 class JobsOutTypedDict(TypedDict):

mistralai/models/legacyjobmetadataout.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-LegacyJobMetadataOutObject = Literal["job.metadata"]
+LegacyJobMetadataOutObject = Literal["job.metadata",]
 
 
 class LegacyJobMetadataOutTypedDict(TypedDict):

mistralai/models/messageinputentry.py

@@ -12,11 +12,17 @@ from typing import List, Literal, Optional, Union
 from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-Object = Literal["entry"]
+Object = Literal["entry",]
 
-MessageInputEntryType = Literal["message.input"]
 
-MessageInputEntryRole = Literal["assistant", "user"]
+MessageInputEntryType = Literal["message.input",]
+
+
+MessageInputEntryRole = Literal[
+    "assistant",
+    "user",
+]
+
 
 MessageInputEntryContentTypedDict = TypeAliasType(
     "MessageInputEntryContentTypedDict",

mistralai/models/messageoutputentry.py

@@ -12,11 +12,14 @@ from typing import List, Literal, Optional, Union
 from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-MessageOutputEntryObject = Literal["entry"]
+MessageOutputEntryObject = Literal["entry",]
 
-MessageOutputEntryType = Literal["message.output"]
 
-MessageOutputEntryRole = Literal["assistant"]
+MessageOutputEntryType = Literal["message.output",]
+
+
+MessageOutputEntryRole = Literal["assistant",]
+
 
 MessageOutputEntryContentTypedDict = TypeAliasType(
     "MessageOutputEntryContentTypedDict",

mistralai/models/messageoutputevent.py

@@ -9,9 +9,11 @@ from typing import Literal, Optional, Union
 from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-MessageOutputEventType = Literal["message.output.delta"]
+MessageOutputEventType = Literal["message.output.delta",]
+
+
+MessageOutputEventRole = Literal["assistant",]
 
-MessageOutputEventRole = Literal["assistant"]
 
 MessageOutputEventContentTypedDict = TypeAliasType(
     "MessageOutputEventContentTypedDict", Union[str, OutputContentChunksTypedDict]

mistralai/models/mistralerror.py

@@ -2,25 +2,29 @@
 
 import httpx
 from typing import Optional
+from dataclasses import dataclass, field
 
 
+@dataclass(unsafe_hash=True)
 class MistralError(Exception):
     """The base class for all HTTP error responses."""
 
     message: str
     status_code: int
     body: str
-    headers: httpx.Headers
-    raw_response: httpx.Response
+    headers: httpx.Headers = field(hash=False)
+    raw_response: httpx.Response = field(hash=False)
 
     def __init__(
         self, message: str, raw_response: httpx.Response, body: Optional[str] = None
     ):
-        self.message = message
-        self.status_code = raw_response.status_code
-        self.body = body if body is not None else raw_response.text
-        self.headers = raw_response.headers
-        self.raw_response = raw_response
+        object.__setattr__(self, "message", message)
+        object.__setattr__(self, "status_code", raw_response.status_code)
+        object.__setattr__(
+            self, "body", body if body is not None else raw_response.text
+        )
+        object.__setattr__(self, "headers", raw_response.headers)
+        object.__setattr__(self, "raw_response", raw_response)
 
     def __str__(self):
         return self.message

mistralai/models/mistralpromptmode.py

@@ -5,4 +5,8 @@ from mistralai.types import UnrecognizedStr
 from typing import Literal, Union
 
 
-MistralPromptMode = Union[Literal["reasoning"], UnrecognizedStr]
+MistralPromptMode = Union[Literal["reasoning",], UnrecognizedStr]
+r"""Available options to the prompt_mode argument on the chat completion endpoint.
+Values represent high-level intent. Assignment to actual SPs is handled internally.
+System prompt may include knowledge cutoff date, model capabilities, tone to use, safety guidelines, etc.
+"""

mistralai/models/modelcapabilities.py

@@ -16,6 +16,7 @@ class ModelCapabilitiesTypedDict(TypedDict):
     classification: NotRequired[bool]
     moderation: NotRequired[bool]
     audio: NotRequired[bool]
+    audio_transcription: NotRequired[bool]
 
 
 class ModelCapabilities(BaseModel):
@@ -36,3 +37,5 @@ class ModelCapabilities(BaseModel):
     moderation: Optional[bool] = False
 
     audio: Optional[bool] = False
+
+    audio_transcription: Optional[bool] = False

mistralai/models/modelconversation.py

@@ -42,7 +42,7 @@ ModelConversationTools = Annotated[
 ]
 
 
-ModelConversationObject = Literal["conversation"]
+ModelConversationObject = Literal["conversation",]
 
 
 class ModelConversationTypedDict(TypedDict):

mistralai/models/no_response_error.py

@@ -1,12 +1,16 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
+from dataclasses import dataclass
+
+
+@dataclass(unsafe_hash=True)
 class NoResponseError(Exception):
     """Error raised when no HTTP response is received from the server."""
 
     message: str
 
     def __init__(self, message: str = "No response received"):
-        self.message = message
+        object.__setattr__(self, "message", message)
         super().__init__(message)
 
     def __str__(self):