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

mistralai/models/transcriptionsegmentchunk.py

@@ -1,20 +1,22 @@
 """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 pydantic import ConfigDict
+from pydantic import ConfigDict, model_serializer
 from typing import Any, Dict, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-Type = Literal["transcription_segment"]
+Type = Literal["transcription_segment",]
 
 
 class TranscriptionSegmentChunkTypedDict(TypedDict):
     text: str
     start: float
     end: float
+    score: NotRequired[Nullable[float]]
+    speaker_id: NotRequired[Nullable[str]]
     type: NotRequired[Type]
 
 
@@ -30,6 +32,10 @@ class TranscriptionSegmentChunk(BaseModel):
 
     end: float
 
+    score: OptionalNullable[float] = UNSET
+
+    speaker_id: OptionalNullable[str] = UNSET
+
     type: Optional[Type] = "transcription_segment"
 
     @property
@@ -39,3 +45,36 @@ class TranscriptionSegmentChunk(BaseModel):
     @additional_properties.setter
     def additional_properties(self, value):
         self.__pydantic_extra__ = value  # pyright: ignore[reportIncompatibleVariableOverride]
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["score", "speaker_id", "type"]
+        nullable_fields = ["score", "speaker_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
+
+        for k, v in serialized.items():
+            m[k] = v
+
+        return m

mistralai/models/transcriptionstreamdone.py

@@ -13,7 +13,7 @@ from typing import Any, Dict, List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-TranscriptionStreamDoneType = Literal["transcription.done"]
+TranscriptionStreamDoneType = Literal["transcription.done",]
 
 
 class TranscriptionStreamDoneTypedDict(TypedDict):

mistralai/models/transcriptionstreamlanguage.py

@@ -8,7 +8,7 @@ from typing import Any, Dict, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-TranscriptionStreamLanguageType = Literal["transcription.language"]
+TranscriptionStreamLanguageType = Literal["transcription.language",]
 
 
 class TranscriptionStreamLanguageTypedDict(TypedDict):

mistralai/models/transcriptionstreamsegmentdelta.py

@@ -1,20 +1,21 @@
 """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 pydantic import ConfigDict
+from pydantic import ConfigDict, model_serializer
 from typing import Any, Dict, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-TranscriptionStreamSegmentDeltaType = Literal["transcription.segment"]
+TranscriptionStreamSegmentDeltaType = Literal["transcription.segment",]
 
 
 class TranscriptionStreamSegmentDeltaTypedDict(TypedDict):
     text: str
     start: float
     end: float
+    speaker_id: NotRequired[Nullable[str]]
     type: NotRequired[TranscriptionStreamSegmentDeltaType]
 
 
@@ -30,6 +31,8 @@ class TranscriptionStreamSegmentDelta(BaseModel):
 
     end: float
 
+    speaker_id: OptionalNullable[str] = UNSET
+
     type: Optional[TranscriptionStreamSegmentDeltaType] = "transcription.segment"
 
     @property
@@ -39,3 +42,36 @@ class TranscriptionStreamSegmentDelta(BaseModel):
     @additional_properties.setter
     def additional_properties(self, value):
         self.__pydantic_extra__ = value  # pyright: ignore[reportIncompatibleVariableOverride]
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["speaker_id", "type"]
+        nullable_fields = ["speaker_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
+
+        for k, v in serialized.items():
+            m[k] = v
+
+        return m

mistralai/models/transcriptionstreamtextdelta.py

@@ -8,7 +8,7 @@ from typing import Any, Dict, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-TranscriptionStreamTextDeltaType = Literal["transcription.text.delta"]
+TranscriptionStreamTextDeltaType = Literal["transcription.text.delta",]
 
 
 class TranscriptionStreamTextDeltaTypedDict(TypedDict):

mistralai/models/unarchiveftmodelout.py

@@ -6,7 +6,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-UnarchiveFTModelOutObject = Literal["model"]
+UnarchiveFTModelOutObject = Literal["model",]
 
 
 class UnarchiveFTModelOutTypedDict(TypedDict):

mistralai/models/uploadfileout.py

@@ -5,10 +5,8 @@ from .filepurpose import FilePurpose
 from .sampletype import SampleType
 from .source import Source
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import validate_open_enum
 import pydantic
 from pydantic import model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -47,11 +45,11 @@ class UploadFileOut(BaseModel):
     filename: str
     r"""The name of the uploaded file."""
 
-    purpose: Annotated[FilePurpose, PlainValidator(validate_open_enum(False))]
+    purpose: FilePurpose
 
-    sample_type: Annotated[SampleType, PlainValidator(validate_open_enum(False))]
+    sample_type: SampleType
 
-    source: Annotated[Source, PlainValidator(validate_open_enum(False))]
+    source: Source
 
     num_lines: OptionalNullable[int] = UNSET
 

mistralai/models/usermessage.py

@@ -16,7 +16,7 @@ UserMessageContentTypedDict = TypeAliasType(
 UserMessageContent = TypeAliasType("UserMessageContent", Union[str, List[ContentChunk]])
 
 
-UserMessageRole = Literal["user"]
+UserMessageRole = Literal["user",]
 
 
 class UserMessageTypedDict(TypedDict):

mistralai/models/wandbintegration.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-WandbIntegrationType = Literal["wandb"]
+WandbIntegrationType = Literal["wandb",]
 
 
 class WandbIntegrationTypedDict(TypedDict):

mistralai/models/wandbintegrationout.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-WandbIntegrationOutType = Literal["wandb"]
+WandbIntegrationOutType = Literal["wandb",]
 
 
 class WandbIntegrationOutTypedDict(TypedDict):

mistralai/models/websearchpremiumtool.py

@@ -6,7 +6,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-WebSearchPremiumToolType = Literal["web_search_premium"]
+WebSearchPremiumToolType = Literal["web_search_premium",]
 
 
 class WebSearchPremiumToolTypedDict(TypedDict):

mistralai/models/websearchtool.py

@@ -6,7 +6,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-WebSearchToolType = Literal["web_search"]
+WebSearchToolType = Literal["web_search",]
 
 
 class WebSearchToolTypedDict(TypedDict):

mistralai/models_.py

@@ -51,6 +51,7 @@ class Models(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -67,7 +68,7 @@ class Models(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="list_models_v1_models_get",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -127,6 +128,7 @@ class Models(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -143,7 +145,7 @@ class Models(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="list_models_v1_models_get",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -210,6 +212,7 @@ class Models(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -226,7 +229,7 @@ class Models(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="retrieve_model_v1_models__model_id__get",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -302,6 +305,7 @@ class Models(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -318,7 +322,7 @@ class Models(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="retrieve_model_v1_models__model_id__get",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -394,6 +398,7 @@ class Models(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -410,7 +415,7 @@ class Models(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete_model_v1_models__model_id__delete",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -483,6 +488,7 @@ class Models(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -499,7 +505,7 @@ class Models(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete_model_v1_models__model_id__delete",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -583,6 +589,7 @@ class Models(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request.update_ft_model_in, False, False, "json", models.UpdateFTModelIn
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -599,7 +606,7 @@ class Models(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_update_fine_tuned_model",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -679,6 +686,7 @@ class Models(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request.update_ft_model_in, False, False, "json", models.UpdateFTModelIn
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -695,7 +703,7 @@ class Models(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_update_fine_tuned_model",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -764,6 +772,7 @@ class Models(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -780,7 +789,7 @@ class Models(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_archive_fine_tuned_model",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -847,6 +856,7 @@ class Models(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -863,7 +873,7 @@ class Models(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_archive_fine_tuned_model",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -930,6 +940,7 @@ class Models(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -946,7 +957,7 @@ class Models(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_unarchive_fine_tuned_model",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -1013,6 +1024,7 @@ class Models(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -1029,7 +1041,7 @@ class Models(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_unarchive_fine_tuned_model",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),

mistralai/ocr.py

@@ -3,6 +3,10 @@
 from .basesdk import BaseSDK
 from mistralai import models, utils
 from mistralai._hooks import HookContext
+from mistralai.models import (
+    ocrrequest as models_ocrrequest,
+    responseformat as models_responseformat,
+)
 from mistralai.types import Nullable, OptionalNullable, UNSET
 from mistralai.utils import get_security_from_env
 from mistralai.utils.unmarshal_json_response import unmarshal_json_response
@@ -16,19 +20,28 @@ class Ocr(BaseSDK):
         self,
         *,
         model: Nullable[str],
-        document: Union[models.Document, models.DocumentTypedDict],
+        document: Union[
+            models_ocrrequest.Document, models_ocrrequest.DocumentTypedDict
+        ],
         id: Optional[str] = None,
         pages: OptionalNullable[List[int]] = UNSET,
         include_image_base64: OptionalNullable[bool] = UNSET,
         image_limit: OptionalNullable[int] = UNSET,
         image_min_size: OptionalNullable[int] = UNSET,
         bbox_annotation_format: OptionalNullable[
-            Union[models.ResponseFormat, models.ResponseFormatTypedDict]
+            Union[
+                models_responseformat.ResponseFormat,
+                models_responseformat.ResponseFormatTypedDict,
+            ]
         ] = UNSET,
         document_annotation_format: OptionalNullable[
-            Union[models.ResponseFormat, models.ResponseFormatTypedDict]
+            Union[
+                models_responseformat.ResponseFormat,
+                models_responseformat.ResponseFormatTypedDict,
+            ]
         ] = UNSET,
-        table_format: OptionalNullable[models.TableFormat] = UNSET,
+        document_annotation_prompt: OptionalNullable[str] = UNSET,
+        table_format: OptionalNullable[models_ocrrequest.TableFormat] = UNSET,
         extract_header: Optional[bool] = None,
         extract_footer: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
@@ -47,6 +60,7 @@ class Ocr(BaseSDK):
         :param image_min_size: Minimum height and width of image to extract
         :param bbox_annotation_format: Structured output class for extracting useful information from each extracted bounding box / image from document. Only json_schema is valid for this field
         :param document_annotation_format: Structured output class for extracting useful information from the entire document. Only json_schema is valid for this field
+        :param document_annotation_prompt: Optional prompt to guide the model in extracting structured output from the entire document. A document_annotation_format must be provided.
         :param table_format:
         :param extract_header:
         :param extract_footer:
@@ -79,6 +93,7 @@ class Ocr(BaseSDK):
             document_annotation_format=utils.get_pydantic_model(
                 document_annotation_format, OptionalNullable[models.ResponseFormat]
             ),
+            document_annotation_prompt=document_annotation_prompt,
             table_format=table_format,
             extract_header=extract_header,
             extract_footer=extract_footer,
@@ -100,6 +115,7 @@ class Ocr(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.OCRRequest
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -116,7 +132,7 @@ class Ocr(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="ocr_v1_ocr_post",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -147,19 +163,28 @@ class Ocr(BaseSDK):
         self,
         *,
         model: Nullable[str],
-        document: Union[models.Document, models.DocumentTypedDict],
+        document: Union[
+            models_ocrrequest.Document, models_ocrrequest.DocumentTypedDict
+        ],
         id: Optional[str] = None,
         pages: OptionalNullable[List[int]] = UNSET,
         include_image_base64: OptionalNullable[bool] = UNSET,
         image_limit: OptionalNullable[int] = UNSET,
         image_min_size: OptionalNullable[int] = UNSET,
         bbox_annotation_format: OptionalNullable[
-            Union[models.ResponseFormat, models.ResponseFormatTypedDict]
+            Union[
+                models_responseformat.ResponseFormat,
+                models_responseformat.ResponseFormatTypedDict,
+            ]
         ] = UNSET,
         document_annotation_format: OptionalNullable[
-            Union[models.ResponseFormat, models.ResponseFormatTypedDict]
+            Union[
+                models_responseformat.ResponseFormat,
+                models_responseformat.ResponseFormatTypedDict,
+            ]
         ] = UNSET,
-        table_format: OptionalNullable[models.TableFormat] = UNSET,
+        document_annotation_prompt: OptionalNullable[str] = UNSET,
+        table_format: OptionalNullable[models_ocrrequest.TableFormat] = UNSET,
         extract_header: Optional[bool] = None,
         extract_footer: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
@@ -178,6 +203,7 @@ class Ocr(BaseSDK):
         :param image_min_size: Minimum height and width of image to extract
         :param bbox_annotation_format: Structured output class for extracting useful information from each extracted bounding box / image from document. Only json_schema is valid for this field
         :param document_annotation_format: Structured output class for extracting useful information from the entire document. Only json_schema is valid for this field
+        :param document_annotation_prompt: Optional prompt to guide the model in extracting structured output from the entire document. A document_annotation_format must be provided.
         :param table_format:
         :param extract_header:
         :param extract_footer:
@@ -210,6 +236,7 @@ class Ocr(BaseSDK):
             document_annotation_format=utils.get_pydantic_model(
                 document_annotation_format, OptionalNullable[models.ResponseFormat]
             ),
+            document_annotation_prompt=document_annotation_prompt,
             table_format=table_format,
             extract_header=extract_header,
             extract_footer=extract_footer,
@@ -231,6 +258,7 @@ class Ocr(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.OCRRequest
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -247,7 +275,7 @@ class Ocr(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="ocr_v1_ocr_post",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),

mistralai/sdk.py

@@ -92,7 +92,7 @@ class Mistral(BaseSDK):
         """
         client_supplied = True
         if client is None:
-            client = httpx.Client()
+            client = httpx.Client(follow_redirects=True)
             client_supplied = False
 
         assert issubclass(
@@ -101,7 +101,7 @@ class Mistral(BaseSDK):
 
         async_client_supplied = True
         if async_client is None:
-            async_client = httpx.AsyncClient()
+            async_client = httpx.AsyncClient(follow_redirects=True)
             async_client_supplied = False
 
         if debug_logger is None: