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

mistralai/models/mistralpromptmode.py

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

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):

mistralai/models/ocrrequest.py

@@ -22,7 +22,10 @@ Document = TypeAliasType("Document", Union[FileChunk, ImageURLChunk, DocumentURL
 r"""Document to run OCR on"""
 
 
-TableFormat = Literal["markdown", "html"]
+TableFormat = Literal[
+    "markdown",
+    "html",
+]
 
 
 class OCRRequestTypedDict(TypedDict):
@@ -42,6 +45,8 @@ class OCRRequestTypedDict(TypedDict):
     r"""Structured output class for extracting useful information from each extracted bounding box / image from document. Only json_schema is valid for this field"""
     document_annotation_format: NotRequired[Nullable[ResponseFormatTypedDict]]
     r"""Structured output class for extracting useful information from the entire document. Only json_schema is valid for this field"""
+    document_annotation_prompt: NotRequired[Nullable[str]]
+    r"""Optional prompt to guide the model in extracting structured output from the entire document. A document_annotation_format must be provided."""
     table_format: NotRequired[Nullable[TableFormat]]
     extract_header: NotRequired[bool]
     extract_footer: NotRequired[bool]
@@ -73,6 +78,9 @@ class OCRRequest(BaseModel):
     document_annotation_format: OptionalNullable[ResponseFormat] = UNSET
     r"""Structured output class for extracting useful information from the entire document. Only json_schema is valid for this field"""
 
+    document_annotation_prompt: OptionalNullable[str] = UNSET
+    r"""Optional prompt to guide the model in extracting structured output from the entire document. A document_annotation_format must be provided."""
+
     table_format: OptionalNullable[TableFormat] = UNSET
 
     extract_header: Optional[bool] = None
@@ -89,6 +97,7 @@ class OCRRequest(BaseModel):
             "image_min_size",
             "bbox_annotation_format",
             "document_annotation_format",
+            "document_annotation_prompt",
             "table_format",
             "extract_header",
             "extract_footer",
@@ -101,6 +110,7 @@ class OCRRequest(BaseModel):
             "image_min_size",
             "bbox_annotation_format",
             "document_annotation_format",
+            "document_annotation_prompt",
             "table_format",
         ]
         null_default_fields = []

mistralai/models/ocrtableobject.py

@@ -7,7 +7,10 @@ from typing import Literal
 from typing_extensions import Annotated, TypedDict
 
 
-Format = Literal["markdown", "html"]
+Format = Literal[
+    "markdown",
+    "html",
+]
 r"""Format of the table"""
 
 

mistralai/models/referencechunk.py

@@ -6,7 +6,7 @@ from typing import List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ReferenceChunkType = Literal["reference"]
+ReferenceChunkType = Literal["reference",]
 
 
 class ReferenceChunkTypedDict(TypedDict):

mistralai/models/requestsource.py

@@ -4,4 +4,8 @@ from __future__ import annotations
 from typing import Literal
 
 
-RequestSource = Literal["api", "playground", "agent_builder_v1"]
+RequestSource = Literal[
+    "api",
+    "playground",
+    "agent_builder_v1",
+]

mistralai/models/responsedoneevent.py

@@ -8,7 +8,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ResponseDoneEventType = Literal["conversation.response.done"]
+ResponseDoneEventType = Literal["conversation.response.done",]
 
 
 class ResponseDoneEventTypedDict(TypedDict):

mistralai/models/responseerrorevent.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ResponseErrorEventType = Literal["conversation.response.error"]
+ResponseErrorEventType = Literal["conversation.response.error",]
 
 
 class ResponseErrorEventTypedDict(TypedDict):

mistralai/models/responseformats.py

@@ -4,4 +4,8 @@ from __future__ import annotations
 from typing import Literal
 
 
-ResponseFormats = Literal["text", "json_object", "json_schema"]
+ResponseFormats = Literal[
+    "text",
+    "json_object",
+    "json_schema",
+]

mistralai/models/responsestartedevent.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ResponseStartedEventType = Literal["conversation.response.started"]
+ResponseStartedEventType = Literal["conversation.response.started",]
 
 
 class ResponseStartedEventTypedDict(TypedDict):

mistralai/models/responsevalidationerror.py

@@ -2,10 +2,12 @@
 
 import httpx
 from typing import Optional
+from dataclasses import dataclass
 
 from mistralai.models import MistralError
 
 
+@dataclass(unsafe_hash=True)
 class ResponseValidationError(MistralError):
     """Error raised when there is a type mismatch between the response data and the expected Pydantic model."""
 

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

@@ -8,7 +8,8 @@ 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]

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

@@ -8,7 +8,8 @@ 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]

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):