mistralai 1.8.0__py3-none-any.whl → 1.8.2__py3-none-any.whl

mistralai/_version.py

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

mistralai/agents.py

@@ -47,6 +47,7 @@ class Agents(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -68,6 +69,7 @@ class Agents(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode:
         :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
@@ -105,6 +107,7 @@ class Agents(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             agent_id=agent_id,
         )
 
@@ -213,6 +216,7 @@ class Agents(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -234,6 +238,7 @@ class Agents(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode:
         :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
@@ -271,6 +276,7 @@ class Agents(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             agent_id=agent_id,
         )
 
@@ -379,6 +385,7 @@ class Agents(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -402,6 +409,7 @@ class Agents(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode:
         :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
@@ -439,6 +447,7 @@ class Agents(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             agent_id=agent_id,
         )
 
@@ -553,6 +562,7 @@ class Agents(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -576,6 +586,7 @@ class Agents(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode:
         :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
@@ -613,6 +624,7 @@ class Agents(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             agent_id=agent_id,
         )
 

mistralai/beta.py

@@ -8,7 +8,9 @@ from mistralai.mistral_agents import MistralAgents
 
 class Beta(BaseSDK):
     conversations: Conversations
+    r"""(beta) Conversations API"""
     agents: MistralAgents
+    r"""(beta) Agents API"""
 
     def __init__(self, sdk_config: SDKConfiguration) -> None:
         BaseSDK.__init__(self, sdk_config)

mistralai/chat.py

@@ -123,6 +123,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -147,6 +148,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode:
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -186,6 +188,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -288,6 +291,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -312,6 +316,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode:
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -351,6 +356,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -461,6 +467,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -487,6 +494,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode:
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -528,6 +536,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -644,6 +653,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -670,6 +680,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode:
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -711,6 +722,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 

mistralai/conversations.py

@@ -34,8 +34,9 @@ if typing.TYPE_CHECKING:
 # endregion imports
 
 
-
 class Conversations(BaseSDK):
+    r"""(beta) Conversations API"""
+
     # region sdk-class-body
     # Custom run code allowing client side execution of code
 
@@ -146,7 +147,9 @@ class Conversations(BaseSDK):
             completion_args=completion_args,
         )
 
-        async def run_generator() -> AsyncGenerator[Union[RunResultEvents, RunResult], None]:
+        async def run_generator() -> (
+            AsyncGenerator[Union[RunResultEvents, RunResult], None]
+        ):
             current_entries = input_entries
             while True:
                 received_event_tracker: defaultdict[

mistralai/embeddings.py

@@ -18,6 +18,8 @@ class Embeddings(BaseSDK):
         inputs: Union[
             models.EmbeddingRequestInputs, models.EmbeddingRequestInputsTypedDict
         ],
+        output_dimension: OptionalNullable[int] = UNSET,
+        output_dtype: Optional[models.EmbeddingDtype] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -29,6 +31,8 @@ class Embeddings(BaseSDK):
 
         :param model: ID of the model to use.
         :param inputs: Text to embed.
+        :param output_dimension: The dimension of the output embeddings.
+        :param output_dtype:
         :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
@@ -47,6 +51,8 @@ class Embeddings(BaseSDK):
         request = models.EmbeddingRequest(
             model=model,
             inputs=inputs,
+            output_dimension=output_dimension,
+            output_dtype=output_dtype,
         )
 
         req = self._build_request(
@@ -125,6 +131,8 @@ class Embeddings(BaseSDK):
         inputs: Union[
             models.EmbeddingRequestInputs, models.EmbeddingRequestInputsTypedDict
         ],
+        output_dimension: OptionalNullable[int] = UNSET,
+        output_dtype: Optional[models.EmbeddingDtype] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -136,6 +144,8 @@ class Embeddings(BaseSDK):
 
         :param model: ID of the model to use.
         :param inputs: Text to embed.
+        :param output_dimension: The dimension of the output embeddings.
+        :param output_dtype:
         :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
@@ -154,6 +164,8 @@ class Embeddings(BaseSDK):
         request = models.EmbeddingRequest(
             model=model,
             inputs=inputs,
+            output_dimension=output_dimension,
+            output_dtype=output_dtype,
         )
 
         req = self._build_request_async(

mistralai/mistral_agents.py

@@ -9,6 +9,8 @@ from typing import Any, List, Mapping, Optional, Union
 
 
 class MistralAgents(BaseSDK):
+    r"""(beta) Agents API"""
+
     def create(
         self,
         *,

mistralai/models/__init__.py

@@ -347,6 +347,7 @@ from .documenturlchunk import (
     DocumentURLChunkType,
     DocumentURLChunkTypedDict,
 )
+from .embeddingdtype import EmbeddingDtype
 from .embeddingrequest import (
     EmbeddingRequest,
     EmbeddingRequestInputs,
@@ -590,6 +591,7 @@ from .messageoutputevent import (
     MessageOutputEventTypedDict,
 )
 from .metricout import MetricOut, MetricOutTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .modelcapabilities import ModelCapabilities, ModelCapabilitiesTypedDict
 from .modelconversation import (
     ModelConversation,
@@ -964,6 +966,7 @@ __all__ = [
     "DocumentURLChunk",
     "DocumentURLChunkType",
     "DocumentURLChunkTypedDict",
+    "EmbeddingDtype",
     "EmbeddingRequest",
     "EmbeddingRequestInputs",
     "EmbeddingRequestInputsTypedDict",
@@ -1152,6 +1155,7 @@ __all__ = [
     "MessagesTypedDict",
     "MetricOut",
     "MetricOutTypedDict",
+    "MistralPromptMode",
     "ModelCapabilities",
     "ModelCapabilitiesTypedDict",
     "ModelConversation",

mistralai/models/agentscompletionrequest.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .prediction import Prediction, PredictionTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
@@ -11,8 +12,9 @@ from .toolchoiceenum import ToolChoiceEnum
 from .toolmessage import ToolMessage, ToolMessageTypedDict
 from .usermessage import UserMessage, UserMessageTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import get_discriminator
+from mistralai.utils import get_discriminator, validate_open_enum
 from pydantic import Discriminator, Tag, model_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -86,6 +88,7 @@ class AgentsCompletionRequestTypedDict(TypedDict):
     r"""Number of completions to return for each request, input tokens are only billed once."""
     prediction: NotRequired[PredictionTypedDict]
     parallel_tool_calls: NotRequired[bool]
+    prompt_mode: NotRequired[Nullable[MistralPromptMode]]
 
 
 class AgentsCompletionRequest(BaseModel):
@@ -126,6 +129,10 @@ class AgentsCompletionRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
+    prompt_mode: Annotated[
+        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
+    ] = UNSET
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = [
@@ -141,8 +148,9 @@ class AgentsCompletionRequest(BaseModel):
             "n",
             "prediction",
             "parallel_tool_calls",
+            "prompt_mode",
         ]
-        nullable_fields = ["max_tokens", "random_seed", "tools", "n"]
+        nullable_fields = ["max_tokens", "random_seed", "tools", "n", "prompt_mode"]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/agentscompletionstreamrequest.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .prediction import Prediction, PredictionTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
@@ -11,8 +12,9 @@ from .toolchoiceenum import ToolChoiceEnum
 from .toolmessage import ToolMessage, ToolMessageTypedDict
 from .usermessage import UserMessage, UserMessageTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import get_discriminator
+from mistralai.utils import get_discriminator, validate_open_enum
 from pydantic import Discriminator, Tag, model_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -85,6 +87,7 @@ class AgentsCompletionStreamRequestTypedDict(TypedDict):
     r"""Number of completions to return for each request, input tokens are only billed once."""
     prediction: NotRequired[PredictionTypedDict]
     parallel_tool_calls: NotRequired[bool]
+    prompt_mode: NotRequired[Nullable[MistralPromptMode]]
 
 
 class AgentsCompletionStreamRequest(BaseModel):
@@ -124,6 +127,10 @@ class AgentsCompletionStreamRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
+    prompt_mode: Annotated[
+        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
+    ] = UNSET
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = [
@@ -139,8 +146,9 @@ class AgentsCompletionStreamRequest(BaseModel):
             "n",
             "prediction",
             "parallel_tool_calls",
+            "prompt_mode",
         ]
-        nullable_fields = ["max_tokens", "random_seed", "tools", "n"]
+        nullable_fields = ["max_tokens", "random_seed", "tools", "n", "prompt_mode"]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/chatcompletionrequest.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .prediction import Prediction, PredictionTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
@@ -11,8 +12,9 @@ from .toolchoiceenum import ToolChoiceEnum
 from .toolmessage import ToolMessage, ToolMessageTypedDict
 from .usermessage import UserMessage, UserMessageTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import get_discriminator
+from mistralai.utils import get_discriminator, validate_open_enum
 from pydantic import Discriminator, Tag, model_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -86,6 +88,7 @@ class ChatCompletionRequestTypedDict(TypedDict):
     r"""Number of completions to return for each request, input tokens are only billed once."""
     prediction: NotRequired[PredictionTypedDict]
     parallel_tool_calls: NotRequired[bool]
+    prompt_mode: NotRequired[Nullable[MistralPromptMode]]
     safe_prompt: NotRequired[bool]
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -134,6 +137,10 @@ class ChatCompletionRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
+    prompt_mode: Annotated[
+        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
+    ] = UNSET
+
     safe_prompt: Optional[bool] = None
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -154,9 +161,17 @@ class ChatCompletionRequest(BaseModel):
             "n",
             "prediction",
             "parallel_tool_calls",
+            "prompt_mode",
             "safe_prompt",
         ]
-        nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]
+        nullable_fields = [
+            "temperature",
+            "max_tokens",
+            "random_seed",
+            "tools",
+            "n",
+            "prompt_mode",
+        ]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/chatcompletionresponse.py

@@ -4,8 +4,8 @@ from __future__ import annotations
 from .chatcompletionchoice import ChatCompletionChoice, ChatCompletionChoiceTypedDict
 from .usageinfo import UsageInfo, UsageInfoTypedDict
 from mistralai.types import BaseModel
-from typing import List, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import List
+from typing_extensions import TypedDict
 
 
 class ChatCompletionResponseTypedDict(TypedDict):
@@ -13,8 +13,8 @@ class ChatCompletionResponseTypedDict(TypedDict):
     object: str
     model: str
     usage: UsageInfoTypedDict
-    created: NotRequired[int]
-    choices: NotRequired[List[ChatCompletionChoiceTypedDict]]
+    created: int
+    choices: List[ChatCompletionChoiceTypedDict]
 
 
 class ChatCompletionResponse(BaseModel):
@@ -26,6 +26,6 @@ class ChatCompletionResponse(BaseModel):
 
     usage: UsageInfo
 
-    created: Optional[int] = None
+    created: int
 
-    choices: Optional[List[ChatCompletionChoice]] = None
+    choices: List[ChatCompletionChoice]

mistralai/models/chatcompletionstreamrequest.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .prediction import Prediction, PredictionTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
@@ -11,8 +12,9 @@ from .toolchoiceenum import ToolChoiceEnum
 from .toolmessage import ToolMessage, ToolMessageTypedDict
 from .usermessage import UserMessage, UserMessageTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import get_discriminator
+from mistralai.utils import get_discriminator, validate_open_enum
 from pydantic import Discriminator, Tag, model_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -89,6 +91,7 @@ class ChatCompletionStreamRequestTypedDict(TypedDict):
     r"""Number of completions to return for each request, input tokens are only billed once."""
     prediction: NotRequired[PredictionTypedDict]
     parallel_tool_calls: NotRequired[bool]
+    prompt_mode: NotRequired[Nullable[MistralPromptMode]]
     safe_prompt: NotRequired[bool]
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -136,6 +139,10 @@ class ChatCompletionStreamRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
+    prompt_mode: Annotated[
+        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
+    ] = UNSET
+
     safe_prompt: Optional[bool] = None
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -156,9 +163,17 @@ class ChatCompletionStreamRequest(BaseModel):
             "n",
             "prediction",
             "parallel_tool_calls",
+            "prompt_mode",
             "safe_prompt",
         ]
-        nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]
+        nullable_fields = [
+            "temperature",
+            "max_tokens",
+            "random_seed",
+            "tools",
+            "n",
+            "prompt_mode",
+        ]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/embeddingdtype.py

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

mistralai/models/embeddingrequest.py

@@ -1,10 +1,12 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from mistralai.types import BaseModel
+from .embeddingdtype import EmbeddingDtype
+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 List, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 EmbeddingRequestInputsTypedDict = TypeAliasType(
@@ -22,6 +24,9 @@ class EmbeddingRequestTypedDict(TypedDict):
     r"""ID of the model to use."""
     inputs: EmbeddingRequestInputsTypedDict
     r"""Text to embed."""
+    output_dimension: NotRequired[Nullable[int]]
+    r"""The dimension of the output embeddings."""
+    output_dtype: NotRequired[EmbeddingDtype]
 
 
 class EmbeddingRequest(BaseModel):
@@ -30,3 +35,38 @@ class EmbeddingRequest(BaseModel):
 
     inputs: Annotated[EmbeddingRequestInputs, pydantic.Field(alias="input")]
     r"""Text to embed."""
+
+    output_dimension: OptionalNullable[int] = UNSET
+    r"""The dimension of the output embeddings."""
+
+    output_dtype: Optional[EmbeddingDtype] = None
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["output_dimension", "output_dtype"]
+        nullable_fields = ["output_dimension"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in 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/fimcompletionresponse.py

@@ -4,8 +4,8 @@ from __future__ import annotations
 from .chatcompletionchoice import ChatCompletionChoice, ChatCompletionChoiceTypedDict
 from .usageinfo import UsageInfo, UsageInfoTypedDict
 from mistralai.types import BaseModel
-from typing import List, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import List
+from typing_extensions import TypedDict
 
 
 class FIMCompletionResponseTypedDict(TypedDict):
@@ -13,8 +13,8 @@ class FIMCompletionResponseTypedDict(TypedDict):
     object: str
     model: str
     usage: UsageInfoTypedDict
-    created: NotRequired[int]
-    choices: NotRequired[List[ChatCompletionChoiceTypedDict]]
+    created: int
+    choices: List[ChatCompletionChoiceTypedDict]
 
 
 class FIMCompletionResponse(BaseModel):
@@ -26,6 +26,6 @@ class FIMCompletionResponse(BaseModel):
 
     usage: UsageInfo
 
-    created: Optional[int] = None
+    created: int
 
-    choices: Optional[List[ChatCompletionChoice]] = None
+    choices: List[ChatCompletionChoice]

mistralai/models/mistralpromptmode.py

@@ -0,0 +1,8 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import UnrecognizedStr
+from typing import Literal, Union
+
+
+MistralPromptMode = Union[Literal["reasoning"], UnrecognizedStr]

{mistralai-1.8.0.dist-info → mistralai-1.8.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mistralai
-Version: 1.8.0
+Version: 1.8.2
 Summary: Python Client SDK for the Mistral AI API.
 Author: Mistral
 Requires-Python: >=3.9

{mistralai-1.8.0.dist-info → mistralai-1.8.2.dist-info}/RECORD

@@ -139,17 +139,17 @@ mistralai/_hooks/deprecation_warning.py,sha256=eyEOf7-o9uqqNWJnufD2RXp3dYrGV4in9
 mistralai/_hooks/registration.py,sha256=ML0W-XbE4WYdJ4eGks_XxF2aLCJTaIWjQATFGzFwvyU,861
 mistralai/_hooks/sdkhooks.py,sha256=s-orhdvnV89TmI3QiPC2LWQtYeM9RrsG1CTll-fYZmQ,2559
 mistralai/_hooks/types.py,sha256=z3AUFDpRJHj2m3h5PklvUeEcGohY0cfph4jL6-nGIzs,2812
-mistralai/_version.py,sha256=2h0W-7sYb3UbVvUO0AVfyy7sRfFO6mJfM7TomiOVI9I,460
-mistralai/agents.py,sha256=o_apyuwiDzxv-U252T84ynAHCb5fn1q7MMXqrZ4oHLo,32652
+mistralai/_version.py,sha256=dj7Yct7sMSL3ozDUgcL3O2a3h43gPQWDexXedEqencU,460
+mistralai/agents.py,sha256=-_Ur2oC2007qM8kXcw1-1gTI5hYjwhPckets08DFi6s,33204
 mistralai/async_client.py,sha256=KUdYxIIqoD6L7vB0EGwUR6lQ0NK5iCTHjnLVR9CVcJY,355
 mistralai/basesdk.py,sha256=GsU5bp8O5fBCl34tKxaYmeYSIIM971eAPeFBBC_BpFo,12191
 mistralai/batch.py,sha256=YN4D0Duwrap9Ysmp_lRpADYp1Znay7THE_z8ERGvDds,501
-mistralai/beta.py,sha256=ZoBpHRinJu_LlNM7q91HC7huXmzKVpgGLL_bRdXjqBg,662
-mistralai/chat.py,sha256=1XVVVvDi726bq6HXCur6-dsmFfzQAEpEWbKT_3sTZ4A,40549
+mistralai/beta.py,sha256=zcRqEzVxBsog2vWbX4X7ZraozAvhNyCo1_oXxLbBuAo,727
+mistralai/chat.py,sha256=D3slKAa_1BlV6G8nHvva--Hfj_zPkmBMLjfECO0mPDU,41101
 mistralai/classifiers.py,sha256=Cbrb6X_eq3-Yz5ZhWkOsFDTGbo3hkgh-vtIEQmU_UdI,33776
 mistralai/client.py,sha256=hrPg-LciKMKiascF0WbRRmqQyCv1lb2yDh6j-aaKVNo,509
-mistralai/conversations.py,sha256=2RTDBKRHPXGtTobMonJUnVWy9zP3i3aVRWE4pcKdF-Y,109256
-mistralai/embeddings.py,sha256=Tq5ZEo6CR0ktQCdxX7mhLy5CfzI5c8mtkuadIVtsWgM,8644
+mistralai/conversations.py,sha256=mLTN8RfsIaykeJYpTOJXHzRNYefPBMgvD4mLceZuN5c,109316
+mistralai/embeddings.py,sha256=SHG4s0yKwCdKhYKGXKdzPXmcD4_WXI9BBnrM7cSI3gw,9258
 mistralai/extra/README.md,sha256=BTS9fy0ijkiUP7ZVoFQ7FVBxHtXIXqucYZyy_ucFjo4,1739
 mistralai/extra/__init__.py,sha256=8DsU_omYYadqcwlmBOoakBwkWKcSohwLmtB8v-jkn2M,392
 mistralai/extra/exceptions.py,sha256=4EEygCfdsniYiroHEFVSVDqerQZkpRG027mlJXvMqns,428
@@ -175,9 +175,9 @@ mistralai/fim.py,sha256=UMx-bFYbaSyANZug8VrCabHsqePdiHoYQO1YMKB2MvY,27935
 mistralai/fine_tuning.py,sha256=UENQqfE054VEsAYxdruV-TBLFIFfO-joXNznH08GUvE,477
 mistralai/httpclient.py,sha256=lC-YQ7q4yiJGKElxBeb3aZnr-4aYxjgEpZ6roeXYlyg,4318
 mistralai/jobs.py,sha256=1DZE14ad348Vg82VHhLRyXhu7SIh8_KgWXc_jP2oFIA,46767
-mistralai/mistral_agents.py,sha256=txjs5YVDV4u5sEHI6WFPcE3KGXb0I2Il3aJS6Eq9iCc,45543
+mistralai/mistral_agents.py,sha256=FrIUGExYpwz3EyMOetLKsD-LtcsoB1xrWR86zp1WwzE,45573
 mistralai/mistral_jobs.py,sha256=EQHFFxFkkx6XvPX-9S8TRZvVSOLUL7z91cg56J8pskQ,31114
-mistralai/models/__init__.py,sha256=3ZtDRtA2WtDv6pOqLkBFkI2M1WmJV7hc4A6ihqudzrw,43330
+mistralai/models/__init__.py,sha256=QNG_8i4IMS1nXu1zRNBCs3jDo80ct4XBaPwJgRkqPrM,43469
 mistralai/models/agent.py,sha256=dx_uXYDNWoENtgK4InGqs1A9eCw1xDphMPCF_Pv43VQ,4185
 mistralai/models/agentconversation.py,sha256=WaPQn3pGiD2c6bNaP7Ya1akVQt54aK7qDtAEnHAjwrk,2086
 mistralai/models/agentcreationrequest.py,sha256=u9WM95ywRmDqnwx0yfEOT1FnJbiPY8lNtju71d3ix4A,3952
@@ -196,8 +196,8 @@ mistralai/models/agents_api_v1_conversations_listop.py,sha256=N7bwRL4VtP3-a9Q-15
 mistralai/models/agents_api_v1_conversations_messagesop.py,sha256=4V1voPx6mGB4Go5uAsYbkOKgHR0HSUYfadDUoFS0sj0,532
 mistralai/models/agents_api_v1_conversations_restart_streamop.py,sha256=7QZ8fEnDPcIRiTqgg68siLZCSPO13_r2-qdQ7bVsxwY,957
 mistralai/models/agents_api_v1_conversations_restartop.py,sha256=QNA8A3ewD3AwxtayUvrCQpqUfqr13-e7-Y9UgrSbp-A,901
-mistralai/models/agentscompletionrequest.py,sha256=gyGoh1KsCGbOpfmaqk9d_hf1CYhWIriH4vaeQoEDfzU,7920
-mistralai/models/agentscompletionstreamrequest.py,sha256=ZI4iFtl6qDJZ5QTIZ7vDIyFQ9n9rqVqN6tJQAdjpQjA,7365
+mistralai/models/agentscompletionrequest.py,sha256=qBBMmyA-Jcmse1zdCtOVARpg1r0weoifJ4tjaMcSwG4,8277
+mistralai/models/agentscompletionstreamrequest.py,sha256=E4xi4062TUk_vE2PW1R_UPPfKfitHWPHAO6vYSU0vg4,7722
 mistralai/models/agentupdaterequest.py,sha256=w3K74yFRqilUDc64bS6ulmnvgY8WMJ0evSbzwu3ukLc,4094
 mistralai/models/apiendpoint.py,sha256=Hvar5leWsJR_FYb0UzRlSw3vRdBZhk_6BR5r2pIb214,400
 mistralai/models/archiveftmodelout.py,sha256=VdppiqIB9JGNB2B0-Y6XQfQgDmB-hOa1Bta3v_StbLs,565
@@ -211,9 +211,9 @@ mistralai/models/batchjobstatus.py,sha256=WlrIl5vWQGfLmgQA91_9CnCMKhWN6Lli458fT-
 mistralai/models/builtinconnectors.py,sha256=cX1M7Q_2tsWeuH-lKWomXED7xN7Du6BJKvYpep1vD30,284
 mistralai/models/chatclassificationrequest.py,sha256=PmU036oOlGqfd75hNESDUJiN4uJNYguACoCt6CzBC2M,534
 mistralai/models/chatcompletionchoice.py,sha256=6iIFLZj2KYx0HFfzS3-E3sNXG6mPEAlDyXxIA5iZI_U,849
-mistralai/models/chatcompletionrequest.py,sha256=6Innwpi7UnKmyauATOJForAVvW0tkSnbjsiQOOp5OKg,9777
-mistralai/models/chatcompletionresponse.py,sha256=sLE-_Bx9W5rH2-HE2fBWPVbJbmBWx_jSY2mJ3KBEn6w,792
-mistralai/models/chatcompletionstreamrequest.py,sha256=0NFa_nMMRmHU66Hsgu1Zm4fggT0AzvW_imrkyZ4sUxc,9465
+mistralai/models/chatcompletionrequest.py,sha256=NgQ_15kcNqLP87TT6nT6_wM_LdKgmewM6VHLnqn0-D4,10217
+mistralai/models/chatcompletionresponse.py,sha256=px0hjCapAtTP50u36hiQdPcC9X6LU81Nq5aJ3AlofjM,709
+mistralai/models/chatcompletionstreamrequest.py,sha256=TKjp0xk9ReNMnT_KVeZ62c_f7n5VRTSeqb0JyY-r5f8,9905
 mistralai/models/chatmoderationrequest.py,sha256=x1eAoxx_GhaxqGRe4wsqNaUi59K39HQakkedLJVUVD8,2236
 mistralai/models/checkpointout.py,sha256=A2kXS8-VT_1lbg3brifVjZD6tXdsET8vLqBm2a-yXgA,1109
 mistralai/models/classificationrequest.py,sha256=FqQfSrGYwLUjVw78Ft7tbmhAkUN0FqolCn4MNArOuR8,922
@@ -256,7 +256,8 @@ mistralai/models/deletemodelout.py,sha256=W_crO0WtksoKUgq5s9Yh8zS8RxSuyKYQCBt1i8
 mistralai/models/deltamessage.py,sha256=7NtvEjdmBOl86rwOx7x2fcCCJSzIF8K6-eu-G9Wr9PI,1939
 mistralai/models/documentlibrarytool.py,sha256=EN50sX8wgfAw7mF6W8BkOwKyqRvEzGvHgINn-evuCcg,654
 mistralai/models/documenturlchunk.py,sha256=j3JB_Cy1eIRY7fTJe8AvQrdrLEA6xsJcM1l9_a1Sh68,1704
-mistralai/models/embeddingrequest.py,sha256=5GUp8OUrIAcoSdhJLO8Ue45_kGuzVaCqz85g0ZQx3gM,864
+mistralai/models/embeddingdtype.py,sha256=c7L-PKhBgPVPZeMGuMub0ZOs0MdxMbpW2ebE0t7oEpU,209
+mistralai/models/embeddingrequest.py,sha256=BE3L0ImjkdmzwOSUQnC_vCsnvyeQqWmqRw4xgzX6vJc,2258
 mistralai/models/embeddingresponse.py,sha256=te6E_LYEzRjHJ9QREmsFp5PeNP2J_8ALVjyb1T20pNA,663
 mistralai/models/embeddingresponsedata.py,sha256=fJ3mrZqyBBBE40a6iegOJX3DVDfgyMRq23ByeGSTLFk,534
 mistralai/models/eventout.py,sha256=TouRJeISBLphMTPHfgSOpuoOmbGDVohPOrdgHyExMpw,1633
@@ -270,7 +271,7 @@ mistralai/models/files_api_routes_upload_fileop.py,sha256=gIGH5xcPryWYkj1FmNv_0-
 mistralai/models/fileschema.py,sha256=n_IjCdNOrC2fuzkv75wJn01XvqGTmPK3JqAFSHaOiMA,2597
 mistralai/models/filesignedurl.py,sha256=VwvuhzhJulAB99Qxz6zr-2F1aINosAfaSxU0IhytDSU,289
 mistralai/models/fimcompletionrequest.py,sha256=wWDCkQ_PMnjB8DrIuIvVJlPGqQtTpVDHt4p7xJ204Ug,6565
-mistralai/models/fimcompletionresponse.py,sha256=_QwzRuL3KuKkyrA4Fxp366SW0H0EzOA7f4FLkWLm-PM,790
+mistralai/models/fimcompletionresponse.py,sha256=qNgb2WFVgkaW7Isfkk2Aol1gTV9UkhQomcDAhoPDeYw,707
 mistralai/models/fimcompletionstreamrequest.py,sha256=fxuR8FDOWMwIqlYU9ttAfGeRdVgTz4l2k26_OEfxelg,5944
 mistralai/models/finetuneablemodeltype.py,sha256=XmTpXeQU8AINnn1kVmXldFUauCaEnRtJNFAXUTVb6RQ,197
 mistralai/models/ftclassifierlossfunction.py,sha256=ApQB8ssAh2yE26-CljxPO7Jc5lxq3OoBPR4rUp-Td9U,203
@@ -317,6 +318,7 @@ mistralai/models/messageoutputcontentchunks.py,sha256=LRvAb-Hn0XSKBBRrBdwW7CtMC_
 mistralai/models/messageoutputentry.py,sha256=Ow-V0HXFMEowBed_T09281m_ysK4Q8jvWYqAHYxVGqI,2936
 mistralai/models/messageoutputevent.py,sha256=64MdFYY92WOWofmRm6amp2yO6W7q-dv3uKZ1npr8ONo,2709
 mistralai/models/metricout.py,sha256=dXQMMU4Nk6-Zr06Jx1TWilFi6cOwiVLjSanCFn0cPxo,2034
+mistralai/models/mistralpromptmode.py,sha256=v0UKuu6N0kcM_gjy3C7pVUWBs9tuMKtbHG6nLF9jtoI,253
 mistralai/models/modelcapabilities.py,sha256=No-Dl09zT1sG4MxsWnx4s8Yo1tUeMQ7k-HR_iQFIMFc,703
 mistralai/models/modelconversation.py,sha256=j8bO1HyDRHp4Dp_Bo68jA11z90CIhdt3L8DjfCscuDo,4437
 mistralai/models/modellist.py,sha256=D4Y784kQkx0ARhofFrpEqGLfxa-jTY8ev0TQMrD_n8I,995
@@ -390,7 +392,7 @@ mistralai/utils/serializers.py,sha256=EGH40Pgp3sSK9uM4PxL7_SYzSHtmo-Uy6QIE5xLVg6
 mistralai/utils/url.py,sha256=BgGPgcTA6MRK4bF8fjP2dUopN3NzEzxWMXPBVg8NQUA,5254
 mistralai/utils/values.py,sha256=CcaCXEa3xHhkUDROyXZocN8f0bdITftv9Y0P9lTf0YM,3517
 mistralai/version.py,sha256=iosXhlXclBwBqlADFKEilxAC2wWKbtuBKi87AmPi7s8,196
-mistralai-1.8.0.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
-mistralai-1.8.0.dist-info/METADATA,sha256=dL-LRZiJrDkGgzpt53FZDtT7zka1PQIEIWGX4sD_Ezo,33379
-mistralai-1.8.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-mistralai-1.8.0.dist-info/RECORD,,
+mistralai-1.8.2.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
+mistralai-1.8.2.dist-info/METADATA,sha256=JmRYmUZoWAa0p5iWdzL3E52JX3-WT9l-E6j2cXkmycA,33379
+mistralai-1.8.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+mistralai-1.8.2.dist-info/RECORD,,