mistralai 0.1.3__py3-none-any.whl → 0.1.7__py3-none-any.whl

mistralai/async_client.py

@@ -14,7 +14,7 @@ from httpx import (
 )
 
 from mistralai.client_base import ClientBase
-from mistralai.constants import ENDPOINT
+from mistralai.constants import ENDPOINT, RETRY_STATUS_CODES
 from mistralai.exceptions import (
     MistralAPIException,
     MistralAPIStatusException,
@@ -52,6 +52,44 @@ class MistralAsyncClient(ClientBase):
     async def close(self) -> None:
         await self._client.aclose()
 
+    async def _check_response_status_codes(self, response: Response) -> None:
+        if response.status_code in RETRY_STATUS_CODES:
+            raise MistralAPIStatusException.from_response(
+                response,
+                message=f"Status: {response.status_code}. Message: {response.text}",
+            )
+        elif 400 <= response.status_code < 500:
+            if response.stream:
+                await response.aread()
+            raise MistralAPIException.from_response(
+                response,
+                message=f"Status: {response.status_code}. Message: {response.text}",
+            )
+        elif response.status_code >= 500:
+            if response.stream:
+                await response.aread()
+            raise MistralException(
+                message=f"Status: {response.status_code}. Message: {response.text}",
+            )
+
+    async def _check_streaming_response(self, response: Response) -> None:
+        await self._check_response_status_codes(response)
+
+    async def _check_response(self, response: Response) -> Dict[str, Any]:
+        await self._check_response_status_codes(response)
+
+        json_response: Dict[str, Any] = response.json()
+
+        if "object" not in json_response:
+            raise MistralException(message=f"Unexpected response: {json_response}")
+        if "error" == json_response["object"]:  # has errors
+            raise MistralAPIException.from_response(
+                response,
+                message=json_response["message"],
+            )
+
+        return json_response
+
     async def _request(
         self,
         method: str,
@@ -82,7 +120,7 @@ class MistralAsyncClient(ClientBase):
                     headers=headers,
                     json=json,
                 ) as response:
-                    self._check_streaming_response(response)
+                    await self._check_streaming_response(response)
 
                     async for line in response.aiter_lines():
                         json_streamed_response = self._process_line(line)
@@ -97,7 +135,7 @@ class MistralAsyncClient(ClientBase):
                     json=json,
                 )
 
-                yield self._check_response(response)
+                yield await self._check_response(response)
 
         except ConnectError as e:
             raise MistralConnectionException(str(e)) from e

mistralai/client.py

@@ -7,7 +7,7 @@ from typing import Any, Dict, Iterable, Iterator, List, Optional, Union
 from httpx import Client, ConnectError, HTTPTransport, RequestError, Response
 
 from mistralai.client_base import ClientBase
-from mistralai.constants import ENDPOINT
+from mistralai.constants import ENDPOINT, RETRY_STATUS_CODES
 from mistralai.exceptions import (
     MistralAPIException,
     MistralAPIStatusException,
@@ -45,6 +45,44 @@ class MistralClient(ClientBase):
     def __del__(self) -> None:
         self._client.close()
 
+    def _check_response_status_codes(self, response: Response) -> None:
+        if response.status_code in RETRY_STATUS_CODES:
+            raise MistralAPIStatusException.from_response(
+                response,
+                message=f"Status: {response.status_code}. Message: {response.text}",
+            )
+        elif 400 <= response.status_code < 500:
+            if response.stream:
+                response.read()
+            raise MistralAPIException.from_response(
+                response,
+                message=f"Status: {response.status_code}. Message: {response.text}",
+            )
+        elif response.status_code >= 500:
+            if response.stream:
+                response.read()
+            raise MistralException(
+                message=f"Status: {response.status_code}. Message: {response.text}",
+            )
+
+    def _check_streaming_response(self, response: Response) -> None:
+        self._check_response_status_codes(response)
+
+    def _check_response(self, response: Response) -> Dict[str, Any]:
+        self._check_response_status_codes(response)
+
+        json_response: Dict[str, Any] = response.json()
+
+        if "object" not in json_response:
+            raise MistralException(message=f"Unexpected response: {json_response}")
+        if "error" == json_response["object"]:  # has errors
+            raise MistralAPIException.from_response(
+                response,
+                message=json_response["message"],
+            )
+
+        return json_response
+
     def _request(
         self,
         method: str,

mistralai/client_base.py

@@ -4,12 +4,8 @@ from abc import ABC
 from typing import Any, Dict, List, Optional, Union
 
 import orjson
-from httpx import Response
 
-from mistralai.constants import RETRY_STATUS_CODES
 from mistralai.exceptions import (
-    MistralAPIException,
-    MistralAPIStatusException,
     MistralException,
 )
 from mistralai.models.chat_completion import ChatMessage, Function, ResponseFormat, ToolChoice
@@ -40,7 +36,7 @@ class ClientBase(ABC):
             self._default_model = "mistral"
 
         # This should be automatically updated by the deploy script
-        self._version = "0.1.3"
+        self._version = "0.1.7"
 
     def _parse_tools(self, tools: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
         parsed_tools: List[Dict[str, Any]] = []
@@ -125,40 +121,6 @@ class ClientBase(ABC):
 
         return request_data
 
-    def _check_response_status_codes(self, response: Response) -> None:
-        if response.status_code in RETRY_STATUS_CODES:
-            raise MistralAPIStatusException.from_response(
-                response,
-                message=f"Status: {response.status_code}. Message: {response.text}",
-            )
-        elif 400 <= response.status_code < 500:
-            raise MistralAPIException.from_response(
-                response,
-                message=f"Status: {response.status_code}. Message: {response.text}",
-            )
-        elif response.status_code >= 500:
-            raise MistralException(
-                message=f"Status: {response.status_code}. Message: {response.text}",
-            )
-
-    def _check_streaming_response(self, response: Response) -> None:
-        self._check_response_status_codes(response)
-
-    def _check_response(self, response: Response) -> Dict[str, Any]:
-        self._check_response_status_codes(response)
-
-        json_response: Dict[str, Any] = response.json()
-
-        if "object" not in json_response:
-            raise MistralException(message=f"Unexpected response: {json_response}")
-        if "error" == json_response["object"]:  # has errors
-            raise MistralAPIException.from_response(
-                response,
-                message=json_response["message"],
-            )
-
-        return json_response
-
     def _process_line(self, line: str) -> Optional[Dict[str, Any]]:
         if line.startswith("data: "):
             line = line[6:].strip()

{mistralai-0.1.3.dist-info → mistralai-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mistralai
-Version: 0.1.3
+Version: 0.1.7
 Summary: 
 Author: Bam4d
 Author-email: bam4d@mistral.ai
@@ -12,8 +12,6 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: httpx (>=0.25.2,<0.26.0)
 Requires-Dist: orjson (>=3.9.10,<4.0.0)
-Requires-Dist: pandas (>=2.2.0,<3.0.0)
-Requires-Dist: pyarrow (>=15.0.0,<16.0.0)
 Requires-Dist: pydantic (>=2.5.2,<3.0.0)
 Description-Content-Type: text/markdown
 

{mistralai-0.1.3.dist-info → mistralai-0.1.7.dist-info}/RECORD

@@ -1,7 +1,7 @@
 mistralai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mistralai/async_client.py,sha256=-qAdHbYdFNmOyBOMnpD3Zdeyh211m1HcbeyEwmqSfgc,9830
-mistralai/client.py,sha256=xrbWUQ-gHKU1XSmYlFiDpy7BBE1uL3-y_8TxlHjWqeU,9639
-mistralai/client_base.py,sha256=OMB-8FbnO-Gmp_phzRJtjzxT2zWLjnawtEB6ik_sE0Q,6156
+mistralai/async_client.py,sha256=dpaZ7W7R2Fmo19vvf1SfyViT6JbmjJUGzaT-keT1vTc,11411
+mistralai/client.py,sha256=cneaYGtFOkrE0sUPh6GQeuX_eXKcCpNjomJjPlVWD4g,11164
+mistralai/client_base.py,sha256=fAOrLK9FSgKMflI6YNCCxslqcPbFKuR2hlMsDfL_7AI,4645
 mistralai/constants.py,sha256=KK286HFjpoTxPKih8xdTp0lW4YMDPyYu2Shi3Nu5Vdw,86
 mistralai/exceptions.py,sha256=d1cli28ZaPEBQy2RKKcSh-dO2T20GXHNoaJrXXBmyIs,1664
 mistralai/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -10,7 +10,7 @@ mistralai/models/common.py,sha256=zatP4aV_LIEpzj3_igsKkJBICwGhmXG0LX3CdO3kn-o,17
 mistralai/models/embeddings.py,sha256=-VthLQBj6wrq7HXJbGmnkQEEanSemA3MAlaMFh94VBg,331
 mistralai/models/models.py,sha256=I4I1kQwP0tJ2_rd4lbSUwOkCJE5E-dSTIExQg2ZNFcw,713
 mistralai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mistralai-0.1.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mistralai-0.1.3.dist-info/METADATA,sha256=FtZmP7OpC1xTs7dv2C8TPMpzN3fLXTlQEkv6-eYIdkg,1917
-mistralai-0.1.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-mistralai-0.1.3.dist-info/RECORD,,
+mistralai-0.1.7.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mistralai-0.1.7.dist-info/METADATA,sha256=-_MdZmZ2d_bj0G4X90l9bICg1dqfAvnhw9iB9PCv8eI,1836
+mistralai-0.1.7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+mistralai-0.1.7.dist-info/RECORD,,