blaxel 0.2.31rc120__py3-none-any.whl → 0.2.32__py3-none-any.whl

blaxel/__init__.py

@@ -4,9 +4,9 @@ from .core.common.autoload import autoload
 from .core.common.env import env
 from .core.common.settings import settings
 
-__version__ = "0.2.31.pre-120"
-__commit__ = "a23256023f1d955467e2ee297a96b3d321172acf"
+__version__ = "0.2.32"
+__commit__ = "72a43488d87cfae83389f5a928dfa9256649396b"
 __sentry_dsn__ = "https://9711de13cd02b285ca4378c01de8dc30@o4508714045276160.ingest.us.sentry.io/4510461121462272"
 __all__ = ["autoload", "settings", "env"]
 
-autoload()
+autoload()

blaxel/core/agents/__init__.py

@@ -1,6 +1,6 @@
 import json
 from logging import getLogger
-from typing import Any, Awaitable
+from typing import Any
 
 from ..cache import find_from_cache
 from ..client import client
@@ -51,10 +51,14 @@ class BlAgent:
             body = json.dumps(body)
 
         # Merge settings headers with provided headers
-        merged_headers = {**settings.headers, "Content-Type": "application/json", **headers}
+        merged_headers = {
+            **settings.headers,
+            "Content-Type": "application/json",
+            **headers,
+        }
 
         return client.get_httpx_client().post(
-            url, headers=merged_headers, data=body, params=params
+            url, headers=merged_headers, content=body, params=params
         )
 
     async def acall(self, url, input_data, headers: dict = {}, params: dict = {}):
@@ -64,10 +68,14 @@ class BlAgent:
             body = json.dumps(body)
 
         # Merge settings headers with provided headers
-        merged_headers = {**settings.headers, "Content-Type": "application/json", **headers}
+        merged_headers = {
+            **settings.headers,
+            "Content-Type": "application/json",
+            **headers,
+        }
 
         return await client.get_async_httpx_client().post(
-            url, headers=merged_headers, data=body, params=params
+            url, headers=merged_headers, content=body, params=params
         )
 
     def run(self, input: Any, headers: dict = {}, params: dict = {}) -> str:
@@ -85,7 +93,7 @@ class BlAgent:
                 )
         return response.text
 
-    async def arun(self, input: Any, headers: dict = {}, params: dict = {}) -> Awaitable[str]:
+    async def arun(self, input: Any, headers: dict = {}, params: dict = {}) -> str:
         logger.debug(f"Agent Calling: {self.name}")
         response = await self.acall(self.url, input, headers, params)
         if response.status_code >= 400:
@@ -120,4 +128,5 @@ async def get_agent_metadata(name):
     except Exception:
         return None
 
-__all__ = ["BlAgent", "bl_agent", "get_agent_metadata"]
+
+__all__ = ["BlAgent", "bl_agent", "get_agent_metadata"]

blaxel/core/authentication/__init__.py

@@ -34,7 +34,8 @@ def get_credentials() -> CredentialsType | None:
 
     if os.environ.get("BL_CLIENT_CREDENTIALS"):
         return CredentialsType(
-            client_credentials=os.environ.get("BL_CLIENT_CREDENTIALS"), workspace=get_workspace()
+            client_credentials=os.environ.get("BL_CLIENT_CREDENTIALS"),
+            workspace=get_workspace(),
         )
 
     try:

blaxel/core/authentication/devicemode.py

@@ -24,6 +24,7 @@ class DeviceLogin:
         client_id (str): The client ID for the device.
         scope (str): The scope of the authentication.
     """
+
     client_id: str
     scope: str
 
@@ -42,6 +43,7 @@ class DeviceLoginResponse:
         verification_uri (str): URI for user to verify device login.
         verification_uri_complete (str): Complete URI including the user code for verification.
     """
+
     client_id: str
     device_code: str
     user_code: str
@@ -61,6 +63,7 @@ class DeviceLoginFinalizeRequest:
         client_id (str): The client ID for finalizing the device login.
         device_code (str): The device code to finalize login.
     """
+
     grant_type: str
     client_id: str
     device_code: str
@@ -78,6 +81,7 @@ class DeviceMode(BlaxelAuth):
     """
     DeviceMode auth that authenticates requests using a device code.
     """
+
     def get_headers(self) -> Dict[str, str]:
         """
         Retrieves the authentication headers containing the Bearer token and workspace information.
@@ -161,7 +165,11 @@ class DeviceMode(BlaxelAuth):
         }
 
         try:
-            response = post(url, json=refresh_data, headers={"Content-Type": "application/json"})
+            response = post(
+                url,
+                json=refresh_data,
+                headers={"Content-Type": "application/json"},
+            )
             response.raise_for_status()
             finalize_response = DeviceLoginFinalizeResponse(**response.json())
 

blaxel/core/authentication/oauth.py

@@ -11,6 +11,7 @@ class OauthTokenData:
     headers: dict[str, str] = field(default_factory=dict)
     authenticated: bool | None = False
 
+
 @dataclass
 class OauthTokenResponse:
     access_token: str
@@ -18,14 +19,16 @@ class OauthTokenResponse:
     expires_in: int
     token_type: str
 
+
 @dataclass
 class OauthTokenError:
     error: str
 
+
 async def oauth_token(
     options: OauthTokenData,
     client: Client | None = None,
-    throw_on_error: bool = False
+    throw_on_error: bool = False,
 ) -> Union[OauthTokenResponse, OauthTokenError]:
     """
     Get a new OAuth token.
@@ -38,13 +41,17 @@ async def oauth_token(
     Returns:
         The OAuth token response or error
     """
-    response = await (client or default_client).get_async_httpx_client().post(
-        url="/oauth/token",
-        json=options.body or {},
-        headers=options.headers or {},
+    response = (
+        await (client or default_client)
+        .get_async_httpx_client()
+        .post(
+            url="/oauth/token",
+            json=options.body or {},
+            headers=options.headers or {},
+        )
     )
     if response.status_code >= 400:
         if throw_on_error:
             raise Exception(f"Failed to get OAuth token: {response.text}")
         return OauthTokenError(error=response.text)
-    return OauthTokenResponse(**response.json())
+    return OauthTokenResponse(**response.json())

blaxel/core/authentication/types.py

@@ -17,6 +17,7 @@ class CredentialsType(BaseModel):
     expires_in: int | None = Field(default=None, description="The expiration time")
     workspace: str | None = Field(default=None, description="The workspace")
 
+
 class BlaxelAuth(Auth):
     def __init__(self, credentials: CredentialsType, workspace_name: str, base_url: str):
         """

blaxel/core/cache/cache.py

@@ -14,19 +14,26 @@ def _load_cache() -> None:
     Load cache data from .cache.yaml file if it exists.
     """
     try:
-        if os.path.exists('.cache.yaml'):
-            with open('.cache.yaml', encoding='utf-8') as file:
+        if os.path.exists(".cache.yaml"):
+            with open(".cache.yaml", encoding="utf-8") as file:
                 cache_string = file.read()
                 cache_data = yaml.safe_load_all(cache_string)
 
                 for doc in cache_data:
-                    if doc and isinstance(doc, dict) and 'kind' in doc and 'metadata' in doc and 'name' in doc['metadata']:
+                    if (
+                        doc
+                        and isinstance(doc, dict)
+                        and "kind" in doc
+                        and "metadata" in doc
+                        and "name" in doc["metadata"]
+                    ):
                         cache_key = f"{doc['kind']}/{doc['metadata']['name']}"
                         _cache[cache_key] = doc
     except Exception:
         # Silently fail if cache file doesn't exist or is invalid
         pass
 
+
 # Initialize cache on module import
 _load_cache()
 

blaxel/core/client/api/agents/create_agent.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Union
+from typing import Any
 
 import httpx
 
@@ -54,7 +54,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[Age
 
 def sync_detailed(
     *,
-    client: Union[Client],
+    client: Client,
     body: Agent,
 ) -> Response[Agent]:
     """Create agent by name
@@ -83,7 +83,7 @@ def sync_detailed(
 
 def sync(
     *,
-    client: Union[Client],
+    client: Client,
     body: Agent,
 ) -> Agent | None:
     """Create agent by name
@@ -107,7 +107,7 @@ def sync(
 
 async def asyncio_detailed(
     *,
-    client: Union[Client],
+    client: Client,
     body: Agent,
 ) -> Response[Agent]:
     """Create agent by name
@@ -134,7 +134,7 @@ async def asyncio_detailed(
 
 async def asyncio(
     *,
-    client: Union[Client],
+    client: Client,
     body: Agent,
 ) -> Agent | None:
     """Create agent by name

blaxel/core/client/api/agents/delete_agent.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Union
+from typing import Any
 
 import httpx
 
@@ -43,7 +43,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[Age
 def sync_detailed(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[Agent]:
     """Delete agent by name
 
@@ -72,7 +72,7 @@ def sync_detailed(
 def sync(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Agent | None:
     """Delete agent by name
 
@@ -96,7 +96,7 @@ def sync(
 async def asyncio_detailed(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[Agent]:
     """Delete agent by name
 
@@ -123,7 +123,7 @@ async def asyncio_detailed(
 async def asyncio(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Agent | None:
     """Delete agent by name
 

blaxel/core/client/api/agents/get_agent.py

@@ -52,7 +52,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[Age
 def sync_detailed(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     show_secrets: Union[Unset, bool] = UNSET,
 ) -> Response[Agent]:
     """Get agent by name
@@ -84,7 +84,7 @@ def sync_detailed(
 def sync(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     show_secrets: Union[Unset, bool] = UNSET,
 ) -> Agent | None:
     """Get agent by name
@@ -111,7 +111,7 @@ def sync(
 async def asyncio_detailed(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     show_secrets: Union[Unset, bool] = UNSET,
 ) -> Response[Agent]:
     """Get agent by name
@@ -141,7 +141,7 @@ async def asyncio_detailed(
 async def asyncio(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     show_secrets: Union[Unset, bool] = UNSET,
 ) -> Agent | None:
     """Get agent by name

blaxel/core/client/api/agents/list_agent_revisions.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Union
+from typing import Any
 
 import httpx
 
@@ -36,7 +36,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> list["Revisi
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[list["RevisionMetadata"]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[list["RevisionMetadata"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -48,7 +50,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[lis
 def sync_detailed(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[list["RevisionMetadata"]]:
     """List all agent revisions
 
@@ -77,7 +79,7 @@ def sync_detailed(
 def sync(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> list["RevisionMetadata"] | None:
     """List all agent revisions
 
@@ -101,7 +103,7 @@ def sync(
 async def asyncio_detailed(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[list["RevisionMetadata"]]:
     """List all agent revisions
 
@@ -128,7 +130,7 @@ async def asyncio_detailed(
 async def asyncio(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> list["RevisionMetadata"] | None:
     """List all agent revisions
 

blaxel/core/client/api/agents/list_agents.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Union
+from typing import Any
 
 import httpx
 
@@ -45,7 +45,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[lis
 
 def sync_detailed(
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[list["Agent"]]:
     """List all agents
 
@@ -68,7 +68,7 @@ def sync_detailed(
 
 def sync(
     *,
-    client: Union[Client],
+    client: Client,
 ) -> list["Agent"] | None:
     """List all agents
 
@@ -87,7 +87,7 @@ def sync(
 
 async def asyncio_detailed(
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[list["Agent"]]:
     """List all agents
 
@@ -108,7 +108,7 @@ async def asyncio_detailed(
 
 async def asyncio(
     *,
-    client: Union[Client],
+    client: Client,
 ) -> list["Agent"] | None:
     """List all agents
 

blaxel/core/client/api/agents/update_agent.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Union
+from typing import Any
 
 import httpx
 
@@ -56,7 +56,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[Age
 def sync_detailed(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: Agent,
 ) -> Response[Agent]:
     """Update agent by name
@@ -88,7 +88,7 @@ def sync_detailed(
 def sync(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: Agent,
 ) -> Agent | None:
     """Update agent by name
@@ -115,7 +115,7 @@ def sync(
 async def asyncio_detailed(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: Agent,
 ) -> Response[Agent]:
     """Update agent by name
@@ -145,7 +145,7 @@ async def asyncio_detailed(
 async def asyncio(
     agent_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: Agent,
 ) -> Agent | None:
     """Update agent by name

blaxel/core/client/api/compute/create_sandbox.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Union
+from typing import Any
 
 import httpx
 
@@ -54,7 +54,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[San
 
 def sync_detailed(
     *,
-    client: Union[Client],
+    client: Client,
     body: Sandbox,
 ) -> Response[Sandbox]:
     """Create Sandbox
@@ -85,7 +85,7 @@ def sync_detailed(
 
 def sync(
     *,
-    client: Union[Client],
+    client: Client,
     body: Sandbox,
 ) -> Sandbox | None:
     """Create Sandbox
@@ -111,7 +111,7 @@ def sync(
 
 async def asyncio_detailed(
     *,
-    client: Union[Client],
+    client: Client,
     body: Sandbox,
 ) -> Response[Sandbox]:
     """Create Sandbox
@@ -140,7 +140,7 @@ async def asyncio_detailed(
 
 async def asyncio(
     *,
-    client: Union[Client],
+    client: Client,
     body: Sandbox,
 ) -> Sandbox | None:
     """Create Sandbox

blaxel/core/client/api/compute/create_sandbox_preview.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Union
+from typing import Any
 
 import httpx
 
@@ -56,7 +56,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[Pre
 def sync_detailed(
     sandbox_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: Preview,
 ) -> Response[Preview]:
     """Create Sandbox Preview
@@ -90,7 +90,7 @@ def sync_detailed(
 def sync(
     sandbox_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: Preview,
 ) -> Preview | None:
     """Create Sandbox Preview
@@ -119,7 +119,7 @@ def sync(
 async def asyncio_detailed(
     sandbox_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: Preview,
 ) -> Response[Preview]:
     """Create Sandbox Preview
@@ -151,7 +151,7 @@ async def asyncio_detailed(
 async def asyncio(
     sandbox_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: Preview,
 ) -> Preview | None:
     """Create Sandbox Preview

blaxel/core/client/api/compute/create_sandbox_preview_token.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Union
+from typing import Any
 
 import httpx
 
@@ -58,7 +58,7 @@ def sync_detailed(
     sandbox_name: str,
     preview_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: PreviewToken,
 ) -> Response[PreviewToken]:
     """Create token for Sandbox Preview
@@ -95,7 +95,7 @@ def sync(
     sandbox_name: str,
     preview_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: PreviewToken,
 ) -> PreviewToken | None:
     """Create token for Sandbox Preview
@@ -127,7 +127,7 @@ async def asyncio_detailed(
     sandbox_name: str,
     preview_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: PreviewToken,
 ) -> Response[PreviewToken]:
     """Create token for Sandbox Preview
@@ -162,7 +162,7 @@ async def asyncio(
     sandbox_name: str,
     preview_name: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: PreviewToken,
 ) -> PreviewToken | None:
     """Create token for Sandbox Preview

blaxel/core/client/api/compute/delete_sandbox.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Union
+from typing import Any
 
 import httpx
 
@@ -43,7 +43,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[San
 def sync_detailed(
     sandbox_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[Sandbox]:
     """Delete Sandbox
 
@@ -74,7 +74,7 @@ def sync_detailed(
 def sync(
     sandbox_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Sandbox | None:
     """Delete Sandbox
 
@@ -100,7 +100,7 @@ def sync(
 async def asyncio_detailed(
     sandbox_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[Sandbox]:
     """Delete Sandbox
 
@@ -129,7 +129,7 @@ async def asyncio_detailed(
 async def asyncio(
     sandbox_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Sandbox | None:
     """Delete Sandbox
 

blaxel/core/client/api/compute/delete_sandbox_preview.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Union
+from typing import Any
 
 import httpx
 
@@ -45,7 +45,7 @@ def sync_detailed(
     sandbox_name: str,
     preview_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[Preview]:
     """Delete Sandbox Preview
 
@@ -79,7 +79,7 @@ def sync(
     sandbox_name: str,
     preview_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Preview | None:
     """Delete Sandbox Preview
 
@@ -108,7 +108,7 @@ async def asyncio_detailed(
     sandbox_name: str,
     preview_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[Preview]:
     """Delete Sandbox Preview
 
@@ -140,7 +140,7 @@ async def asyncio(
     sandbox_name: str,
     preview_name: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Preview | None:
     """Delete Sandbox Preview