mistralai 1.5.2rc1__py3-none-any.whl → 1.7.0__py3-none-any.whl

mistralai_azure/models/chatcompletionrequest.py

@@ -95,6 +95,7 @@ class ChatCompletionRequestTypedDict(TypedDict):
     n: NotRequired[Nullable[int]]
     r"""Number of completions to return for each request, input tokens are only billed once."""
     prediction: NotRequired[PredictionTypedDict]
+    parallel_tool_calls: NotRequired[bool]
     safe_prompt: NotRequired[bool]
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -141,6 +142,8 @@ class ChatCompletionRequest(BaseModel):
 
     prediction: Optional[Prediction] = None
 
+    parallel_tool_calls: Optional[bool] = None
+
     safe_prompt: Optional[bool] = None
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -161,6 +164,7 @@ class ChatCompletionRequest(BaseModel):
             "frequency_penalty",
             "n",
             "prediction",
+            "parallel_tool_calls",
             "safe_prompt",
         ]
         nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]

mistralai_azure/models/chatcompletionstreamrequest.py

@@ -90,6 +90,7 @@ class ChatCompletionStreamRequestTypedDict(TypedDict):
     n: NotRequired[Nullable[int]]
     r"""Number of completions to return for each request, input tokens are only billed once."""
     prediction: NotRequired[PredictionTypedDict]
+    parallel_tool_calls: NotRequired[bool]
     safe_prompt: NotRequired[bool]
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -135,6 +136,8 @@ class ChatCompletionStreamRequest(BaseModel):
 
     prediction: Optional[Prediction] = None
 
+    parallel_tool_calls: Optional[bool] = None
+
     safe_prompt: Optional[bool] = None
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -155,6 +158,7 @@ class ChatCompletionStreamRequest(BaseModel):
             "frequency_penalty",
             "n",
             "prediction",
+            "parallel_tool_calls",
             "safe_prompt",
         ]
         nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]

mistralai_azure/models/contentchunk.py

@@ -1,6 +1,7 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .imageurlchunk import ImageURLChunk, ImageURLChunkTypedDict
 from .referencechunk import ReferenceChunk, ReferenceChunkTypedDict
 from .textchunk import TextChunk, TextChunkTypedDict
 from mistralai_azure.utils import get_discriminator
@@ -10,13 +11,16 @@ from typing_extensions import Annotated, TypeAliasType
 
 
 ContentChunkTypedDict = TypeAliasType(
-    "ContentChunkTypedDict", Union[TextChunkTypedDict, ReferenceChunkTypedDict]
+    "ContentChunkTypedDict",
+    Union[TextChunkTypedDict, ImageURLChunkTypedDict, ReferenceChunkTypedDict],
 )
 
 
 ContentChunk = Annotated[
     Union[
-        Annotated[TextChunk, Tag("text")], Annotated[ReferenceChunk, Tag("reference")]
+        Annotated[ImageURLChunk, Tag("image_url")],
+        Annotated[TextChunk, Tag("text")],
+        Annotated[ReferenceChunk, Tag("reference")],
     ],
     Discriminator(lambda m: get_discriminator(m, "type", "type")),
 ]

mistralai_azure/models/function.py

@@ -18,6 +18,6 @@ class Function(BaseModel):
 
     parameters: Dict[str, Any]
 
-    description: Optional[str] = ""
+    description: Optional[str] = None
 
-    strict: Optional[bool] = False
+    strict: Optional[bool] = None

mistralai_azure/models/imageurl.py

@@ -0,0 +1,53 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai_azure.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
+from pydantic import model_serializer
+from typing_extensions import NotRequired, TypedDict
+
+
+class ImageURLTypedDict(TypedDict):
+    url: str
+    detail: NotRequired[Nullable[str]]
+
+
+class ImageURL(BaseModel):
+    url: str
+
+    detail: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["detail"]
+        nullable_fields = ["detail"]
+        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_azure/models/imageurlchunk.py

@@ -0,0 +1,33 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .imageurl import ImageURL, ImageURLTypedDict
+from mistralai_azure.types import BaseModel
+from typing import Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
+
+
+ImageURLChunkImageURLTypedDict = TypeAliasType(
+    "ImageURLChunkImageURLTypedDict", Union[ImageURLTypedDict, str]
+)
+
+
+ImageURLChunkImageURL = TypeAliasType("ImageURLChunkImageURL", Union[ImageURL, str])
+
+
+ImageURLChunkType = Literal["image_url"]
+
+
+class ImageURLChunkTypedDict(TypedDict):
+    r"""{\"type\":\"image_url\",\"image_url\":{\"url\":\"data:image/png;base64,iVBORw0"""
+
+    image_url: ImageURLChunkImageURLTypedDict
+    type: NotRequired[ImageURLChunkType]
+
+
+class ImageURLChunk(BaseModel):
+    r"""{\"type\":\"image_url\",\"image_url\":{\"url\":\"data:image/png;base64,iVBORw0"""
+
+    image_url: ImageURLChunkImageURL
+
+    type: Optional[ImageURLChunkType] = "image_url"

mistralai_azure/models/jsonschema.py

@@ -28,7 +28,7 @@ class JSONSchema(BaseModel):
 
     description: OptionalNullable[str] = UNSET
 
-    strict: Optional[bool] = False
+    strict: Optional[bool] = None
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):

mistralai_gcp/_version.py

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

mistralai_gcp/chat.py

@@ -40,6 +40,7 @@ class Chat(BaseSDK):
         prediction: Optional[
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
+        parallel_tool_calls: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -64,6 +65,7 @@ class Chat(BaseSDK):
         :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
+        :param parallel_tool_calls:
         :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
@@ -101,6 +103,7 @@ class Chat(BaseSDK):
             prediction=utils.get_pydantic_model(
                 prediction, Optional[models.Prediction]
             ),
+            parallel_tool_calls=parallel_tool_calls,
         )
 
         req = self._build_request(
@@ -205,6 +208,7 @@ class Chat(BaseSDK):
         prediction: Optional[
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
+        parallel_tool_calls: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -229,6 +233,7 @@ class Chat(BaseSDK):
         :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
+        :param parallel_tool_calls:
         :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
@@ -266,6 +271,7 @@ class Chat(BaseSDK):
             prediction=utils.get_pydantic_model(
                 prediction, Optional[models.Prediction]
             ),
+            parallel_tool_calls=parallel_tool_calls,
         )
 
         req = self._build_request_async(
@@ -378,6 +384,7 @@ class Chat(BaseSDK):
         prediction: Optional[
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
+        parallel_tool_calls: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -400,6 +407,7 @@ class Chat(BaseSDK):
         :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
+        :param parallel_tool_calls:
         :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 Chat(BaseSDK):
             prediction=utils.get_pydantic_model(
                 prediction, Optional[models.Prediction]
             ),
+            parallel_tool_calls=parallel_tool_calls,
         )
 
         req = self._build_request(
@@ -547,6 +556,7 @@ class Chat(BaseSDK):
         prediction: Optional[
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
+        parallel_tool_calls: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -569,6 +579,7 @@ class Chat(BaseSDK):
         :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
+        :param parallel_tool_calls:
         :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
@@ -608,6 +619,7 @@ class Chat(BaseSDK):
             prediction=utils.get_pydantic_model(
                 prediction, Optional[models.Prediction]
             ),
+            parallel_tool_calls=parallel_tool_calls,
         )
 
         req = self._build_request_async(

mistralai_gcp/models/__init__.py

@@ -67,6 +67,14 @@ from .functioncall import (
 )
 from .functionname import FunctionName, FunctionNameTypedDict
 from .httpvalidationerror import HTTPValidationError, HTTPValidationErrorData
+from .imageurl import ImageURL, ImageURLTypedDict
+from .imageurlchunk import (
+    ImageURLChunk,
+    ImageURLChunkImageURL,
+    ImageURLChunkImageURLTypedDict,
+    ImageURLChunkType,
+    ImageURLChunkTypedDict,
+)
 from .jsonschema import JSONSchema, JSONSchemaTypedDict
 from .prediction import Prediction, PredictionTypedDict
 from .referencechunk import ReferenceChunk, ReferenceChunkType, ReferenceChunkTypedDict
@@ -166,6 +174,13 @@ __all__ = [
     "FunctionTypedDict",
     "HTTPValidationError",
     "HTTPValidationErrorData",
+    "ImageURL",
+    "ImageURLChunk",
+    "ImageURLChunkImageURL",
+    "ImageURLChunkImageURLTypedDict",
+    "ImageURLChunkType",
+    "ImageURLChunkTypedDict",
+    "ImageURLTypedDict",
     "JSONSchema",
     "JSONSchemaTypedDict",
     "Loc",

mistralai_gcp/models/chatcompletionrequest.py

@@ -95,6 +95,7 @@ class ChatCompletionRequestTypedDict(TypedDict):
     n: NotRequired[Nullable[int]]
     r"""Number of completions to return for each request, input tokens are only billed once."""
     prediction: NotRequired[PredictionTypedDict]
+    parallel_tool_calls: NotRequired[bool]
 
 
 class ChatCompletionRequest(BaseModel):
@@ -139,6 +140,8 @@ class ChatCompletionRequest(BaseModel):
 
     prediction: Optional[Prediction] = None
 
+    parallel_tool_calls: Optional[bool] = None
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = [
@@ -155,6 +158,7 @@ class ChatCompletionRequest(BaseModel):
             "frequency_penalty",
             "n",
             "prediction",
+            "parallel_tool_calls",
         ]
         nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]
         null_default_fields = []

mistralai_gcp/models/chatcompletionstreamrequest.py

@@ -90,6 +90,7 @@ class ChatCompletionStreamRequestTypedDict(TypedDict):
     n: NotRequired[Nullable[int]]
     r"""Number of completions to return for each request, input tokens are only billed once."""
     prediction: NotRequired[PredictionTypedDict]
+    parallel_tool_calls: NotRequired[bool]
 
 
 class ChatCompletionStreamRequest(BaseModel):
@@ -133,6 +134,8 @@ class ChatCompletionStreamRequest(BaseModel):
 
     prediction: Optional[Prediction] = None
 
+    parallel_tool_calls: Optional[bool] = None
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = [
@@ -149,6 +152,7 @@ class ChatCompletionStreamRequest(BaseModel):
             "frequency_penalty",
             "n",
             "prediction",
+            "parallel_tool_calls",
         ]
         nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]
         null_default_fields = []

mistralai_gcp/models/contentchunk.py

@@ -1,6 +1,7 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .imageurlchunk import ImageURLChunk, ImageURLChunkTypedDict
 from .referencechunk import ReferenceChunk, ReferenceChunkTypedDict
 from .textchunk import TextChunk, TextChunkTypedDict
 from mistralai_gcp.utils import get_discriminator
@@ -10,13 +11,16 @@ from typing_extensions import Annotated, TypeAliasType
 
 
 ContentChunkTypedDict = TypeAliasType(
-    "ContentChunkTypedDict", Union[TextChunkTypedDict, ReferenceChunkTypedDict]
+    "ContentChunkTypedDict",
+    Union[TextChunkTypedDict, ImageURLChunkTypedDict, ReferenceChunkTypedDict],
 )
 
 
 ContentChunk = Annotated[
     Union[
-        Annotated[TextChunk, Tag("text")], Annotated[ReferenceChunk, Tag("reference")]
+        Annotated[ImageURLChunk, Tag("image_url")],
+        Annotated[TextChunk, Tag("text")],
+        Annotated[ReferenceChunk, Tag("reference")],
     ],
     Discriminator(lambda m: get_discriminator(m, "type", "type")),
 ]

mistralai_gcp/models/function.py

@@ -18,6 +18,6 @@ class Function(BaseModel):
 
     parameters: Dict[str, Any]
 
-    description: Optional[str] = ""
+    description: Optional[str] = None
 
-    strict: Optional[bool] = False
+    strict: Optional[bool] = None

mistralai_gcp/models/imageurl.py

@@ -0,0 +1,53 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai_gcp.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
+from pydantic import model_serializer
+from typing_extensions import NotRequired, TypedDict
+
+
+class ImageURLTypedDict(TypedDict):
+    url: str
+    detail: NotRequired[Nullable[str]]
+
+
+class ImageURL(BaseModel):
+    url: str
+
+    detail: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["detail"]
+        nullable_fields = ["detail"]
+        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_gcp/models/imageurlchunk.py

@@ -0,0 +1,33 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .imageurl import ImageURL, ImageURLTypedDict
+from mistralai_gcp.types import BaseModel
+from typing import Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
+
+
+ImageURLChunkImageURLTypedDict = TypeAliasType(
+    "ImageURLChunkImageURLTypedDict", Union[ImageURLTypedDict, str]
+)
+
+
+ImageURLChunkImageURL = TypeAliasType("ImageURLChunkImageURL", Union[ImageURL, str])
+
+
+ImageURLChunkType = Literal["image_url"]
+
+
+class ImageURLChunkTypedDict(TypedDict):
+    r"""{\"type\":\"image_url\",\"image_url\":{\"url\":\"data:image/png;base64,iVBORw0"""
+
+    image_url: ImageURLChunkImageURLTypedDict
+    type: NotRequired[ImageURLChunkType]
+
+
+class ImageURLChunk(BaseModel):
+    r"""{\"type\":\"image_url\",\"image_url\":{\"url\":\"data:image/png;base64,iVBORw0"""
+
+    image_url: ImageURLChunkImageURL
+
+    type: Optional[ImageURLChunkType] = "image_url"

mistralai_gcp/models/jsonschema.py

@@ -28,7 +28,7 @@ class JSONSchema(BaseModel):
 
     description: OptionalNullable[str] = UNSET
 
-    strict: Optional[bool] = False
+    strict: Optional[bool] = None
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):