blaxel 0.2.34__py3-none-any.whl → 0.2.35__py3-none-any.whl

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

@@ -1,11 +1,12 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
 from ...models.agent import Agent
+from ...models.error import Error
 from ...types import Response
 
 
@@ -33,18 +34,38 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Agent | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Agent, Error] | None:
     if response.status_code == 200:
         response_200 = Agent.from_dict(response.json())
 
         return response_200
+    if response.status_code == 400:
+        response_400 = Error.from_dict(response.json())
+
+        return response_400
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+    if response.status_code == 404:
+        response_404 = Error.from_dict(response.json())
+
+        return response_404
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
     if client.raise_on_unexpected_status:
         raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Agent]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Agent, Error]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -58,19 +79,24 @@ def sync_detailed(
     *,
     client: Client,
     body: Agent,
-) -> Response[Agent]:
-    """Update agent by name
+) -> Response[Union[Agent, Error]]:
+    """Update agent
+
+     Updates an agent's configuration and triggers a new deployment. Changes to runtime settings,
+    environment variables, or scaling parameters will be applied on the next deployment.
 
     Args:
         agent_name (str):
-        body (Agent): Agent
+        body (Agent): Serverless AI agent deployment that runs your custom agent code as an auto-
+            scaling API endpoint. Agents are deployed from your code repository and expose a global
+            inference URL for querying.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Agent]
+        Response[Union[Agent, Error]]
     """
 
     kwargs = _get_kwargs(
@@ -90,19 +116,24 @@ def sync(
     *,
     client: Client,
     body: Agent,
-) -> Agent | None:
-    """Update agent by name
+) -> Union[Agent, Error] | None:
+    """Update agent
+
+     Updates an agent's configuration and triggers a new deployment. Changes to runtime settings,
+    environment variables, or scaling parameters will be applied on the next deployment.
 
     Args:
         agent_name (str):
-        body (Agent): Agent
+        body (Agent): Serverless AI agent deployment that runs your custom agent code as an auto-
+            scaling API endpoint. Agents are deployed from your code repository and expose a global
+            inference URL for querying.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Agent
+        Union[Agent, Error]
     """
 
     return sync_detailed(
@@ -117,19 +148,24 @@ async def asyncio_detailed(
     *,
     client: Client,
     body: Agent,
-) -> Response[Agent]:
-    """Update agent by name
+) -> Response[Union[Agent, Error]]:
+    """Update agent
+
+     Updates an agent's configuration and triggers a new deployment. Changes to runtime settings,
+    environment variables, or scaling parameters will be applied on the next deployment.
 
     Args:
         agent_name (str):
-        body (Agent): Agent
+        body (Agent): Serverless AI agent deployment that runs your custom agent code as an auto-
+            scaling API endpoint. Agents are deployed from your code repository and expose a global
+            inference URL for querying.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Agent]
+        Response[Union[Agent, Error]]
     """
 
     kwargs = _get_kwargs(
@@ -147,19 +183,24 @@ async def asyncio(
     *,
     client: Client,
     body: Agent,
-) -> Agent | None:
-    """Update agent by name
+) -> Union[Agent, Error] | None:
+    """Update agent
+
+     Updates an agent's configuration and triggers a new deployment. Changes to runtime settings,
+    environment variables, or scaling parameters will be applied on the next deployment.
 
     Args:
         agent_name (str):
-        body (Agent): Agent
+        body (Agent): Serverless AI agent deployment that runs your custom agent code as an auto-
+            scaling API endpoint. Agents are deployed from your code repository and expose a global
+            inference URL for querying.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Agent
+        Union[Agent, Error]
     """
 
     return (

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

@@ -1,11 +1,12 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
 from ...models.sandbox import Sandbox
+from ...models.sandbox_error import SandboxError
 from ...types import Response
 
 
@@ -32,18 +33,42 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Sandbox | None:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[Sandbox, SandboxError] | None:
     if response.status_code == 200:
         response_200 = Sandbox.from_dict(response.json())
 
         return response_200
+    if response.status_code == 400:
+        response_400 = SandboxError.from_dict(response.json())
+
+        return response_400
+    if response.status_code == 401:
+        response_401 = SandboxError.from_dict(response.json())
+
+        return response_401
+    if response.status_code == 403:
+        response_403 = SandboxError.from_dict(response.json())
+
+        return response_403
+    if response.status_code == 409:
+        response_409 = SandboxError.from_dict(response.json())
+
+        return response_409
+    if response.status_code == 500:
+        response_500 = SandboxError.from_dict(response.json())
+
+        return response_500
     if client.raise_on_unexpected_status:
         raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Sandbox]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[Sandbox, SandboxError]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -56,20 +81,23 @@ def sync_detailed(
     *,
     client: Client,
     body: Sandbox,
-) -> Response[Sandbox]:
-    """Create Sandbox
+) -> Response[Union[Sandbox, SandboxError]]:
+    """Create sandbox
 
-     Creates a Sandbox.
+     Creates a new sandbox VM for secure AI code execution. Sandboxes automatically scale to zero when
+    idle and resume instantly, preserving memory state including running processes and filesystem.
 
     Args:
-        body (Sandbox): Micro VM for running agentic tasks
+        body (Sandbox): Lightweight virtual machine for secure AI code execution. Sandboxes resume
+            from standby in under 25ms and automatically scale to zero after inactivity, preserving
+            memory state including running processes and filesystem.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Sandbox]
+        Response[Union[Sandbox, SandboxError]]
     """
 
     kwargs = _get_kwargs(
@@ -87,20 +115,23 @@ def sync(
     *,
     client: Client,
     body: Sandbox,
-) -> Sandbox | None:
-    """Create Sandbox
+) -> Union[Sandbox, SandboxError] | None:
+    """Create sandbox
 
-     Creates a Sandbox.
+     Creates a new sandbox VM for secure AI code execution. Sandboxes automatically scale to zero when
+    idle and resume instantly, preserving memory state including running processes and filesystem.
 
     Args:
-        body (Sandbox): Micro VM for running agentic tasks
+        body (Sandbox): Lightweight virtual machine for secure AI code execution. Sandboxes resume
+            from standby in under 25ms and automatically scale to zero after inactivity, preserving
+            memory state including running processes and filesystem.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Sandbox
+        Union[Sandbox, SandboxError]
     """
 
     return sync_detailed(
@@ -113,20 +144,23 @@ async def asyncio_detailed(
     *,
     client: Client,
     body: Sandbox,
-) -> Response[Sandbox]:
-    """Create Sandbox
+) -> Response[Union[Sandbox, SandboxError]]:
+    """Create sandbox
 
-     Creates a Sandbox.
+     Creates a new sandbox VM for secure AI code execution. Sandboxes automatically scale to zero when
+    idle and resume instantly, preserving memory state including running processes and filesystem.
 
     Args:
-        body (Sandbox): Micro VM for running agentic tasks
+        body (Sandbox): Lightweight virtual machine for secure AI code execution. Sandboxes resume
+            from standby in under 25ms and automatically scale to zero after inactivity, preserving
+            memory state including running processes and filesystem.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Sandbox]
+        Response[Union[Sandbox, SandboxError]]
     """
 
     kwargs = _get_kwargs(
@@ -142,20 +176,23 @@ async def asyncio(
     *,
     client: Client,
     body: Sandbox,
-) -> Sandbox | None:
-    """Create Sandbox
+) -> Union[Sandbox, SandboxError] | None:
+    """Create sandbox
 
-     Creates a Sandbox.
+     Creates a new sandbox VM for secure AI code execution. Sandboxes automatically scale to zero when
+    idle and resume instantly, preserving memory state including running processes and filesystem.
 
     Args:
-        body (Sandbox): Micro VM for running agentic tasks
+        body (Sandbox): Lightweight virtual machine for secure AI code execution. Sandboxes resume
+            from standby in under 25ms and automatically scale to zero after inactivity, preserving
+            memory state including running processes and filesystem.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Sandbox
+        Union[Sandbox, SandboxError]
     """
 
     return (

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

@@ -1,10 +1,11 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
+from ...models.error import Error
 from ...models.sandbox import Sandbox
 from ...types import Response
 
@@ -20,18 +21,34 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Sandbox | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Error, Sandbox] | None:
     if response.status_code == 200:
         response_200 = Sandbox.from_dict(response.json())
 
         return response_200
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+    if response.status_code == 404:
+        response_404 = Error.from_dict(response.json())
+
+        return response_404
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
     if client.raise_on_unexpected_status:
         raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Sandbox]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Error, Sandbox]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -44,10 +61,11 @@ def sync_detailed(
     sandbox_name: str,
     *,
     client: Client,
-) -> Response[Sandbox]:
-    """Delete Sandbox
+) -> Response[Union[Error, Sandbox]]:
+    """Delete sandbox
 
-     Deletes a Sandbox by name.
+     Permanently deletes a sandbox and all its data. If no volumes are attached, this guarantees zero
+    data retention (ZDR). This action cannot be undone.
 
     Args:
         sandbox_name (str):
@@ -57,7 +75,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Sandbox]
+        Response[Union[Error, Sandbox]]
     """
 
     kwargs = _get_kwargs(
@@ -75,10 +93,11 @@ def sync(
     sandbox_name: str,
     *,
     client: Client,
-) -> Sandbox | None:
-    """Delete Sandbox
+) -> Union[Error, Sandbox] | None:
+    """Delete sandbox
 
-     Deletes a Sandbox by name.
+     Permanently deletes a sandbox and all its data. If no volumes are attached, this guarantees zero
+    data retention (ZDR). This action cannot be undone.
 
     Args:
         sandbox_name (str):
@@ -88,7 +107,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Sandbox
+        Union[Error, Sandbox]
     """
 
     return sync_detailed(
@@ -101,10 +120,11 @@ async def asyncio_detailed(
     sandbox_name: str,
     *,
     client: Client,
-) -> Response[Sandbox]:
-    """Delete Sandbox
+) -> Response[Union[Error, Sandbox]]:
+    """Delete sandbox
 
-     Deletes a Sandbox by name.
+     Permanently deletes a sandbox and all its data. If no volumes are attached, this guarantees zero
+    data retention (ZDR). This action cannot be undone.
 
     Args:
         sandbox_name (str):
@@ -114,7 +134,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Sandbox]
+        Response[Union[Error, Sandbox]]
     """
 
     kwargs = _get_kwargs(
@@ -130,10 +150,11 @@ async def asyncio(
     sandbox_name: str,
     *,
     client: Client,
-) -> Sandbox | None:
-    """Delete Sandbox
+) -> Union[Error, Sandbox] | None:
+    """Delete sandbox
 
-     Deletes a Sandbox by name.
+     Permanently deletes a sandbox and all its data. If no volumes are attached, this guarantees zero
+    data retention (ZDR). This action cannot be undone.
 
     Args:
         sandbox_name (str):
@@ -143,7 +164,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Sandbox
+        Union[Error, Sandbox]
     """
 
     return (

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

@@ -5,6 +5,7 @@ import httpx
 
 from ... import errors
 from ...client import Client
+from ...models.error import Error
 from ...models.sandbox import Sandbox
 from ...types import UNSET, Response, Unset
 
@@ -29,18 +30,34 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Sandbox | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Error, Sandbox] | None:
     if response.status_code == 200:
         response_200 = Sandbox.from_dict(response.json())
 
         return response_200
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+    if response.status_code == 404:
+        response_404 = Error.from_dict(response.json())
+
+        return response_404
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
     if client.raise_on_unexpected_status:
         raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Sandbox]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Error, Sandbox]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -54,10 +71,11 @@ def sync_detailed(
     *,
     client: Client,
     show_secrets: Union[Unset, bool] = UNSET,
-) -> Response[Sandbox]:
-    """Get Sandbox
+) -> Response[Union[Error, Sandbox]]:
+    """Get sandbox
 
-     Returns a Sandbox by name.
+     Returns detailed information about a sandbox including its configuration, attached volumes,
+    lifecycle policies, and API endpoint URL.
 
     Args:
         sandbox_name (str):
@@ -68,7 +86,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Sandbox]
+        Response[Union[Error, Sandbox]]
     """
 
     kwargs = _get_kwargs(
@@ -88,10 +106,11 @@ def sync(
     *,
     client: Client,
     show_secrets: Union[Unset, bool] = UNSET,
-) -> Sandbox | None:
-    """Get Sandbox
+) -> Union[Error, Sandbox] | None:
+    """Get sandbox
 
-     Returns a Sandbox by name.
+     Returns detailed information about a sandbox including its configuration, attached volumes,
+    lifecycle policies, and API endpoint URL.
 
     Args:
         sandbox_name (str):
@@ -102,7 +121,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Sandbox
+        Union[Error, Sandbox]
     """
 
     return sync_detailed(
@@ -117,10 +136,11 @@ async def asyncio_detailed(
     *,
     client: Client,
     show_secrets: Union[Unset, bool] = UNSET,
-) -> Response[Sandbox]:
-    """Get Sandbox
+) -> Response[Union[Error, Sandbox]]:
+    """Get sandbox
 
-     Returns a Sandbox by name.
+     Returns detailed information about a sandbox including its configuration, attached volumes,
+    lifecycle policies, and API endpoint URL.
 
     Args:
         sandbox_name (str):
@@ -131,7 +151,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Sandbox]
+        Response[Union[Error, Sandbox]]
     """
 
     kwargs = _get_kwargs(
@@ -149,10 +169,11 @@ async def asyncio(
     *,
     client: Client,
     show_secrets: Union[Unset, bool] = UNSET,
-) -> Sandbox | None:
-    """Get Sandbox
+) -> Union[Error, Sandbox] | None:
+    """Get sandbox
 
-     Returns a Sandbox by name.
+     Returns detailed information about a sandbox including its configuration, attached volumes,
+    lifecycle policies, and API endpoint URL.
 
     Args:
         sandbox_name (str):
@@ -163,7 +184,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Sandbox
+        Union[Error, Sandbox]
     """
 
     return (