mistralai 1.9.1__py3-none-any.whl → 1.9.3__py3-none-any.whl

mistralai/models/transcriptionresponse.py

@@ -0,0 +1,79 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .transcriptionsegmentchunk import (
+    TranscriptionSegmentChunk,
+    TranscriptionSegmentChunkTypedDict,
+)
+from .usageinfo import UsageInfo, UsageInfoTypedDict
+from mistralai.types import BaseModel, Nullable, UNSET_SENTINEL
+import pydantic
+from pydantic import ConfigDict, model_serializer
+from typing import Any, Dict, List, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+class TranscriptionResponseTypedDict(TypedDict):
+    model: str
+    text: str
+    usage: UsageInfoTypedDict
+    language: Nullable[str]
+    segments: NotRequired[List[TranscriptionSegmentChunkTypedDict]]
+
+
+class TranscriptionResponse(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True, arbitrary_types_allowed=True, extra="allow"
+    )
+    __pydantic_extra__: Dict[str, Any] = pydantic.Field(init=False)
+
+    model: str
+
+    text: str
+
+    usage: UsageInfo
+
+    language: Nullable[str]
+
+    segments: Optional[List[TranscriptionSegmentChunk]] = None
+
+    @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 = ["segments"]
+        nullable_fields = ["language"]
+        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/models/transcriptionsegmentchunk.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
+
+
+Type = Literal["transcription_segment"]
+
+
+class TranscriptionSegmentChunkTypedDict(TypedDict):
+    text: str
+    start: float
+    end: float
+    type: NotRequired[Type]
+
+
+class TranscriptionSegmentChunk(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[Type] = "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/transcriptionstreamdone.py

@@ -0,0 +1,85 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .transcriptionsegmentchunk import (
+    TranscriptionSegmentChunk,
+    TranscriptionSegmentChunkTypedDict,
+)
+from .usageinfo import UsageInfo, UsageInfoTypedDict
+from mistralai.types import BaseModel, Nullable, UNSET_SENTINEL
+import pydantic
+from pydantic import ConfigDict, model_serializer
+from typing import Any, Dict, List, Literal, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+TranscriptionStreamDoneType = Literal["transcription.done"]
+
+
+class TranscriptionStreamDoneTypedDict(TypedDict):
+    model: str
+    text: str
+    usage: UsageInfoTypedDict
+    language: Nullable[str]
+    segments: NotRequired[List[TranscriptionSegmentChunkTypedDict]]
+    type: NotRequired[TranscriptionStreamDoneType]
+
+
+class TranscriptionStreamDone(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True, arbitrary_types_allowed=True, extra="allow"
+    )
+    __pydantic_extra__: Dict[str, Any] = pydantic.Field(init=False)
+
+    model: str
+
+    text: str
+
+    usage: UsageInfo
+
+    language: Nullable[str]
+
+    segments: Optional[List[TranscriptionSegmentChunk]] = None
+
+    type: Optional[TranscriptionStreamDoneType] = "transcription.done"
+
+    @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 = ["segments", "type"]
+        nullable_fields = ["language"]
+        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/models/transcriptionstreamevents.py

@@ -0,0 +1,58 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .transcriptionstreamdone import (
+    TranscriptionStreamDone,
+    TranscriptionStreamDoneTypedDict,
+)
+from .transcriptionstreameventtypes import TranscriptionStreamEventTypes
+from .transcriptionstreamlanguage import (
+    TranscriptionStreamLanguage,
+    TranscriptionStreamLanguageTypedDict,
+)
+from .transcriptionstreamsegmentdelta import (
+    TranscriptionStreamSegmentDelta,
+    TranscriptionStreamSegmentDeltaTypedDict,
+)
+from .transcriptionstreamtextdelta import (
+    TranscriptionStreamTextDelta,
+    TranscriptionStreamTextDeltaTypedDict,
+)
+from mistralai.types import BaseModel
+from mistralai.utils import get_discriminator
+from pydantic import Discriminator, Tag
+from typing import Union
+from typing_extensions import Annotated, TypeAliasType, TypedDict
+
+
+TranscriptionStreamEventsDataTypedDict = TypeAliasType(
+    "TranscriptionStreamEventsDataTypedDict",
+    Union[
+        TranscriptionStreamTextDeltaTypedDict,
+        TranscriptionStreamLanguageTypedDict,
+        TranscriptionStreamSegmentDeltaTypedDict,
+        TranscriptionStreamDoneTypedDict,
+    ],
+)
+
+
+TranscriptionStreamEventsData = Annotated[
+    Union[
+        Annotated[TranscriptionStreamDone, Tag("transcription.done")],
+        Annotated[TranscriptionStreamLanguage, Tag("transcription.language")],
+        Annotated[TranscriptionStreamSegmentDelta, Tag("transcription.segment")],
+        Annotated[TranscriptionStreamTextDelta, Tag("transcription.text.delta")],
+    ],
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]
+
+
+class TranscriptionStreamEventsTypedDict(TypedDict):
+    event: TranscriptionStreamEventTypes
+    data: TranscriptionStreamEventsDataTypedDict
+
+
+class TranscriptionStreamEvents(BaseModel):
+    event: TranscriptionStreamEventTypes
+
+    data: TranscriptionStreamEventsData

mistralai/models/transcriptionstreameventtypes.py

@@ -0,0 +1,12 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from typing import Literal
+
+
+TranscriptionStreamEventTypes = Literal[
+    "transcription.language",
+    "transcription.segment",
+    "transcription.text.delta",
+    "transcription.done",
+]

mistralai/models/transcriptionstreamlanguage.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
+
+
+TranscriptionStreamLanguageType = Literal["transcription.language"]
+
+
+class TranscriptionStreamLanguageTypedDict(TypedDict):
+    audio_language: str
+    type: NotRequired[TranscriptionStreamLanguageType]
+
+
+class TranscriptionStreamLanguage(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True, arbitrary_types_allowed=True, extra="allow"
+    )
+    __pydantic_extra__: Dict[str, Any] = pydantic.Field(init=False)
+
+    audio_language: str
+
+    type: Optional[TranscriptionStreamLanguageType] = "transcription.language"
+
+    @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/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__(