blaxel 0.1.14rc52__tar.gz → 0.1.14rc54__tar.gz

{blaxel-0.1.14rc52 → blaxel-0.1.14rc54}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.1.14rc52
+Version: 0.1.14rc54
 Summary: Add your description here
 Author-email: cploujoux <cploujoux@blaxel.ai>
 License-File: LICENSE
 Requires-Python: >=3.10
 Requires-Dist: attrs>=21.3.0
 Requires-Dist: httpx>=0.27.0
-Requires-Dist: mcp>=1.2.1
+Requires-Dist: mcp<=1.7.1
 Requires-Dist: opentelemetry-api>=1.28.0
 Requires-Dist: opentelemetry-exporter-otlp>=1.28.0
 Requires-Dist: opentelemetry-instrumentation-anthropic==0.35.0

{blaxel-0.1.14rc52 → blaxel-0.1.14rc54}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "blaxel"
-version = "0.1.14.pre-52"
+version = "0.1.14.pre-54"
 description = "Add your description here"
 readme = "README.md"
 authors = [{ name = "cploujoux", email = "cploujoux@blaxel.ai" }]
@@ -8,7 +8,7 @@ requires-python = ">=3.10"
 dependencies = [
     "attrs>=21.3.0",
     "httpx>=0.27.0",
-    "mcp>=1.2.1",
+    "mcp<=1.7.1",
     # OpenTelemetry dependencies
     "opentelemetry-api>=1.28.0",
     "opentelemetry-exporter-otlp>=1.28.0",

{blaxel-0.1.14rc52 → blaxel-0.1.14rc54}/src/blaxel/authentication/clientcredentials.py

@@ -7,6 +7,7 @@ client credentials and refresh tokens.
 from dataclasses import dataclass
 from datetime import datetime, timedelta
 from typing import Generator, Optional
+import time
 
 import requests
 from httpx import Request, Response
@@ -46,14 +47,17 @@ class ClientCredentials(BlaxelAuth):
             "X-Blaxel-Workspace": self.workspace_name,
         }
 
-    def get_token(self) -> Optional[Exception]:
+    def _request_token(self, remaining_retries: int = 3) -> Optional[Exception]:
         """
-        Checks if the access token needs to be refreshed and performs the refresh if necessary.
+        Makes the token request with recursive retry logic.
+
+        Args:
+            remaining_retries (int): Number of retry attempts remaining.
 
         Returns:
-            Optional[Exception]: An exception if refreshing fails, otherwise None.
+            Optional[Exception]: An exception if refreshing fails after all retries, otherwise None.
         """
-        if self.need_token():
+        try:
             headers = {"Authorization": f"Basic {self.credentials.client_credentials}", "Content-Type": "application/json"}
             body = {"grant_type": "client_credentials"}
             response = requests.post(f"{self.base_url}/oauth/token", headers=headers, json=body)
@@ -63,6 +67,23 @@ class ClientCredentials(BlaxelAuth):
             self.credentials.refresh_token = creds["refresh_token"]
             self.credentials.expires_in = creds["expires_in"]
             self.expires_at = datetime.now() + timedelta(seconds=self.credentials.expires_in)
+            return None
+        except Exception as e:
+            if remaining_retries > 0:
+                time.sleep(1)
+                return self._request_token(remaining_retries - 1)
+            return e
+
+    def get_token(self) -> Optional[Exception]:
+        """
+        Checks if the access token needs to be refreshed and performs the refresh if necessary.
+        Uses recursive retry logic for up to 3 attempts.
+
+        Returns:
+            Optional[Exception]: An exception if refreshing fails after all retries, otherwise None.
+        """
+        if self.need_token():
+            return self._request_token()
         return None
 
     def need_token(self):