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

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,6 +44,10 @@ 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,
@@ -52,6 +59,12 @@ if TYPE_CHECKING:
     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,
@@ -340,6 +353,8 @@ if TYPE_CHECKING:
         ConversationMessagesTypedDict,
     )
     from .conversationrequest import (
+        AgentVersion,
+        AgentVersionTypedDict,
         ConversationRequest,
         ConversationRequestTypedDict,
         HandoffExecution,
@@ -355,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,
@@ -758,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,
@@ -951,7 +994,11 @@ if TYPE_CHECKING:
 __all__ = [
     "APIEndpoint",
     "Agent",
+    "AgentAliasResponse",
+    "AgentAliasResponseTypedDict",
     "AgentConversation",
+    "AgentConversationAgentVersion",
+    "AgentConversationAgentVersionTypedDict",
     "AgentConversationObject",
     "AgentConversationTypedDict",
     "AgentCreationRequest",
@@ -976,6 +1023,10 @@ __all__ = [
     "AgentUpdateRequestTools",
     "AgentUpdateRequestToolsTypedDict",
     "AgentUpdateRequestTypedDict",
+    "AgentVersion",
+    "AgentVersionTypedDict",
+    "AgentsAPIV1AgentsCreateOrUpdateAliasRequest",
+    "AgentsAPIV1AgentsCreateOrUpdateAliasRequestTypedDict",
     "AgentsAPIV1AgentsDeleteRequest",
     "AgentsAPIV1AgentsDeleteRequestTypedDict",
     "AgentsAPIV1AgentsGetRequest",
@@ -984,6 +1035,8 @@ __all__ = [
     "AgentsAPIV1AgentsGetVersionRequestTypedDict",
     "AgentsAPIV1AgentsListRequest",
     "AgentsAPIV1AgentsListRequestTypedDict",
+    "AgentsAPIV1AgentsListVersionAliasesRequest",
+    "AgentsAPIV1AgentsListVersionAliasesRequestTypedDict",
     "AgentsAPIV1AgentsListVersionsRequest",
     "AgentsAPIV1AgentsListVersionsRequestTypedDict",
     "AgentsAPIV1AgentsUpdateRequest",
@@ -1184,12 +1237,18 @@ __all__ = [
     "ConversationResponseObject",
     "ConversationResponseTypedDict",
     "ConversationRestartRequest",
+    "ConversationRestartRequestAgentVersion",
+    "ConversationRestartRequestAgentVersionTypedDict",
     "ConversationRestartRequestHandoffExecution",
     "ConversationRestartRequestTypedDict",
     "ConversationRestartStreamRequest",
+    "ConversationRestartStreamRequestAgentVersion",
+    "ConversationRestartStreamRequestAgentVersionTypedDict",
     "ConversationRestartStreamRequestHandoffExecution",
     "ConversationRestartStreamRequestTypedDict",
     "ConversationStreamRequest",
+    "ConversationStreamRequestAgentVersion",
+    "ConversationStreamRequestAgentVersionTypedDict",
     "ConversationStreamRequestHandoffExecution",
     "ConversationStreamRequestTools",
     "ConversationStreamRequestToolsTypedDict",
@@ -1431,6 +1490,7 @@ __all__ = [
     "ListSharingOutTypedDict",
     "Loc",
     "LocTypedDict",
+    "Message",
     "MessageEntries",
     "MessageEntriesTypedDict",
     "MessageInputContentChunks",
@@ -1456,6 +1516,7 @@ __all__ = [
     "MessageOutputEventRole",
     "MessageOutputEventType",
     "MessageOutputEventTypedDict",
+    "MessageTypedDict",
     "Messages",
     "MessagesTypedDict",
     "MetricOut",
@@ -1506,7 +1567,19 @@ __all__ = [
     "PredictionTypedDict",
     "ProcessingStatusOut",
     "ProcessingStatusOutTypedDict",
+    "QueryParamAgentVersion",
+    "QueryParamAgentVersionTypedDict",
     "QueryParamStatus",
+    "RealtimeTranscriptionError",
+    "RealtimeTranscriptionErrorDetail",
+    "RealtimeTranscriptionErrorDetailTypedDict",
+    "RealtimeTranscriptionErrorTypedDict",
+    "RealtimeTranscriptionSession",
+    "RealtimeTranscriptionSessionCreated",
+    "RealtimeTranscriptionSessionCreatedTypedDict",
+    "RealtimeTranscriptionSessionTypedDict",
+    "RealtimeTranscriptionSessionUpdated",
+    "RealtimeTranscriptionSessionUpdatedTypedDict",
     "ReferenceChunk",
     "ReferenceChunkType",
     "ReferenceChunkTypedDict",
@@ -1675,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",
@@ -1692,12 +1769,18 @@ _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",
@@ -1913,6 +1996,8 @@ _dynamic_imports: dict[str, str] = {
     "ConversationMessages": ".conversationmessages",
     "ConversationMessagesObject": ".conversationmessages",
     "ConversationMessagesTypedDict": ".conversationmessages",
+    "AgentVersion": ".conversationrequest",
+    "AgentVersionTypedDict": ".conversationrequest",
     "ConversationRequest": ".conversationrequest",
     "ConversationRequestTypedDict": ".conversationrequest",
     "HandoffExecution": ".conversationrequest",
@@ -1924,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",
@@ -2237,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/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,13 +4,23 @@ 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",]
 
 
+AgentConversationAgentVersionTypedDict = TypeAliasType(
+    "AgentConversationAgentVersionTypedDict", Union[str, int]
+)
+
+
+AgentConversationAgentVersion = TypeAliasType(
+    "AgentConversationAgentVersion", Union[str, int]
+)
+
+
 class AgentConversationTypedDict(TypedDict):
     id: str
     created_at: datetime
@@ -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/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

@@ -8,7 +8,7 @@ from typing_extensions import Annotated, TypedDict
 
 class AgentsAPIV1AgentsGetVersionRequestTypedDict(TypedDict):
     agent_id: str
-    version: int
+    version: str
 
 
 class AgentsAPIV1AgentsGetVersionRequest(BaseModel):
@@ -17,5 +17,5 @@ class AgentsAPIV1AgentsGetVersionRequest(BaseModel):
     ]
 
     version: Annotated[
-        int, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+        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))
+    ]

mistralai/models/audiotranscriptionrequest.py

@@ -24,6 +24,8 @@ class AudioTranscriptionRequestTypedDict(TypedDict):
     r"""Language of the audio, e.g. 'en'. Providing the language can boost accuracy."""
     temperature: NotRequired[Nullable[float]]
     stream: Literal[False]
+    diarize: NotRequired[bool]
+    context_bias: NotRequired[List[str]]
     timestamp_granularities: NotRequired[List[TimestampGranularity]]
     r"""Granularities of timestamps to include in the response."""
 
@@ -55,6 +57,10 @@ class AudioTranscriptionRequest(BaseModel):
         FieldMetadata(multipart=True),
     ] = False
 
+    diarize: Annotated[Optional[bool], FieldMetadata(multipart=True)] = False
+
+    context_bias: Annotated[Optional[List[str]], FieldMetadata(multipart=True)] = None
+
     timestamp_granularities: Annotated[
         Optional[List[TimestampGranularity]], FieldMetadata(multipart=True)
     ] = None
@@ -69,6 +75,8 @@ class AudioTranscriptionRequest(BaseModel):
             "language",
             "temperature",
             "stream",
+            "diarize",
+            "context_bias",
             "timestamp_granularities",
         ]
         nullable_fields = ["file_url", "file_id", "language", "temperature"]

mistralai/models/audiotranscriptionrequeststream.py

@@ -23,6 +23,8 @@ class AudioTranscriptionRequestStreamTypedDict(TypedDict):
     r"""Language of the audio, e.g. 'en'. Providing the language can boost accuracy."""
     temperature: NotRequired[Nullable[float]]
     stream: Literal[True]
+    diarize: NotRequired[bool]
+    context_bias: NotRequired[List[str]]
     timestamp_granularities: NotRequired[List[TimestampGranularity]]
     r"""Granularities of timestamps to include in the response."""
 
@@ -53,6 +55,10 @@ class AudioTranscriptionRequestStream(BaseModel):
         FieldMetadata(multipart=True),
     ] = True
 
+    diarize: Annotated[Optional[bool], FieldMetadata(multipart=True)] = False
+
+    context_bias: Annotated[Optional[List[str]], FieldMetadata(multipart=True)] = None
+
     timestamp_granularities: Annotated[
         Optional[List[TimestampGranularity]], FieldMetadata(multipart=True)
     ] = None
@@ -67,6 +73,8 @@ class AudioTranscriptionRequestStream(BaseModel):
             "language",
             "temperature",
             "stream",
+            "diarize",
+            "context_bias",
             "timestamp_granularities",
         ]
         nullable_fields = ["file_url", "file_id", "language", "temperature"]

mistralai/models/conversationrequest.py

@@ -48,6 +48,12 @@ Tools = Annotated[
 ]
 
 
+AgentVersionTypedDict = TypeAliasType("AgentVersionTypedDict", Union[str, int])
+
+
+AgentVersion = TypeAliasType("AgentVersion", Union[str, int])
+
+
 class ConversationRequestTypedDict(TypedDict):
     inputs: ConversationInputsTypedDict
     stream: NotRequired[bool]
@@ -61,7 +67,7 @@ class ConversationRequestTypedDict(TypedDict):
     description: NotRequired[Nullable[str]]
     metadata: NotRequired[Nullable[Dict[str, Any]]]
     agent_id: NotRequired[Nullable[str]]
-    agent_version: NotRequired[Nullable[int]]
+    agent_version: NotRequired[Nullable[AgentVersionTypedDict]]
     model: NotRequired[Nullable[str]]
 
 
@@ -89,7 +95,7 @@ class ConversationRequest(BaseModel):
 
     agent_id: OptionalNullable[str] = UNSET
 
-    agent_version: OptionalNullable[int] = UNSET
+    agent_version: OptionalNullable[AgentVersion] = UNSET
 
     model: OptionalNullable[str] = UNSET
 

mistralai/models/conversationrestartrequest.py

@@ -5,8 +5,8 @@ from .completionargs import CompletionArgs, CompletionArgsTypedDict
 from .conversationinputs import ConversationInputs, ConversationInputsTypedDict
 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
 
 
 ConversationRestartRequestHandoffExecution = Literal[
@@ -15,6 +15,18 @@ ConversationRestartRequestHandoffExecution = Literal[
 ]
 
 
+ConversationRestartRequestAgentVersionTypedDict = TypeAliasType(
+    "ConversationRestartRequestAgentVersionTypedDict", Union[str, int]
+)
+r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
+
+
+ConversationRestartRequestAgentVersion = TypeAliasType(
+    "ConversationRestartRequestAgentVersion", Union[str, int]
+)
+r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
+
+
 class ConversationRestartRequestTypedDict(TypedDict):
     r"""Request to restart a new conversation from a given entry in the conversation."""
 
@@ -28,7 +40,9 @@ class ConversationRestartRequestTypedDict(TypedDict):
     r"""White-listed arguments from the completion API"""
     metadata: NotRequired[Nullable[Dict[str, Any]]]
     r"""Custom metadata for the conversation."""
-    agent_version: NotRequired[Nullable[int]]
+    agent_version: NotRequired[
+        Nullable[ConversationRestartRequestAgentVersionTypedDict]
+    ]
     r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
 
 
@@ -52,7 +66,7 @@ class ConversationRestartRequest(BaseModel):
     metadata: OptionalNullable[Dict[str, Any]] = UNSET
     r"""Custom metadata for the conversation."""
 
-    agent_version: OptionalNullable[int] = UNSET
+    agent_version: OptionalNullable[ConversationRestartRequestAgentVersion] = UNSET
     r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
 
     @model_serializer(mode="wrap")

mistralai/models/conversationrestartstreamrequest.py

@@ -5,8 +5,8 @@ from .completionargs import CompletionArgs, CompletionArgsTypedDict
 from .conversationinputs import ConversationInputs, ConversationInputsTypedDict
 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
 
 
 ConversationRestartStreamRequestHandoffExecution = Literal[
@@ -15,6 +15,18 @@ ConversationRestartStreamRequestHandoffExecution = Literal[
 ]
 
 
+ConversationRestartStreamRequestAgentVersionTypedDict = TypeAliasType(
+    "ConversationRestartStreamRequestAgentVersionTypedDict", Union[str, int]
+)
+r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
+
+
+ConversationRestartStreamRequestAgentVersion = TypeAliasType(
+    "ConversationRestartStreamRequestAgentVersion", Union[str, int]
+)
+r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
+
+
 class ConversationRestartStreamRequestTypedDict(TypedDict):
     r"""Request to restart a new conversation from a given entry in the conversation."""
 
@@ -28,7 +40,9 @@ class ConversationRestartStreamRequestTypedDict(TypedDict):
     r"""White-listed arguments from the completion API"""
     metadata: NotRequired[Nullable[Dict[str, Any]]]
     r"""Custom metadata for the conversation."""
-    agent_version: NotRequired[Nullable[int]]
+    agent_version: NotRequired[
+        Nullable[ConversationRestartStreamRequestAgentVersionTypedDict]
+    ]
     r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
 
 
@@ -54,7 +68,9 @@ class ConversationRestartStreamRequest(BaseModel):
     metadata: OptionalNullable[Dict[str, Any]] = UNSET
     r"""Custom metadata for the conversation."""
 
-    agent_version: OptionalNullable[int] = UNSET
+    agent_version: OptionalNullable[ConversationRestartStreamRequestAgentVersion] = (
+        UNSET
+    )
     r"""Specific version of the agent to use when restarting. If not provided, uses the current version."""
 
     @model_serializer(mode="wrap")

mistralai/models/conversationstreamrequest.py

@@ -48,6 +48,16 @@ ConversationStreamRequestTools = Annotated[
 ]
 
 
+ConversationStreamRequestAgentVersionTypedDict = TypeAliasType(
+    "ConversationStreamRequestAgentVersionTypedDict", Union[str, int]
+)
+
+
+ConversationStreamRequestAgentVersion = TypeAliasType(
+    "ConversationStreamRequestAgentVersion", Union[str, int]
+)
+
+
 class ConversationStreamRequestTypedDict(TypedDict):
     inputs: ConversationInputsTypedDict
     stream: NotRequired[bool]
@@ -61,7 +71,7 @@ class ConversationStreamRequestTypedDict(TypedDict):
     description: NotRequired[Nullable[str]]
     metadata: NotRequired[Nullable[Dict[str, Any]]]
     agent_id: NotRequired[Nullable[str]]
-    agent_version: NotRequired[Nullable[int]]
+    agent_version: NotRequired[Nullable[ConversationStreamRequestAgentVersionTypedDict]]
     model: NotRequired[Nullable[str]]
 
 
@@ -91,7 +101,7 @@ class ConversationStreamRequest(BaseModel):
 
     agent_id: OptionalNullable[str] = UNSET
 
-    agent_version: OptionalNullable[int] = UNSET
+    agent_version: OptionalNullable[ConversationStreamRequestAgentVersion] = UNSET
 
     model: OptionalNullable[str] = UNSET