mistralai 1.11.1__tar.gz → 1.12.0__tar.gz

{mistralai-1.11.1 → mistralai-1.12.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mistralai
-Version: 1.11.1
+Version: 1.12.0
 Summary: Python Client SDK for the Mistral AI API.
 Project-URL: Repository, https://github.com/mistralai/client-python.git
 Author: Mistral
@@ -24,6 +24,8 @@ Requires-Dist: mcp<2.0,>=1.0; extra == 'agents'
 Provides-Extra: gcp
 Requires-Dist: google-auth>=2.27.0; extra == 'gcp'
 Requires-Dist: requests>=2.32.3; extra == 'gcp'
+Provides-Extra: realtime
+Requires-Dist: websockets>=13.0; extra == 'realtime'
 Description-Content-Type: text/markdown
 
 # Mistral Python Client
@@ -503,6 +505,8 @@ The documentation for the GCP SDK is available [here](packages/mistralai_gcp/REA
 * [update_version](docs/sdks/mistralagents/README.md#update_version) - Update an agent version.
 * [list_versions](docs/sdks/mistralagents/README.md#list_versions) - List all versions of an agent.
 * [get_version](docs/sdks/mistralagents/README.md#get_version) - Retrieve a specific version of an agent.
+* [create_version_alias](docs/sdks/mistralagents/README.md#create_version_alias) - Create or update an agent version alias.
+* [list_version_aliases](docs/sdks/mistralagents/README.md#list_version_aliases) - List all aliases for an agent.
 
 ### [Beta.Conversations](docs/sdks/conversations/README.md)
 
@@ -780,7 +784,7 @@ with Mistral(
 
 
 **Inherit from [`MistralError`](./src/mistralai/models/mistralerror.py)**:
-* [`HTTPValidationError`](./src/mistralai/models/httpvalidationerror.py): Validation Error. Status code `422`. Applicable to 50 of 72 methods.*
+* [`HTTPValidationError`](./src/mistralai/models/httpvalidationerror.py): Validation Error. Status code `422`. Applicable to 52 of 74 methods.*
 * [`ResponseValidationError`](./src/mistralai/models/responsevalidationerror.py): Type mismatch between the response data and the expected Pydantic model. Provides access to the Pydantic validation error via the `cause` attribute.
 
 </details>

{mistralai-1.11.1 → mistralai-1.12.0}/README.md

@@ -475,6 +475,8 @@ The documentation for the GCP SDK is available [here](packages/mistralai_gcp/REA
 * [update_version](docs/sdks/mistralagents/README.md#update_version) - Update an agent version.
 * [list_versions](docs/sdks/mistralagents/README.md#list_versions) - List all versions of an agent.
 * [get_version](docs/sdks/mistralagents/README.md#get_version) - Retrieve a specific version of an agent.
+* [create_version_alias](docs/sdks/mistralagents/README.md#create_version_alias) - Create or update an agent version alias.
+* [list_version_aliases](docs/sdks/mistralagents/README.md#list_version_aliases) - List all aliases for an agent.
 
 ### [Beta.Conversations](docs/sdks/conversations/README.md)
 
@@ -752,7 +754,7 @@ with Mistral(
 
 
 **Inherit from [`MistralError`](./src/mistralai/models/mistralerror.py)**:
-* [`HTTPValidationError`](./src/mistralai/models/httpvalidationerror.py): Validation Error. Status code `422`. Applicable to 50 of 72 methods.*
+* [`HTTPValidationError`](./src/mistralai/models/httpvalidationerror.py): Validation Error. Status code `422`. Applicable to 52 of 74 methods.*
 * [`ResponseValidationError`](./src/mistralai/models/responsevalidationerror.py): Type mismatch between the response data and the expected Pydantic model. Provides access to the Pydantic validation error via the `cause` attribute.
 
 </details>

{mistralai-1.11.1 → mistralai-1.12.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mistralai"
-version = "1.11.1"
+version = "1.12.0"
 description = "Python Client SDK for the Mistral AI API."
 authors = [{ name = "Mistral" }]
 requires-python = ">=3.10"
@@ -29,6 +29,9 @@ agents = [
     "griffe >=1.7.3,<2.0",
     "authlib >=1.5.2,<2.0",
 ]
+realtime = [
+    "websockets >=13.0",
+]
 
 [project.urls]
 Repository = "https://github.com/mistralai/client-python.git"
@@ -45,6 +48,7 @@ dev = [
     "mcp>=1.0,<2",
     "griffe>=1.7.3,<2",
     "authlib>=1.5.2,<2",
+    "websockets >=13.0",
 ]
 lint = [
     "ruff>=0.11.10,<0.12",
@@ -103,6 +107,7 @@ module = [
     "jsonpath.*",
     "typing_inspect.*",
     "authlib.*",
+    "websockets.*",
     "mcp.*",
     "griffe.*"
 ]

{mistralai-1.11.1 → mistralai-1.12.0}/src/mistralai/_version.py

@@ -3,10 +3,10 @@
 import importlib.metadata
 
 __title__: str = "mistralai"
-__version__: str = "1.11.1"
+__version__: str = "1.12.0"
 __openapi_doc_version__: str = "1.0.0"
 __gen_version__: str = "2.794.1"
-__user_agent__: str = "speakeasy-sdk/python 1.11.1 2.794.1 1.0.0 mistralai"
+__user_agent__: str = "speakeasy-sdk/python 1.12.0 2.794.1 1.0.0 mistralai"
 
 try:
     if __package__ is not None:

{mistralai-1.11.1 → mistralai-1.12.0}/src/mistralai/audio.py

@@ -5,6 +5,13 @@ from .sdkconfiguration import SDKConfiguration
 from mistralai.transcriptions import Transcriptions
 from typing import Optional
 
+# region imports
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from mistralai.extra.realtime import RealtimeTranscription
+# endregion imports
+
 
 class Audio(BaseSDK):
     transcriptions: Transcriptions
@@ -21,3 +28,16 @@ class Audio(BaseSDK):
         self.transcriptions = Transcriptions(
             self.sdk_configuration, parent_ref=self.parent_ref
         )
+
+    # region sdk-class-body
+    @property
+    def realtime(self) -> "RealtimeTranscription":
+        """Returns a client for real-time audio transcription via WebSocket."""
+        if not hasattr(self, "_realtime"):
+            from mistralai.extra.realtime import RealtimeTranscription  # pylint: disable=import-outside-toplevel
+
+            self._realtime = RealtimeTranscription(self.sdk_configuration)  # pylint: disable=attribute-defined-outside-init
+
+        return self._realtime
+
+    # endregion sdk-class-body

{mistralai-1.11.1 → mistralai-1.12.0}/src/mistralai/conversations.py

@@ -259,7 +259,12 @@ class Conversations(BaseSDK):
         description: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         agent_id: OptionalNullable[str] = UNSET,
-        agent_version: OptionalNullable[int] = UNSET,
+        agent_version: OptionalNullable[
+            Union[
+                models_conversationrequest.AgentVersion,
+                models_conversationrequest.AgentVersionTypedDict,
+            ]
+        ] = UNSET,
         model: OptionalNullable[str] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -405,7 +410,12 @@ class Conversations(BaseSDK):
         description: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         agent_id: OptionalNullable[str] = UNSET,
-        agent_version: OptionalNullable[int] = UNSET,
+        agent_version: OptionalNullable[
+            Union[
+                models_conversationrequest.AgentVersion,
+                models_conversationrequest.AgentVersionTypedDict,
+            ]
+        ] = UNSET,
         model: OptionalNullable[str] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -1711,7 +1721,12 @@ class Conversations(BaseSDK):
             ]
         ] = None,
         metadata: OptionalNullable[Dict[str, Any]] = UNSET,
-        agent_version: OptionalNullable[int] = UNSET,
+        agent_version: OptionalNullable[
+            Union[
+                models_conversationrestartrequest.ConversationRestartRequestAgentVersion,
+                models_conversationrestartrequest.ConversationRestartRequestAgentVersionTypedDict,
+            ]
+        ] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -1846,7 +1861,12 @@ class Conversations(BaseSDK):
             ]
         ] = None,
         metadata: OptionalNullable[Dict[str, Any]] = UNSET,
-        agent_version: OptionalNullable[int] = UNSET,
+        agent_version: OptionalNullable[
+            Union[
+                models_conversationrestartrequest.ConversationRestartRequestAgentVersion,
+                models_conversationrestartrequest.ConversationRestartRequestAgentVersionTypedDict,
+            ]
+        ] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -1991,7 +2011,12 @@ class Conversations(BaseSDK):
         description: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         agent_id: OptionalNullable[str] = UNSET,
-        agent_version: OptionalNullable[int] = UNSET,
+        agent_version: OptionalNullable[
+            Union[
+                models_conversationstreamrequest.ConversationStreamRequestAgentVersion,
+                models_conversationstreamrequest.ConversationStreamRequestAgentVersionTypedDict,
+            ]
+        ] = UNSET,
         model: OptionalNullable[str] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -2148,7 +2173,12 @@ class Conversations(BaseSDK):
         description: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         agent_id: OptionalNullable[str] = UNSET,
-        agent_version: OptionalNullable[int] = UNSET,
+        agent_version: OptionalNullable[
+            Union[
+                models_conversationstreamrequest.ConversationStreamRequestAgentVersion,
+                models_conversationstreamrequest.ConversationStreamRequestAgentVersionTypedDict,
+            ]
+        ] = UNSET,
         model: OptionalNullable[str] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -2561,7 +2591,12 @@ class Conversations(BaseSDK):
             ]
         ] = None,
         metadata: OptionalNullable[Dict[str, Any]] = UNSET,
-        agent_version: OptionalNullable[int] = UNSET,
+        agent_version: OptionalNullable[
+            Union[
+                models_conversationrestartstreamrequest.ConversationRestartStreamRequestAgentVersion,
+                models_conversationrestartstreamrequest.ConversationRestartStreamRequestAgentVersionTypedDict,
+            ]
+        ] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -2703,7 +2738,12 @@ class Conversations(BaseSDK):
             ]
         ] = None,
         metadata: OptionalNullable[Dict[str, Any]] = UNSET,
-        agent_version: OptionalNullable[int] = UNSET,
+        agent_version: OptionalNullable[
+            Union[
+                models_conversationrestartstreamrequest.ConversationRestartStreamRequestAgentVersion,
+                models_conversationrestartstreamrequest.ConversationRestartStreamRequestAgentVersionTypedDict,
+            ]
+        ] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,

mistralai-1.12.0/src/mistralai/extra/__init__.py

@@ -0,0 +1,61 @@
+from typing import TYPE_CHECKING
+
+from .struct_chat import (
+    ParsedChatCompletionResponse,
+    convert_to_parsed_chat_completion_response,
+)
+from .utils import response_format_from_pydantic_model
+from .utils.response_format import CustomPydanticModel
+
+if TYPE_CHECKING:
+    from .realtime import (
+        AudioEncoding,
+        AudioFormat,
+        RealtimeConnection,
+        RealtimeTranscriptionError,
+        RealtimeTranscriptionErrorDetail,
+        RealtimeTranscriptionSession,
+        RealtimeTranscriptionSessionCreated,
+        RealtimeTranscriptionSessionUpdated,
+        RealtimeTranscription,
+        UnknownRealtimeEvent,
+    )
+
+_REALTIME_EXPORTS = {
+    "RealtimeTranscription",
+    "RealtimeConnection",
+    "AudioEncoding",
+    "AudioFormat",
+    "UnknownRealtimeEvent",
+    "RealtimeTranscriptionError",
+    "RealtimeTranscriptionErrorDetail",
+    "RealtimeTranscriptionSession",
+    "RealtimeTranscriptionSessionCreated",
+    "RealtimeTranscriptionSessionUpdated",
+}
+
+
+def __getattr__(name: str):
+    if name in _REALTIME_EXPORTS:
+        from . import realtime
+
+        return getattr(realtime, name)
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+__all__ = [
+    "convert_to_parsed_chat_completion_response",
+    "response_format_from_pydantic_model",
+    "CustomPydanticModel",
+    "ParsedChatCompletionResponse",
+    "RealtimeTranscription",
+    "RealtimeConnection",
+    "AudioEncoding",
+    "AudioFormat",
+    "UnknownRealtimeEvent",
+    "RealtimeTranscriptionError",
+    "RealtimeTranscriptionErrorDetail",
+    "RealtimeTranscriptionSession",
+    "RealtimeTranscriptionSessionCreated",
+    "RealtimeTranscriptionSessionUpdated",
+]

mistralai-1.12.0/src/mistralai/extra/exceptions.py

@@ -0,0 +1,59 @@
+from typing import Optional, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from mistralai.models import RealtimeTranscriptionError
+
+
+class MistralClientException(Exception):
+    """Base exception for client errors."""
+
+
+class RunException(MistralClientException):
+    """Conversation run errors."""
+
+
+class MCPException(MistralClientException):
+    """MCP operation errors."""
+
+
+class MCPAuthException(MCPException):
+    """MCP authentication errors."""
+
+
+class RealtimeTranscriptionException(MistralClientException):
+    """Base realtime transcription exception."""
+
+    def __init__(
+        self,
+        message: str,
+        *,
+        code: Optional[int] = None,
+        payload: Optional[object] = None,
+    ) -> None:
+        super().__init__(message)
+        self.code = code
+        self.payload = payload
+
+
+class RealtimeTranscriptionWSError(RealtimeTranscriptionException):
+    def __init__(
+        self,
+        message: str,
+        *,
+        payload: Optional["RealtimeTranscriptionError"] = None,
+        raw: Optional[object] = None,
+    ) -> None:
+        code: Optional[int] = None
+        if payload is not None:
+            try:
+                maybe_code = getattr(payload.error, "code", None)
+                if isinstance(maybe_code, int):
+                    code = maybe_code
+            except Exception:
+                code = None
+
+        super().__init__(
+            message, code=code, payload=payload if payload is not None else raw
+        )
+        self.payload_typed = payload
+        self.payload_raw = raw

mistralai-1.12.0/src/mistralai/extra/realtime/__init__.py

@@ -0,0 +1,25 @@
+from mistralai.models import (
+    AudioEncoding,
+    AudioFormat,
+    RealtimeTranscriptionError,
+    RealtimeTranscriptionErrorDetail,
+    RealtimeTranscriptionSession,
+    RealtimeTranscriptionSessionCreated,
+    RealtimeTranscriptionSessionUpdated,
+)
+
+from .connection import UnknownRealtimeEvent, RealtimeConnection
+from .transcription import RealtimeTranscription
+
+__all__ = [
+    "AudioEncoding",
+    "AudioFormat",
+    "RealtimeTranscriptionError",
+    "RealtimeTranscriptionErrorDetail",
+    "RealtimeTranscriptionSession",
+    "RealtimeTranscriptionSessionCreated",
+    "RealtimeTranscriptionSessionUpdated",
+    "RealtimeConnection",
+    "RealtimeTranscription",
+    "UnknownRealtimeEvent",
+]

mistralai-1.12.0/src/mistralai/extra/realtime/connection.py

@@ -0,0 +1,207 @@
+from __future__ import annotations
+
+import base64
+import json
+from asyncio import CancelledError
+from collections import deque
+from typing import Any, AsyncIterator, Deque, Optional, Union
+
+from pydantic import ValidationError, BaseModel
+
+try:
+    from websockets.asyncio.client import ClientConnection  # websockets >= 13.0
+except ImportError as exc:
+    raise ImportError(
+        "The `websockets` package (>=13.0) is required for real-time transcription. "
+        "Install with: pip install 'mistralai[realtime]'"
+    ) from exc
+
+from mistralai.models import (
+    AudioFormat,
+    RealtimeTranscriptionError,
+    RealtimeTranscriptionSession,
+    RealtimeTranscriptionSessionCreated,
+    RealtimeTranscriptionSessionUpdated,
+    TranscriptionStreamDone,
+    TranscriptionStreamLanguage,
+    TranscriptionStreamSegmentDelta,
+    TranscriptionStreamTextDelta,
+)
+
+
+class UnknownRealtimeEvent(BaseModel):
+    """
+    Forward-compat fallback event:
+    - unknown message type
+    - invalid JSON payload
+    - schema validation failure
+    """
+    type: Optional[str]
+    content: Any
+    error: Optional[str] = None
+
+
+RealtimeEvent = Union[
+    # session lifecycle
+    RealtimeTranscriptionSessionCreated,
+    RealtimeTranscriptionSessionUpdated,
+    # server errors
+    RealtimeTranscriptionError,
+    # transcription events
+    TranscriptionStreamLanguage,
+    TranscriptionStreamSegmentDelta,
+    TranscriptionStreamTextDelta,
+    TranscriptionStreamDone,
+    # forward-compat fallback
+    UnknownRealtimeEvent,
+]
+
+
+_MESSAGE_MODELS: dict[str, Any] = {
+    "session.created": RealtimeTranscriptionSessionCreated,
+    "session.updated": RealtimeTranscriptionSessionUpdated,
+    "error": RealtimeTranscriptionError,
+    "transcription.language": TranscriptionStreamLanguage,
+    "transcription.segment": TranscriptionStreamSegmentDelta,
+    "transcription.text.delta": TranscriptionStreamTextDelta,
+    "transcription.done": TranscriptionStreamDone,
+}
+
+
+def parse_realtime_event(payload: Any) -> RealtimeEvent:
+    """
+    Tolerant parser:
+    - unknown event type -> UnknownRealtimeEvent
+    - validation failures -> UnknownRealtimeEvent (includes error string)
+    - invalid payload -> UnknownRealtimeEvent
+    """
+    if not isinstance(payload, dict):
+        return UnknownRealtimeEvent(
+            type=None, content=payload, error="expected JSON object"
+        )
+
+    msg_type = payload.get("type")
+    if not isinstance(msg_type, str):
+        return UnknownRealtimeEvent(
+            type=None, content=payload, error="missing/invalid 'type'"
+        )
+
+    model_cls = _MESSAGE_MODELS.get(msg_type)
+    if model_cls is None:
+        return UnknownRealtimeEvent(
+            type=msg_type, content=payload, error="unknown event type"
+        )
+    try:
+        parsed = model_cls.model_validate(payload)
+        return parsed
+    except ValidationError as exc:
+        return UnknownRealtimeEvent(type=msg_type, content=payload, error=str(exc))
+
+
+class RealtimeConnection:
+    def __init__(
+        self,
+        websocket: ClientConnection,
+        session: RealtimeTranscriptionSession,
+        *,
+        initial_events: Optional[list[RealtimeEvent]] = None,
+    ) -> None:
+        self._websocket = websocket
+        self._session = session
+        self._audio_format = session.audio_format
+        self._closed = False
+        self._initial_events: Deque[RealtimeEvent] = deque(initial_events or [])
+
+    @property
+    def request_id(self) -> str:
+        return self._session.request_id
+
+    @property
+    def session(self) -> RealtimeTranscriptionSession:
+        return self._session
+
+    @property
+    def audio_format(self) -> AudioFormat:
+        return self._audio_format
+
+    @property
+    def is_closed(self) -> bool:
+        return self._closed
+
+    async def send_audio(
+        self, audio_bytes: Union[bytes, bytearray, memoryview]
+    ) -> None:
+        if self._closed:
+            raise RuntimeError("Connection is closed")
+
+        message = {
+            "type": "input_audio.append",
+            "audio": base64.b64encode(bytes(audio_bytes)).decode("ascii"),
+        }
+        await self._websocket.send(json.dumps(message))
+
+    async def update_session(self, audio_format: AudioFormat) -> None:
+        if self._closed:
+            raise RuntimeError("Connection is closed")
+
+        self._audio_format = audio_format
+        message = {
+            "type": "session.update",
+            "session": {"audio_format": audio_format.model_dump(mode="json")},
+        }
+        await self._websocket.send(json.dumps(message))
+
+    async def end_audio(self) -> None:
+        if self._closed:
+            return
+        await self._websocket.send(json.dumps({"type": "input_audio.end"}))
+
+    async def close(self, *, code: int = 1000, reason: str = "") -> None:
+        if self._closed:
+            return
+        self._closed = True
+        await self._websocket.close(code=code, reason=reason)
+
+    async def __aenter__(self) -> "RealtimeConnection":
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb) -> None:
+        await self.close()
+
+    def __aiter__(self) -> AsyncIterator[RealtimeEvent]:
+        return self.events()
+
+    async def events(self) -> AsyncIterator[RealtimeEvent]:
+        # replay any handshake/prelude events (including session.created)
+        while self._initial_events:
+            ev = self._initial_events.popleft()
+            self._apply_session_updates(ev)
+            yield ev
+
+        try:
+            async for msg in self._websocket:
+                text = (
+                    msg.decode("utf-8", errors="replace")
+                    if isinstance(msg, (bytes, bytearray))
+                    else msg
+                )
+                try:
+                    data = json.loads(text)
+                except Exception as exc:
+                    yield UnknownRealtimeEvent(
+                        type=None, content=text, error=f"invalid JSON: {exc}"
+                    )
+                    continue
+
+                ev = parse_realtime_event(data)
+                self._apply_session_updates(ev)
+                yield ev
+        except CancelledError:
+            pass
+        finally:
+            await self.close()
+
+    def _apply_session_updates(self, ev: RealtimeEvent) -> None:
+        if isinstance(ev, RealtimeTranscriptionSessionCreated) or isinstance(ev, RealtimeTranscriptionSessionUpdated):
+            self._session = ev.session
+            self._audio_format = ev.session.audio_format