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

mistralai/models/agent.py

@@ -108,7 +108,7 @@ class Agent(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/agentconversation.py

@@ -50,7 +50,7 @@ class AgentConversation(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/agentcreationrequest.py

@@ -88,7 +88,7 @@ class AgentCreationRequest(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/agenthandoffentry.py

@@ -54,7 +54,7 @@ class AgentHandoffEntry(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/agents_api_v1_conversations_getop.py

@@ -11,12 +11,14 @@ from typing_extensions import Annotated, TypeAliasType, TypedDict
 
 class AgentsAPIV1ConversationsGetRequestTypedDict(TypedDict):
     conversation_id: str
+    r"""ID of the conversation from which we are fetching metadata."""
 
 
 class AgentsAPIV1ConversationsGetRequest(BaseModel):
     conversation_id: Annotated[
         str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
     ]
+    r"""ID of the conversation from which we are fetching metadata."""
 
 
 AgentsAPIV1ConversationsGetResponseV1ConversationsGetTypedDict = TypeAliasType(

mistralai/models/agents_api_v1_conversations_historyop.py

@@ -8,9 +8,11 @@ from typing_extensions import Annotated, TypedDict
 
 class AgentsAPIV1ConversationsHistoryRequestTypedDict(TypedDict):
     conversation_id: str
+    r"""ID of the conversation from which we are fetching entries."""
 
 
 class AgentsAPIV1ConversationsHistoryRequest(BaseModel):
     conversation_id: Annotated[
         str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
     ]
+    r"""ID of the conversation from which we are fetching entries."""

mistralai/models/agents_api_v1_conversations_messagesop.py

@@ -8,9 +8,11 @@ from typing_extensions import Annotated, TypedDict
 
 class AgentsAPIV1ConversationsMessagesRequestTypedDict(TypedDict):
     conversation_id: str
+    r"""ID of the conversation from which we are fetching messages."""
 
 
 class AgentsAPIV1ConversationsMessagesRequest(BaseModel):
     conversation_id: Annotated[
         str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
     ]
+    r"""ID of the conversation from which we are fetching messages."""

mistralai/models/agents_api_v1_conversations_restart_streamop.py

@@ -12,6 +12,7 @@ from typing_extensions import Annotated, TypedDict
 
 class AgentsAPIV1ConversationsRestartStreamRequestTypedDict(TypedDict):
     conversation_id: str
+    r"""ID of the original conversation which is being restarted."""
     conversation_restart_stream_request: ConversationRestartStreamRequestTypedDict
 
 
@@ -19,6 +20,7 @@ class AgentsAPIV1ConversationsRestartStreamRequest(BaseModel):
     conversation_id: Annotated[
         str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
     ]
+    r"""ID of the original conversation which is being restarted."""
 
     conversation_restart_stream_request: Annotated[
         ConversationRestartStreamRequest,

mistralai/models/agents_api_v1_conversations_restartop.py

@@ -12,6 +12,7 @@ from typing_extensions import Annotated, TypedDict
 
 class AgentsAPIV1ConversationsRestartRequestTypedDict(TypedDict):
     conversation_id: str
+    r"""ID of the original conversation which is being restarted."""
     conversation_restart_request: ConversationRestartRequestTypedDict
 
 
@@ -19,6 +20,7 @@ class AgentsAPIV1ConversationsRestartRequest(BaseModel):
     conversation_id: Annotated[
         str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
     ]
+    r"""ID of the original conversation which is being restarted."""
 
     conversation_restart_request: Annotated[
         ConversationRestartRequest,

mistralai/models/agentscompletionrequest.py

@@ -89,6 +89,7 @@ class AgentsCompletionRequestTypedDict(TypedDict):
     prediction: NotRequired[PredictionTypedDict]
     parallel_tool_calls: NotRequired[bool]
     prompt_mode: NotRequired[Nullable[MistralPromptMode]]
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
 
 
 class AgentsCompletionRequest(BaseModel):
@@ -132,6 +133,7 @@ class AgentsCompletionRequest(BaseModel):
     prompt_mode: Annotated[
         OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
     ] = UNSET
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
@@ -157,7 +159,7 @@ class AgentsCompletionRequest(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/agentscompletionstreamrequest.py

@@ -88,6 +88,7 @@ class AgentsCompletionStreamRequestTypedDict(TypedDict):
     prediction: NotRequired[PredictionTypedDict]
     parallel_tool_calls: NotRequired[bool]
     prompt_mode: NotRequired[Nullable[MistralPromptMode]]
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
 
 
 class AgentsCompletionStreamRequest(BaseModel):
@@ -130,6 +131,7 @@ class AgentsCompletionStreamRequest(BaseModel):
     prompt_mode: Annotated[
         OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
     ] = UNSET
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
@@ -155,7 +157,7 @@ class AgentsCompletionStreamRequest(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/agentupdaterequest.py

@@ -90,7 +90,7 @@ class AgentUpdateRequest(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/assistantmessage.py

@@ -50,7 +50,7 @@ class AssistantMessage(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/basemodelcard.py

@@ -12,9 +12,6 @@ from typing import List, Literal, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-Type = Literal["base"]
-
-
 class BaseModelCardTypedDict(TypedDict):
     id: str
     capabilities: ModelCapabilitiesTypedDict
@@ -26,8 +23,9 @@ class BaseModelCardTypedDict(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: Type
+    type: Literal["base"]
 
 
 class BaseModelCard(BaseModel):
@@ -51,10 +49,12 @@ class BaseModelCard(BaseModel):
 
     deprecation: OptionalNullable[datetime] = UNSET
 
+    deprecation_replacement_model: OptionalNullable[str] = UNSET
+
     default_model_temperature: OptionalNullable[float] = UNSET
 
     TYPE: Annotated[
-        Annotated[Optional[Type], AfterValidator(validate_const("base"))],
+        Annotated[Optional[Literal["base"]], AfterValidator(validate_const("base"))],
         pydantic.Field(alias="type"),
     ] = "base"
 
@@ -69,6 +69,7 @@ class BaseModelCard(BaseModel):
             "max_context_length",
             "aliases",
             "deprecation",
+            "deprecation_replacement_model",
             "default_model_temperature",
             "type",
         ]
@@ -76,6 +77,7 @@ class BaseModelCard(BaseModel):
             "name",
             "description",
             "deprecation",
+            "deprecation_replacement_model",
             "default_model_temperature",
         ]
         null_default_fields = []
@@ -84,7 +86,7 @@ class BaseModelCard(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/batchjobin.py

@@ -39,7 +39,7 @@ class BatchJobIn(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/batchjobout.py

@@ -90,7 +90,7 @@ class BatchJobOut(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/chatcompletionrequest.py

@@ -89,6 +89,7 @@ class ChatCompletionRequestTypedDict(TypedDict):
     prediction: NotRequired[PredictionTypedDict]
     parallel_tool_calls: NotRequired[bool]
     prompt_mode: NotRequired[Nullable[MistralPromptMode]]
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
     safe_prompt: NotRequired[bool]
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -140,6 +141,7 @@ class ChatCompletionRequest(BaseModel):
     prompt_mode: Annotated[
         OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
     ] = UNSET
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
 
     safe_prompt: Optional[bool] = None
     r"""Whether to inject a safety prompt before all conversations."""
@@ -178,7 +180,7 @@ class ChatCompletionRequest(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/chatcompletionstreamrequest.py

@@ -92,6 +92,7 @@ class ChatCompletionStreamRequestTypedDict(TypedDict):
     prediction: NotRequired[PredictionTypedDict]
     parallel_tool_calls: NotRequired[bool]
     prompt_mode: NotRequired[Nullable[MistralPromptMode]]
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
     safe_prompt: NotRequired[bool]
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -142,6 +143,7 @@ class ChatCompletionStreamRequest(BaseModel):
     prompt_mode: Annotated[
         OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
     ] = UNSET
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
 
     safe_prompt: Optional[bool] = None
     r"""Whether to inject a safety prompt before all conversations."""
@@ -180,7 +182,7 @@ class ChatCompletionStreamRequest(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/classifierdetailedjobout.py

@@ -135,7 +135,7 @@ class ClassifierDetailedJobOut(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/classifierftmodelout.py

@@ -80,7 +80,7 @@ class ClassifierFTModelOut(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/classifierjobout.py

@@ -144,7 +144,7 @@ class ClassifierJobOut(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/classifiertargetin.py

@@ -34,7 +34,7 @@ class ClassifierTargetIn(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/classifiertrainingparameters.py

@@ -52,7 +52,7 @@ class ClassifierTrainingParameters(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/classifiertrainingparametersin.py

@@ -64,7 +64,7 @@ class ClassifierTrainingParametersIn(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/completionargs.py

@@ -79,7 +79,7 @@ class CompletionArgs(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/completiondetailedjobout.py

@@ -141,7 +141,7 @@ class CompletionDetailedJobOut(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/completionftmodelout.py

@@ -76,7 +76,7 @@ class CompletionFTModelOut(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/completionjobout.py

@@ -154,7 +154,7 @@ class CompletionJobOut(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/completionresponsestreamchoice.py

@@ -41,7 +41,7 @@ class CompletionResponseStreamChoice(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/completiontrainingparameters.py

@@ -57,7 +57,7 @@ class CompletionTrainingParameters(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/completiontrainingparametersin.py

@@ -69,7 +69,7 @@ class CompletionTrainingParametersIn(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/contentchunk.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 .referencechunk import ReferenceChunk, ReferenceChunkTypedDict
 from .textchunk import TextChunk, TextChunkTypedDict
@@ -17,6 +18,7 @@ ContentChunkTypedDict = TypeAliasType(
         TextChunkTypedDict,
         ImageURLChunkTypedDict,
         ReferenceChunkTypedDict,
+        FileChunkTypedDict,
         DocumentURLChunkTypedDict,
     ],
 )
@@ -28,6 +30,7 @@ ContentChunk = Annotated[
         Annotated[DocumentURLChunk, Tag("document_url")],
         Annotated[TextChunk, Tag("text")],
         Annotated[ReferenceChunk, Tag("reference")],
+        Annotated[FileChunk, Tag("file")],
     ],
     Discriminator(lambda m: get_discriminator(m, "type", "type")),
 ]

mistralai/models/conversationrequest.py

@@ -112,7 +112,7 @@ class ConversationRequest(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/conversationstreamrequest.py

@@ -114,7 +114,7 @@ class ConversationStreamRequest(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/conversationusageinfo.py

@@ -42,7 +42,7 @@ class ConversationUsageInfo(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/deltamessage.py

@@ -40,7 +40,7 @@ class DeltaMessage(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/documenturlchunk.py

@@ -35,7 +35,7 @@ class DocumentURLChunk(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/embeddingrequest.py

@@ -51,7 +51,7 @@ class EmbeddingRequest(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/eventout.py

@@ -34,7 +34,7 @@ class EventOut(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/filechunk.py

@@ -0,0 +1,23 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from mistralai.utils import validate_const
+import pydantic
+from pydantic.functional_validators import AfterValidator
+from typing import Literal, Optional
+from typing_extensions import Annotated, TypedDict
+
+
+class FileChunkTypedDict(TypedDict):
+    file_id: str
+    type: Literal["file"]
+
+
+class FileChunk(BaseModel):
+    file_id: str
+
+    TYPE: Annotated[
+        Annotated[Optional[Literal["file"]], AfterValidator(validate_const("file"))],
+        pydantic.Field(alias="type"),
+    ] = "file"

mistralai/models/files_api_routes_list_filesop.py

@@ -75,7 +75,7 @@ class FilesAPIRoutesListFilesRequest(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/fileschema.py

@@ -63,7 +63,7 @@ class FileSchema(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/fimcompletionrequest.py

@@ -104,7 +104,7 @@ class FIMCompletionRequest(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/fimcompletionstreamrequest.py

@@ -102,7 +102,7 @@ class FIMCompletionStreamRequest(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)