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

mistralai/models/audiotranscriptionrequest.py

@@ -0,0 +1,97 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .file import File, FileTypedDict
+from .timestampgranularity import TimestampGranularity
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from mistralai.utils import FieldMetadata, MultipartFormMetadata, validate_const
+import pydantic
+from pydantic import model_serializer
+from pydantic.functional_validators import AfterValidator
+from typing import List, Literal, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class AudioTranscriptionRequestTypedDict(TypedDict):
+    model: str
+    file: NotRequired[FileTypedDict]
+    file_url: NotRequired[Nullable[str]]
+    r"""Url of a file to be transcribed"""
+    file_id: NotRequired[Nullable[str]]
+    r"""ID of a file uploaded to /v1/files"""
+    language: NotRequired[Nullable[str]]
+    r"""Language of the audio, e.g. 'en'. Providing the language can boost accuracy."""
+    temperature: NotRequired[Nullable[float]]
+    stream: Literal[False]
+    timestamp_granularities: NotRequired[List[TimestampGranularity]]
+    r"""Granularities of timestamps to include in the response."""
+
+
+class AudioTranscriptionRequest(BaseModel):
+    model: Annotated[str, FieldMetadata(multipart=True)]
+
+    file: Annotated[
+        Optional[File], FieldMetadata(multipart=MultipartFormMetadata(file=True))
+    ] = None
+
+    file_url: Annotated[OptionalNullable[str], FieldMetadata(multipart=True)] = UNSET
+    r"""Url of a file to be transcribed"""
+
+    file_id: Annotated[OptionalNullable[str], FieldMetadata(multipart=True)] = UNSET
+    r"""ID of a file uploaded to /v1/files"""
+
+    language: Annotated[OptionalNullable[str], FieldMetadata(multipart=True)] = UNSET
+    r"""Language of the audio, e.g. 'en'. Providing the language can boost accuracy."""
+
+    temperature: Annotated[OptionalNullable[float], FieldMetadata(multipart=True)] = (
+        UNSET
+    )
+
+    STREAM: Annotated[
+        Annotated[Optional[Literal[False]], AfterValidator(validate_const(False))],
+        pydantic.Field(alias="stream"),
+        FieldMetadata(multipart=True),
+    ] = False
+
+    timestamp_granularities: Annotated[
+        Optional[List[TimestampGranularity]], FieldMetadata(multipart=True)
+    ] = None
+    r"""Granularities of timestamps to include in the response."""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "file",
+            "file_url",
+            "file_id",
+            "language",
+            "temperature",
+            "stream",
+            "timestamp_granularities",
+        ]
+        nullable_fields = ["file_url", "file_id", "language", "temperature"]
+        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
+
+        return m

mistralai/models/audiotranscriptionrequeststream.py

@@ -0,0 +1,97 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .file import File, FileTypedDict
+from .timestampgranularity import TimestampGranularity
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from mistralai.utils import FieldMetadata, MultipartFormMetadata, validate_const
+import pydantic
+from pydantic import model_serializer
+from pydantic.functional_validators import AfterValidator
+from typing import List, Literal, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class AudioTranscriptionRequestStreamTypedDict(TypedDict):
+    model: str
+    file: NotRequired[FileTypedDict]
+    file_url: NotRequired[Nullable[str]]
+    r"""Url of a file to be transcribed"""
+    file_id: NotRequired[Nullable[str]]
+    r"""ID of a file uploaded to /v1/files"""
+    language: NotRequired[Nullable[str]]
+    r"""Language of the audio, e.g. 'en'. Providing the language can boost accuracy."""
+    temperature: NotRequired[Nullable[float]]
+    stream: Literal[True]
+    timestamp_granularities: NotRequired[List[TimestampGranularity]]
+    r"""Granularities of timestamps to include in the response."""
+
+
+class AudioTranscriptionRequestStream(BaseModel):
+    model: Annotated[str, FieldMetadata(multipart=True)]
+
+    file: Annotated[
+        Optional[File], FieldMetadata(multipart=MultipartFormMetadata(file=True))
+    ] = None
+
+    file_url: Annotated[OptionalNullable[str], FieldMetadata(multipart=True)] = UNSET
+    r"""Url of a file to be transcribed"""
+
+    file_id: Annotated[OptionalNullable[str], FieldMetadata(multipart=True)] = UNSET
+    r"""ID of a file uploaded to /v1/files"""
+
+    language: Annotated[OptionalNullable[str], FieldMetadata(multipart=True)] = UNSET
+    r"""Language of the audio, e.g. 'en'. Providing the language can boost accuracy."""
+
+    temperature: Annotated[OptionalNullable[float], FieldMetadata(multipart=True)] = (
+        UNSET
+    )
+
+    STREAM: Annotated[
+        Annotated[Optional[Literal[True]], AfterValidator(validate_const(True))],
+        pydantic.Field(alias="stream"),
+        FieldMetadata(multipart=True),
+    ] = True
+
+    timestamp_granularities: Annotated[
+        Optional[List[TimestampGranularity]], FieldMetadata(multipart=True)
+    ] = None
+    r"""Granularities of timestamps to include in the response."""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "file",
+            "file_url",
+            "file_id",
+            "language",
+            "temperature",
+            "stream",
+            "timestamp_granularities",
+        ]
+        nullable_fields = ["file_url", "file_id", "language", "temperature"]
+        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
+
+        return m

mistralai/models/basemodelcard.py

@@ -12,6 +12,9 @@ from typing import List, Literal, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
+BaseModelCardType = Literal["base"]
+
+
 class BaseModelCardTypedDict(TypedDict):
     id: str
     capabilities: ModelCapabilitiesTypedDict
@@ -25,7 +28,7 @@ class BaseModelCardTypedDict(TypedDict):
     deprecation: NotRequired[Nullable[datetime]]
     deprecation_replacement_model: NotRequired[Nullable[str]]
     default_model_temperature: NotRequired[Nullable[float]]
-    type: Literal["base"]
+    type: BaseModelCardType
 
 
 class BaseModelCard(BaseModel):
@@ -54,7 +57,7 @@ class BaseModelCard(BaseModel):
     default_model_temperature: OptionalNullable[float] = UNSET
 
     TYPE: Annotated[
-        Annotated[Optional[Literal["base"]], AfterValidator(validate_const("base"))],
+        Annotated[Optional[BaseModelCardType], AfterValidator(validate_const("base"))],
         pydantic.Field(alias="type"),
     ] = "base"
 

mistralai/models/batchjobin.py

@@ -13,7 +13,8 @@ from typing_extensions import Annotated, NotRequired, TypedDict
 class BatchJobInTypedDict(TypedDict):
     input_files: List[str]
     endpoint: APIEndpoint
-    model: str
+    model: NotRequired[Nullable[str]]
+    agent_id: NotRequired[Nullable[str]]
     metadata: NotRequired[Nullable[Dict[str, str]]]
     timeout_hours: NotRequired[int]
 
@@ -23,7 +24,9 @@ class BatchJobIn(BaseModel):
 
     endpoint: Annotated[APIEndpoint, PlainValidator(validate_open_enum(False))]
 
-    model: str
+    model: OptionalNullable[str] = UNSET
+
+    agent_id: OptionalNullable[str] = UNSET
 
     metadata: OptionalNullable[Dict[str, str]] = UNSET
 
@@ -31,8 +34,8 @@ class BatchJobIn(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["metadata", "timeout_hours"]
-        nullable_fields = ["metadata"]
+        optional_fields = ["model", "agent_id", "metadata", "timeout_hours"]
+        nullable_fields = ["model", "agent_id", "metadata"]
         null_default_fields = []
 
         serialized = handler(self)

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/conversationevents.py

@@ -12,6 +12,10 @@ from .responsedoneevent import ResponseDoneEvent, ResponseDoneEventTypedDict
 from .responseerrorevent import ResponseErrorEvent, ResponseErrorEventTypedDict
 from .responsestartedevent import ResponseStartedEvent, ResponseStartedEventTypedDict
 from .ssetypes import SSETypes
+from .toolexecutiondeltaevent import (
+    ToolExecutionDeltaEvent,
+    ToolExecutionDeltaEventTypedDict,
+)
 from .toolexecutiondoneevent import (
     ToolExecutionDoneEvent,
     ToolExecutionDoneEventTypedDict,
@@ -34,6 +38,7 @@ ConversationEventsDataTypedDict = TypeAliasType(
         ResponseDoneEventTypedDict,
         ResponseErrorEventTypedDict,
         ToolExecutionStartedEventTypedDict,
+        ToolExecutionDeltaEventTypedDict,
         ToolExecutionDoneEventTypedDict,
         AgentHandoffStartedEventTypedDict,
         AgentHandoffDoneEventTypedDict,
@@ -52,6 +57,7 @@ ConversationEventsData = Annotated[
         Annotated[ResponseStartedEvent, Tag("conversation.response.started")],
         Annotated[FunctionCallEvent, Tag("function.call.delta")],
         Annotated[MessageOutputEvent, Tag("message.output.delta")],
+        Annotated[ToolExecutionDeltaEvent, Tag("tool.execution.delta")],
         Annotated[ToolExecutionDoneEvent, Tag("tool.execution.done")],
         Annotated[ToolExecutionStartedEvent, Tag("tool.execution.started")],
     ],

mistralai/models/conversationhistory.py

@@ -17,10 +17,10 @@ ConversationHistoryObject = Literal["conversation.history"]
 EntriesTypedDict = TypeAliasType(
     "EntriesTypedDict",
     Union[
-        MessageInputEntryTypedDict,
         FunctionResultEntryTypedDict,
-        ToolExecutionEntryTypedDict,
+        MessageInputEntryTypedDict,
         FunctionCallEntryTypedDict,
+        ToolExecutionEntryTypedDict,
         MessageOutputEntryTypedDict,
         AgentHandoffEntryTypedDict,
     ],
@@ -30,10 +30,10 @@ EntriesTypedDict = TypeAliasType(
 Entries = TypeAliasType(
     "Entries",
     Union[
-        MessageInputEntry,
         FunctionResultEntry,
-        ToolExecutionEntry,
+        MessageInputEntry,
         FunctionCallEntry,
+        ToolExecutionEntry,
         MessageOutputEntry,
         AgentHandoffEntry,
     ],

mistralai/models/documentout.py

@@ -0,0 +1,105 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+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_extensions import NotRequired, TypedDict
+
+
+class DocumentOutTypedDict(TypedDict):
+    id: str
+    library_id: str
+    hash: str
+    mime_type: str
+    extension: str
+    size: int
+    name: str
+    created_at: datetime
+    processing_status: str
+    uploaded_by_id: str
+    uploaded_by_type: str
+    tokens_processing_total: int
+    summary: NotRequired[Nullable[str]]
+    last_processed_at: NotRequired[Nullable[datetime]]
+    number_of_pages: NotRequired[Nullable[int]]
+    tokens_processing_main_content: NotRequired[Nullable[int]]
+    tokens_processing_summary: NotRequired[Nullable[int]]
+
+
+class DocumentOut(BaseModel):
+    id: str
+
+    library_id: str
+
+    hash: str
+
+    mime_type: str
+
+    extension: str
+
+    size: int
+
+    name: str
+
+    created_at: datetime
+
+    processing_status: str
+
+    uploaded_by_id: str
+
+    uploaded_by_type: str
+
+    tokens_processing_total: int
+
+    summary: OptionalNullable[str] = UNSET
+
+    last_processed_at: OptionalNullable[datetime] = UNSET
+
+    number_of_pages: OptionalNullable[int] = UNSET
+
+    tokens_processing_main_content: OptionalNullable[int] = UNSET
+
+    tokens_processing_summary: OptionalNullable[int] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "summary",
+            "last_processed_at",
+            "number_of_pages",
+            "tokens_processing_main_content",
+            "tokens_processing_summary",
+        ]
+        nullable_fields = [
+            "summary",
+            "last_processed_at",
+            "number_of_pages",
+            "tokens_processing_main_content",
+            "tokens_processing_summary",
+        ]
+        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
+
+        return m

mistralai/models/documenttextcontent.py

@@ -0,0 +1,13 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from typing_extensions import TypedDict
+
+
+class DocumentTextContentTypedDict(TypedDict):
+    text: str
+
+
+class DocumentTextContent(BaseModel):
+    text: str

mistralai/models/documentupdatein.py

@@ -0,0 +1,44 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing_extensions import NotRequired, TypedDict
+
+
+class DocumentUpdateInTypedDict(TypedDict):
+    name: NotRequired[Nullable[str]]
+
+
+class DocumentUpdateIn(BaseModel):
+    name: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["name"]
+        nullable_fields = ["name"]
+        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
+
+        return m

mistralai/models/entitytype.py

@@ -0,0 +1,9 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import UnrecognizedStr
+from typing import Literal, Union
+
+
+EntityType = Union[Literal["User", "Workspace", "Org"], UnrecognizedStr]
+r"""The type of entity, used to share a library."""

mistralai/models/file.py

@@ -0,0 +1,33 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+import io
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, MultipartFormMetadata
+import pydantic
+from typing import IO, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class FileTypedDict(TypedDict):
+    file_name: str
+    content: Union[bytes, IO[bytes], io.BufferedReader]
+    content_type: NotRequired[str]
+
+
+class File(BaseModel):
+    file_name: Annotated[
+        str, pydantic.Field(alias="fileName"), FieldMetadata(multipart=True)
+    ]
+
+    content: Annotated[
+        Union[bytes, IO[bytes], io.BufferedReader],
+        pydantic.Field(alias=""),
+        FieldMetadata(multipart=MultipartFormMetadata(content=True)),
+    ]
+
+    content_type: Annotated[
+        Optional[str],
+        pydantic.Field(alias="Content-Type"),
+        FieldMetadata(multipart=True),
+    ] = None