mistralai 1.8.1__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

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .prediction import Prediction, PredictionTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
@@ -11,8 +12,9 @@ from .toolchoiceenum import ToolChoiceEnum
 from .toolmessage import ToolMessage, ToolMessageTypedDict
 from .usermessage import UserMessage, UserMessageTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import get_discriminator
+from mistralai.utils import get_discriminator, validate_open_enum
 from pydantic import Discriminator, Tag, model_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -86,6 +88,8 @@ class AgentsCompletionRequestTypedDict(TypedDict):
     r"""Number of completions to return for each request, input tokens are only billed once."""
     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):
@@ -126,6 +130,11 @@ class AgentsCompletionRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
+    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):
         optional_fields = [
@@ -141,15 +150,16 @@ class AgentsCompletionRequest(BaseModel):
             "n",
             "prediction",
             "parallel_tool_calls",
+            "prompt_mode",
         ]
-        nullable_fields = ["max_tokens", "random_seed", "tools", "n"]
+        nullable_fields = ["max_tokens", "random_seed", "tools", "n", "prompt_mode"]
         null_default_fields = []
 
         serialized = handler(self)
 
         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

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .prediction import Prediction, PredictionTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
@@ -11,8 +12,9 @@ from .toolchoiceenum import ToolChoiceEnum
 from .toolmessage import ToolMessage, ToolMessageTypedDict
 from .usermessage import UserMessage, UserMessageTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import get_discriminator
+from mistralai.utils import get_discriminator, validate_open_enum
 from pydantic import Discriminator, Tag, model_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -85,6 +87,8 @@ class AgentsCompletionStreamRequestTypedDict(TypedDict):
     r"""Number of completions to return for each request, input tokens are only billed once."""
     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):
@@ -124,6 +128,11 @@ class AgentsCompletionStreamRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
+    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):
         optional_fields = [
@@ -139,15 +148,16 @@ class AgentsCompletionStreamRequest(BaseModel):
             "n",
             "prediction",
             "parallel_tool_calls",
+            "prompt_mode",
         ]
-        nullable_fields = ["max_tokens", "random_seed", "tools", "n"]
+        nullable_fields = ["max_tokens", "random_seed", "tools", "n", "prompt_mode"]
         null_default_fields = []
 
         serialized = handler(self)
 
         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

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .prediction import Prediction, PredictionTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
@@ -11,8 +12,9 @@ from .toolchoiceenum import ToolChoiceEnum
 from .toolmessage import ToolMessage, ToolMessageTypedDict
 from .usermessage import UserMessage, UserMessageTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import get_discriminator
+from mistralai.utils import get_discriminator, validate_open_enum
 from pydantic import Discriminator, Tag, model_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -86,6 +88,8 @@ class ChatCompletionRequestTypedDict(TypedDict):
     r"""Number of completions to return for each request, input tokens are only billed once."""
     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."""
 
@@ -134,6 +138,11 @@ class ChatCompletionRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
+    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."""
 
@@ -154,16 +163,24 @@ class ChatCompletionRequest(BaseModel):
             "n",
             "prediction",
             "parallel_tool_calls",
+            "prompt_mode",
             "safe_prompt",
         ]
-        nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]
+        nullable_fields = [
+            "temperature",
+            "max_tokens",
+            "random_seed",
+            "tools",
+            "n",
+            "prompt_mode",
+        ]
         null_default_fields = []
 
         serialized = handler(self)
 
         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

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .prediction import Prediction, PredictionTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
@@ -11,8 +12,9 @@ from .toolchoiceenum import ToolChoiceEnum
 from .toolmessage import ToolMessage, ToolMessageTypedDict
 from .usermessage import UserMessage, UserMessageTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import get_discriminator
+from mistralai.utils import get_discriminator, validate_open_enum
 from pydantic import Discriminator, Tag, model_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -89,6 +91,8 @@ class ChatCompletionStreamRequestTypedDict(TypedDict):
     r"""Number of completions to return for each request, input tokens are only billed once."""
     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."""
 
@@ -136,6 +140,11 @@ class ChatCompletionStreamRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
+    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."""
 
@@ -156,16 +165,24 @@ class ChatCompletionStreamRequest(BaseModel):
             "n",
             "prediction",
             "parallel_tool_calls",
+            "prompt_mode",
             "safe_prompt",
         ]
-        nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]
+        nullable_fields = [
+            "temperature",
+            "max_tokens",
+            "random_seed",
+            "tools",
+            "n",
+            "prompt_mode",
+        ]
         null_default_fields = []
 
         serialized = handler(self)
 
         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)