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

mistralai/models/classifierjobout.py

@@ -27,16 +27,18 @@ ClassifierJobOutStatus = Literal[
 ]
 r"""The current status of the fine-tuning job."""
 
-ClassifierJobOutObject = Literal["job"]
+
+ClassifierJobOutObject = Literal["job",]
 r"""The object type of the fine-tuning job."""
 
+
 ClassifierJobOutIntegrationsTypedDict = WandbIntegrationOutTypedDict
 
 
 ClassifierJobOutIntegrations = WandbIntegrationOut
 
 
-ClassifierJobOutJobType = Literal["classifier"]
+ClassifierJobOutJobType = Literal["classifier",]
 r"""The type of job (`FT` for fine-tuning)."""
 
 

mistralai/models/codeinterpretertool.py

@@ -6,7 +6,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-CodeInterpreterToolType = Literal["code_interpreter"]
+CodeInterpreterToolType = Literal["code_interpreter",]
 
 
 class CodeInterpreterToolTypedDict(TypedDict):

mistralai/models/completiondetailedjobout.py

@@ -29,7 +29,9 @@ CompletionDetailedJobOutStatus = Literal[
     "CANCELLATION_REQUESTED",
 ]
 
-CompletionDetailedJobOutObject = Literal["job"]
+
+CompletionDetailedJobOutObject = Literal["job",]
+
 
 CompletionDetailedJobOutIntegrationsTypedDict = WandbIntegrationOutTypedDict
 
@@ -37,7 +39,8 @@ CompletionDetailedJobOutIntegrationsTypedDict = WandbIntegrationOutTypedDict
 CompletionDetailedJobOutIntegrations = WandbIntegrationOut
 
 
-CompletionDetailedJobOutJobType = Literal["completion"]
+CompletionDetailedJobOutJobType = Literal["completion",]
+
 
 CompletionDetailedJobOutRepositoriesTypedDict = GithubRepositoryOutTypedDict
 

mistralai/models/completionftmodelout.py

@@ -11,9 +11,10 @@ from typing import List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-CompletionFTModelOutObject = Literal["model"]
+CompletionFTModelOutObject = Literal["model",]
 
-ModelType = Literal["completion"]
+
+ModelType = Literal["completion",]
 
 
 class CompletionFTModelOutTypedDict(TypedDict):

mistralai/models/completionjobout.py

@@ -28,18 +28,21 @@ Status = Literal[
 ]
 r"""The current status of the fine-tuning job."""
 
-CompletionJobOutObject = Literal["job"]
+
+CompletionJobOutObject = Literal["job",]
 r"""The object type of the fine-tuning job."""
 
+
 IntegrationsTypedDict = WandbIntegrationOutTypedDict
 
 
 Integrations = WandbIntegrationOut
 
 
-JobType = Literal["completion"]
+JobType = Literal["completion",]
 r"""The type of job (`FT` for fine-tuning)."""
 
+
 RepositoriesTypedDict = GithubRepositoryOutTypedDict
 
 

mistralai/models/completionresponsestreamchoice.py

@@ -3,15 +3,19 @@
 from __future__ import annotations
 from .deltamessage import DeltaMessage, DeltaMessageTypedDict
 from mistralai.types import BaseModel, Nullable, UNSET_SENTINEL, UnrecognizedStr
-from mistralai.utils import validate_open_enum
 from pydantic import model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing import Literal, Union
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import TypedDict
 
 
 CompletionResponseStreamChoiceFinishReason = Union[
-    Literal["stop", "length", "error", "tool_calls"], UnrecognizedStr
+    Literal[
+        "stop",
+        "length",
+        "error",
+        "tool_calls",
+    ],
+    UnrecognizedStr,
 ]
 
 
@@ -26,10 +30,7 @@ class CompletionResponseStreamChoice(BaseModel):
 
     delta: DeltaMessage
 
-    finish_reason: Annotated[
-        Nullable[CompletionResponseStreamChoiceFinishReason],
-        PlainValidator(validate_open_enum(False)),
-    ]
+    finish_reason: Nullable[CompletionResponseStreamChoiceFinishReason]
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):

mistralai/models/conversationappendrequest.py

@@ -8,7 +8,10 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ConversationAppendRequestHandoffExecution = Literal["client", "server"]
+ConversationAppendRequestHandoffExecution = Literal[
+    "client",
+    "server",
+]
 
 
 class ConversationAppendRequestTypedDict(TypedDict):

mistralai/models/conversationappendstreamrequest.py

@@ -8,7 +8,10 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ConversationAppendStreamRequestHandoffExecution = Literal["client", "server"]
+ConversationAppendStreamRequestHandoffExecution = Literal[
+    "client",
+    "server",
+]
 
 
 class ConversationAppendStreamRequestTypedDict(TypedDict):

mistralai/models/conversationhistory.py

@@ -12,7 +12,8 @@ from typing import List, Literal, Optional, Union
 from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ConversationHistoryObject = Literal["conversation.history"]
+ConversationHistoryObject = Literal["conversation.history",]
+
 
 EntriesTypedDict = TypeAliasType(
     "EntriesTypedDict",

mistralai/models/conversationmessages.py

@@ -7,7 +7,7 @@ from typing import List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ConversationMessagesObject = Literal["conversation.messages"]
+ConversationMessagesObject = Literal["conversation.messages",]
 
 
 class ConversationMessagesTypedDict(TypedDict):

mistralai/models/conversationrequest.py

@@ -16,7 +16,11 @@ from typing import Any, Dict, List, Literal, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-HandoffExecution = Literal["client", "server"]
+HandoffExecution = Literal[
+    "client",
+    "server",
+]
+
 
 ToolsTypedDict = TypeAliasType(
     "ToolsTypedDict",

mistralai/models/conversationresponse.py

@@ -11,7 +11,8 @@ from typing import List, Literal, Optional, Union
 from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ConversationResponseObject = Literal["conversation.response"]
+ConversationResponseObject = Literal["conversation.response",]
+
 
 OutputsTypedDict = TypeAliasType(
     "OutputsTypedDict",

mistralai/models/conversationrestartrequest.py

@@ -9,7 +9,10 @@ from typing import Any, Dict, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ConversationRestartRequestHandoffExecution = Literal["client", "server"]
+ConversationRestartRequestHandoffExecution = Literal[
+    "client",
+    "server",
+]
 
 
 class ConversationRestartRequestTypedDict(TypedDict):

mistralai/models/conversationrestartstreamrequest.py

@@ -9,7 +9,10 @@ from typing import Any, Dict, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ConversationRestartStreamRequestHandoffExecution = Literal["client", "server"]
+ConversationRestartStreamRequestHandoffExecution = Literal[
+    "client",
+    "server",
+]
 
 
 class ConversationRestartStreamRequestTypedDict(TypedDict):

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",

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
 
@@ -39,16 +38,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 +53,7 @@ class FilesAPIRoutesListFilesRequest(BaseModel):
     ] = UNSET
 
     purpose: Annotated[
-        Annotated[
-            OptionalNullable[FilePurpose], PlainValidator(validate_open_enum(False))
-        ],
+        OptionalNullable[FilePurpose],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = UNSET
 

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