mistralai 1.9.2__py3-none-any.whl → 1.9.6__py3-none-any.whl

mistralai/models/transcriptionstreamsegmentdelta.py

@@ -0,0 +1,41 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+import pydantic
+from pydantic import ConfigDict
+from typing import Any, Dict, Literal, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+TranscriptionStreamSegmentDeltaType = Literal["transcription.segment"]
+
+
+class TranscriptionStreamSegmentDeltaTypedDict(TypedDict):
+    text: str
+    start: float
+    end: float
+    type: NotRequired[TranscriptionStreamSegmentDeltaType]
+
+
+class TranscriptionStreamSegmentDelta(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True, arbitrary_types_allowed=True, extra="allow"
+    )
+    __pydantic_extra__: Dict[str, Any] = pydantic.Field(init=False)
+
+    text: str
+
+    start: float
+
+    end: float
+
+    type: Optional[TranscriptionStreamSegmentDeltaType] = "transcription.segment"
+
+    @property
+    def additional_properties(self):
+        return self.__pydantic_extra__
+
+    @additional_properties.setter
+    def additional_properties(self, value):
+        self.__pydantic_extra__ = value  # pyright: ignore[reportIncompatibleVariableOverride]

mistralai/models/transcriptionstreamtextdelta.py

@@ -0,0 +1,35 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+import pydantic
+from pydantic import ConfigDict
+from typing import Any, Dict, Literal, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+TranscriptionStreamTextDeltaType = Literal["transcription.text.delta"]
+
+
+class TranscriptionStreamTextDeltaTypedDict(TypedDict):
+    text: str
+    type: NotRequired[TranscriptionStreamTextDeltaType]
+
+
+class TranscriptionStreamTextDelta(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True, arbitrary_types_allowed=True, extra="allow"
+    )
+    __pydantic_extra__: Dict[str, Any] = pydantic.Field(init=False)
+
+    text: str
+
+    type: Optional[TranscriptionStreamTextDeltaType] = "transcription.text.delta"
+
+    @property
+    def additional_properties(self):
+        return self.__pydantic_extra__
+
+    @additional_properties.setter
+    def additional_properties(self, value):
+        self.__pydantic_extra__ = value  # pyright: ignore[reportIncompatibleVariableOverride]

mistralai/models/uploadfileout.py

@@ -27,6 +27,8 @@ class UploadFileOutTypedDict(TypedDict):
     sample_type: SampleType
     source: Source
     num_lines: NotRequired[Nullable[int]]
+    mimetype: NotRequired[Nullable[str]]
+    signature: NotRequired[Nullable[str]]
 
 
 class UploadFileOut(BaseModel):
@@ -53,10 +55,14 @@ class UploadFileOut(BaseModel):
 
     num_lines: OptionalNullable[int] = UNSET
 
+    mimetype: OptionalNullable[str] = UNSET
+
+    signature: OptionalNullable[str] = UNSET
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["num_lines"]
-        nullable_fields = ["num_lines"]
+        optional_fields = ["num_lines", "mimetype", "signature"]
+        nullable_fields = ["num_lines", "mimetype", "signature"]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/usageinfo.py

@@ -1,19 +1,76 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from mistralai.types import BaseModel
-from typing_extensions import TypedDict
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+import pydantic
+from pydantic import ConfigDict, model_serializer
+from typing import Any, Dict, Optional
+from typing_extensions import NotRequired, TypedDict
 
 
 class UsageInfoTypedDict(TypedDict):
-    prompt_tokens: int
-    completion_tokens: int
-    total_tokens: int
+    prompt_tokens: NotRequired[int]
+    completion_tokens: NotRequired[int]
+    total_tokens: NotRequired[int]
+    prompt_audio_seconds: NotRequired[Nullable[int]]
 
 
 class UsageInfo(BaseModel):
-    prompt_tokens: int
+    model_config = ConfigDict(
+        populate_by_name=True, arbitrary_types_allowed=True, extra="allow"
+    )
+    __pydantic_extra__: Dict[str, Any] = pydantic.Field(init=False)
 
-    completion_tokens: int
+    prompt_tokens: Optional[int] = 0
 
-    total_tokens: int
+    completion_tokens: Optional[int] = 0
+
+    total_tokens: Optional[int] = 0
+
+    prompt_audio_seconds: OptionalNullable[int] = UNSET
+
+    @property
+    def additional_properties(self):
+        return self.__pydantic_extra__
+
+    @additional_properties.setter
+    def additional_properties(self, value):
+        self.__pydantic_extra__ = value  # pyright: ignore[reportIncompatibleVariableOverride]
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "prompt_tokens",
+            "completion_tokens",
+            "total_tokens",
+            "prompt_audio_seconds",
+        ]
+        nullable_fields = ["prompt_audio_seconds"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+            serialized.pop(k, None)
+
+            optional_nullable = k in optional_fields and k in nullable_fields
+            is_set = (
+                self.__pydantic_fields_set__.intersection({n})
+                or k in null_default_fields
+            )  # pylint: disable=no-member
+
+            if val is not None and val != UNSET_SENTINEL:
+                m[k] = val
+            elif val != UNSET_SENTINEL and (
+                not k in optional_fields or (optional_nullable and is_set)
+            ):
+                m[k] = val
+
+        for k, v in serialized.items():
+            m[k] = v
+
+        return m

mistralai/sdk.py

@@ -15,6 +15,7 @@ import weakref
 
 if TYPE_CHECKING:
     from mistralai.agents import Agents
+    from mistralai.audio import Audio
     from mistralai.batch import Batch
     from mistralai.beta import Beta
     from mistralai.chat import Chat
@@ -49,6 +50,7 @@ class Mistral(BaseSDK):
     r"""Classifiers API."""
     ocr: "Ocr"
     r"""OCR API"""
+    audio: "Audio"
     _sub_sdk_map = {
         "models": ("mistralai.models_", "Models"),
         "beta": ("mistralai.beta", "Beta"),
@@ -61,6 +63,7 @@ class Mistral(BaseSDK):
         "embeddings": ("mistralai.embeddings", "Embeddings"),
         "classifiers": ("mistralai.classifiers", "Classifiers"),
         "ocr": ("mistralai.ocr", "Ocr"),
+        "audio": ("mistralai.audio", "Audio"),
     }
 
     def __init__(

mistralai/transcriptions.py

@@ -0,0 +1,480 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from .basesdk import BaseSDK
+from mistralai import models, utils
+from mistralai._hooks import HookContext
+from mistralai.types import OptionalNullable, UNSET
+from mistralai.utils import eventstreaming, get_security_from_env
+from typing import List, Mapping, Optional, Union
+
+
+class Transcriptions(BaseSDK):
+    r"""API for audio transcription."""
+
+    def complete(
+        self,
+        *,
+        model: str,
+        file: Optional[Union[models.File, models.FileTypedDict]] = None,
+        file_url: OptionalNullable[str] = UNSET,
+        file_id: OptionalNullable[str] = UNSET,
+        language: OptionalNullable[str] = UNSET,
+        temperature: OptionalNullable[float] = UNSET,
+        timestamp_granularities: Optional[List[models.TimestampGranularity]] = None,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> models.TranscriptionResponse:
+        r"""Create Transcription
+
+        :param model:
+        :param file:
+        :param file_url: Url of a file to be transcribed
+        :param file_id: ID of a file uploaded to /v1/files
+        :param language: Language of the audio, e.g. 'en'. Providing the language can boost accuracy.
+        :param temperature:
+        :param timestamp_granularities: Granularities of timestamps to include in the response.
+        :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
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
+
+        request = models.AudioTranscriptionRequest(
+            model=model,
+            file=utils.get_pydantic_model(file, Optional[models.File]),
+            file_url=file_url,
+            file_id=file_id,
+            language=language,
+            temperature=temperature,
+            timestamp_granularities=timestamp_granularities,
+        )
+
+        req = self._build_request(
+            method="POST",
+            path="/v1/audio/transcriptions",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=True,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request, False, False, "multipart", models.AudioTranscriptionRequest
+            ),
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = self.do_request(
+            hook_ctx=HookContext(
+                config=self.sdk_configuration,
+                base_url=base_url or "",
+                operation_id="audio_api_v1_transcriptions_post",
+                oauth2_scopes=[],
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        if utils.match_response(http_res, "200", "application/json"):
+            return utils.unmarshal_json(http_res.text, models.TranscriptionResponse)
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+
+        content_type = http_res.headers.get("Content-Type")
+        http_res_text = utils.stream_to_text(http_res)
+        raise models.SDKError(
+            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
+            http_res.status_code,
+            http_res_text,
+            http_res,
+        )
+
+    async def complete_async(
+        self,
+        *,
+        model: str,
+        file: Optional[Union[models.File, models.FileTypedDict]] = None,
+        file_url: OptionalNullable[str] = UNSET,
+        file_id: OptionalNullable[str] = UNSET,
+        language: OptionalNullable[str] = UNSET,
+        temperature: OptionalNullable[float] = UNSET,
+        timestamp_granularities: Optional[List[models.TimestampGranularity]] = None,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> models.TranscriptionResponse:
+        r"""Create Transcription
+
+        :param model:
+        :param file:
+        :param file_url: Url of a file to be transcribed
+        :param file_id: ID of a file uploaded to /v1/files
+        :param language: Language of the audio, e.g. 'en'. Providing the language can boost accuracy.
+        :param temperature:
+        :param timestamp_granularities: Granularities of timestamps to include in the response.
+        :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
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
+
+        request = models.AudioTranscriptionRequest(
+            model=model,
+            file=utils.get_pydantic_model(file, Optional[models.File]),
+            file_url=file_url,
+            file_id=file_id,
+            language=language,
+            temperature=temperature,
+            timestamp_granularities=timestamp_granularities,
+        )
+
+        req = self._build_request_async(
+            method="POST",
+            path="/v1/audio/transcriptions",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=True,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request, False, False, "multipart", models.AudioTranscriptionRequest
+            ),
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = await self.do_request_async(
+            hook_ctx=HookContext(
+                config=self.sdk_configuration,
+                base_url=base_url or "",
+                operation_id="audio_api_v1_transcriptions_post",
+                oauth2_scopes=[],
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        if utils.match_response(http_res, "200", "application/json"):
+            return utils.unmarshal_json(http_res.text, models.TranscriptionResponse)
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+
+        content_type = http_res.headers.get("Content-Type")
+        http_res_text = await utils.stream_to_text_async(http_res)
+        raise models.SDKError(
+            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
+            http_res.status_code,
+            http_res_text,
+            http_res,
+        )
+
+    def stream(
+        self,
+        *,
+        model: str,
+        file: Optional[Union[models.File, models.FileTypedDict]] = None,
+        file_url: OptionalNullable[str] = UNSET,
+        file_id: OptionalNullable[str] = UNSET,
+        language: OptionalNullable[str] = UNSET,
+        temperature: OptionalNullable[float] = UNSET,
+        timestamp_granularities: Optional[List[models.TimestampGranularity]] = None,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> eventstreaming.EventStream[models.TranscriptionStreamEvents]:
+        r"""Create streaming transcription (SSE)
+
+        :param model:
+        :param file:
+        :param file_url: Url of a file to be transcribed
+        :param file_id: ID of a file uploaded to /v1/files
+        :param language: Language of the audio, e.g. 'en'. Providing the language can boost accuracy.
+        :param temperature:
+        :param timestamp_granularities: Granularities of timestamps to include in the response.
+        :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
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
+
+        request = models.AudioTranscriptionRequestStream(
+            model=model,
+            file=utils.get_pydantic_model(file, Optional[models.File]),
+            file_url=file_url,
+            file_id=file_id,
+            language=language,
+            temperature=temperature,
+            timestamp_granularities=timestamp_granularities,
+        )
+
+        req = self._build_request(
+            method="POST",
+            path="/v1/audio/transcriptions#stream",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=True,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="text/event-stream",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request,
+                False,
+                False,
+                "multipart",
+                models.AudioTranscriptionRequestStream,
+            ),
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = self.do_request(
+            hook_ctx=HookContext(
+                config=self.sdk_configuration,
+                base_url=base_url or "",
+                operation_id="audio_api_v1_transcriptions_post_stream",
+                oauth2_scopes=[],
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["4XX", "5XX"],
+            stream=True,
+            retry_config=retry_config,
+        )
+
+        if utils.match_response(http_res, "200", "text/event-stream"):
+            return eventstreaming.EventStream(
+                http_res,
+                lambda raw: utils.unmarshal_json(raw, models.TranscriptionStreamEvents),
+            )
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+
+        content_type = http_res.headers.get("Content-Type")
+        http_res_text = utils.stream_to_text(http_res)
+        raise models.SDKError(
+            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
+            http_res.status_code,
+            http_res_text,
+            http_res,
+        )
+
+    async def stream_async(
+        self,
+        *,
+        model: str,
+        file: Optional[Union[models.File, models.FileTypedDict]] = None,
+        file_url: OptionalNullable[str] = UNSET,
+        file_id: OptionalNullable[str] = UNSET,
+        language: OptionalNullable[str] = UNSET,
+        temperature: OptionalNullable[float] = UNSET,
+        timestamp_granularities: Optional[List[models.TimestampGranularity]] = None,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> eventstreaming.EventStreamAsync[models.TranscriptionStreamEvents]:
+        r"""Create streaming transcription (SSE)
+
+        :param model:
+        :param file:
+        :param file_url: Url of a file to be transcribed
+        :param file_id: ID of a file uploaded to /v1/files
+        :param language: Language of the audio, e.g. 'en'. Providing the language can boost accuracy.
+        :param temperature:
+        :param timestamp_granularities: Granularities of timestamps to include in the response.
+        :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
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
+
+        request = models.AudioTranscriptionRequestStream(
+            model=model,
+            file=utils.get_pydantic_model(file, Optional[models.File]),
+            file_url=file_url,
+            file_id=file_id,
+            language=language,
+            temperature=temperature,
+            timestamp_granularities=timestamp_granularities,
+        )
+
+        req = self._build_request_async(
+            method="POST",
+            path="/v1/audio/transcriptions#stream",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=True,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="text/event-stream",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request,
+                False,
+                False,
+                "multipart",
+                models.AudioTranscriptionRequestStream,
+            ),
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = await self.do_request_async(
+            hook_ctx=HookContext(
+                config=self.sdk_configuration,
+                base_url=base_url or "",
+                operation_id="audio_api_v1_transcriptions_post_stream",
+                oauth2_scopes=[],
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["4XX", "5XX"],
+            stream=True,
+            retry_config=retry_config,
+        )
+
+        if utils.match_response(http_res, "200", "text/event-stream"):
+            return eventstreaming.EventStreamAsync(
+                http_res,
+                lambda raw: utils.unmarshal_json(raw, models.TranscriptionStreamEvents),
+            )
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+
+        content_type = http_res.headers.get("Content-Type")
+        http_res_text = await utils.stream_to_text_async(http_res)
+        raise models.SDKError(
+            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
+            http_res.status_code,
+            http_res_text,
+            http_res,
+        )