mistralai 1.10.0__py3-none-any.whl → 1.10.1__py3-none-any.whl

mistralai/mistral_jobs.py

@@ -7,7 +7,7 @@ from mistralai._hooks import HookContext
 from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import get_security_from_env
 from mistralai.utils.unmarshal_json_response import unmarshal_json_response
-from typing import Any, Dict, List, Mapping, Optional
+from typing import Any, Dict, List, Mapping, Optional, Union
 
 
 class MistralJobs(BaseSDK):
@@ -222,8 +222,11 @@ class MistralJobs(BaseSDK):
     def create(
         self,
         *,
-        input_files: List[str],
         endpoint: models.APIEndpoint,
+        input_files: OptionalNullable[List[str]] = UNSET,
+        requests: OptionalNullable[
+            Union[List[models.BatchRequest], List[models.BatchRequestTypedDict]]
+        ] = UNSET,
         model: OptionalNullable[str] = UNSET,
         agent_id: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, str]] = UNSET,
@@ -237,8 +240,9 @@ class MistralJobs(BaseSDK):
 
         Create a new batch job, it will be queued for processing.
 
-        :param input_files: The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```
         :param endpoint:
+        :param input_files: The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```
+        :param requests:
         :param model: The model to be used for batch inference.
         :param agent_id: In case you want to use a specific agent from the **deprecated** agents api for batch inference, you can specify the agent ID here.
         :param metadata: The metadata of your choice to be associated with the batch inference job.
@@ -260,6 +264,9 @@ class MistralJobs(BaseSDK):
 
         request = models.BatchJobIn(
             input_files=input_files,
+            requests=utils.get_pydantic_model(
+                requests, OptionalNullable[List[models.BatchRequest]]
+            ),
             endpoint=endpoint,
             model=model,
             agent_id=agent_id,
@@ -323,8 +330,11 @@ class MistralJobs(BaseSDK):
     async def create_async(
         self,
         *,
-        input_files: List[str],
         endpoint: models.APIEndpoint,
+        input_files: OptionalNullable[List[str]] = UNSET,
+        requests: OptionalNullable[
+            Union[List[models.BatchRequest], List[models.BatchRequestTypedDict]]
+        ] = UNSET,
         model: OptionalNullable[str] = UNSET,
         agent_id: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, str]] = UNSET,
@@ -338,8 +348,9 @@ class MistralJobs(BaseSDK):
 
         Create a new batch job, it will be queued for processing.
 
-        :param input_files: The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```
         :param endpoint:
+        :param input_files: The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```
+        :param requests:
         :param model: The model to be used for batch inference.
         :param agent_id: In case you want to use a specific agent from the **deprecated** agents api for batch inference, you can specify the agent ID here.
         :param metadata: The metadata of your choice to be associated with the batch inference job.
@@ -361,6 +372,9 @@ class MistralJobs(BaseSDK):
 
         request = models.BatchJobIn(
             input_files=input_files,
+            requests=utils.get_pydantic_model(
+                requests, OptionalNullable[List[models.BatchRequest]]
+            ),
             endpoint=endpoint,
             model=model,
             agent_id=agent_id,
@@ -425,6 +439,7 @@ class MistralJobs(BaseSDK):
         self,
         *,
         job_id: str,
+        inline: OptionalNullable[bool] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -434,7 +449,11 @@ class MistralJobs(BaseSDK):
 
         Get a batch job details by its UUID.
 
+        Args:
+        inline: If True, return results inline in the response.
+
         :param job_id:
+        :param inline:
         :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
@@ -452,6 +471,7 @@ class MistralJobs(BaseSDK):
 
         request = models.JobsAPIRoutesBatchGetBatchJobRequest(
             job_id=job_id,
+            inline=inline,
         )
 
         req = self._build_request(
@@ -508,6 +528,7 @@ class MistralJobs(BaseSDK):
         self,
         *,
         job_id: str,
+        inline: OptionalNullable[bool] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -517,7 +538,11 @@ class MistralJobs(BaseSDK):
 
         Get a batch job details by its UUID.
 
+        Args:
+        inline: If True, return results inline in the response.
+
         :param job_id:
+        :param inline:
         :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
@@ -535,6 +560,7 @@ class MistralJobs(BaseSDK):
 
         request = models.JobsAPIRoutesBatchGetBatchJobRequest(
             job_id=job_id,
+            inline=inline,
         )
 
         req = self._build_request_async(

mistralai/models/__init__.py

@@ -141,6 +141,8 @@ if TYPE_CHECKING:
         AssistantMessageTypedDict,
     )
     from .audiochunk import AudioChunk, AudioChunkType, AudioChunkTypedDict
+    from .audioencoding import AudioEncoding
+    from .audioformat import AudioFormat, AudioFormatTypedDict
     from .audiotranscriptionrequest import (
         AudioTranscriptionRequest,
         AudioTranscriptionRequestTypedDict,
@@ -155,6 +157,7 @@ if TYPE_CHECKING:
     from .batchjobout import BatchJobOut, BatchJobOutObject, BatchJobOutTypedDict
     from .batchjobsout import BatchJobsOut, BatchJobsOutObject, BatchJobsOutTypedDict
     from .batchjobstatus import BatchJobStatus
+    from .batchrequest import BatchRequest, BatchRequestTypedDict
     from .builtinconnectors import BuiltInConnectors
     from .chatclassificationrequest import (
         ChatClassificationRequest,
@@ -840,7 +843,13 @@ if TYPE_CHECKING:
         ToolExecutionStartedEventType,
         ToolExecutionStartedEventTypedDict,
     )
-    from .toolfilechunk import ToolFileChunk, ToolFileChunkType, ToolFileChunkTypedDict
+    from .toolfilechunk import (
+        ToolFileChunk,
+        ToolFileChunkTool,
+        ToolFileChunkToolTypedDict,
+        ToolFileChunkType,
+        ToolFileChunkTypedDict,
+    )
     from .toolmessage import (
         ToolMessage,
         ToolMessageContent,
@@ -850,6 +859,8 @@ if TYPE_CHECKING:
     )
     from .toolreferencechunk import (
         ToolReferenceChunk,
+        ToolReferenceChunkTool,
+        ToolReferenceChunkToolTypedDict,
         ToolReferenceChunkType,
         ToolReferenceChunkTypedDict,
     )
@@ -1018,6 +1029,9 @@ __all__ = [
     "AudioChunk",
     "AudioChunkType",
     "AudioChunkTypedDict",
+    "AudioEncoding",
+    "AudioFormat",
+    "AudioFormatTypedDict",
     "AudioTranscriptionRequest",
     "AudioTranscriptionRequestStream",
     "AudioTranscriptionRequestStreamTypedDict",
@@ -1036,6 +1050,8 @@ __all__ = [
     "BatchJobsOut",
     "BatchJobsOutObject",
     "BatchJobsOutTypedDict",
+    "BatchRequest",
+    "BatchRequestTypedDict",
     "BuiltInConnectors",
     "ChatClassificationRequest",
     "ChatClassificationRequestTypedDict",
@@ -1567,6 +1583,8 @@ __all__ = [
     "ToolExecutionStartedEventType",
     "ToolExecutionStartedEventTypedDict",
     "ToolFileChunk",
+    "ToolFileChunkTool",
+    "ToolFileChunkToolTypedDict",
     "ToolFileChunkType",
     "ToolFileChunkTypedDict",
     "ToolMessage",
@@ -1575,6 +1593,8 @@ __all__ = [
     "ToolMessageRole",
     "ToolMessageTypedDict",
     "ToolReferenceChunk",
+    "ToolReferenceChunkTool",
+    "ToolReferenceChunkToolTypedDict",
     "ToolReferenceChunkType",
     "ToolReferenceChunkTypedDict",
     "ToolTypedDict",
@@ -1724,6 +1744,9 @@ _dynamic_imports: dict[str, str] = {
     "AudioChunk": ".audiochunk",
     "AudioChunkType": ".audiochunk",
     "AudioChunkTypedDict": ".audiochunk",
+    "AudioEncoding": ".audioencoding",
+    "AudioFormat": ".audioformat",
+    "AudioFormatTypedDict": ".audioformat",
     "AudioTranscriptionRequest": ".audiotranscriptionrequest",
     "AudioTranscriptionRequestTypedDict": ".audiotranscriptionrequest",
     "AudioTranscriptionRequestStream": ".audiotranscriptionrequeststream",
@@ -1742,6 +1765,8 @@ _dynamic_imports: dict[str, str] = {
     "BatchJobsOutObject": ".batchjobsout",
     "BatchJobsOutTypedDict": ".batchjobsout",
     "BatchJobStatus": ".batchjobstatus",
+    "BatchRequest": ".batchrequest",
+    "BatchRequestTypedDict": ".batchrequest",
     "BuiltInConnectors": ".builtinconnectors",
     "ChatClassificationRequest": ".chatclassificationrequest",
     "ChatClassificationRequestTypedDict": ".chatclassificationrequest",
@@ -2277,6 +2302,8 @@ _dynamic_imports: dict[str, str] = {
     "ToolExecutionStartedEventType": ".toolexecutionstartedevent",
     "ToolExecutionStartedEventTypedDict": ".toolexecutionstartedevent",
     "ToolFileChunk": ".toolfilechunk",
+    "ToolFileChunkTool": ".toolfilechunk",
+    "ToolFileChunkToolTypedDict": ".toolfilechunk",
     "ToolFileChunkType": ".toolfilechunk",
     "ToolFileChunkTypedDict": ".toolfilechunk",
     "ToolMessage": ".toolmessage",
@@ -2285,6 +2312,8 @@ _dynamic_imports: dict[str, str] = {
     "ToolMessageRole": ".toolmessage",
     "ToolMessageTypedDict": ".toolmessage",
     "ToolReferenceChunk": ".toolreferencechunk",
+    "ToolReferenceChunkTool": ".toolreferencechunk",
+    "ToolReferenceChunkToolTypedDict": ".toolreferencechunk",
     "ToolReferenceChunkType": ".toolreferencechunk",
     "ToolReferenceChunkTypedDict": ".toolreferencechunk",
     "ToolTypes": ".tooltypes",

mistralai/models/agents_api_v1_agents_listop.py

@@ -52,7 +52,7 @@ class AgentsAPIV1AgentsListRequest(BaseModel):
 
     metadata: Annotated[
         OptionalNullable[Dict[str, Any]],
-        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+        FieldMetadata(query=QueryParamMetadata(serialization="json")),
     ] = UNSET
 
     @model_serializer(mode="wrap")

mistralai/models/agents_api_v1_conversations_listop.py

@@ -29,7 +29,7 @@ class AgentsAPIV1ConversationsListRequest(BaseModel):
 
     metadata: Annotated[
         OptionalNullable[Dict[str, Any]],
-        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+        FieldMetadata(query=QueryParamMetadata(serialization="json")),
     ] = UNSET
 
     @model_serializer(mode="wrap")

mistralai/models/audioencoding.py

@@ -0,0 +1,13 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import UnrecognizedStr
+from typing import Literal, Union
+
+
+AudioEncoding = Union[
+    Literal[
+        "pcm_s16le", "pcm_s32le", "pcm_f16le", "pcm_f32le", "pcm_mulaw", "pcm_alaw"
+    ],
+    UnrecognizedStr,
+]

mistralai/models/audioformat.py

@@ -0,0 +1,19 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .audioencoding import AudioEncoding
+from mistralai.types import BaseModel
+from mistralai.utils import validate_open_enum
+from pydantic.functional_validators import PlainValidator
+from typing_extensions import Annotated, TypedDict
+
+
+class AudioFormatTypedDict(TypedDict):
+    encoding: AudioEncoding
+    sample_rate: int
+
+
+class AudioFormat(BaseModel):
+    encoding: Annotated[AudioEncoding, PlainValidator(validate_open_enum(False))]
+
+    sample_rate: int

mistralai/models/batchjobin.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .apiendpoint import APIEndpoint
+from .batchrequest import BatchRequest, BatchRequestTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mistralai.utils import validate_open_enum
 from pydantic import model_serializer
@@ -11,9 +12,10 @@ from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class BatchJobInTypedDict(TypedDict):
-    input_files: List[str]
-    r"""The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```"""
     endpoint: APIEndpoint
+    input_files: NotRequired[Nullable[List[str]]]
+    r"""The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```"""
+    requests: NotRequired[Nullable[List[BatchRequestTypedDict]]]
     model: NotRequired[Nullable[str]]
     r"""The model to be used for batch inference."""
     agent_id: NotRequired[Nullable[str]]
@@ -25,10 +27,12 @@ class BatchJobInTypedDict(TypedDict):
 
 
 class BatchJobIn(BaseModel):
-    input_files: List[str]
+    endpoint: Annotated[APIEndpoint, PlainValidator(validate_open_enum(False))]
+
+    input_files: OptionalNullable[List[str]] = UNSET
     r"""The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```"""
 
-    endpoint: Annotated[APIEndpoint, PlainValidator(validate_open_enum(False))]
+    requests: OptionalNullable[List[BatchRequest]] = UNSET
 
     model: OptionalNullable[str] = UNSET
     r"""The model to be used for batch inference."""
@@ -44,8 +48,15 @@ class BatchJobIn(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["model", "agent_id", "metadata", "timeout_hours"]
-        nullable_fields = ["model", "agent_id", "metadata"]
+        optional_fields = [
+            "input_files",
+            "requests",
+            "model",
+            "agent_id",
+            "metadata",
+            "timeout_hours",
+        ]
+        nullable_fields = ["input_files", "requests", "model", "agent_id", "metadata"]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/batchjobout.py

@@ -29,6 +29,7 @@ class BatchJobOutTypedDict(TypedDict):
     agent_id: NotRequired[Nullable[str]]
     output_file: NotRequired[Nullable[str]]
     error_file: NotRequired[Nullable[str]]
+    outputs: NotRequired[Nullable[List[Dict[str, Any]]]]
     started_at: NotRequired[Nullable[int]]
     completed_at: NotRequired[Nullable[int]]
 
@@ -66,6 +67,8 @@ class BatchJobOut(BaseModel):
 
     error_file: OptionalNullable[str] = UNSET
 
+    outputs: OptionalNullable[List[Dict[str, Any]]] = UNSET
+
     started_at: OptionalNullable[int] = UNSET
 
     completed_at: OptionalNullable[int] = UNSET
@@ -79,6 +82,7 @@ class BatchJobOut(BaseModel):
             "agent_id",
             "output_file",
             "error_file",
+            "outputs",
             "started_at",
             "completed_at",
         ]
@@ -88,6 +92,7 @@ class BatchJobOut(BaseModel):
             "agent_id",
             "output_file",
             "error_file",
+            "outputs",
             "started_at",
             "completed_at",
         ]

mistralai/models/batchrequest.py

@@ -0,0 +1,48 @@
+"""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 import Any, Dict
+from typing_extensions import NotRequired, TypedDict
+
+
+class BatchRequestTypedDict(TypedDict):
+    body: Dict[str, Any]
+    custom_id: NotRequired[Nullable[str]]
+
+
+class BatchRequest(BaseModel):
+    body: Dict[str, Any]
+
+    custom_id: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["custom_id"]
+        nullable_fields = ["custom_id"]
+        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/classificationrequest.py

@@ -1,10 +1,11 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from mistralai.types import BaseModel
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 import pydantic
-from typing import List, Union
-from typing_extensions import Annotated, TypeAliasType, TypedDict
+from pydantic import model_serializer
+from typing import Any, Dict, List, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 ClassificationRequestInputsTypedDict = TypeAliasType(
@@ -24,6 +25,7 @@ class ClassificationRequestTypedDict(TypedDict):
     r"""ID of the model to use."""
     inputs: ClassificationRequestInputsTypedDict
     r"""Text to classify."""
+    metadata: NotRequired[Nullable[Dict[str, Any]]]
 
 
 class ClassificationRequest(BaseModel):
@@ -32,3 +34,35 @@ class ClassificationRequest(BaseModel):
 
     inputs: Annotated[ClassificationRequestInputs, pydantic.Field(alias="input")]
     r"""Text to classify."""
+
+    metadata: OptionalNullable[Dict[str, Any]] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["metadata"]
+        nullable_fields = ["metadata"]
+        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/embeddingrequest.py

@@ -6,7 +6,7 @@ from .encodingformat import EncodingFormat
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 import pydantic
 from pydantic import model_serializer
-from typing import List, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
@@ -25,6 +25,7 @@ class EmbeddingRequestTypedDict(TypedDict):
     r"""The ID of the model to be used for embedding."""
     inputs: EmbeddingRequestInputsTypedDict
     r"""The text content to be embedded, can be a string or an array of strings for fast processing in bulk."""
+    metadata: NotRequired[Nullable[Dict[str, Any]]]
     output_dimension: NotRequired[Nullable[int]]
     r"""The dimension of the output embeddings when feature available. If not provided, a default output dimension will be used."""
     output_dtype: NotRequired[EmbeddingDtype]
@@ -38,6 +39,8 @@ class EmbeddingRequest(BaseModel):
     inputs: Annotated[EmbeddingRequestInputs, pydantic.Field(alias="input")]
     r"""The text content to be embedded, can be a string or an array of strings for fast processing in bulk."""
 
+    metadata: OptionalNullable[Dict[str, Any]] = UNSET
+
     output_dimension: OptionalNullable[int] = UNSET
     r"""The dimension of the output embeddings when feature available. If not provided, a default output dimension will be used."""
 
@@ -47,8 +50,13 @@ class EmbeddingRequest(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["output_dimension", "output_dtype", "encoding_format"]
-        nullable_fields = ["output_dimension"]
+        optional_fields = [
+            "metadata",
+            "output_dimension",
+            "output_dtype",
+            "encoding_format",
+        ]
+        nullable_fields = ["metadata", "output_dimension"]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/jobs_api_routes_batch_get_batch_jobop.py

@@ -1,16 +1,53 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from mistralai.types import BaseModel
-from mistralai.utils import FieldMetadata, PathParamMetadata
-from typing_extensions import Annotated, TypedDict
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from mistralai.utils import FieldMetadata, PathParamMetadata, QueryParamMetadata
+from pydantic import model_serializer
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class JobsAPIRoutesBatchGetBatchJobRequestTypedDict(TypedDict):
     job_id: str
+    inline: NotRequired[Nullable[bool]]
 
 
 class JobsAPIRoutesBatchGetBatchJobRequest(BaseModel):
     job_id: Annotated[
         str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
     ]
+
+    inline: Annotated[
+        OptionalNullable[bool],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["inline"]
+        nullable_fields = ["inline"]
+        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/toolfilechunk.py

@@ -4,15 +4,22 @@ from __future__ import annotations
 from .builtinconnectors import BuiltInConnectors
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Literal, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
 ToolFileChunkType = Literal["tool_file"]
 
+ToolFileChunkToolTypedDict = TypeAliasType(
+    "ToolFileChunkToolTypedDict", Union[BuiltInConnectors, str]
+)
+
+
+ToolFileChunkTool = TypeAliasType("ToolFileChunkTool", Union[BuiltInConnectors, str])
+
 
 class ToolFileChunkTypedDict(TypedDict):
-    tool: BuiltInConnectors
+    tool: ToolFileChunkToolTypedDict
     file_id: str
     type: NotRequired[ToolFileChunkType]
     file_name: NotRequired[Nullable[str]]
@@ -20,7 +27,7 @@ class ToolFileChunkTypedDict(TypedDict):
 
 
 class ToolFileChunk(BaseModel):
-    tool: BuiltInConnectors
+    tool: ToolFileChunkTool
 
     file_id: str
 

mistralai/models/toolreferencechunk.py

@@ -4,15 +4,24 @@ from __future__ import annotations
 from .builtinconnectors import BuiltInConnectors
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Literal, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
 ToolReferenceChunkType = Literal["tool_reference"]
 
+ToolReferenceChunkToolTypedDict = TypeAliasType(
+    "ToolReferenceChunkToolTypedDict", Union[BuiltInConnectors, str]
+)
+
+
+ToolReferenceChunkTool = TypeAliasType(
+    "ToolReferenceChunkTool", Union[BuiltInConnectors, str]
+)
+
 
 class ToolReferenceChunkTypedDict(TypedDict):
-    tool: BuiltInConnectors
+    tool: ToolReferenceChunkToolTypedDict
     title: str
     type: NotRequired[ToolReferenceChunkType]
     url: NotRequired[Nullable[str]]
@@ -21,7 +30,7 @@ class ToolReferenceChunkTypedDict(TypedDict):
 
 
 class ToolReferenceChunk(BaseModel):
-    tool: BuiltInConnectors
+    tool: ToolReferenceChunkTool
 
     title: str