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

mistralai/models/batchjobout.py

@@ -16,7 +16,6 @@ class BatchJobOutTypedDict(TypedDict):
     id: str
     input_files: List[str]
     endpoint: str
-    model: str
     errors: List[BatchErrorTypedDict]
     status: BatchJobStatus
     created_at: int
@@ -26,6 +25,8 @@ class BatchJobOutTypedDict(TypedDict):
     failed_requests: int
     object: NotRequired[BatchJobOutObject]
     metadata: NotRequired[Nullable[Dict[str, Any]]]
+    model: NotRequired[Nullable[str]]
+    agent_id: NotRequired[Nullable[str]]
     output_file: NotRequired[Nullable[str]]
     error_file: NotRequired[Nullable[str]]
     started_at: NotRequired[Nullable[int]]
@@ -39,8 +40,6 @@ class BatchJobOut(BaseModel):
 
     endpoint: str
 
-    model: str
-
     errors: List[BatchError]
 
     status: BatchJobStatus
@@ -59,6 +58,10 @@ class BatchJobOut(BaseModel):
 
     metadata: OptionalNullable[Dict[str, Any]] = UNSET
 
+    model: OptionalNullable[str] = UNSET
+
+    agent_id: OptionalNullable[str] = UNSET
+
     output_file: OptionalNullable[str] = UNSET
 
     error_file: OptionalNullable[str] = UNSET
@@ -72,6 +75,8 @@ class BatchJobOut(BaseModel):
         optional_fields = [
             "object",
             "metadata",
+            "model",
+            "agent_id",
             "output_file",
             "error_file",
             "started_at",
@@ -79,6 +84,8 @@ class BatchJobOut(BaseModel):
         ]
         nullable_fields = [
             "metadata",
+            "model",
+            "agent_id",
             "output_file",
             "error_file",
             "started_at",

mistralai/models/classifierftmodelout.py

@@ -21,7 +21,9 @@ class ClassifierFTModelOutTypedDict(TypedDict):
     id: str
     created: int
     owned_by: str
+    workspace_id: str
     root: str
+    root_version: str
     archived: bool
     capabilities: FTModelCapabilitiesOutTypedDict
     job: str
@@ -41,8 +43,12 @@ class ClassifierFTModelOut(BaseModel):
 
     owned_by: str
 
+    workspace_id: str
+
     root: str
 
+    root_version: str
+
     archived: bool
 
     capabilities: FTModelCapabilitiesOut

mistralai/models/completionargs.py

@@ -17,7 +17,7 @@ class CompletionArgsTypedDict(TypedDict):
     stop: NotRequired[Nullable[CompletionArgsStopTypedDict]]
     presence_penalty: NotRequired[Nullable[float]]
     frequency_penalty: NotRequired[Nullable[float]]
-    temperature: NotRequired[float]
+    temperature: NotRequired[Nullable[float]]
     top_p: NotRequired[Nullable[float]]
     max_tokens: NotRequired[Nullable[int]]
     random_seed: NotRequired[Nullable[int]]
@@ -35,7 +35,7 @@ class CompletionArgs(BaseModel):
 
     frequency_penalty: OptionalNullable[float] = UNSET
 
-    temperature: Optional[float] = 0.3
+    temperature: OptionalNullable[float] = UNSET
 
     top_p: OptionalNullable[float] = UNSET
 
@@ -67,6 +67,7 @@ class CompletionArgs(BaseModel):
             "stop",
             "presence_penalty",
             "frequency_penalty",
+            "temperature",
             "top_p",
             "max_tokens",
             "random_seed",

mistralai/models/completionftmodelout.py

@@ -20,7 +20,9 @@ class CompletionFTModelOutTypedDict(TypedDict):
     id: str
     created: int
     owned_by: str
+    workspace_id: str
     root: str
+    root_version: str
     archived: bool
     capabilities: FTModelCapabilitiesOutTypedDict
     job: str
@@ -39,8 +41,12 @@ class CompletionFTModelOut(BaseModel):
 
     owned_by: str
 
+    workspace_id: str
+
     root: str
 
+    root_version: str
+
     archived: bool
 
     capabilities: FTModelCapabilitiesOut

mistralai/models/contentchunk.py

@@ -1,11 +1,13 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .audiochunk import AudioChunk, AudioChunkTypedDict
 from .documenturlchunk import DocumentURLChunk, DocumentURLChunkTypedDict
 from .filechunk import FileChunk, FileChunkTypedDict
 from .imageurlchunk import ImageURLChunk, ImageURLChunkTypedDict
 from .referencechunk import ReferenceChunk, ReferenceChunkTypedDict
 from .textchunk import TextChunk, TextChunkTypedDict
+from .thinkchunk import ThinkChunk, ThinkChunkTypedDict
 from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag
 from typing import Union
@@ -19,7 +21,9 @@ ContentChunkTypedDict = TypeAliasType(
         ImageURLChunkTypedDict,
         ReferenceChunkTypedDict,
         FileChunkTypedDict,
+        AudioChunkTypedDict,
         DocumentURLChunkTypedDict,
+        ThinkChunkTypedDict,
     ],
 )
 
@@ -31,6 +35,8 @@ ContentChunk = Annotated[
         Annotated[TextChunk, Tag("text")],
         Annotated[ReferenceChunk, Tag("reference")],
         Annotated[FileChunk, Tag("file")],
+        Annotated[ThinkChunk, Tag("thinking")],
+        Annotated[AudioChunk, Tag("input_audio")],
     ],
     Discriminator(lambda m: get_discriminator(m, "type", "type")),
 ]

mistralai/models/fileschema.py

@@ -27,6 +27,8 @@ class FileSchemaTypedDict(TypedDict):
     sample_type: SampleType
     source: Source
     num_lines: NotRequired[Nullable[int]]
+    mimetype: NotRequired[Nullable[str]]
+    signature: NotRequired[Nullable[str]]
 
 
 class FileSchema(BaseModel):
@@ -53,10 +55,14 @@ class FileSchema(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/jobs_api_routes_batch_get_batch_jobsop.py

@@ -14,6 +14,7 @@ class JobsAPIRoutesBatchGetBatchJobsRequestTypedDict(TypedDict):
     page: NotRequired[int]
     page_size: NotRequired[int]
     model: NotRequired[Nullable[str]]
+    agent_id: NotRequired[Nullable[str]]
     metadata: NotRequired[Nullable[Dict[str, Any]]]
     created_after: NotRequired[Nullable[datetime]]
     created_by_me: NotRequired[bool]
@@ -36,6 +37,11 @@ class JobsAPIRoutesBatchGetBatchJobsRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = UNSET
 
+    agent_id: Annotated[
+        OptionalNullable[str],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = UNSET
+
     metadata: Annotated[
         OptionalNullable[Dict[str, Any]],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
@@ -62,12 +68,13 @@ class JobsAPIRoutesBatchGetBatchJobsRequest(BaseModel):
             "page",
             "page_size",
             "model",
+            "agent_id",
             "metadata",
             "created_after",
             "created_by_me",
             "status",
         ]
-        nullable_fields = ["model", "metadata", "created_after", "status"]
+        nullable_fields = ["model", "agent_id", "metadata", "created_after", "status"]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/messageoutputcontentchunks.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 from .documenturlchunk import DocumentURLChunk, DocumentURLChunkTypedDict
 from .imageurlchunk import ImageURLChunk, ImageURLChunkTypedDict
 from .textchunk import TextChunk, TextChunkTypedDict
+from .thinkchunk import ThinkChunk, ThinkChunkTypedDict
 from .toolfilechunk import ToolFileChunk, ToolFileChunkTypedDict
 from .toolreferencechunk import ToolReferenceChunk, ToolReferenceChunkTypedDict
 from typing import Union
@@ -16,6 +17,7 @@ MessageOutputContentChunksTypedDict = TypeAliasType(
         TextChunkTypedDict,
         ImageURLChunkTypedDict,
         DocumentURLChunkTypedDict,
+        ThinkChunkTypedDict,
         ToolFileChunkTypedDict,
         ToolReferenceChunkTypedDict,
     ],
@@ -25,6 +27,11 @@ MessageOutputContentChunksTypedDict = TypeAliasType(
 MessageOutputContentChunks = TypeAliasType(
     "MessageOutputContentChunks",
     Union[
-        TextChunk, ImageURLChunk, DocumentURLChunk, ToolFileChunk, ToolReferenceChunk
+        TextChunk,
+        ImageURLChunk,
+        DocumentURLChunk,
+        ThinkChunk,
+        ToolFileChunk,
+        ToolReferenceChunk,
     ],
 )

mistralai/models/outputcontentchunks.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 from .documenturlchunk import DocumentURLChunk, DocumentURLChunkTypedDict
 from .imageurlchunk import ImageURLChunk, ImageURLChunkTypedDict
 from .textchunk import TextChunk, TextChunkTypedDict
+from .thinkchunk import ThinkChunk, ThinkChunkTypedDict
 from .toolfilechunk import ToolFileChunk, ToolFileChunkTypedDict
 from .toolreferencechunk import ToolReferenceChunk, ToolReferenceChunkTypedDict
 from typing import Union
@@ -16,6 +17,7 @@ OutputContentChunksTypedDict = TypeAliasType(
         TextChunkTypedDict,
         ImageURLChunkTypedDict,
         DocumentURLChunkTypedDict,
+        ThinkChunkTypedDict,
         ToolFileChunkTypedDict,
         ToolReferenceChunkTypedDict,
     ],
@@ -25,6 +27,11 @@ OutputContentChunksTypedDict = TypeAliasType(
 OutputContentChunks = TypeAliasType(
     "OutputContentChunks",
     Union[
-        TextChunk, ImageURLChunk, DocumentURLChunk, ToolFileChunk, ToolReferenceChunk
+        TextChunk,
+        ImageURLChunk,
+        DocumentURLChunk,
+        ThinkChunk,
+        ToolFileChunk,
+        ToolReferenceChunk,
     ],
 )

mistralai/models/retrievefileout.py

@@ -28,6 +28,8 @@ class RetrieveFileOutTypedDict(TypedDict):
     source: Source
     deleted: bool
     num_lines: NotRequired[Nullable[int]]
+    mimetype: NotRequired[Nullable[str]]
+    signature: NotRequired[Nullable[str]]
 
 
 class RetrieveFileOut(BaseModel):
@@ -56,10 +58,14 @@ class RetrieveFileOut(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/thinkchunk.py

@@ -0,0 +1,35 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .referencechunk import ReferenceChunk, ReferenceChunkTypedDict
+from .textchunk import TextChunk, TextChunkTypedDict
+from mistralai.types import BaseModel
+from typing import List, Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
+
+
+ThinkingTypedDict = TypeAliasType(
+    "ThinkingTypedDict", Union[ReferenceChunkTypedDict, TextChunkTypedDict]
+)
+
+
+Thinking = TypeAliasType("Thinking", Union[ReferenceChunk, TextChunk])
+
+
+ThinkChunkType = Literal["thinking"]
+
+
+class ThinkChunkTypedDict(TypedDict):
+    thinking: List[ThinkingTypedDict]
+    closed: NotRequired[bool]
+    r"""Whether the thinking chunk is closed or not. Currently only used for prefixing."""
+    type: NotRequired[ThinkChunkType]
+
+
+class ThinkChunk(BaseModel):
+    thinking: List[Thinking]
+
+    closed: Optional[bool] = None
+    r"""Whether the thinking chunk is closed or not. Currently only used for prefixing."""
+
+    type: Optional[ThinkChunkType] = "thinking"

mistralai/models/timestampgranularity.py

@@ -0,0 +1,7 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from typing import Literal
+
+
+TimestampGranularity = Literal["segment"]

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]