mistralai 1.0.0rc1__py3-none-any.whl → 1.0.1__py3-none-any.whl

mistralai_gcp/utils/__init__.py

@@ -34,6 +34,7 @@ from .serializers import (
 )
 from .url import generate_url, template_url, remove_suffix
 from .values import get_global_from_env, match_content_type, match_status_codes, match_response
+from .logger import Logger, get_body_content, NoOpLogger
 
 __all__ = [
     "BackoffStrategy",
@@ -41,6 +42,7 @@ __all__ = [
     "find_metadata",
     "FormMetadata",
     "generate_url",
+    "get_body_content",
     "get_discriminator",
     "get_global_from_env",
     "get_headers",
@@ -49,11 +51,13 @@ __all__ = [
     "get_response_headers",
     "get_security",
     "HeaderMetadata",
+    "Logger",
     "marshal_json",
     "match_content_type",
     "match_status_codes",
     "match_response",
     "MultipartFormMetadata",
+    "NoOpLogger",
     "OpenEnumMeta",
     "PathParamMetadata",
     "QueryParamMetadata",

mistralai_gcp/utils/eventstreaming.py

@@ -147,15 +147,14 @@ def _parse_event(
         data = data[:-1]
         event.data = data
 
-        if (
-            data.isnumeric()
-            or data == "true"
-            or data == "false"
-            or data == "null"
-            or data.startswith("{")
-            or data.startswith("[")
-            or data.startswith('"')
-        ):
+        data_is_primitive = (
+            data.isnumeric() or data == "true" or data == "false" or data == "null"
+        )
+        data_is_json = (
+            data.startswith("{") or data.startswith("[") or data.startswith('"')
+        )
+
+        if data_is_primitive or data_is_json:
             try:
                 event.data = json.loads(data)
             except Exception:

mistralai_gcp/utils/logger.py

@@ -0,0 +1,16 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+import httpx
+from typing import Any, Protocol
+
+class Logger(Protocol):
+    def debug(self, msg: str, *args: Any, **kwargs: Any) -> None:
+        pass
+
+class NoOpLogger:
+    def debug(self, msg: str, *args: Any, **kwargs: Any) -> None:
+        pass
+
+def get_body_content(req: httpx.Request) -> str:
+    return "<streaming body>" if not hasattr(req, "_content") else str(req.content)
+

mistralai_gcp/utils/retries.py

@@ -76,7 +76,7 @@ def retry(func, retries: Retries):
 
                         status_major = res.status_code / 100
 
-                        if status_major >= code_range and status_major < code_range + 1:
+                        if code_range <= status_major < code_range + 1:
                             raise TemporaryError(res)
                     else:
                         parsed_code = int(code)
@@ -125,7 +125,7 @@ async def retry_async(func, retries: Retries):
 
                         status_major = res.status_code / 100
 
-                        if status_major >= code_range and status_major < code_range + 1:
+                        if code_range <= status_major < code_range + 1:
                             raise TemporaryError(res)
                     else:
                         parsed_code = int(code)