mistralai 1.4.0__py3-none-any.whl → 1.5.1__py3-none-any.whl

mistralai/ocr.py

@@ -0,0 +1,238 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from .basesdk import BaseSDK
+from mistralai import models, utils
+from mistralai._hooks import HookContext
+from mistralai.types import Nullable, OptionalNullable, UNSET
+from mistralai.utils import get_security_from_env
+from typing import Any, List, Mapping, Optional, Union
+
+
+class Ocr(BaseSDK):
+    r"""OCR API"""
+
+    def process(
+        self,
+        *,
+        model: Nullable[str],
+        document: Union[models.Document, models.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,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> models.OCRResponse:
+        r"""OCR
+
+        :param model:
+        :param document: Document to run OCR on
+        :param id:
+        :param pages: Specific pages user wants to process in various formats: single number, range, or list of both. Starts from 0
+        :param include_image_base64: Include image URLs in response
+        :param image_limit: Max images to extract
+        :param image_min_size: Minimum height and width of image to extract
+        :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
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+
+        request = models.OCRRequest(
+            model=model,
+            id=id,
+            document=utils.get_pydantic_model(document, models.Document),
+            pages=pages,
+            include_image_base64=include_image_base64,
+            image_limit=image_limit,
+            image_min_size=image_min_size,
+        )
+
+        req = self._build_request(
+            method="POST",
+            path="/v1/ocr",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=True,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request, False, False, "json", models.OCRRequest
+            ),
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = self.do_request(
+            hook_ctx=HookContext(
+                operation_id="ocr_v1_ocr_post",
+                oauth2_scopes=[],
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["422", "4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        data: Any = None
+        if utils.match_response(http_res, "200", "application/json"):
+            return utils.unmarshal_json(http_res.text, models.OCRResponse)
+        if utils.match_response(http_res, "422", "application/json"):
+            data = utils.unmarshal_json(http_res.text, models.HTTPValidationErrorData)
+            raise models.HTTPValidationError(data=data)
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+
+        content_type = http_res.headers.get("Content-Type")
+        http_res_text = utils.stream_to_text(http_res)
+        raise models.SDKError(
+            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
+            http_res.status_code,
+            http_res_text,
+            http_res,
+        )
+
+    async def process_async(
+        self,
+        *,
+        model: Nullable[str],
+        document: Union[models.Document, models.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,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> models.OCRResponse:
+        r"""OCR
+
+        :param model:
+        :param document: Document to run OCR on
+        :param id:
+        :param pages: Specific pages user wants to process in various formats: single number, range, or list of both. Starts from 0
+        :param include_image_base64: Include image URLs in response
+        :param image_limit: Max images to extract
+        :param image_min_size: Minimum height and width of image to extract
+        :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
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+
+        request = models.OCRRequest(
+            model=model,
+            id=id,
+            document=utils.get_pydantic_model(document, models.Document),
+            pages=pages,
+            include_image_base64=include_image_base64,
+            image_limit=image_limit,
+            image_min_size=image_min_size,
+        )
+
+        req = self._build_request_async(
+            method="POST",
+            path="/v1/ocr",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=True,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request, False, False, "json", models.OCRRequest
+            ),
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = await self.do_request_async(
+            hook_ctx=HookContext(
+                operation_id="ocr_v1_ocr_post",
+                oauth2_scopes=[],
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["422", "4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        data: Any = None
+        if utils.match_response(http_res, "200", "application/json"):
+            return utils.unmarshal_json(http_res.text, models.OCRResponse)
+        if utils.match_response(http_res, "422", "application/json"):
+            data = utils.unmarshal_json(http_res.text, models.HTTPValidationErrorData)
+            raise models.HTTPValidationError(data=data)
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+
+        content_type = http_res.headers.get("Content-Type")
+        http_res_text = await utils.stream_to_text_async(http_res)
+        raise models.SDKError(
+            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
+            http_res.status_code,
+            http_res_text,
+            http_res,
+        )

mistralai/sdk.py

@@ -1,7 +1,7 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from .basesdk import BaseSDK
-from .httpclient import AsyncHttpClient, HttpClient
+from .httpclient import AsyncHttpClient, ClientOwner, HttpClient, close_clients
 from .sdkconfiguration import SDKConfiguration
 from .utils.logger import Logger, get_default_logger
 from .utils.retries import RetryConfig
@@ -17,8 +17,10 @@ from mistralai.files import Files
 from mistralai.fim import Fim
 from mistralai.fine_tuning import FineTuning
 from mistralai.models_ import Models
+from mistralai.ocr import Ocr
 from mistralai.types import OptionalNullable, UNSET
-from typing import Any, Callable, Dict, Optional, Union
+from typing import Any, Callable, Dict, Optional, Union, cast
+import weakref
 
 
 class Mistral(BaseSDK):
@@ -40,6 +42,8 @@ class Mistral(BaseSDK):
     r"""Embeddings API."""
     classifiers: Classifiers
     r"""Classifiers API."""
+    ocr: Ocr
+    r"""OCR API"""
 
     def __init__(
         self,
@@ -118,6 +122,14 @@ class Mistral(BaseSDK):
         # pylint: disable=protected-access
         self.sdk_configuration.__dict__["_hooks"] = hooks
 
+        weakref.finalize(
+            self,
+            close_clients,
+            cast(ClientOwner, self.sdk_configuration),
+            self.sdk_configuration.client,
+            self.sdk_configuration.async_client,
+        )
+
         self._init_sdks()
 
     def _init_sdks(self):
@@ -130,6 +142,7 @@ class Mistral(BaseSDK):
         self.agents = Agents(self.sdk_configuration)
         self.embeddings = Embeddings(self.sdk_configuration)
         self.classifiers = Classifiers(self.sdk_configuration)
+        self.ocr = Ocr(self.sdk_configuration)
 
     def __enter__(self):
         return self

{mistralai-1.4.0.dist-info → mistralai-1.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mistralai
-Version: 1.4.0
+Version: 1.5.1
 Summary: Python Client SDK for the Mistral AI API.
 Author: Mistral
 Requires-Python: >=3.8
@@ -67,6 +67,7 @@ Mistral AI API: Our Chat Completion and Embeddings APIs specification. Create yo
   * [Server Selection](https://github.com/mistralai/client-python/blob/master/#server-selection)
   * [Custom HTTP Client](https://github.com/mistralai/client-python/blob/master/#custom-http-client)
   * [Authentication](https://github.com/mistralai/client-python/blob/master/#authentication)
+  * [Resource Management](https://github.com/mistralai/client-python/blob/master/#resource-management)
   * [Debugging](https://github.com/mistralai/client-python/blob/master/#debugging)
   * [IDE Support](https://github.com/mistralai/client-python/blob/master/#ide-support)
 * [Development](https://github.com/mistralai/client-python/blob/master/#development)
@@ -77,6 +78,11 @@ Mistral AI API: Our Chat Completion and Embeddings APIs specification. Create yo
 <!-- Start SDK Installation [installation] -->
 ## SDK Installation
 
+> [!NOTE]
+> **Python version upgrade policy**
+>
+> Once a Python version reaches its [official end of life date](https://devguide.python.org/versions/), a 3-month grace period is provided for users to upgrade. Following this grace period, the minimum python version supported in the SDK will be updated.
+
 The SDK can be installed with either *pip* or *poetry* package managers.
 
 ### PIP
@@ -461,6 +467,10 @@ The documentation for the GCP SDK is available [here](https://github.com/mistral
 * [archive](https://github.com/mistralai/client-python/blob/master/docs/sdks/models/README.md#archive) - Archive Fine Tuned Model
 * [unarchive](https://github.com/mistralai/client-python/blob/master/docs/sdks/models/README.md#unarchive) - Unarchive Fine Tuned Model
 
+### [ocr](https://github.com/mistralai/client-python/blob/master/docs/sdks/ocr/README.md)
+
+* [process](https://github.com/mistralai/client-python/blob/master/docs/sdks/ocr/README.md#process) - OCR
+
 </details>
 <!-- End Available Resources and Operations [operations] -->
 
@@ -779,6 +789,32 @@ with Mistral(
 ```
 <!-- End Authentication [security] -->
 
+<!-- Start Resource Management [resource-management] -->
+## Resource Management
+
+The `Mistral` class implements the context manager protocol and registers a finalizer function to close the underlying sync and async HTTPX clients it uses under the hood. This will close HTTP connections, release memory and free up other resources held by the SDK. In short-lived Python programs and notebooks that make a few SDK method calls, resource management may not be a concern. However, in longer-lived programs, it is beneficial to create a single SDK instance via a [context manager][context-manager] and reuse it across the application.
+
+[context-manager]: https://docs.python.org/3/reference/datamodel.html#context-managers
+
+```python
+from mistralai import Mistral
+import os
+def main():
+    with Mistral(
+        api_key=os.getenv("MISTRAL_API_KEY", ""),
+    ) as mistral:
+        # Rest of application here...
+
+
+# Or when using async:
+async def amain():
+    async with Mistral(
+        api_key=os.getenv("MISTRAL_API_KEY", ""),
+    ) as mistral:
+        # Rest of application here...
+```
+<!-- End Resource Management [resource-management] -->
+
 <!-- Start Debugging [debug] -->
 ## Debugging
 

{mistralai-1.4.0.dist-info → mistralai-1.5.1.dist-info}/RECORD

@@ -131,52 +131,62 @@ 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=vUkTVk_TSaK10aEj368KYWvnd4T5EsawixMcTro_UHc,2570
-mistralai/_version.py,sha256=Ks6KHk739TqW4iKtkhh7FmwsiwVVsnyyC6K3ETq80ls,462
+mistralai/_version.py,sha256=pBAiPZarkMWJ_MM3cNcwKrR6I5rxAjX8exNV9MXZ99Y,460
 mistralai/agents.py,sha256=63-FJfMJ_9Qb0O0-uorAJM8km4FHpCEjjIk14margM8,31392
 mistralai/async_client.py,sha256=KUdYxIIqoD6L7vB0EGwUR6lQ0NK5iCTHjnLVR9CVcJY,355
 mistralai/basesdk.py,sha256=da0sFeLR-ztU5-fuGJ4TMqUnnqSzXbPAjpNgKI52tBk,11999
 mistralai/batch.py,sha256=YN4D0Duwrap9Ysmp_lRpADYp1Znay7THE_z8ERGvDds,501
-mistralai/chat.py,sha256=wO3VoQYzEbBDly8sNKj1NKxlNMz51LN4E5toul-UKUs,35575
-mistralai/classifiers.py,sha256=oxGQEj6QfIWvV2GIAhfev9DVbFTG79uR5ODKTmOiJxs,16594
+mistralai/chat.py,sha256=TpFLC7pprrJ3rf3pL3RQRbFeruuihu_Vo9sRKLr2koU,39289
+mistralai/classifiers.py,sha256=VpnCqw1H4saN_6-xjT8592lQIs5GNfZ9s5UvcacG5v0,16808
 mistralai/client.py,sha256=hrPg-LciKMKiascF0WbRRmqQyCv1lb2yDh6j-aaKVNo,509
-mistralai/embeddings.py,sha256=oT6SgsC3ODtn9mAWfpHN9Eli6puLmz9dEe-ZrY0QiPA,8590
+mistralai/embeddings.py,sha256=aCjn43G2gzV2GSNap0xmNxAN5_xygC_9QFT4lMbZhAM,8244
+mistralai/extra/README.md,sha256=BTS9fy0ijkiUP7ZVoFQ7FVBxHtXIXqucYZyy_ucFjo4,1739
+mistralai/extra/__init__.py,sha256=MHf0pUgLc9Sb7eTUE31JlE2FKMxfQupmJ_iR8UkgQ9w,360
+mistralai/extra/struct_chat.py,sha256=ZkpdExC5rgC-nBZ44hQIVhQmK6lYMk36RBSFPZMFaIg,2157
+mistralai/extra/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mistralai/extra/tests/test_struct_chat.py,sha256=WT6GGfcbXCok8UkEny19u4q1g2QOgkekvmAb3ZinQZ8,4343
+mistralai/extra/tests/test_utils.py,sha256=VesGDR_IiE6u0iY7yOi1iERd7esdJgi2aL4xZp0vKVI,5113
+mistralai/extra/utils/__init__.py,sha256=SExo5t_hx0ybiQhVJIG3r3hOA-Pfny3lIO_WsqNXlN8,116
+mistralai/extra/utils/_pydantic_helper.py,sha256=kU_HbsSl1qGXnrrHnBcxun2MtHowu8eBp3jYMyFsPWw,859
+mistralai/extra/utils/response_format.py,sha256=OiIpNXMODKJ6U2QDCXxPHBoVtXzXF7jtBzCLmI4t_CU,907
 mistralai/files.py,sha256=E1-MxJc-84IdKrc-0k-bvYNa7OSoNFCQ7wBX9tMnFb8,44359
-mistralai/fim.py,sha256=WFlELwTAT_dT7HcBRuby8xqoQsUsw-IXqzqufyyqz_g,27289
+mistralai/fim.py,sha256=14IVw1I6OkNVHx6g7cgXlwp5ZxuTku_8BNRL8FYntG0,27239
 mistralai/fine_tuning.py,sha256=UENQqfE054VEsAYxdruV-TBLFIFfO-joXNznH08GUvE,477
-mistralai/httpclient.py,sha256=WDbLpMzo7qmWki_ryOJcCAYNI1T4uyWKV08rRuCdNII,2688
+mistralai/httpclient.py,sha256=N-D-srtDBykpfyVKacTY4upDGvNLqdWlEYqhJvta99E,4194
 mistralai/jobs.py,sha256=SVNlvn8XGaCkNJ5tKOKci5QLavacmkNqoYeIGF4ik0Q,43481
 mistralai/mistral_jobs.py,sha256=2ScKd2Tv79-MWxEQkrqr53Ikya8rmTbSiJ96judp7DY,30166
-mistralai/models/__init__.py,sha256=W9QnG5egzQoSpFCy5Oiz9wYHvbmN1EkVvwW173XeLsY,21852
+mistralai/models/__init__.py,sha256=39Owsya5LkbCsZIu_EUeXPq2WWcKVl0Fg1A6Ce-6vxI,22771
 mistralai/models/agentscompletionrequest.py,sha256=2tV6_p33zLCfBD7EzlAVeSjG0E_pknbVZlbQFM3Lulc,7794
 mistralai/models/agentscompletionstreamrequest.py,sha256=D1fla5nnnwNKXwHG1w4XVGnqaEvx6sOuhTXdP_e65sM,7239
 mistralai/models/apiendpoint.py,sha256=Hvar5leWsJR_FYb0UzRlSw3vRdBZhk_6BR5r2pIb214,400
 mistralai/models/archiveftmodelout.py,sha256=dQx1J91UA06pjk2r7okhKMyBBePmHal7SPpn6Y_wEsY,820
-mistralai/models/assistantmessage.py,sha256=BbvwC7W4Zk_8GN3w1QnEj7UygVGl0pfCtr6eosNUAgw,2204
+mistralai/models/assistantmessage.py,sha256=pmOhSINRB8sJ11lNtfKEL0k6-JnTEJ7cjlWW9D0pIMM,2624
 mistralai/models/basemodelcard.py,sha256=nv-xjoZFCuIdjKBl98dVH5puT7qh0AC2MaE7WTsMxfs,2985
 mistralai/models/batcherror.py,sha256=tThkO9B-g-6eDSBCm1Emd-zDI4B3mk2vAl0L1MI3pdQ,390
 mistralai/models/batchjobin.py,sha256=1GDaaHJeGVo71F4HVZkTdX92bmr3DieNB0ZuHFzBIeE,1850
 mistralai/models/batchjobout.py,sha256=AXFSgDFGY_1LFVHtVdp66y08DUZCxsZtZ_NzTAJYDWM,3067
 mistralai/models/batchjobsout.py,sha256=Tq6bcb4_-fcW8C9AOnfJN8_sTy1NQhDn82qFOKdFPcg,868
 mistralai/models/batchjobstatus.py,sha256=WlrIl5vWQGfLmgQA91_9CnCMKhWN6Lli458fT-4Asj4,294
-mistralai/models/chatclassificationrequest.py,sha256=WYthWiE7euFSeJYJRQJMXmxbWzGDl8kpF4yyLO9ZJaE,3225
 mistralai/models/chatcompletionchoice.py,sha256=6iIFLZj2KYx0HFfzS3-E3sNXG6mPEAlDyXxIA5iZI_U,849
-mistralai/models/chatcompletionrequest.py,sha256=MlAYYz5-VK9i81Teb89D_KxqQ34lip0eU3DlHkv05aM,9763
+mistralai/models/chatcompletionrequest.py,sha256=Qm2b3Ohv3hQi4N6bGn53EsdiHDmwK8BL4JFT06R3VT4,9651
 mistralai/models/chatcompletionresponse.py,sha256=sLE-_Bx9W5rH2-HE2fBWPVbJbmBWx_jSY2mJ3KBEn6w,792
-mistralai/models/chatcompletionstreamrequest.py,sha256=3Td59jU_GDoioTjcrx5pteOqiRJxlFF0lshVpRsx7V0,9451
+mistralai/models/chatcompletionstreamrequest.py,sha256=5IPSsJC51NdNdQ3uZrVnHx0eIZUvFEIeJ9CD6T53Bz4,9339
+mistralai/models/chatmoderationrequest.py,sha256=LX-dhlYxecEzChSTt4jo4DA8lC4DEp5brgaiksTGF-o,2367
 mistralai/models/checkpointout.py,sha256=A2kXS8-VT_1lbg3brifVjZD6tXdsET8vLqBm2a-yXgA,1109
 mistralai/models/classificationobject.py,sha256=JqaKo3AQD4t5X12ZnHjJ6K3Y6LXUn94uGdLJSoGr8vY,665
-mistralai/models/classificationrequest.py,sha256=UFbjwNFtLZmylX3DqQSH3FoOvvD1UGBapK6f9b9LGYE,1903
+mistralai/models/classificationrequest.py,sha256=FqQfSrGYwLUjVw78Ft7tbmhAkUN0FqolCn4MNArOuR8,922
 mistralai/models/classificationresponse.py,sha256=TeBV7mH0nvLswPdLhv4ZaDTsXvuVYJqSe5Uci7qVSMA,649
 mistralai/models/completionchunk.py,sha256=Cdq-FBWa1oBhrxapoOAj8qr6xmkeIPsfPQSbjeK6NLY,871
 mistralai/models/completionevent.py,sha256=rFc5dJBRnNOzspI95Jhkjd9WyM476u48cN0T1Vh-Cxw,399
 mistralai/models/completionresponsestreamchoice.py,sha256=gw5-_iOyznuimumDtBV65E3zwUW0KH1OHP55dGCAsAA,1927
-mistralai/models/contentchunk.py,sha256=SR4M8E4u44_r5YNFVTuWmiDHCU3wp4kWBhVj_HRxRGs,875
+mistralai/models/contentchunk.py,sha256=V8-d2u9ReICA6uXZwJTUXu88VfKRIAsLRO6o113Mcw8,1073
 mistralai/models/delete_model_v1_models_model_id_deleteop.py,sha256=lnVRFX-G0jkn1dCFC89sXY2Pj_w4QfMDeF1tPjS4hWE,602
 mistralai/models/deletefileout.py,sha256=s3a-H2RgFQ9HX0kYSmP6GwmwE1jghz7dBj-3G0NBVSY,587
 mistralai/models/deletemodelout.py,sha256=W_crO0WtksoKUgq5s9Yh8zS8RxSuyKYQCBt1i8vB1sE,693
 mistralai/models/deltamessage.py,sha256=7NtvEjdmBOl86rwOx7x2fcCCJSzIF8K6-eu-G9Wr9PI,1939
 mistralai/models/detailedjobout.py,sha256=aw3xmCM6E2kE1cPI5MtLZOdbtaP7FLBeHZG7ACtlEKg,4862
-mistralai/models/embeddingrequest.py,sha256=9SZGc7QOMfYd_KRGSG-fXapaKisN4hZlqmtvo6sodX0,2046
+mistralai/models/documenturlchunk.py,sha256=NbUo4DOQteGfA5koBA_jj1DfyLwxoXq43QE5RVrAhoM,1934
+mistralai/models/embeddingrequest.py,sha256=d8l83jwIfTBtiXYWxCwAnZaTp_fptmZ9SbPBFr2mK5g,826
 mistralai/models/embeddingresponse.py,sha256=te6E_LYEzRjHJ9QREmsFp5PeNP2J_8ALVjyb1T20pNA,663
 mistralai/models/embeddingresponsedata.py,sha256=fJ3mrZqyBBBE40a6iegOJX3DVDfgyMRq23ByeGSTLFk,534
 mistralai/models/eventout.py,sha256=TouRJeISBLphMTPHfgSOpuoOmbGDVohPOrdgHyExMpw,1633
@@ -189,9 +199,9 @@ mistralai/models/files_api_routes_retrieve_fileop.py,sha256=8DjSbYqUqFoPq-qcNXyV
 mistralai/models/files_api_routes_upload_fileop.py,sha256=gIGH5xcPryWYkj1FmNv_0-9He-QB_rRf1a5nUYBNDTU,2213
 mistralai/models/fileschema.py,sha256=n_IjCdNOrC2fuzkv75wJn01XvqGTmPK3JqAFSHaOiMA,2597
 mistralai/models/filesignedurl.py,sha256=VwvuhzhJulAB99Qxz6zr-2F1aINosAfaSxU0IhytDSU,289
-mistralai/models/fimcompletionrequest.py,sha256=DidQBt_s7o-VkfGNZpN6MUi19Qs_WXVCvA7ek45Hcnc,6606
+mistralai/models/fimcompletionrequest.py,sha256=wWDCkQ_PMnjB8DrIuIvVJlPGqQtTpVDHt4p7xJ204Ug,6565
 mistralai/models/fimcompletionresponse.py,sha256=_QwzRuL3KuKkyrA4Fxp366SW0H0EzOA7f4FLkWLm-PM,790
-mistralai/models/fimcompletionstreamrequest.py,sha256=n-uzWyaSk4-dfBLM8rbmhZOESLKG_JCcaoc1aLdrbqk,5985
+mistralai/models/fimcompletionstreamrequest.py,sha256=fxuR8FDOWMwIqlYU9ttAfGeRdVgTz4l2k26_OEfxelg,5944
 mistralai/models/ftmodelcapabilitiesout.py,sha256=H1kKEChUPgYT31ZQUz0tn9NRa7Z3hRZlh-sFfDYvBos,648
 mistralai/models/ftmodelcard.py,sha256=G3dioHDMOhbI5HIw5gCaxZDb1sCthBzkXIAYMNHwya8,3300
 mistralai/models/ftmodelout.py,sha256=dw-y8KKT_7rzW6tu10gfc1YKB8-Kpw4e4zy23z_U1d4,2530
@@ -218,15 +228,22 @@ mistralai/models/jobs_api_routes_fine_tuning_start_fine_tuning_jobop.py,sha256=h
 mistralai/models/jobs_api_routes_fine_tuning_unarchive_fine_tuned_modelop.py,sha256=_pkyhD7OzG-59fgcajI9NmSLTLDktkCxXo_IuvWeyfs,636
 mistralai/models/jobs_api_routes_fine_tuning_update_fine_tuned_modelop.py,sha256=s-EYS-Hw0NExYeIyN-3JlHbKmnTmtyB8ljVSfOylqYk,907
 mistralai/models/jobsout.py,sha256=uCKt0aw7yXzI4oLDGeAAEhsRjdRg3g7lPopg0__czTA,818
+mistralai/models/jsonschema.py,sha256=Sh6VCcRxJffd-vXX61GVElbxKNi-j2MKvMUZQcW5EUs,1653
 mistralai/models/legacyjobmetadataout.py,sha256=08zAGNTSrICsK8u2SFFUXiNWF7MCQvezmFQeMQzxsys,4762
 mistralai/models/listfilesout.py,sha256=tW2fNabLKcftc5kytkjwVaChlOzWRL4FKtNzDak9MNs,468
 mistralai/models/metricout.py,sha256=dXQMMU4Nk6-Zr06Jx1TWilFi6cOwiVLjSanCFn0cPxo,2034
 mistralai/models/modelcapabilities.py,sha256=No-Dl09zT1sG4MxsWnx4s8Yo1tUeMQ7k-HR_iQFIMFc,703
 mistralai/models/modellist.py,sha256=D4Y784kQkx0ARhofFrpEqGLfxa-jTY8ev0TQMrD_n8I,995
-mistralai/models/prediction.py,sha256=54P1n4Y5pXu4YsFKAcmThj0GyUrWxyOlIfjG5K2htB8,726
+mistralai/models/ocrimageobject.py,sha256=QDylsNCXy78rWLHuE4CrnQAPXnubz0-FN1wQXtsVsDA,2534
+mistralai/models/ocrpagedimensions.py,sha256=oP4v80I8d6ZELSZt6cRoECd6uIONgdyCeeFalm-4OvM,609
+mistralai/models/ocrpageobject.py,sha256=s0OzcCA0cLTFYahNOr0-r4ds7WOsXYhEx-QcNLngW7M,2085
+mistralai/models/ocrrequest.py,sha256=KKFFvxsGsOQKAYFpuZ02-VsiJWNawlIIz-7dPPtNklU,3112
+mistralai/models/ocrresponse.py,sha256=JpI9Yl6fYeGf6Wsn4M4Z-i6Ir2qAlMKFQnVXqsLiC5Y,752
+mistralai/models/ocrusageinfo.py,sha256=cWtxHxXyJn3U1qJgG-XU-1xfC7poGHvhVtPqdrFrfII,1571
+mistralai/models/prediction.py,sha256=BgWbbeSi1eD9Rh1xk8srXlRgD7Xooj8nLsbSQ21pNRo,718
 mistralai/models/referencechunk.py,sha256=A9vV5pZv-tUqGlswdu0HOyCYy0Q-UIJY0Oc9ZfM6XJA,519
-mistralai/models/responseformat.py,sha256=C_zO6X4cbT1qSS_q9Qxq64AmtfK10i9tqEz_39ZcFzo,1045
-mistralai/models/responseformats.py,sha256=oeXHoVUoZrZwrz-0cm-rHj5sHygv9MpeqgdummGa8ww,488
+mistralai/models/responseformat.py,sha256=-TAPGth3_FAiNl-kuE4COI5cSP5fxQ7xewFSV989i58,2225
+mistralai/models/responseformats.py,sha256=O9lwS2M9m53DsRxTC4uRP12SvRhgaQoMjIYsDys5A7s,503
 mistralai/models/retrieve_model_v1_models_model_id_getop.py,sha256=N9_JFwiz9tz4zRXJ9c1V0c_anFEVxVzPDoFt2Wrer4M,1388
 mistralai/models/retrievefileout.py,sha256=nAjSITJCHj0daChhpwOZTmps-74mmYZO4IckGA0yIvQ,2644
 mistralai/models/sampletype.py,sha256=zowUiTFxum8fltBs6j__BrFPio-dQdG0CIyLj-5icG8,316
@@ -253,8 +270,9 @@ mistralai/models/validationerror.py,sha256=DouDBJmBhbW4KPsF5rZEyBdnB_adC-l32kuHC
 mistralai/models/wandbintegration.py,sha256=BkLD3r08ToZkMAhPXdnC7bfOGr3banKqt1wVKMGehUQ,2406
 mistralai/models/wandbintegrationout.py,sha256=C0HpS8jJGnACs7eWnuIq0qJEroIUAbjkvzfSSkSKS7Q,2274
 mistralai/models_.py,sha256=r0etSSUChK7hxxf7ZyhoeloyE8TyPRL1s5Jh4Jgukbw,44394
+mistralai/ocr.py,sha256=aMro4iLNC7Jr65N2A-3DVcs1AIWNUOG_9n_5zAD4V2M,9703
 mistralai/py.typed,sha256=zrp19r0G21lr2yRiMC0f8MFkQFGj9wMpSbboePMg8KM,59
-mistralai/sdk.py,sha256=9eHH11No72LdVYs7uJXyUqyiZ4hX-qXRtoCxq0LCn3M,5455
+mistralai/sdk.py,sha256=ZStm1fV-bzMxVgnkmsHpi1hyY0603xTTwSecv0oPyzg,5842
 mistralai/sdkconfiguration.py,sha256=sx6U1xgxFbflhJdOBAeI7isId-8wlMd8NK9pi-JGs1o,1789
 mistralai/types/__init__.py,sha256=RArOwSgeeTIva6h-4ttjXwMUeCkz10nAFBL9D-QljI4,377
 mistralai/types/basemodel.py,sha256=PexI39iKiOkIlobB8Ueo0yn8PLHp6_wb-WO-zelNDZY,1170
@@ -274,7 +292,7 @@ mistralai/utils/serializers.py,sha256=BSJT7kBOkNBFyP7KREyMoe14JGbgijD1M6AXFMbdmc
 mistralai/utils/url.py,sha256=BgGPgcTA6MRK4bF8fjP2dUopN3NzEzxWMXPBVg8NQUA,5254
 mistralai/utils/values.py,sha256=_89YXPTI_BU6SXJBzFR4pIzTCBPQW9tsOTN1jeBBIDs,3428
 mistralai/version.py,sha256=iosXhlXclBwBqlADFKEilxAC2wWKbtuBKi87AmPi7s8,196
-mistralai-1.4.0.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
-mistralai-1.4.0.dist-info/METADATA,sha256=6Gs8wKfT4XGnX9TsM5l63GXbcRcI6EnPo-23DI7YIS4,27705
-mistralai-1.4.0.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-mistralai-1.4.0.dist-info/RECORD,,
+mistralai-1.5.1.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
+mistralai-1.5.1.dist-info/METADATA,sha256=3SyFjKQU-IBkJmJuUV-2BUDEPXSRaS5Z9PmJvmK65Og,29449
+mistralai-1.5.1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+mistralai-1.5.1.dist-info/RECORD,,

{mistralai-1.4.0.dist-info → mistralai-1.5.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.0.1
+Generator: poetry-core 2.1.1
 Root-Is-Purelib: true
 Tag: py3-none-any

mistralai/models/chatclassificationrequest.py

@@ -1,113 +0,0 @@
-"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
-
-from __future__ import annotations
-from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
-from .systemmessage import SystemMessage, SystemMessageTypedDict
-from .toolmessage import ToolMessage, ToolMessageTypedDict
-from .usermessage import UserMessage, UserMessageTypedDict
-from mistralai.types import BaseModel, Nullable, UNSET_SENTINEL
-from mistralai.utils import get_discriminator
-import pydantic
-from pydantic import Discriminator, Tag, model_serializer
-from typing import List, Union
-from typing_extensions import Annotated, TypeAliasType, TypedDict
-
-
-TwoTypedDict = TypeAliasType(
-    "TwoTypedDict",
-    Union[
-        SystemMessageTypedDict,
-        UserMessageTypedDict,
-        AssistantMessageTypedDict,
-        ToolMessageTypedDict,
-    ],
-)
-
-
-Two = Annotated[
-    Union[
-        Annotated[AssistantMessage, Tag("assistant")],
-        Annotated[SystemMessage, Tag("system")],
-        Annotated[ToolMessage, Tag("tool")],
-        Annotated[UserMessage, Tag("user")],
-    ],
-    Discriminator(lambda m: get_discriminator(m, "role", "role")),
-]
-
-
-OneTypedDict = TypeAliasType(
-    "OneTypedDict",
-    Union[
-        SystemMessageTypedDict,
-        UserMessageTypedDict,
-        AssistantMessageTypedDict,
-        ToolMessageTypedDict,
-    ],
-)
-
-
-One = Annotated[
-    Union[
-        Annotated[AssistantMessage, Tag("assistant")],
-        Annotated[SystemMessage, Tag("system")],
-        Annotated[ToolMessage, Tag("tool")],
-        Annotated[UserMessage, Tag("user")],
-    ],
-    Discriminator(lambda m: get_discriminator(m, "role", "role")),
-]
-
-
-ChatClassificationRequestInputsTypedDict = TypeAliasType(
-    "ChatClassificationRequestInputsTypedDict",
-    Union[List[OneTypedDict], List[List[TwoTypedDict]]],
-)
-r"""Chat to classify"""
-
-
-ChatClassificationRequestInputs = TypeAliasType(
-    "ChatClassificationRequestInputs", Union[List[One], List[List[Two]]]
-)
-r"""Chat to classify"""
-
-
-class ChatClassificationRequestTypedDict(TypedDict):
-    inputs: ChatClassificationRequestInputsTypedDict
-    r"""Chat to classify"""
-    model: Nullable[str]
-
-
-class ChatClassificationRequest(BaseModel):
-    inputs: Annotated[ChatClassificationRequestInputs, pydantic.Field(alias="input")]
-    r"""Chat to classify"""
-
-    model: Nullable[str]
-
-    @model_serializer(mode="wrap")
-    def serialize_model(self, handler):
-        optional_fields = []
-        nullable_fields = ["model"]
-        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