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

mistralai/models/retrievefileout.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
 
 
@@ -48,11 +46,11 @@ class RetrieveFileOut(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
 
     deleted: bool
 

mistralai/models/sampletype.py

@@ -6,6 +6,12 @@ from typing import Literal, Union
 
 
 SampleType = Union[
-    Literal["pretrain", "instruct", "batch_request", "batch_result", "batch_error"],
+    Literal[
+        "pretrain",
+        "instruct",
+        "batch_request",
+        "batch_result",
+        "batch_error",
+    ],
     UnrecognizedStr,
 ]

mistralai/models/sdkerror.py

@@ -2,12 +2,14 @@
 
 import httpx
 from typing import Optional
+from dataclasses import dataclass
 
 from mistralai.models import MistralError
 
 MAX_MESSAGE_LEN = 10_000
 
 
+@dataclass(unsafe_hash=True)
 class SDKError(MistralError):
     """The fallback error class if no more specific error class is matched."""
 

mistralai/models/shareenum.py

@@ -5,4 +5,10 @@ from mistralai.types import UnrecognizedStr
 from typing import Literal, Union
 
 
-ShareEnum = Union[Literal["Viewer", "Editor"], UnrecognizedStr]
+ShareEnum = Union[
+    Literal[
+        "Viewer",
+        "Editor",
+    ],
+    UnrecognizedStr,
+]

mistralai/models/sharingdelete.py

@@ -3,10 +3,8 @@
 from __future__ import annotations
 from .entitytype import EntityType
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import validate_open_enum
 from pydantic import model_serializer
-from pydantic.functional_validators import PlainValidator
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 class SharingDeleteTypedDict(TypedDict):
@@ -21,7 +19,7 @@ class SharingDelete(BaseModel):
     share_with_uuid: str
     r"""The id of the entity (user, workspace or organization) to share with"""
 
-    share_with_type: Annotated[EntityType, PlainValidator(validate_open_enum(False))]
+    share_with_type: EntityType
     r"""The type of entity, used to share a library."""
 
     org_id: OptionalNullable[str] = UNSET

mistralai/models/sharingin.py

@@ -4,10 +4,8 @@ from __future__ import annotations
 from .entitytype import EntityType
 from .shareenum import ShareEnum
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import validate_open_enum
 from pydantic import model_serializer
-from pydantic.functional_validators import PlainValidator
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 class SharingInTypedDict(TypedDict):
@@ -20,12 +18,12 @@ class SharingInTypedDict(TypedDict):
 
 
 class SharingIn(BaseModel):
-    level: Annotated[ShareEnum, PlainValidator(validate_open_enum(False))]
+    level: ShareEnum
 
     share_with_uuid: str
     r"""The id of the entity (user, workspace or organization) to share with"""
 
-    share_with_type: Annotated[EntityType, PlainValidator(validate_open_enum(False))]
+    share_with_type: EntityType
     r"""The type of entity, used to share a library."""
 
     org_id: OptionalNullable[str] = UNSET

mistralai/models/source.py

@@ -5,4 +5,11 @@ from mistralai.types import UnrecognizedStr
 from typing import Literal, Union
 
 
-Source = Union[Literal["upload", "repository", "mistral"], UnrecognizedStr]
+Source = Union[
+    Literal[
+        "upload",
+        "repository",
+        "mistral",
+    ],
+    UnrecognizedStr,
+]

mistralai/models/systemmessage.py

@@ -21,7 +21,7 @@ SystemMessageContent = TypeAliasType(
 )
 
 
-Role = Literal["system"]
+Role = Literal["system",]
 
 
 class SystemMessageTypedDict(TypedDict):

mistralai/models/textchunk.py

@@ -6,7 +6,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-TextChunkType = Literal["text"]
+TextChunkType = Literal["text",]
 
 
 class TextChunkTypedDict(TypedDict):

mistralai/models/thinkchunk.py

@@ -16,7 +16,7 @@ ThinkingTypedDict = TypeAliasType(
 Thinking = TypeAliasType("Thinking", Union[ReferenceChunk, TextChunk])
 
 
-ThinkChunkType = Literal["thinking"]
+ThinkChunkType = Literal["thinking",]
 
 
 class ThinkChunkTypedDict(TypedDict):

mistralai/models/timestampgranularity.py

@@ -4,4 +4,4 @@ from __future__ import annotations
 from typing import Literal
 
 
-TimestampGranularity = Literal["segment"]
+TimestampGranularity = Literal["segment",]

mistralai/models/tool.py

@@ -4,10 +4,8 @@ from __future__ import annotations
 from .function import Function, FunctionTypedDict
 from .tooltypes import ToolTypes
 from mistralai.types import BaseModel
-from mistralai.utils import validate_open_enum
-from pydantic.functional_validators import PlainValidator
 from typing import Optional
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 class ToolTypedDict(TypedDict):
@@ -18,6 +16,4 @@ class ToolTypedDict(TypedDict):
 class Tool(BaseModel):
     function: Function
 
-    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = (
-        None
-    )
+    type: Optional[ToolTypes] = None

mistralai/models/toolcall.py

@@ -4,10 +4,8 @@ from __future__ import annotations
 from .functioncall import FunctionCall, FunctionCallTypedDict
 from .tooltypes import ToolTypes
 from mistralai.types import BaseModel
-from mistralai.utils import validate_open_enum
-from pydantic.functional_validators import PlainValidator
 from typing import Optional
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 class ToolCallTypedDict(TypedDict):
@@ -22,8 +20,6 @@ class ToolCall(BaseModel):
 
     id: Optional[str] = "null"
 
-    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = (
-        None
-    )
+    type: Optional[ToolTypes] = None
 
     index: Optional[int] = 0

mistralai/models/toolchoice.py

@@ -4,10 +4,8 @@ from __future__ import annotations
 from .functionname import FunctionName, FunctionNameTypedDict
 from .tooltypes import ToolTypes
 from mistralai.types import BaseModel
-from mistralai.utils import validate_open_enum
-from pydantic.functional_validators import PlainValidator
 from typing import Optional
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 class ToolChoiceTypedDict(TypedDict):
@@ -24,6 +22,4 @@ class ToolChoice(BaseModel):
     function: FunctionName
     r"""this restriction of `Function` is used to select a specific function to call"""
 
-    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = (
-        None
-    )
+    type: Optional[ToolTypes] = None

mistralai/models/toolchoiceenum.py

@@ -4,4 +4,9 @@ from __future__ import annotations
 from typing import Literal
 
 
-ToolChoiceEnum = Literal["auto", "none", "any", "required"]
+ToolChoiceEnum = Literal[
+    "auto",
+    "none",
+    "any",
+    "required",
+]

mistralai/models/toolexecutiondeltaevent.py

@@ -8,7 +8,8 @@ from typing import Literal, Optional, Union
 from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ToolExecutionDeltaEventType = Literal["tool.execution.delta"]
+ToolExecutionDeltaEventType = Literal["tool.execution.delta",]
+
 
 ToolExecutionDeltaEventNameTypedDict = TypeAliasType(
     "ToolExecutionDeltaEventNameTypedDict", Union[BuiltInConnectors, str]

mistralai/models/toolexecutiondoneevent.py

@@ -8,7 +8,8 @@ from typing import Any, Dict, Literal, Optional, Union
 from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ToolExecutionDoneEventType = Literal["tool.execution.done"]
+ToolExecutionDoneEventType = Literal["tool.execution.done",]
+
 
 ToolExecutionDoneEventNameTypedDict = TypeAliasType(
     "ToolExecutionDoneEventNameTypedDict", Union[BuiltInConnectors, str]

mistralai/models/toolexecutionentry.py

@@ -9,9 +9,11 @@ from typing import Any, Dict, Literal, Optional, Union
 from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ToolExecutionEntryObject = Literal["entry"]
+ToolExecutionEntryObject = Literal["entry",]
+
+
+ToolExecutionEntryType = Literal["tool.execution",]
 
-ToolExecutionEntryType = Literal["tool.execution"]
 
 NameTypedDict = TypeAliasType("NameTypedDict", Union[BuiltInConnectors, str])
 

mistralai/models/toolexecutionstartedevent.py

@@ -8,7 +8,8 @@ from typing import Literal, Optional, Union
 from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ToolExecutionStartedEventType = Literal["tool.execution.started"]
+ToolExecutionStartedEventType = Literal["tool.execution.started",]
+
 
 ToolExecutionStartedEventNameTypedDict = TypeAliasType(
     "ToolExecutionStartedEventNameTypedDict", Union[BuiltInConnectors, str]

mistralai/models/toolfilechunk.py

@@ -4,15 +4,23 @@ from __future__ import annotations
 from .builtinconnectors import BuiltInConnectors
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Literal, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ToolFileChunkType = Literal["tool_file"]
+ToolFileChunkType = Literal["tool_file",]
+
+
+ToolFileChunkToolTypedDict = TypeAliasType(
+    "ToolFileChunkToolTypedDict", Union[BuiltInConnectors, str]
+)
+
+
+ToolFileChunkTool = TypeAliasType("ToolFileChunkTool", Union[BuiltInConnectors, str])
 
 
 class ToolFileChunkTypedDict(TypedDict):
-    tool: BuiltInConnectors
+    tool: ToolFileChunkToolTypedDict
     file_id: str
     type: NotRequired[ToolFileChunkType]
     file_name: NotRequired[Nullable[str]]
@@ -20,7 +28,7 @@ class ToolFileChunkTypedDict(TypedDict):
 
 
 class ToolFileChunk(BaseModel):
-    tool: BuiltInConnectors
+    tool: ToolFileChunkTool
 
     file_id: str
 

mistralai/models/toolmessage.py

@@ -16,7 +16,7 @@ ToolMessageContentTypedDict = TypeAliasType(
 ToolMessageContent = TypeAliasType("ToolMessageContent", Union[str, List[ContentChunk]])
 
 
-ToolMessageRole = Literal["tool"]
+ToolMessageRole = Literal["tool",]
 
 
 class ToolMessageTypedDict(TypedDict):

mistralai/models/toolreferencechunk.py

@@ -4,15 +4,25 @@ from __future__ import annotations
 from .builtinconnectors import BuiltInConnectors
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Literal, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ToolReferenceChunkType = Literal["tool_reference"]
+ToolReferenceChunkType = Literal["tool_reference",]
+
+
+ToolReferenceChunkToolTypedDict = TypeAliasType(
+    "ToolReferenceChunkToolTypedDict", Union[BuiltInConnectors, str]
+)
+
+
+ToolReferenceChunkTool = TypeAliasType(
+    "ToolReferenceChunkTool", Union[BuiltInConnectors, str]
+)
 
 
 class ToolReferenceChunkTypedDict(TypedDict):
-    tool: BuiltInConnectors
+    tool: ToolReferenceChunkToolTypedDict
     title: str
     type: NotRequired[ToolReferenceChunkType]
     url: NotRequired[Nullable[str]]
@@ -21,7 +31,7 @@ class ToolReferenceChunkTypedDict(TypedDict):
 
 
 class ToolReferenceChunk(BaseModel):
-    tool: BuiltInConnectors
+    tool: ToolReferenceChunkTool
 
     title: str
 

mistralai/models/tooltypes.py

@@ -5,4 +5,4 @@ from mistralai.types import UnrecognizedStr
 from typing import Literal, Union
 
 
-ToolTypes = Union[Literal["function"], UnrecognizedStr]
+ToolTypes = Union[Literal["function",], UnrecognizedStr]

mistralai/models/transcriptionsegmentchunk.py

@@ -8,7 +8,7 @@ from typing import Any, Dict, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-Type = Literal["transcription_segment"]
+Type = Literal["transcription_segment",]
 
 
 class TranscriptionSegmentChunkTypedDict(TypedDict):

mistralai/models/transcriptionstreamdone.py

@@ -13,7 +13,7 @@ from typing import Any, Dict, List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-TranscriptionStreamDoneType = Literal["transcription.done"]
+TranscriptionStreamDoneType = Literal["transcription.done",]
 
 
 class TranscriptionStreamDoneTypedDict(TypedDict):

mistralai/models/transcriptionstreamlanguage.py

@@ -8,7 +8,7 @@ from typing import Any, Dict, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-TranscriptionStreamLanguageType = Literal["transcription.language"]
+TranscriptionStreamLanguageType = Literal["transcription.language",]
 
 
 class TranscriptionStreamLanguageTypedDict(TypedDict):

mistralai/models/transcriptionstreamsegmentdelta.py

@@ -8,7 +8,7 @@ from typing import Any, Dict, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-TranscriptionStreamSegmentDeltaType = Literal["transcription.segment"]
+TranscriptionStreamSegmentDeltaType = Literal["transcription.segment",]
 
 
 class TranscriptionStreamSegmentDeltaTypedDict(TypedDict):

mistralai/models/transcriptionstreamtextdelta.py

@@ -8,7 +8,7 @@ from typing import Any, Dict, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-TranscriptionStreamTextDeltaType = Literal["transcription.text.delta"]
+TranscriptionStreamTextDeltaType = Literal["transcription.text.delta",]
 
 
 class TranscriptionStreamTextDeltaTypedDict(TypedDict):

mistralai/models/unarchiveftmodelout.py

@@ -6,7 +6,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-UnarchiveFTModelOutObject = Literal["model"]
+UnarchiveFTModelOutObject = Literal["model",]
 
 
 class UnarchiveFTModelOutTypedDict(TypedDict):

mistralai/models/uploadfileout.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 UploadFileOut(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/usermessage.py

@@ -16,7 +16,7 @@ UserMessageContentTypedDict = TypeAliasType(
 UserMessageContent = TypeAliasType("UserMessageContent", Union[str, List[ContentChunk]])
 
 
-UserMessageRole = Literal["user"]
+UserMessageRole = Literal["user",]
 
 
 class UserMessageTypedDict(TypedDict):

mistralai/models/wandbintegration.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-WandbIntegrationType = Literal["wandb"]
+WandbIntegrationType = Literal["wandb",]
 
 
 class WandbIntegrationTypedDict(TypedDict):

mistralai/models/wandbintegrationout.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-WandbIntegrationOutType = Literal["wandb"]
+WandbIntegrationOutType = Literal["wandb",]
 
 
 class WandbIntegrationOutTypedDict(TypedDict):

mistralai/models/websearchpremiumtool.py

@@ -6,7 +6,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-WebSearchPremiumToolType = Literal["web_search_premium"]
+WebSearchPremiumToolType = Literal["web_search_premium",]
 
 
 class WebSearchPremiumToolTypedDict(TypedDict):

mistralai/models/websearchtool.py

@@ -6,7 +6,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-WebSearchToolType = Literal["web_search"]
+WebSearchToolType = Literal["web_search",]
 
 
 class WebSearchToolTypedDict(TypedDict):