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

mistralai/mistral_jobs.py

@@ -4,6 +4,11 @@ from .basesdk import BaseSDK
 from datetime import datetime
 from mistralai import models, utils
 from mistralai._hooks import HookContext
+from mistralai.models import (
+    apiendpoint as models_apiendpoint,
+    batchjobstatus as models_batchjobstatus,
+    batchrequest as models_batchrequest,
+)
 from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import get_security_from_env
 from mistralai.utils.unmarshal_json_response import unmarshal_json_response
@@ -21,7 +26,7 @@ class MistralJobs(BaseSDK):
         metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         created_after: OptionalNullable[datetime] = UNSET,
         created_by_me: Optional[bool] = False,
-        status: OptionalNullable[List[models.BatchJobStatus]] = UNSET,
+        status: OptionalNullable[List[models_batchjobstatus.BatchJobStatus]] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -78,6 +83,7 @@ class MistralJobs(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -94,7 +100,7 @@ class MistralJobs(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_get_batch_jobs",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -125,7 +131,7 @@ class MistralJobs(BaseSDK):
         metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         created_after: OptionalNullable[datetime] = UNSET,
         created_by_me: Optional[bool] = False,
-        status: OptionalNullable[List[models.BatchJobStatus]] = UNSET,
+        status: OptionalNullable[List[models_batchjobstatus.BatchJobStatus]] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -182,6 +188,7 @@ class MistralJobs(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -198,7 +205,7 @@ class MistralJobs(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_get_batch_jobs",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -222,10 +229,13 @@ class MistralJobs(BaseSDK):
     def create(
         self,
         *,
-        endpoint: models.APIEndpoint,
+        endpoint: models_apiendpoint.APIEndpoint,
         input_files: OptionalNullable[List[str]] = UNSET,
         requests: OptionalNullable[
-            Union[List[models.BatchRequest], List[models.BatchRequestTypedDict]]
+            Union[
+                List[models_batchrequest.BatchRequest],
+                List[models_batchrequest.BatchRequestTypedDict],
+            ]
         ] = UNSET,
         model: OptionalNullable[str] = UNSET,
         agent_id: OptionalNullable[str] = UNSET,
@@ -290,6 +300,7 @@ class MistralJobs(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.BatchJobIn
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -306,7 +317,7 @@ class MistralJobs(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_create_batch_job",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -330,10 +341,13 @@ class MistralJobs(BaseSDK):
     async def create_async(
         self,
         *,
-        endpoint: models.APIEndpoint,
+        endpoint: models_apiendpoint.APIEndpoint,
         input_files: OptionalNullable[List[str]] = UNSET,
         requests: OptionalNullable[
-            Union[List[models.BatchRequest], List[models.BatchRequestTypedDict]]
+            Union[
+                List[models_batchrequest.BatchRequest],
+                List[models_batchrequest.BatchRequestTypedDict],
+            ]
         ] = UNSET,
         model: OptionalNullable[str] = UNSET,
         agent_id: OptionalNullable[str] = UNSET,
@@ -398,6 +412,7 @@ class MistralJobs(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.BatchJobIn
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -414,7 +429,7 @@ class MistralJobs(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_create_batch_job",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -487,6 +502,7 @@ class MistralJobs(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -503,7 +519,7 @@ class MistralJobs(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_get_batch_job",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -576,6 +592,7 @@ class MistralJobs(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -592,7 +609,7 @@ class MistralJobs(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_get_batch_job",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -659,6 +676,7 @@ class MistralJobs(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -675,7 +693,7 @@ class MistralJobs(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_cancel_batch_job",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -742,6 +760,7 @@ class MistralJobs(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -758,7 +777,7 @@ class MistralJobs(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_cancel_batch_job",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),

mistralai/models/__init__.py

@@ -45,10 +45,18 @@ if TYPE_CHECKING:
         AgentsAPIV1AgentsDeleteRequest,
         AgentsAPIV1AgentsDeleteRequestTypedDict,
     )
+    from .agents_api_v1_agents_get_versionop import (
+        AgentsAPIV1AgentsGetVersionRequest,
+        AgentsAPIV1AgentsGetVersionRequestTypedDict,
+    )
     from .agents_api_v1_agents_getop import (
         AgentsAPIV1AgentsGetRequest,
         AgentsAPIV1AgentsGetRequestTypedDict,
     )
+    from .agents_api_v1_agents_list_versionsop import (
+        AgentsAPIV1AgentsListVersionsRequest,
+        AgentsAPIV1AgentsListVersionsRequestTypedDict,
+    )
     from .agents_api_v1_agents_listop import (
         AgentsAPIV1AgentsListRequest,
         AgentsAPIV1AgentsListRequestTypedDict,
@@ -972,8 +980,12 @@ __all__ = [
     "AgentsAPIV1AgentsDeleteRequestTypedDict",
     "AgentsAPIV1AgentsGetRequest",
     "AgentsAPIV1AgentsGetRequestTypedDict",
+    "AgentsAPIV1AgentsGetVersionRequest",
+    "AgentsAPIV1AgentsGetVersionRequestTypedDict",
     "AgentsAPIV1AgentsListRequest",
     "AgentsAPIV1AgentsListRequestTypedDict",
+    "AgentsAPIV1AgentsListVersionsRequest",
+    "AgentsAPIV1AgentsListVersionsRequestTypedDict",
     "AgentsAPIV1AgentsUpdateRequest",
     "AgentsAPIV1AgentsUpdateRequestTypedDict",
     "AgentsAPIV1AgentsUpdateVersionRequest",
@@ -1682,8 +1694,12 @@ _dynamic_imports: dict[str, str] = {
     "AgentHandoffStartedEventTypedDict": ".agenthandoffstartedevent",
     "AgentsAPIV1AgentsDeleteRequest": ".agents_api_v1_agents_deleteop",
     "AgentsAPIV1AgentsDeleteRequestTypedDict": ".agents_api_v1_agents_deleteop",
+    "AgentsAPIV1AgentsGetVersionRequest": ".agents_api_v1_agents_get_versionop",
+    "AgentsAPIV1AgentsGetVersionRequestTypedDict": ".agents_api_v1_agents_get_versionop",
     "AgentsAPIV1AgentsGetRequest": ".agents_api_v1_agents_getop",
     "AgentsAPIV1AgentsGetRequestTypedDict": ".agents_api_v1_agents_getop",
+    "AgentsAPIV1AgentsListVersionsRequest": ".agents_api_v1_agents_list_versionsop",
+    "AgentsAPIV1AgentsListVersionsRequestTypedDict": ".agents_api_v1_agents_list_versionsop",
     "AgentsAPIV1AgentsListRequest": ".agents_api_v1_agents_listop",
     "AgentsAPIV1AgentsListRequestTypedDict": ".agents_api_v1_agents_listop",
     "AgentsAPIV1AgentsUpdateVersionRequest": ".agents_api_v1_agents_update_versionop",

mistralai/models/agent.py

@@ -42,7 +42,7 @@ AgentTools = Annotated[
 ]
 
 
-AgentObject = Literal["agent"]
+AgentObject = Literal["agent",]
 
 
 class AgentTypedDict(TypedDict):

mistralai/models/agentconversation.py

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

mistralai/models/agenthandoffdoneevent.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-AgentHandoffDoneEventType = Literal["agent.handoff.done"]
+AgentHandoffDoneEventType = Literal["agent.handoff.done",]
 
 
 class AgentHandoffDoneEventTypedDict(TypedDict):

mistralai/models/agenthandoffentry.py

@@ -8,9 +8,10 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-AgentHandoffEntryObject = Literal["entry"]
+AgentHandoffEntryObject = Literal["entry",]
 
-AgentHandoffEntryType = Literal["agent.handoff"]
+
+AgentHandoffEntryType = Literal["agent.handoff",]
 
 
 class AgentHandoffEntryTypedDict(TypedDict):

mistralai/models/agenthandoffstartedevent.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-AgentHandoffStartedEventType = Literal["agent.handoff.started"]
+AgentHandoffStartedEventType = Literal["agent.handoff.started",]
 
 
 class AgentHandoffStartedEventTypedDict(TypedDict):

mistralai/models/agents_api_v1_agents_get_versionop.py

@@ -0,0 +1,21 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class AgentsAPIV1AgentsGetVersionRequestTypedDict(TypedDict):
+    agent_id: str
+    version: int
+
+
+class AgentsAPIV1AgentsGetVersionRequest(BaseModel):
+    agent_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    version: Annotated[
+        int, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]

mistralai/models/agents_api_v1_agents_list_versionsop.py

@@ -0,0 +1,33 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata, QueryParamMetadata
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class AgentsAPIV1AgentsListVersionsRequestTypedDict(TypedDict):
+    agent_id: str
+    page: NotRequired[int]
+    r"""Page number (0-indexed)"""
+    page_size: NotRequired[int]
+    r"""Number of versions per page"""
+
+
+class AgentsAPIV1AgentsListVersionsRequest(BaseModel):
+    agent_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    page: Annotated[
+        Optional[int],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = 0
+    r"""Page number (0-indexed)"""
+
+    page_size: Annotated[
+        Optional[int],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = 20
+    r"""Number of versions per page"""

mistralai/models/agents_api_v1_agents_listop.py

@@ -11,7 +11,9 @@ from typing_extensions import Annotated, NotRequired, TypedDict
 
 class AgentsAPIV1AgentsListRequestTypedDict(TypedDict):
     page: NotRequired[int]
+    r"""Page number (0-indexed)"""
     page_size: NotRequired[int]
+    r"""Number of agents per page"""
     deployment_chat: NotRequired[Nullable[bool]]
     sources: NotRequired[Nullable[List[RequestSource]]]
     name: NotRequired[Nullable[str]]
@@ -24,11 +26,13 @@ class AgentsAPIV1AgentsListRequest(BaseModel):
         Optional[int],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = 0
+    r"""Page number (0-indexed)"""
 
     page_size: Annotated[
         Optional[int],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = 20
+    r"""Number of agents per page"""
 
     deployment_chat: Annotated[
         OptionalNullable[bool],

mistralai/models/agentscompletionrequest.py

@@ -12,9 +12,8 @@ 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, validate_open_enum
+from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing import Any, Dict, List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -137,9 +136,7 @@ class AgentsCompletionRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
-    prompt_mode: Annotated[
-        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
-    ] = UNSET
+    prompt_mode: OptionalNullable[MistralPromptMode] = 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")

mistralai/models/agentscompletionstreamrequest.py

@@ -12,9 +12,8 @@ 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, validate_open_enum
+from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing import Any, Dict, List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -135,9 +134,7 @@ class AgentsCompletionStreamRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
-    prompt_mode: Annotated[
-        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
-    ] = UNSET
+    prompt_mode: OptionalNullable[MistralPromptMode] = 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")

mistralai/models/archiveftmodelout.py

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

mistralai/models/assistantmessage.py

@@ -19,7 +19,7 @@ AssistantMessageContent = TypeAliasType(
 )
 
 
-AssistantMessageRole = Literal["assistant"]
+AssistantMessageRole = Literal["assistant",]
 
 
 class AssistantMessageTypedDict(TypedDict):

mistralai/models/audiochunk.py

@@ -6,7 +6,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-AudioChunkType = Literal["input_audio"]
+AudioChunkType = Literal["input_audio",]
 
 
 class AudioChunkTypedDict(TypedDict):

mistralai/models/audioencoding.py

@@ -7,7 +7,12 @@ from typing import Literal, Union
 
 AudioEncoding = Union[
     Literal[
-        "pcm_s16le", "pcm_s32le", "pcm_f16le", "pcm_f32le", "pcm_mulaw", "pcm_alaw"
+        "pcm_s16le",
+        "pcm_s32le",
+        "pcm_f16le",
+        "pcm_f32le",
+        "pcm_mulaw",
+        "pcm_alaw",
     ],
     UnrecognizedStr,
 ]

mistralai/models/audioformat.py

@@ -3,9 +3,7 @@
 from __future__ import annotations
 from .audioencoding import AudioEncoding
 from mistralai.types import BaseModel
-from mistralai.utils import validate_open_enum
-from pydantic.functional_validators import PlainValidator
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import TypedDict
 
 
 class AudioFormatTypedDict(TypedDict):
@@ -14,6 +12,6 @@ class AudioFormatTypedDict(TypedDict):
 
 
 class AudioFormat(BaseModel):
-    encoding: Annotated[AudioEncoding, PlainValidator(validate_open_enum(False))]
+    encoding: AudioEncoding
 
     sample_rate: int

mistralai/models/basemodelcard.py

@@ -12,7 +12,7 @@ from typing import List, Literal, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-BaseModelCardType = Literal["base"]
+BaseModelCardType = Literal["base",]
 
 
 class BaseModelCardTypedDict(TypedDict):

mistralai/models/batchjobin.py

@@ -4,11 +4,9 @@ from __future__ import annotations
 from .apiendpoint import APIEndpoint
 from .batchrequest import BatchRequest, BatchRequestTypedDict
 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 import Dict, List, Optional
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 class BatchJobInTypedDict(TypedDict):
@@ -27,7 +25,7 @@ class BatchJobInTypedDict(TypedDict):
 
 
 class BatchJobIn(BaseModel):
-    endpoint: Annotated[APIEndpoint, PlainValidator(validate_open_enum(False))]
+    endpoint: APIEndpoint
 
     input_files: OptionalNullable[List[str]] = UNSET
     r"""The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```"""

mistralai/models/batchjobout.py

@@ -9,7 +9,7 @@ from typing import Any, Dict, List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-BatchJobOutObject = Literal["batch"]
+BatchJobOutObject = Literal["batch",]
 
 
 class BatchJobOutTypedDict(TypedDict):

mistralai/models/batchjobsout.py

@@ -7,7 +7,7 @@ from typing import List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-BatchJobsOutObject = Literal["list"]
+BatchJobsOutObject = Literal["list",]
 
 
 class BatchJobsOutTypedDict(TypedDict):

mistralai/models/chatcompletionchoice.py

@@ -3,14 +3,19 @@
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
 from mistralai.types import BaseModel, UnrecognizedStr
-from mistralai.utils import validate_open_enum
-from pydantic.functional_validators import PlainValidator
 from typing import Literal, Union
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import TypedDict
 
 
 FinishReason = Union[
-    Literal["stop", "length", "model_length", "error", "tool_calls"], UnrecognizedStr
+    Literal[
+        "stop",
+        "length",
+        "model_length",
+        "error",
+        "tool_calls",
+    ],
+    UnrecognizedStr,
 ]
 
 
@@ -25,4 +30,4 @@ class ChatCompletionChoice(BaseModel):
 
     message: AssistantMessage
 
-    finish_reason: Annotated[FinishReason, PlainValidator(validate_open_enum(False))]
+    finish_reason: FinishReason

mistralai/models/chatcompletionrequest.py

@@ -12,9 +12,8 @@ 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, validate_open_enum
+from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing import Any, Dict, List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -153,9 +152,7 @@ class ChatCompletionRequest(BaseModel):
     parallel_tool_calls: Optional[bool] = None
     r"""Whether to enable parallel function calling during tool use, when enabled the model can call multiple tools in parallel."""
 
-    prompt_mode: Annotated[
-        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
-    ] = UNSET
+    prompt_mode: OptionalNullable[MistralPromptMode] = 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

mistralai/models/chatcompletionstreamrequest.py

@@ -12,9 +12,8 @@ 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, validate_open_enum
+from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing import Any, Dict, List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -155,9 +154,7 @@ class ChatCompletionStreamRequest(BaseModel):
     parallel_tool_calls: Optional[bool] = None
     r"""Whether to enable parallel function calling during tool use, when enabled the model can call multiple tools in parallel."""
 
-    prompt_mode: Annotated[
-        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
-    ] = UNSET
+    prompt_mode: OptionalNullable[MistralPromptMode] = 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

mistralai/models/classifierdetailedjobout.py

@@ -29,7 +29,9 @@ ClassifierDetailedJobOutStatus = Literal[
     "CANCELLATION_REQUESTED",
 ]
 
-ClassifierDetailedJobOutObject = Literal["job"]
+
+ClassifierDetailedJobOutObject = Literal["job",]
+
 
 ClassifierDetailedJobOutIntegrationsTypedDict = WandbIntegrationOutTypedDict
 
@@ -37,7 +39,7 @@ ClassifierDetailedJobOutIntegrationsTypedDict = WandbIntegrationOutTypedDict
 ClassifierDetailedJobOutIntegrations = WandbIntegrationOut
 
 
-ClassifierDetailedJobOutJobType = Literal["classifier"]
+ClassifierDetailedJobOutJobType = Literal["classifier",]
 
 
 class ClassifierDetailedJobOutTypedDict(TypedDict):

mistralai/models/classifierftmodelout.py

@@ -12,9 +12,10 @@ from typing import List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ClassifierFTModelOutObject = Literal["model"]
+ClassifierFTModelOutObject = Literal["model",]
 
-ClassifierFTModelOutModelType = Literal["classifier"]
+
+ClassifierFTModelOutModelType = Literal["classifier",]
 
 
 class ClassifierFTModelOutTypedDict(TypedDict):