mistralai 1.8.2__py3-none-any.whl → 1.9.1__py3-none-any.whl

mistralai/models/ftmodelcard.py

@@ -12,9 +12,6 @@ from typing import List, Literal, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-FTModelCardType = Literal["fine-tuned"]
-
-
 class FTModelCardTypedDict(TypedDict):
     r"""Extra fields for fine-tuned models."""
 
@@ -30,8 +27,9 @@ class FTModelCardTypedDict(TypedDict):
     max_context_length: NotRequired[int]
     aliases: NotRequired[List[str]]
     deprecation: NotRequired[Nullable[datetime]]
+    deprecation_replacement_model: NotRequired[Nullable[str]]
     default_model_temperature: NotRequired[Nullable[float]]
-    type: FTModelCardType
+    type: Literal["fine-tuned"]
     archived: NotRequired[bool]
 
 
@@ -62,11 +60,14 @@ class FTModelCard(BaseModel):
 
     deprecation: OptionalNullable[datetime] = UNSET
 
+    deprecation_replacement_model: OptionalNullable[str] = UNSET
+
     default_model_temperature: OptionalNullable[float] = UNSET
 
     TYPE: Annotated[
         Annotated[
-            Optional[FTModelCardType], AfterValidator(validate_const("fine-tuned"))
+            Optional[Literal["fine-tuned"]],
+            AfterValidator(validate_const("fine-tuned")),
         ],
         pydantic.Field(alias="type"),
     ] = "fine-tuned"
@@ -84,6 +85,7 @@ class FTModelCard(BaseModel):
             "max_context_length",
             "aliases",
             "deprecation",
+            "deprecation_replacement_model",
             "default_model_temperature",
             "type",
             "archived",
@@ -92,6 +94,7 @@ class FTModelCard(BaseModel):
             "name",
             "description",
             "deprecation",
+            "deprecation_replacement_model",
             "default_model_temperature",
         ]
         null_default_fields = []
@@ -100,7 +103,7 @@ class FTModelCard(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/functioncallentry.py

@@ -55,7 +55,7 @@ class FunctionCallEntry(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/functionresultentry.py

@@ -48,7 +48,7 @@ class FunctionResultEntry(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/githubrepositoryin.py

@@ -42,7 +42,7 @@ class GithubRepositoryIn(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/githubrepositoryout.py

@@ -42,7 +42,7 @@ class GithubRepositoryOut(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/imageurl.py

@@ -26,7 +26,7 @@ class ImageURL(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/inputentries.py

@@ -1,18 +1,37 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .agenthandoffentry import AgentHandoffEntry, AgentHandoffEntryTypedDict
+from .functioncallentry import FunctionCallEntry, FunctionCallEntryTypedDict
 from .functionresultentry import FunctionResultEntry, FunctionResultEntryTypedDict
 from .messageinputentry import MessageInputEntry, MessageInputEntryTypedDict
+from .messageoutputentry import MessageOutputEntry, MessageOutputEntryTypedDict
+from .toolexecutionentry import ToolExecutionEntry, ToolExecutionEntryTypedDict
 from typing import Union
 from typing_extensions import TypeAliasType
 
 
 InputEntriesTypedDict = TypeAliasType(
     "InputEntriesTypedDict",
-    Union[MessageInputEntryTypedDict, FunctionResultEntryTypedDict],
+    Union[
+        MessageInputEntryTypedDict,
+        FunctionResultEntryTypedDict,
+        ToolExecutionEntryTypedDict,
+        FunctionCallEntryTypedDict,
+        MessageOutputEntryTypedDict,
+        AgentHandoffEntryTypedDict,
+    ],
 )
 
 
 InputEntries = TypeAliasType(
-    "InputEntries", Union[MessageInputEntry, FunctionResultEntry]
+    "InputEntries",
+    Union[
+        MessageInputEntry,
+        FunctionResultEntry,
+        ToolExecutionEntry,
+        FunctionCallEntry,
+        MessageOutputEntry,
+        AgentHandoffEntry,
+    ],
 )

mistralai/models/jobin.py

@@ -120,7 +120,7 @@ class JobIn(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/jobmetadataout.py

@@ -57,7 +57,7 @@ class JobMetadataOut(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/jobs_api_routes_batch_get_batch_jobsop.py

@@ -74,7 +74,7 @@ class JobsAPIRoutesBatchGetBatchJobsRequest(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/jobs_api_routes_fine_tuning_get_fine_tuning_jobsop.py

@@ -135,7 +135,7 @@ class JobsAPIRoutesFineTuningGetFineTuningJobsRequest(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/jsonschema.py

@@ -34,7 +34,7 @@ class JSONSchema(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/legacyjobmetadataout.py

@@ -98,7 +98,7 @@ class LegacyJobMetadataOut(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/messageinputentry.py

@@ -14,7 +14,7 @@ from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 Object = Literal["entry"]
 
-MessageInputEntryType = Literal["message.input"]
+Type = Literal["message.input"]
 
 MessageInputEntryRole = Literal["assistant", "user"]
 
@@ -35,7 +35,7 @@ class MessageInputEntryTypedDict(TypedDict):
     role: MessageInputEntryRole
     content: MessageInputEntryContentTypedDict
     object: NotRequired[Object]
-    type: NotRequired[MessageInputEntryType]
+    type: NotRequired[Type]
     created_at: NotRequired[datetime]
     completed_at: NotRequired[Nullable[datetime]]
     id: NotRequired[str]
@@ -50,7 +50,7 @@ class MessageInputEntry(BaseModel):
 
     object: Optional[Object] = "entry"
 
-    type: Optional[MessageInputEntryType] = "message.input"
+    type: Optional[Type] = "message.input"
 
     created_at: Optional[datetime] = None
 
@@ -68,7 +68,7 @@ class MessageInputEntry(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/messageoutputentry.py

@@ -79,7 +79,7 @@ class MessageOutputEntry(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/messageoutputevent.py

@@ -72,7 +72,7 @@ class MessageOutputEvent(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/metricout.py

@@ -33,7 +33,7 @@ class MetricOut(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/modelcapabilities.py

@@ -12,6 +12,7 @@ class ModelCapabilitiesTypedDict(TypedDict):
     function_calling: NotRequired[bool]
     fine_tuning: NotRequired[bool]
     vision: NotRequired[bool]
+    classification: NotRequired[bool]
 
 
 class ModelCapabilities(BaseModel):
@@ -24,3 +25,5 @@ class ModelCapabilities(BaseModel):
     fine_tuning: Optional[bool] = False
 
     vision: Optional[bool] = False
+
+    classification: Optional[bool] = False

mistralai/models/modelconversation.py

@@ -106,7 +106,7 @@ class ModelConversation(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/ocrimageobject.py

@@ -62,7 +62,7 @@ class OCRImageObject(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/ocrpageobject.py

@@ -43,7 +43,7 @@ class OCRPageObject(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/ocrrequest.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .documenturlchunk import DocumentURLChunk, DocumentURLChunkTypedDict
+from .filechunk import FileChunk, FileChunkTypedDict
 from .imageurlchunk import ImageURLChunk, ImageURLChunkTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
@@ -11,12 +12,13 @@ from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
 DocumentTypedDict = TypeAliasType(
-    "DocumentTypedDict", Union[ImageURLChunkTypedDict, DocumentURLChunkTypedDict]
+    "DocumentTypedDict",
+    Union[FileChunkTypedDict, ImageURLChunkTypedDict, DocumentURLChunkTypedDict],
 )
 r"""Document to run OCR on"""
 
 
-Document = TypeAliasType("Document", Union[ImageURLChunk, DocumentURLChunk])
+Document = TypeAliasType("Document", Union[FileChunk, ImageURLChunk, DocumentURLChunk])
 r"""Document to run OCR on"""
 
 
@@ -91,7 +93,7 @@ class OCRRequest(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/ocrresponse.py

@@ -41,7 +41,7 @@ class OCRResponse(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/ocrusageinfo.py

@@ -30,7 +30,7 @@ class OCRUsageInfo(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/responseformat.py

@@ -31,7 +31,7 @@ class ResponseFormat(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/retrievefileout.py

@@ -66,7 +66,7 @@ class RetrieveFileOut(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/toolexecutionentry.py

@@ -49,7 +49,7 @@ class ToolExecutionEntry(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/toolfilechunk.py

@@ -40,7 +40,7 @@ class ToolFileChunk(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/toolmessage.py

@@ -45,7 +45,7 @@ class ToolMessage(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/toolreferencechunk.py

@@ -40,7 +40,7 @@ class ToolReferenceChunk(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/updateftmodelin.py

@@ -26,7 +26,7 @@ class UpdateFTModelIn(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/uploadfileout.py

@@ -63,7 +63,7 @@ class UploadFileOut(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/usermessage.py

@@ -39,7 +39,7 @@ class UserMessage(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/wandbintegration.py

@@ -45,7 +45,7 @@ class WandbIntegration(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models/wandbintegrationout.py

@@ -43,7 +43,7 @@ class WandbIntegrationOut(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai/models_.py

@@ -63,6 +63,7 @@ class Models(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="list_models_v1_models_get",
                 oauth2_scopes=[],
@@ -155,6 +156,7 @@ class Models(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="list_models_v1_models_get",
                 oauth2_scopes=[],
@@ -206,7 +208,7 @@ class Models(BaseSDK):
     ) -> models.RetrieveModelV1ModelsModelIDGetResponseRetrieveModelV1ModelsModelIDGet:
         r"""Retrieve Model
 
-        Retrieve a model information.
+        Retrieve information about a model.
 
         :param model_id: The ID of the model to retrieve.
         :param retries: Override the default retry configuration for this method
@@ -254,6 +256,7 @@ class Models(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="retrieve_model_v1_models__model_id__get",
                 oauth2_scopes=[],
@@ -308,7 +311,7 @@ class Models(BaseSDK):
     ) -> models.RetrieveModelV1ModelsModelIDGetResponseRetrieveModelV1ModelsModelIDGet:
         r"""Retrieve Model
 
-        Retrieve a model information.
+        Retrieve information about a model.
 
         :param model_id: The ID of the model to retrieve.
         :param retries: Override the default retry configuration for this method
@@ -356,6 +359,7 @@ class Models(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="retrieve_model_v1_models__model_id__get",
                 oauth2_scopes=[],
@@ -458,6 +462,7 @@ class Models(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete_model_v1_models__model_id__delete",
                 oauth2_scopes=[],
@@ -557,6 +562,7 @@ class Models(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete_model_v1_models__model_id__delete",
                 oauth2_scopes=[],
@@ -667,6 +673,7 @@ class Models(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_update_fine_tuned_model",
                 oauth2_scopes=[],
@@ -774,6 +781,7 @@ class Models(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_update_fine_tuned_model",
                 oauth2_scopes=[],
@@ -870,6 +878,7 @@ class Models(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_archive_fine_tuned_model",
                 oauth2_scopes=[],
@@ -963,6 +972,7 @@ class Models(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_archive_fine_tuned_model",
                 oauth2_scopes=[],
@@ -1056,6 +1066,7 @@ class Models(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_unarchive_fine_tuned_model",
                 oauth2_scopes=[],
@@ -1149,6 +1160,7 @@ class Models(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_unarchive_fine_tuned_model",
                 oauth2_scopes=[],

mistralai/ocr.py

@@ -103,6 +103,7 @@ class Ocr(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="ocr_v1_ocr_post",
                 oauth2_scopes=[],
@@ -235,6 +236,7 @@ class Ocr(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="ocr_v1_ocr_post",
                 oauth2_scopes=[],