mistralai 1.10.1__py3-none-any.whl → 1.12.0__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

@@ -14,8 +14,11 @@ if TYPE_CHECKING:
         AgentToolsTypedDict,
         AgentTypedDict,
     )
+    from .agentaliasresponse import AgentAliasResponse, AgentAliasResponseTypedDict
     from .agentconversation import (
         AgentConversation,
+        AgentConversationAgentVersion,
+        AgentConversationAgentVersionTypedDict,
         AgentConversationObject,
         AgentConversationTypedDict,
     )
@@ -41,13 +44,31 @@ if TYPE_CHECKING:
         AgentHandoffStartedEventType,
         AgentHandoffStartedEventTypedDict,
     )
+    from .agents_api_v1_agents_create_or_update_aliasop import (
+        AgentsAPIV1AgentsCreateOrUpdateAliasRequest,
+        AgentsAPIV1AgentsCreateOrUpdateAliasRequestTypedDict,
+    )
     from .agents_api_v1_agents_deleteop import (
         AgentsAPIV1AgentsDeleteRequest,
         AgentsAPIV1AgentsDeleteRequestTypedDict,
     )
+    from .agents_api_v1_agents_get_versionop import (
+        AgentsAPIV1AgentsGetVersionRequest,
+        AgentsAPIV1AgentsGetVersionRequestTypedDict,
+    )
     from .agents_api_v1_agents_getop import (
         AgentsAPIV1AgentsGetRequest,
         AgentsAPIV1AgentsGetRequestTypedDict,
+        QueryParamAgentVersion,
+        QueryParamAgentVersionTypedDict,
+    )
+    from .agents_api_v1_agents_list_version_aliasesop import (
+        AgentsAPIV1AgentsListVersionAliasesRequest,
+        AgentsAPIV1AgentsListVersionAliasesRequestTypedDict,
+    )
+    from .agents_api_v1_agents_list_versionsop import (
+        AgentsAPIV1AgentsListVersionsRequest,
+        AgentsAPIV1AgentsListVersionsRequestTypedDict,
     )
     from .agents_api_v1_agents_listop import (
         AgentsAPIV1AgentsListRequest,
@@ -332,6 +353,8 @@ if TYPE_CHECKING:
         ConversationMessagesTypedDict,
     )
     from .conversationrequest import (
+        AgentVersion,
+        AgentVersionTypedDict,
         ConversationRequest,
         ConversationRequestTypedDict,
         HandoffExecution,
@@ -347,16 +370,22 @@ if TYPE_CHECKING:
     )
     from .conversationrestartrequest import (
         ConversationRestartRequest,
+        ConversationRestartRequestAgentVersion,
+        ConversationRestartRequestAgentVersionTypedDict,
         ConversationRestartRequestHandoffExecution,
         ConversationRestartRequestTypedDict,
     )
     from .conversationrestartstreamrequest import (
         ConversationRestartStreamRequest,
+        ConversationRestartStreamRequestAgentVersion,
+        ConversationRestartStreamRequestAgentVersionTypedDict,
         ConversationRestartStreamRequestHandoffExecution,
         ConversationRestartStreamRequestTypedDict,
     )
     from .conversationstreamrequest import (
         ConversationStreamRequest,
+        ConversationStreamRequestAgentVersion,
+        ConversationStreamRequestAgentVersionTypedDict,
         ConversationStreamRequestHandoffExecution,
         ConversationStreamRequestTools,
         ConversationStreamRequestToolsTypedDict,
@@ -750,6 +779,28 @@ if TYPE_CHECKING:
     from .paginationinfo import PaginationInfo, PaginationInfoTypedDict
     from .prediction import Prediction, PredictionTypedDict
     from .processingstatusout import ProcessingStatusOut, ProcessingStatusOutTypedDict
+    from .realtimetranscriptionerror import (
+        RealtimeTranscriptionError,
+        RealtimeTranscriptionErrorTypedDict,
+    )
+    from .realtimetranscriptionerrordetail import (
+        Message,
+        MessageTypedDict,
+        RealtimeTranscriptionErrorDetail,
+        RealtimeTranscriptionErrorDetailTypedDict,
+    )
+    from .realtimetranscriptionsession import (
+        RealtimeTranscriptionSession,
+        RealtimeTranscriptionSessionTypedDict,
+    )
+    from .realtimetranscriptionsessioncreated import (
+        RealtimeTranscriptionSessionCreated,
+        RealtimeTranscriptionSessionCreatedTypedDict,
+    )
+    from .realtimetranscriptionsessionupdated import (
+        RealtimeTranscriptionSessionUpdated,
+        RealtimeTranscriptionSessionUpdatedTypedDict,
+    )
     from .referencechunk import (
         ReferenceChunk,
         ReferenceChunkType,
@@ -943,7 +994,11 @@ if TYPE_CHECKING:
 __all__ = [
     "APIEndpoint",
     "Agent",
+    "AgentAliasResponse",
+    "AgentAliasResponseTypedDict",
     "AgentConversation",
+    "AgentConversationAgentVersion",
+    "AgentConversationAgentVersionTypedDict",
     "AgentConversationObject",
     "AgentConversationTypedDict",
     "AgentCreationRequest",
@@ -968,12 +1023,22 @@ __all__ = [
     "AgentUpdateRequestTools",
     "AgentUpdateRequestToolsTypedDict",
     "AgentUpdateRequestTypedDict",
+    "AgentVersion",
+    "AgentVersionTypedDict",
+    "AgentsAPIV1AgentsCreateOrUpdateAliasRequest",
+    "AgentsAPIV1AgentsCreateOrUpdateAliasRequestTypedDict",
     "AgentsAPIV1AgentsDeleteRequest",
     "AgentsAPIV1AgentsDeleteRequestTypedDict",
     "AgentsAPIV1AgentsGetRequest",
     "AgentsAPIV1AgentsGetRequestTypedDict",
+    "AgentsAPIV1AgentsGetVersionRequest",
+    "AgentsAPIV1AgentsGetVersionRequestTypedDict",
     "AgentsAPIV1AgentsListRequest",
     "AgentsAPIV1AgentsListRequestTypedDict",
+    "AgentsAPIV1AgentsListVersionAliasesRequest",
+    "AgentsAPIV1AgentsListVersionAliasesRequestTypedDict",
+    "AgentsAPIV1AgentsListVersionsRequest",
+    "AgentsAPIV1AgentsListVersionsRequestTypedDict",
     "AgentsAPIV1AgentsUpdateRequest",
     "AgentsAPIV1AgentsUpdateRequestTypedDict",
     "AgentsAPIV1AgentsUpdateVersionRequest",
@@ -1172,12 +1237,18 @@ __all__ = [
     "ConversationResponseObject",
     "ConversationResponseTypedDict",
     "ConversationRestartRequest",
+    "ConversationRestartRequestAgentVersion",
+    "ConversationRestartRequestAgentVersionTypedDict",
     "ConversationRestartRequestHandoffExecution",
     "ConversationRestartRequestTypedDict",
     "ConversationRestartStreamRequest",
+    "ConversationRestartStreamRequestAgentVersion",
+    "ConversationRestartStreamRequestAgentVersionTypedDict",
     "ConversationRestartStreamRequestHandoffExecution",
     "ConversationRestartStreamRequestTypedDict",
     "ConversationStreamRequest",
+    "ConversationStreamRequestAgentVersion",
+    "ConversationStreamRequestAgentVersionTypedDict",
     "ConversationStreamRequestHandoffExecution",
     "ConversationStreamRequestTools",
     "ConversationStreamRequestToolsTypedDict",
@@ -1419,6 +1490,7 @@ __all__ = [
     "ListSharingOutTypedDict",
     "Loc",
     "LocTypedDict",
+    "Message",
     "MessageEntries",
     "MessageEntriesTypedDict",
     "MessageInputContentChunks",
@@ -1444,6 +1516,7 @@ __all__ = [
     "MessageOutputEventRole",
     "MessageOutputEventType",
     "MessageOutputEventTypedDict",
+    "MessageTypedDict",
     "Messages",
     "MessagesTypedDict",
     "MetricOut",
@@ -1494,7 +1567,19 @@ __all__ = [
     "PredictionTypedDict",
     "ProcessingStatusOut",
     "ProcessingStatusOutTypedDict",
+    "QueryParamAgentVersion",
+    "QueryParamAgentVersionTypedDict",
     "QueryParamStatus",
+    "RealtimeTranscriptionError",
+    "RealtimeTranscriptionErrorDetail",
+    "RealtimeTranscriptionErrorDetailTypedDict",
+    "RealtimeTranscriptionErrorTypedDict",
+    "RealtimeTranscriptionSession",
+    "RealtimeTranscriptionSessionCreated",
+    "RealtimeTranscriptionSessionCreatedTypedDict",
+    "RealtimeTranscriptionSessionTypedDict",
+    "RealtimeTranscriptionSessionUpdated",
+    "RealtimeTranscriptionSessionUpdatedTypedDict",
     "ReferenceChunk",
     "ReferenceChunkType",
     "ReferenceChunkTypedDict",
@@ -1663,7 +1748,11 @@ _dynamic_imports: dict[str, str] = {
     "AgentTools": ".agent",
     "AgentToolsTypedDict": ".agent",
     "AgentTypedDict": ".agent",
+    "AgentAliasResponse": ".agentaliasresponse",
+    "AgentAliasResponseTypedDict": ".agentaliasresponse",
     "AgentConversation": ".agentconversation",
+    "AgentConversationAgentVersion": ".agentconversation",
+    "AgentConversationAgentVersionTypedDict": ".agentconversation",
     "AgentConversationObject": ".agentconversation",
     "AgentConversationTypedDict": ".agentconversation",
     "AgentCreationRequest": ".agentcreationrequest",
@@ -1680,10 +1769,20 @@ _dynamic_imports: dict[str, str] = {
     "AgentHandoffStartedEvent": ".agenthandoffstartedevent",
     "AgentHandoffStartedEventType": ".agenthandoffstartedevent",
     "AgentHandoffStartedEventTypedDict": ".agenthandoffstartedevent",
+    "AgentsAPIV1AgentsCreateOrUpdateAliasRequest": ".agents_api_v1_agents_create_or_update_aliasop",
+    "AgentsAPIV1AgentsCreateOrUpdateAliasRequestTypedDict": ".agents_api_v1_agents_create_or_update_aliasop",
     "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",
+    "QueryParamAgentVersion": ".agents_api_v1_agents_getop",
+    "QueryParamAgentVersionTypedDict": ".agents_api_v1_agents_getop",
+    "AgentsAPIV1AgentsListVersionAliasesRequest": ".agents_api_v1_agents_list_version_aliasesop",
+    "AgentsAPIV1AgentsListVersionAliasesRequestTypedDict": ".agents_api_v1_agents_list_version_aliasesop",
+    "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",
@@ -1897,6 +1996,8 @@ _dynamic_imports: dict[str, str] = {
     "ConversationMessages": ".conversationmessages",
     "ConversationMessagesObject": ".conversationmessages",
     "ConversationMessagesTypedDict": ".conversationmessages",
+    "AgentVersion": ".conversationrequest",
+    "AgentVersionTypedDict": ".conversationrequest",
     "ConversationRequest": ".conversationrequest",
     "ConversationRequestTypedDict": ".conversationrequest",
     "HandoffExecution": ".conversationrequest",
@@ -1908,12 +2009,18 @@ _dynamic_imports: dict[str, str] = {
     "Outputs": ".conversationresponse",
     "OutputsTypedDict": ".conversationresponse",
     "ConversationRestartRequest": ".conversationrestartrequest",
+    "ConversationRestartRequestAgentVersion": ".conversationrestartrequest",
+    "ConversationRestartRequestAgentVersionTypedDict": ".conversationrestartrequest",
     "ConversationRestartRequestHandoffExecution": ".conversationrestartrequest",
     "ConversationRestartRequestTypedDict": ".conversationrestartrequest",
     "ConversationRestartStreamRequest": ".conversationrestartstreamrequest",
+    "ConversationRestartStreamRequestAgentVersion": ".conversationrestartstreamrequest",
+    "ConversationRestartStreamRequestAgentVersionTypedDict": ".conversationrestartstreamrequest",
     "ConversationRestartStreamRequestHandoffExecution": ".conversationrestartstreamrequest",
     "ConversationRestartStreamRequestTypedDict": ".conversationrestartstreamrequest",
     "ConversationStreamRequest": ".conversationstreamrequest",
+    "ConversationStreamRequestAgentVersion": ".conversationstreamrequest",
+    "ConversationStreamRequestAgentVersionTypedDict": ".conversationstreamrequest",
     "ConversationStreamRequestHandoffExecution": ".conversationstreamrequest",
     "ConversationStreamRequestTools": ".conversationstreamrequest",
     "ConversationStreamRequestToolsTypedDict": ".conversationstreamrequest",
@@ -2221,6 +2328,18 @@ _dynamic_imports: dict[str, str] = {
     "PredictionTypedDict": ".prediction",
     "ProcessingStatusOut": ".processingstatusout",
     "ProcessingStatusOutTypedDict": ".processingstatusout",
+    "RealtimeTranscriptionError": ".realtimetranscriptionerror",
+    "RealtimeTranscriptionErrorTypedDict": ".realtimetranscriptionerror",
+    "Message": ".realtimetranscriptionerrordetail",
+    "MessageTypedDict": ".realtimetranscriptionerrordetail",
+    "RealtimeTranscriptionErrorDetail": ".realtimetranscriptionerrordetail",
+    "RealtimeTranscriptionErrorDetailTypedDict": ".realtimetranscriptionerrordetail",
+    "RealtimeTranscriptionSession": ".realtimetranscriptionsession",
+    "RealtimeTranscriptionSessionTypedDict": ".realtimetranscriptionsession",
+    "RealtimeTranscriptionSessionCreated": ".realtimetranscriptionsessioncreated",
+    "RealtimeTranscriptionSessionCreatedTypedDict": ".realtimetranscriptionsessioncreated",
+    "RealtimeTranscriptionSessionUpdated": ".realtimetranscriptionsessionupdated",
+    "RealtimeTranscriptionSessionUpdatedTypedDict": ".realtimetranscriptionsessionupdated",
     "ReferenceChunk": ".referencechunk",
     "ReferenceChunkType": ".referencechunk",
     "ReferenceChunkTypedDict": ".referencechunk",

mistralai/models/agent.py

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

mistralai/models/agentaliasresponse.py

@@ -0,0 +1,23 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from datetime import datetime
+from mistralai.types import BaseModel
+from typing_extensions import TypedDict
+
+
+class AgentAliasResponseTypedDict(TypedDict):
+    alias: str
+    version: int
+    created_at: datetime
+    updated_at: datetime
+
+
+class AgentAliasResponse(BaseModel):
+    alias: str
+
+    version: int
+
+    created_at: datetime
+
+    updated_at: datetime

mistralai/models/agentconversation.py

@@ -4,11 +4,21 @@ from __future__ import annotations
 from datetime import datetime
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Any, Dict, Literal, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import Any, Dict, Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-AgentConversationObject = Literal["conversation"]
+AgentConversationObject = Literal["conversation",]
+
+
+AgentConversationAgentVersionTypedDict = TypeAliasType(
+    "AgentConversationAgentVersionTypedDict", Union[str, int]
+)
+
+
+AgentConversationAgentVersion = TypeAliasType(
+    "AgentConversationAgentVersion", Union[str, int]
+)
 
 
 class AgentConversationTypedDict(TypedDict):
@@ -23,7 +33,7 @@ class AgentConversationTypedDict(TypedDict):
     metadata: NotRequired[Nullable[Dict[str, Any]]]
     r"""Custom metadata for the conversation."""
     object: NotRequired[AgentConversationObject]
-    agent_version: NotRequired[Nullable[int]]
+    agent_version: NotRequired[Nullable[AgentConversationAgentVersionTypedDict]]
 
 
 class AgentConversation(BaseModel):
@@ -46,7 +56,7 @@ class AgentConversation(BaseModel):
 
     object: Optional[AgentConversationObject] = "conversation"
 
-    agent_version: OptionalNullable[int] = UNSET
+    agent_version: OptionalNullable[AgentConversationAgentVersion] = UNSET
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):

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_create_or_update_aliasop.py

@@ -0,0 +1,26 @@
+"""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_extensions import Annotated, TypedDict
+
+
+class AgentsAPIV1AgentsCreateOrUpdateAliasRequestTypedDict(TypedDict):
+    agent_id: str
+    alias: str
+    version: int
+
+
+class AgentsAPIV1AgentsCreateOrUpdateAliasRequest(BaseModel):
+    agent_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    alias: Annotated[
+        str, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
+    ]
+
+    version: Annotated[
+        int, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
+    ]

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: str
+
+
+class AgentsAPIV1AgentsGetVersionRequest(BaseModel):
+    agent_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    version: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]

mistralai/models/agents_api_v1_agents_getop.py

@@ -4,12 +4,21 @@ from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mistralai.utils import FieldMetadata, PathParamMetadata, QueryParamMetadata
 from pydantic import model_serializer
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing import Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
+
+
+QueryParamAgentVersionTypedDict = TypeAliasType(
+    "QueryParamAgentVersionTypedDict", Union[int, str]
+)
+
+
+QueryParamAgentVersion = TypeAliasType("QueryParamAgentVersion", Union[int, str])
 
 
 class AgentsAPIV1AgentsGetRequestTypedDict(TypedDict):
     agent_id: str
-    agent_version: NotRequired[Nullable[int]]
+    agent_version: NotRequired[Nullable[QueryParamAgentVersionTypedDict]]
 
 
 class AgentsAPIV1AgentsGetRequest(BaseModel):
@@ -18,7 +27,7 @@ class AgentsAPIV1AgentsGetRequest(BaseModel):
     ]
 
     agent_version: Annotated[
-        OptionalNullable[int],
+        OptionalNullable[QueryParamAgentVersion],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = UNSET
 

mistralai/models/agents_api_v1_agents_list_version_aliasesop.py

@@ -0,0 +1,16 @@
+"""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 AgentsAPIV1AgentsListVersionAliasesRequestTypedDict(TypedDict):
+    agent_id: str
+
+
+class AgentsAPIV1AgentsListVersionAliasesRequest(BaseModel):
+    agent_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]