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

mistralai/mistral_jobs.py

@@ -4,10 +4,15 @@ 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
-from typing import Any, Dict, List, Mapping, Optional
+from typing import Any, Dict, List, Mapping, Optional, Union
 
 
 class MistralJobs(BaseSDK):
@@ -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,8 +229,14 @@ class MistralJobs(BaseSDK):
     def create(
         self,
         *,
-        input_files: List[str],
-        endpoint: models.APIEndpoint,
+        endpoint: models_apiendpoint.APIEndpoint,
+        input_files: OptionalNullable[List[str]] = UNSET,
+        requests: OptionalNullable[
+            Union[
+                List[models_batchrequest.BatchRequest],
+                List[models_batchrequest.BatchRequestTypedDict],
+            ]
+        ] = UNSET,
         model: OptionalNullable[str] = UNSET,
         agent_id: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, str]] = UNSET,
@@ -237,8 +250,9 @@ class MistralJobs(BaseSDK):
 
         Create a new batch job, it will be queued for processing.
 
-        :param input_files: 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?\"}]}} ```
         :param endpoint:
+        :param input_files: 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?\"}]}} ```
+        :param requests:
         :param model: The model to be used for batch inference.
         :param agent_id: In case you want to use a specific agent from the **deprecated** agents api for batch inference, you can specify the agent ID here.
         :param metadata: The metadata of your choice to be associated with the batch inference job.
@@ -260,6 +274,9 @@ class MistralJobs(BaseSDK):
 
         request = models.BatchJobIn(
             input_files=input_files,
+            requests=utils.get_pydantic_model(
+                requests, OptionalNullable[List[models.BatchRequest]]
+            ),
             endpoint=endpoint,
             model=model,
             agent_id=agent_id,
@@ -283,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,
         )
 
@@ -299,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
                 ),
@@ -323,8 +341,14 @@ class MistralJobs(BaseSDK):
     async def create_async(
         self,
         *,
-        input_files: List[str],
-        endpoint: models.APIEndpoint,
+        endpoint: models_apiendpoint.APIEndpoint,
+        input_files: OptionalNullable[List[str]] = UNSET,
+        requests: OptionalNullable[
+            Union[
+                List[models_batchrequest.BatchRequest],
+                List[models_batchrequest.BatchRequestTypedDict],
+            ]
+        ] = UNSET,
         model: OptionalNullable[str] = UNSET,
         agent_id: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, str]] = UNSET,
@@ -338,8 +362,9 @@ class MistralJobs(BaseSDK):
 
         Create a new batch job, it will be queued for processing.
 
-        :param input_files: 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?\"}]}} ```
         :param endpoint:
+        :param input_files: 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?\"}]}} ```
+        :param requests:
         :param model: The model to be used for batch inference.
         :param agent_id: In case you want to use a specific agent from the **deprecated** agents api for batch inference, you can specify the agent ID here.
         :param metadata: The metadata of your choice to be associated with the batch inference job.
@@ -361,6 +386,9 @@ class MistralJobs(BaseSDK):
 
         request = models.BatchJobIn(
             input_files=input_files,
+            requests=utils.get_pydantic_model(
+                requests, OptionalNullable[List[models.BatchRequest]]
+            ),
             endpoint=endpoint,
             model=model,
             agent_id=agent_id,
@@ -384,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,
         )
 
@@ -400,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
                 ),
@@ -425,6 +454,7 @@ class MistralJobs(BaseSDK):
         self,
         *,
         job_id: str,
+        inline: OptionalNullable[bool] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -434,7 +464,11 @@ class MistralJobs(BaseSDK):
 
         Get a batch job details by its UUID.
 
+        Args:
+        inline: If True, return results inline in the response.
+
         :param job_id:
+        :param inline:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -452,6 +486,7 @@ class MistralJobs(BaseSDK):
 
         request = models.JobsAPIRoutesBatchGetBatchJobRequest(
             job_id=job_id,
+            inline=inline,
         )
 
         req = self._build_request(
@@ -467,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,
         )
 
@@ -483,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
                 ),
@@ -508,6 +544,7 @@ class MistralJobs(BaseSDK):
         self,
         *,
         job_id: str,
+        inline: OptionalNullable[bool] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -517,7 +554,11 @@ class MistralJobs(BaseSDK):
 
         Get a batch job details by its UUID.
 
+        Args:
+        inline: If True, return results inline in the response.
+
         :param job_id:
+        :param inline:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -535,6 +576,7 @@ class MistralJobs(BaseSDK):
 
         request = models.JobsAPIRoutesBatchGetBatchJobRequest(
             job_id=job_id,
+            inline=inline,
         )
 
         req = self._build_request_async(
@@ -550,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,
         )
 
@@ -566,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
                 ),
@@ -633,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,
         )
 
@@ -649,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
                 ),
@@ -716,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,
         )
 
@@ -732,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,
@@ -141,6 +149,8 @@ if TYPE_CHECKING:
         AssistantMessageTypedDict,
     )
     from .audiochunk import AudioChunk, AudioChunkType, AudioChunkTypedDict
+    from .audioencoding import AudioEncoding
+    from .audioformat import AudioFormat, AudioFormatTypedDict
     from .audiotranscriptionrequest import (
         AudioTranscriptionRequest,
         AudioTranscriptionRequestTypedDict,
@@ -155,6 +165,7 @@ if TYPE_CHECKING:
     from .batchjobout import BatchJobOut, BatchJobOutObject, BatchJobOutTypedDict
     from .batchjobsout import BatchJobsOut, BatchJobsOutObject, BatchJobsOutTypedDict
     from .batchjobstatus import BatchJobStatus
+    from .batchrequest import BatchRequest, BatchRequestTypedDict
     from .builtinconnectors import BuiltInConnectors
     from .chatclassificationrequest import (
         ChatClassificationRequest,
@@ -840,7 +851,13 @@ if TYPE_CHECKING:
         ToolExecutionStartedEventType,
         ToolExecutionStartedEventTypedDict,
     )
-    from .toolfilechunk import ToolFileChunk, ToolFileChunkType, ToolFileChunkTypedDict
+    from .toolfilechunk import (
+        ToolFileChunk,
+        ToolFileChunkTool,
+        ToolFileChunkToolTypedDict,
+        ToolFileChunkType,
+        ToolFileChunkTypedDict,
+    )
     from .toolmessage import (
         ToolMessage,
         ToolMessageContent,
@@ -850,6 +867,8 @@ if TYPE_CHECKING:
     )
     from .toolreferencechunk import (
         ToolReferenceChunk,
+        ToolReferenceChunkTool,
+        ToolReferenceChunkToolTypedDict,
         ToolReferenceChunkType,
         ToolReferenceChunkTypedDict,
     )
@@ -961,8 +980,12 @@ __all__ = [
     "AgentsAPIV1AgentsDeleteRequestTypedDict",
     "AgentsAPIV1AgentsGetRequest",
     "AgentsAPIV1AgentsGetRequestTypedDict",
+    "AgentsAPIV1AgentsGetVersionRequest",
+    "AgentsAPIV1AgentsGetVersionRequestTypedDict",
     "AgentsAPIV1AgentsListRequest",
     "AgentsAPIV1AgentsListRequestTypedDict",
+    "AgentsAPIV1AgentsListVersionsRequest",
+    "AgentsAPIV1AgentsListVersionsRequestTypedDict",
     "AgentsAPIV1AgentsUpdateRequest",
     "AgentsAPIV1AgentsUpdateRequestTypedDict",
     "AgentsAPIV1AgentsUpdateVersionRequest",
@@ -1018,6 +1041,9 @@ __all__ = [
     "AudioChunk",
     "AudioChunkType",
     "AudioChunkTypedDict",
+    "AudioEncoding",
+    "AudioFormat",
+    "AudioFormatTypedDict",
     "AudioTranscriptionRequest",
     "AudioTranscriptionRequestStream",
     "AudioTranscriptionRequestStreamTypedDict",
@@ -1036,6 +1062,8 @@ __all__ = [
     "BatchJobsOut",
     "BatchJobsOutObject",
     "BatchJobsOutTypedDict",
+    "BatchRequest",
+    "BatchRequestTypedDict",
     "BuiltInConnectors",
     "ChatClassificationRequest",
     "ChatClassificationRequestTypedDict",
@@ -1567,6 +1595,8 @@ __all__ = [
     "ToolExecutionStartedEventType",
     "ToolExecutionStartedEventTypedDict",
     "ToolFileChunk",
+    "ToolFileChunkTool",
+    "ToolFileChunkToolTypedDict",
     "ToolFileChunkType",
     "ToolFileChunkTypedDict",
     "ToolMessage",
@@ -1575,6 +1605,8 @@ __all__ = [
     "ToolMessageRole",
     "ToolMessageTypedDict",
     "ToolReferenceChunk",
+    "ToolReferenceChunkTool",
+    "ToolReferenceChunkToolTypedDict",
     "ToolReferenceChunkType",
     "ToolReferenceChunkTypedDict",
     "ToolTypedDict",
@@ -1662,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",
@@ -1724,6 +1760,9 @@ _dynamic_imports: dict[str, str] = {
     "AudioChunk": ".audiochunk",
     "AudioChunkType": ".audiochunk",
     "AudioChunkTypedDict": ".audiochunk",
+    "AudioEncoding": ".audioencoding",
+    "AudioFormat": ".audioformat",
+    "AudioFormatTypedDict": ".audioformat",
     "AudioTranscriptionRequest": ".audiotranscriptionrequest",
     "AudioTranscriptionRequestTypedDict": ".audiotranscriptionrequest",
     "AudioTranscriptionRequestStream": ".audiotranscriptionrequeststream",
@@ -1742,6 +1781,8 @@ _dynamic_imports: dict[str, str] = {
     "BatchJobsOutObject": ".batchjobsout",
     "BatchJobsOutTypedDict": ".batchjobsout",
     "BatchJobStatus": ".batchjobstatus",
+    "BatchRequest": ".batchrequest",
+    "BatchRequestTypedDict": ".batchrequest",
     "BuiltInConnectors": ".builtinconnectors",
     "ChatClassificationRequest": ".chatclassificationrequest",
     "ChatClassificationRequestTypedDict": ".chatclassificationrequest",
@@ -2277,6 +2318,8 @@ _dynamic_imports: dict[str, str] = {
     "ToolExecutionStartedEventType": ".toolexecutionstartedevent",
     "ToolExecutionStartedEventTypedDict": ".toolexecutionstartedevent",
     "ToolFileChunk": ".toolfilechunk",
+    "ToolFileChunkTool": ".toolfilechunk",
+    "ToolFileChunkToolTypedDict": ".toolfilechunk",
     "ToolFileChunkType": ".toolfilechunk",
     "ToolFileChunkTypedDict": ".toolfilechunk",
     "ToolMessage": ".toolmessage",
@@ -2285,6 +2328,8 @@ _dynamic_imports: dict[str, str] = {
     "ToolMessageRole": ".toolmessage",
     "ToolMessageTypedDict": ".toolmessage",
     "ToolReferenceChunk": ".toolreferencechunk",
+    "ToolReferenceChunkTool": ".toolreferencechunk",
+    "ToolReferenceChunkToolTypedDict": ".toolreferencechunk",
     "ToolReferenceChunkType": ".toolreferencechunk",
     "ToolReferenceChunkTypedDict": ".toolreferencechunk",
     "ToolTypes": ".tooltypes",

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],
@@ -52,7 +56,7 @@ class AgentsAPIV1AgentsListRequest(BaseModel):
 
     metadata: Annotated[
         OptionalNullable[Dict[str, Any]],
-        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+        FieldMetadata(query=QueryParamMetadata(serialization="json")),
     ] = UNSET
 
     @model_serializer(mode="wrap")

mistralai/models/agents_api_v1_conversations_listop.py

@@ -29,7 +29,7 @@ class AgentsAPIV1ConversationsListRequest(BaseModel):
 
     metadata: Annotated[
         OptionalNullable[Dict[str, Any]],
-        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+        FieldMetadata(query=QueryParamMetadata(serialization="json")),
     ] = UNSET
 
     @model_serializer(mode="wrap")

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