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

mistralai/_version.py

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

mistralai/accesses.py

@@ -9,6 +9,8 @@ from typing import Any, Mapping, Optional
 
 
 class Accesses(BaseSDK):
+    r"""(beta) Libraries API - manage access to a library."""
+
     def list(
         self,
         *,

mistralai/audio.py

@@ -0,0 +1,18 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from .basesdk import BaseSDK
+from .sdkconfiguration import SDKConfiguration
+from mistralai.transcriptions import Transcriptions
+
+
+class Audio(BaseSDK):
+    transcriptions: Transcriptions
+    r"""API for audio transcription."""
+
+    def __init__(self, sdk_config: SDKConfiguration) -> None:
+        BaseSDK.__init__(self, sdk_config)
+        self.sdk_configuration = sdk_config
+        self._init_sdks()
+
+    def _init_sdks(self):
+        self.transcriptions = Transcriptions(self.sdk_configuration)

mistralai/beta.py

@@ -13,7 +13,7 @@ class Beta(BaseSDK):
     agents: MistralAgents
     r"""(beta) Agents API"""
     libraries: Libraries
-    r"""(beta) Libraries API for indexing documents to enhance agent capabilities."""
+    r"""(beta) Libraries API to create and manage libraries - index your documents to enhance agent capabilities."""
 
     def __init__(self, sdk_config: SDKConfiguration) -> None:
         BaseSDK.__init__(self, sdk_config)

mistralai/conversations.py

@@ -85,7 +85,7 @@ class Conversations(BaseSDK):
                     retries=retries,
                     server_url=server_url,
                     timeout_ms=timeout_ms,
-                    **req,
+                    **req,  # type: ignore
                 )
                 run_result.conversation_id = res.conversation_id
                 run_ctx.conversation_id = res.conversation_id
@@ -164,7 +164,7 @@ class Conversations(BaseSDK):
                         retries=retries,
                         server_url=server_url,
                         timeout_ms=timeout_ms,
-                        **req,
+                        **req,  # type: ignore
                     )
                 else:
                     res = await self.append_stream_async(

mistralai/documents.py

@@ -9,6 +9,8 @@ from typing import Any, Mapping, Optional, Union
 
 
 class Documents(BaseSDK):
+    r"""(beta) Libraries API - manage documents in a library."""
+
     def list(
         self,
         *,

mistralai/libraries.py

@@ -12,10 +12,12 @@ from typing import Any, Mapping, Optional
 
 
 class Libraries(BaseSDK):
-    r"""(beta) Libraries API for indexing documents to enhance agent capabilities."""
+    r"""(beta) Libraries API to create and manage libraries - index your documents to enhance agent capabilities."""
 
     documents: Documents
+    r"""(beta) Libraries API - manage documents in a library."""
     accesses: Accesses
+    r"""(beta) Libraries API - manage access to a library."""
 
     def __init__(self, sdk_config: SDKConfiguration) -> None:
         BaseSDK.__init__(self, sdk_config)

mistralai/mistral_jobs.py

@@ -16,6 +16,7 @@ class MistralJobs(BaseSDK):
         page: Optional[int] = 0,
         page_size: Optional[int] = 100,
         model: OptionalNullable[str] = UNSET,
+        agent_id: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         created_after: OptionalNullable[datetime] = UNSET,
         created_by_me: Optional[bool] = False,
@@ -32,6 +33,7 @@ class MistralJobs(BaseSDK):
         :param page:
         :param page_size:
         :param model:
+        :param agent_id:
         :param metadata:
         :param created_after:
         :param created_by_me:
@@ -55,6 +57,7 @@ class MistralJobs(BaseSDK):
             page=page,
             page_size=page_size,
             model=model,
+            agent_id=agent_id,
             metadata=metadata,
             created_after=created_after,
             created_by_me=created_by_me,
@@ -128,6 +131,7 @@ class MistralJobs(BaseSDK):
         page: Optional[int] = 0,
         page_size: Optional[int] = 100,
         model: OptionalNullable[str] = UNSET,
+        agent_id: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         created_after: OptionalNullable[datetime] = UNSET,
         created_by_me: Optional[bool] = False,
@@ -144,6 +148,7 @@ class MistralJobs(BaseSDK):
         :param page:
         :param page_size:
         :param model:
+        :param agent_id:
         :param metadata:
         :param created_after:
         :param created_by_me:
@@ -167,6 +172,7 @@ class MistralJobs(BaseSDK):
             page=page,
             page_size=page_size,
             model=model,
+            agent_id=agent_id,
             metadata=metadata,
             created_after=created_after,
             created_by_me=created_by_me,
@@ -239,7 +245,8 @@ class MistralJobs(BaseSDK):
         *,
         input_files: List[str],
         endpoint: models.APIEndpoint,
-        model: str,
+        model: OptionalNullable[str] = UNSET,
+        agent_id: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, str]] = UNSET,
         timeout_hours: Optional[int] = 24,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
@@ -254,6 +261,7 @@ class MistralJobs(BaseSDK):
         :param input_files:
         :param endpoint:
         :param model:
+        :param agent_id:
         :param metadata:
         :param timeout_hours:
         :param retries: Override the default retry configuration for this method
@@ -275,6 +283,7 @@ class MistralJobs(BaseSDK):
             input_files=input_files,
             endpoint=endpoint,
             model=model,
+            agent_id=agent_id,
             metadata=metadata,
             timeout_hours=timeout_hours,
         )
@@ -348,7 +357,8 @@ class MistralJobs(BaseSDK):
         *,
         input_files: List[str],
         endpoint: models.APIEndpoint,
-        model: str,
+        model: OptionalNullable[str] = UNSET,
+        agent_id: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, str]] = UNSET,
         timeout_hours: Optional[int] = 24,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
@@ -363,6 +373,7 @@ class MistralJobs(BaseSDK):
         :param input_files:
         :param endpoint:
         :param model:
+        :param agent_id:
         :param metadata:
         :param timeout_hours:
         :param retries: Override the default retry configuration for this method
@@ -384,6 +395,7 @@ class MistralJobs(BaseSDK):
             input_files=input_files,
             endpoint=endpoint,
             model=model,
+            agent_id=agent_id,
             metadata=metadata,
             timeout_hours=timeout_hours,
         )

mistralai/models/__init__.py

@@ -129,7 +129,16 @@ if TYPE_CHECKING:
         AssistantMessageRole,
         AssistantMessageTypedDict,
     )
-    from .basemodelcard import BaseModelCard, BaseModelCardTypedDict, Type
+    from .audiochunk import AudioChunk, AudioChunkType, AudioChunkTypedDict
+    from .audiotranscriptionrequest import (
+        AudioTranscriptionRequest,
+        AudioTranscriptionRequestTypedDict,
+    )
+    from .audiotranscriptionrequeststream import (
+        AudioTranscriptionRequestStream,
+        AudioTranscriptionRequestStreamTypedDict,
+    )
+    from .basemodelcard import BaseModelCard, BaseModelCardType, BaseModelCardTypedDict
     from .batcherror import BatchError, BatchErrorTypedDict
     from .batchjobin import BatchJobIn, BatchJobInTypedDict
     from .batchjobout import BatchJobOut, BatchJobOutObject, BatchJobOutTypedDict
@@ -759,6 +768,14 @@ if TYPE_CHECKING:
         SystemMessageTypedDict,
     )
     from .textchunk import TextChunk, TextChunkType, TextChunkTypedDict
+    from .thinkchunk import (
+        ThinkChunk,
+        ThinkChunkType,
+        ThinkChunkTypedDict,
+        Thinking,
+        ThinkingTypedDict,
+    )
+    from .timestampgranularity import TimestampGranularity
     from .tool import Tool, ToolTypedDict
     from .toolcall import ToolCall, ToolCallTypedDict
     from .toolchoice import ToolChoice, ToolChoiceTypedDict
@@ -799,6 +816,42 @@ if TYPE_CHECKING:
     )
     from .tooltypes import ToolTypes
     from .trainingfile import TrainingFile, TrainingFileTypedDict
+    from .transcriptionresponse import (
+        TranscriptionResponse,
+        TranscriptionResponseTypedDict,
+    )
+    from .transcriptionsegmentchunk import (
+        TranscriptionSegmentChunk,
+        TranscriptionSegmentChunkTypedDict,
+        Type,
+    )
+    from .transcriptionstreamdone import (
+        TranscriptionStreamDone,
+        TranscriptionStreamDoneType,
+        TranscriptionStreamDoneTypedDict,
+    )
+    from .transcriptionstreamevents import (
+        TranscriptionStreamEvents,
+        TranscriptionStreamEventsData,
+        TranscriptionStreamEventsDataTypedDict,
+        TranscriptionStreamEventsTypedDict,
+    )
+    from .transcriptionstreameventtypes import TranscriptionStreamEventTypes
+    from .transcriptionstreamlanguage import (
+        TranscriptionStreamLanguage,
+        TranscriptionStreamLanguageType,
+        TranscriptionStreamLanguageTypedDict,
+    )
+    from .transcriptionstreamsegmentdelta import (
+        TranscriptionStreamSegmentDelta,
+        TranscriptionStreamSegmentDeltaType,
+        TranscriptionStreamSegmentDeltaTypedDict,
+    )
+    from .transcriptionstreamtextdelta import (
+        TranscriptionStreamTextDelta,
+        TranscriptionStreamTextDeltaType,
+        TranscriptionStreamTextDeltaTypedDict,
+    )
     from .unarchiveftmodelout import (
         UnarchiveFTModelOut,
         UnarchiveFTModelOutObject,
@@ -917,7 +970,15 @@ __all__ = [
     "AssistantMessageContentTypedDict",
     "AssistantMessageRole",
     "AssistantMessageTypedDict",
+    "AudioChunk",
+    "AudioChunkType",
+    "AudioChunkTypedDict",
+    "AudioTranscriptionRequest",
+    "AudioTranscriptionRequestStream",
+    "AudioTranscriptionRequestStreamTypedDict",
+    "AudioTranscriptionRequestTypedDict",
     "BaseModelCard",
+    "BaseModelCardType",
     "BaseModelCardTypedDict",
     "BatchError",
     "BatchErrorTypedDict",
@@ -1416,6 +1477,12 @@ __all__ = [
     "TextChunk",
     "TextChunkType",
     "TextChunkTypedDict",
+    "ThinkChunk",
+    "ThinkChunkType",
+    "ThinkChunkTypedDict",
+    "Thinking",
+    "ThinkingTypedDict",
+    "TimestampGranularity",
     "Tool",
     "ToolCall",
     "ToolCallTypedDict",
@@ -1452,6 +1519,27 @@ __all__ = [
     "ToolsTypedDict",
     "TrainingFile",
     "TrainingFileTypedDict",
+    "TranscriptionResponse",
+    "TranscriptionResponseTypedDict",
+    "TranscriptionSegmentChunk",
+    "TranscriptionSegmentChunkTypedDict",
+    "TranscriptionStreamDone",
+    "TranscriptionStreamDoneType",
+    "TranscriptionStreamDoneTypedDict",
+    "TranscriptionStreamEventTypes",
+    "TranscriptionStreamEvents",
+    "TranscriptionStreamEventsData",
+    "TranscriptionStreamEventsDataTypedDict",
+    "TranscriptionStreamEventsTypedDict",
+    "TranscriptionStreamLanguage",
+    "TranscriptionStreamLanguageType",
+    "TranscriptionStreamLanguageTypedDict",
+    "TranscriptionStreamSegmentDelta",
+    "TranscriptionStreamSegmentDeltaType",
+    "TranscriptionStreamSegmentDeltaTypedDict",
+    "TranscriptionStreamTextDelta",
+    "TranscriptionStreamTextDeltaType",
+    "TranscriptionStreamTextDeltaTypedDict",
     "Two",
     "TwoTypedDict",
     "Type",
@@ -1565,9 +1653,16 @@ _dynamic_imports: dict[str, str] = {
     "AssistantMessageContentTypedDict": ".assistantmessage",
     "AssistantMessageRole": ".assistantmessage",
     "AssistantMessageTypedDict": ".assistantmessage",
+    "AudioChunk": ".audiochunk",
+    "AudioChunkType": ".audiochunk",
+    "AudioChunkTypedDict": ".audiochunk",
+    "AudioTranscriptionRequest": ".audiotranscriptionrequest",
+    "AudioTranscriptionRequestTypedDict": ".audiotranscriptionrequest",
+    "AudioTranscriptionRequestStream": ".audiotranscriptionrequeststream",
+    "AudioTranscriptionRequestStreamTypedDict": ".audiotranscriptionrequeststream",
     "BaseModelCard": ".basemodelcard",
+    "BaseModelCardType": ".basemodelcard",
     "BaseModelCardTypedDict": ".basemodelcard",
-    "Type": ".basemodelcard",
     "BatchError": ".batcherror",
     "BatchErrorTypedDict": ".batcherror",
     "BatchJobIn": ".batchjobin",
@@ -2067,6 +2162,12 @@ _dynamic_imports: dict[str, str] = {
     "TextChunk": ".textchunk",
     "TextChunkType": ".textchunk",
     "TextChunkTypedDict": ".textchunk",
+    "ThinkChunk": ".thinkchunk",
+    "ThinkChunkType": ".thinkchunk",
+    "ThinkChunkTypedDict": ".thinkchunk",
+    "Thinking": ".thinkchunk",
+    "ThinkingTypedDict": ".thinkchunk",
+    "TimestampGranularity": ".timestampgranularity",
     "Tool": ".tool",
     "ToolTypedDict": ".tool",
     "ToolCall": ".toolcall",
@@ -2101,6 +2202,28 @@ _dynamic_imports: dict[str, str] = {
     "ToolTypes": ".tooltypes",
     "TrainingFile": ".trainingfile",
     "TrainingFileTypedDict": ".trainingfile",
+    "TranscriptionResponse": ".transcriptionresponse",
+    "TranscriptionResponseTypedDict": ".transcriptionresponse",
+    "TranscriptionSegmentChunk": ".transcriptionsegmentchunk",
+    "TranscriptionSegmentChunkTypedDict": ".transcriptionsegmentchunk",
+    "Type": ".transcriptionsegmentchunk",
+    "TranscriptionStreamDone": ".transcriptionstreamdone",
+    "TranscriptionStreamDoneType": ".transcriptionstreamdone",
+    "TranscriptionStreamDoneTypedDict": ".transcriptionstreamdone",
+    "TranscriptionStreamEvents": ".transcriptionstreamevents",
+    "TranscriptionStreamEventsData": ".transcriptionstreamevents",
+    "TranscriptionStreamEventsDataTypedDict": ".transcriptionstreamevents",
+    "TranscriptionStreamEventsTypedDict": ".transcriptionstreamevents",
+    "TranscriptionStreamEventTypes": ".transcriptionstreameventtypes",
+    "TranscriptionStreamLanguage": ".transcriptionstreamlanguage",
+    "TranscriptionStreamLanguageType": ".transcriptionstreamlanguage",
+    "TranscriptionStreamLanguageTypedDict": ".transcriptionstreamlanguage",
+    "TranscriptionStreamSegmentDelta": ".transcriptionstreamsegmentdelta",
+    "TranscriptionStreamSegmentDeltaType": ".transcriptionstreamsegmentdelta",
+    "TranscriptionStreamSegmentDeltaTypedDict": ".transcriptionstreamsegmentdelta",
+    "TranscriptionStreamTextDelta": ".transcriptionstreamtextdelta",
+    "TranscriptionStreamTextDeltaType": ".transcriptionstreamtextdelta",
+    "TranscriptionStreamTextDeltaTypedDict": ".transcriptionstreamtextdelta",
     "UnarchiveFTModelOut": ".unarchiveftmodelout",
     "UnarchiveFTModelOutObject": ".unarchiveftmodelout",
     "UnarchiveFTModelOutTypedDict": ".unarchiveftmodelout",

mistralai/models/audiochunk.py

@@ -0,0 +1,20 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from typing import Literal, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+AudioChunkType = Literal["input_audio"]
+
+
+class AudioChunkTypedDict(TypedDict):
+    input_audio: str
+    type: NotRequired[AudioChunkType]
+
+
+class AudioChunk(BaseModel):
+    input_audio: str
+
+    type: Optional[AudioChunkType] = "input_audio"

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,7 +12,7 @@ from typing import List, Literal, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-Type = Literal["base"]
+BaseModelCardType = Literal["base"]
 
 
 class BaseModelCardTypedDict(TypedDict):
@@ -28,7 +28,7 @@ class BaseModelCardTypedDict(TypedDict):
     deprecation: NotRequired[Nullable[datetime]]
     deprecation_replacement_model: NotRequired[Nullable[str]]
     default_model_temperature: NotRequired[Nullable[float]]
-    type: Type
+    type: BaseModelCardType
 
 
 class BaseModelCard(BaseModel):
@@ -57,7 +57,7 @@ class BaseModelCard(BaseModel):
     default_model_temperature: OptionalNullable[float] = UNSET
 
     TYPE: Annotated[
-        Annotated[Optional[Type], 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)