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

blaxel/core/client/api/volumes/delete_volume.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.volume import Volume
 from ...types import Response
 
@@ -20,18 +21,38 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Volume | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Error, Volume] | None:
     if response.status_code == 200:
         response_200 = Volume.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 == 409:
+        response_409 = Error.from_dict(response.json())
+
+        return response_409
+    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[Volume]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Error, Volume]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -44,10 +65,11 @@ def sync_detailed(
     volume_name: str,
     *,
     client: Client,
-) -> Response[Volume]:
-    """Delete volume
+) -> Response[Union[Error, Volume]]:
+    """Delete persistent volume
 
-     Deletes a volume by name.
+     Permanently deletes a volume and all its data. The volume must not be attached to any sandbox. This
+    action cannot be undone.
 
     Args:
         volume_name (str):
@@ -57,7 +79,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Volume]
+        Response[Union[Error, Volume]]
     """
 
     kwargs = _get_kwargs(
@@ -75,10 +97,11 @@ def sync(
     volume_name: str,
     *,
     client: Client,
-) -> Volume | None:
-    """Delete volume
+) -> Union[Error, Volume] | None:
+    """Delete persistent volume
 
-     Deletes a volume by name.
+     Permanently deletes a volume and all its data. The volume must not be attached to any sandbox. This
+    action cannot be undone.
 
     Args:
         volume_name (str):
@@ -88,7 +111,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Volume
+        Union[Error, Volume]
     """
 
     return sync_detailed(
@@ -101,10 +124,11 @@ async def asyncio_detailed(
     volume_name: str,
     *,
     client: Client,
-) -> Response[Volume]:
-    """Delete volume
+) -> Response[Union[Error, Volume]]:
+    """Delete persistent volume
 
-     Deletes a volume by name.
+     Permanently deletes a volume and all its data. The volume must not be attached to any sandbox. This
+    action cannot be undone.
 
     Args:
         volume_name (str):
@@ -114,7 +138,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Volume]
+        Response[Union[Error, Volume]]
     """
 
     kwargs = _get_kwargs(
@@ -130,10 +154,11 @@ async def asyncio(
     volume_name: str,
     *,
     client: Client,
-) -> Volume | None:
-    """Delete volume
+) -> Union[Error, Volume] | None:
+    """Delete persistent volume
 
-     Deletes a volume by name.
+     Permanently deletes a volume and all its data. The volume must not be attached to any sandbox. This
+    action cannot be undone.
 
     Args:
         volume_name (str):
@@ -143,7 +168,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Volume
+        Union[Error, Volume]
     """
 
     return (

blaxel/core/client/api/volumes/get_volume.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.volume import Volume
 from ...types import Response
 
@@ -20,18 +21,34 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Volume | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Error, Volume] | None:
     if response.status_code == 200:
         response_200 = Volume.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[Volume]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Error, Volume]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -44,10 +61,11 @@ def sync_detailed(
     volume_name: str,
     *,
     client: Client,
-) -> Response[Volume]:
-    """Get volume
+) -> Response[Union[Error, Volume]]:
+    """Get persistent volume
 
-     Returns a volume by name.
+     Returns detailed information about a volume including its size, region, attachment status, and any
+    events history.
 
     Args:
         volume_name (str):
@@ -57,7 +75,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Volume]
+        Response[Union[Error, Volume]]
     """
 
     kwargs = _get_kwargs(
@@ -75,10 +93,11 @@ def sync(
     volume_name: str,
     *,
     client: Client,
-) -> Volume | None:
-    """Get volume
+) -> Union[Error, Volume] | None:
+    """Get persistent volume
 
-     Returns a volume by name.
+     Returns detailed information about a volume including its size, region, attachment status, and any
+    events history.
 
     Args:
         volume_name (str):
@@ -88,7 +107,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Volume
+        Union[Error, Volume]
     """
 
     return sync_detailed(
@@ -101,10 +120,11 @@ async def asyncio_detailed(
     volume_name: str,
     *,
     client: Client,
-) -> Response[Volume]:
-    """Get volume
+) -> Response[Union[Error, Volume]]:
+    """Get persistent volume
 
-     Returns a volume by name.
+     Returns detailed information about a volume including its size, region, attachment status, and any
+    events history.
 
     Args:
         volume_name (str):
@@ -114,7 +134,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Volume]
+        Response[Union[Error, Volume]]
     """
 
     kwargs = _get_kwargs(
@@ -130,10 +150,11 @@ async def asyncio(
     volume_name: str,
     *,
     client: Client,
-) -> Volume | None:
-    """Get volume
+) -> Union[Error, Volume] | None:
+    """Get persistent volume
 
-     Returns a volume by name.
+     Returns detailed information about a volume including its size, region, attachment status, and any
+    events history.
 
     Args:
         volume_name (str):
@@ -143,7 +164,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Volume
+        Union[Error, Volume]
     """
 
     return (

blaxel/core/client/api/volumes/list_volumes.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.volume import Volume
 from ...types import Response
 
@@ -18,7 +19,9 @@ def _get_kwargs() -> dict[str, Any]:
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> list["Volume"] | None:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[Error, list["Volume"]] | None:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
@@ -28,13 +31,27 @@ def _parse_response(*, client: Client, response: httpx.Response) -> list["Volume
             response_200.append(response_200_item)
 
         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 == 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[list["Volume"]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[Error, list["Volume"]]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -46,17 +63,18 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[lis
 def sync_detailed(
     *,
     client: Client,
-) -> Response[list["Volume"]]:
-    """List volumes
+) -> Response[Union[Error, list["Volume"]]]:
+    """List persistent volumes
 
-     Returns a list of all volumes in the workspace.
+     Returns all persistent storage volumes in the workspace. Volumes can be attached to sandboxes for
+    durable file storage that persists across sessions and sandbox deletions.
 
     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[list['Volume']]
+        Response[Union[Error, list['Volume']]]
     """
 
     kwargs = _get_kwargs()
@@ -71,17 +89,18 @@ def sync_detailed(
 def sync(
     *,
     client: Client,
-) -> list["Volume"] | None:
-    """List volumes
+) -> Union[Error, list["Volume"]] | None:
+    """List persistent volumes
 
-     Returns a list of all volumes in the workspace.
+     Returns all persistent storage volumes in the workspace. Volumes can be attached to sandboxes for
+    durable file storage that persists across sessions and sandbox deletions.
 
     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:
-        list['Volume']
+        Union[Error, list['Volume']]
     """
 
     return sync_detailed(
@@ -92,17 +111,18 @@ def sync(
 async def asyncio_detailed(
     *,
     client: Client,
-) -> Response[list["Volume"]]:
-    """List volumes
+) -> Response[Union[Error, list["Volume"]]]:
+    """List persistent volumes
 
-     Returns a list of all volumes in the workspace.
+     Returns all persistent storage volumes in the workspace. Volumes can be attached to sandboxes for
+    durable file storage that persists across sessions and sandbox deletions.
 
     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[list['Volume']]
+        Response[Union[Error, list['Volume']]]
     """
 
     kwargs = _get_kwargs()
@@ -115,17 +135,18 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: Client,
-) -> list["Volume"] | None:
-    """List volumes
+) -> Union[Error, list["Volume"]] | None:
+    """List persistent volumes
 
-     Returns a list of all volumes in the workspace.
+     Returns all persistent storage volumes in the workspace. Volumes can be attached to sandboxes for
+    durable file storage that persists across sessions and sandbox deletions.
 
     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:
-        list['Volume']
+        Union[Error, list['Volume']]
     """
 
     return (

blaxel/core/client/api/workspaces/create_workspace.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.workspace import Workspace
 from ...types import Response
 
@@ -32,18 +33,40 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Workspace | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Error, Workspace] | None:
     if response.status_code == 200:
         response_200 = Workspace.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 == 409:
+        response_409 = Error.from_dict(response.json())
+
+        return response_409
+    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[Workspace]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[Error, Workspace]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -56,20 +79,22 @@ def sync_detailed(
     *,
     client: Client,
     body: Workspace,
-) -> Response[Workspace]:
-    """Create worspace
+) -> Response[Union[Error, Workspace]]:
+    """Create workspace tenant
 
-     Creates a workspace.
+     Creates a new workspace tenant. The authenticated user becomes the workspace admin. Requires a
+    linked billing account.
 
     Args:
-        body (Workspace): Workspace
+        body (Workspace): Tenant container that groups all Blaxel resources (agents, functions,
+            models, etc.) with shared team access control and billing.
 
     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[Workspace]
+        Response[Union[Error, Workspace]]
     """
 
     kwargs = _get_kwargs(
@@ -87,20 +112,22 @@ def sync(
     *,
     client: Client,
     body: Workspace,
-) -> Workspace | None:
-    """Create worspace
+) -> Union[Error, Workspace] | None:
+    """Create workspace tenant
 
-     Creates a workspace.
+     Creates a new workspace tenant. The authenticated user becomes the workspace admin. Requires a
+    linked billing account.
 
     Args:
-        body (Workspace): Workspace
+        body (Workspace): Tenant container that groups all Blaxel resources (agents, functions,
+            models, etc.) with shared team access control and billing.
 
     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:
-        Workspace
+        Union[Error, Workspace]
     """
 
     return sync_detailed(
@@ -113,20 +140,22 @@ async def asyncio_detailed(
     *,
     client: Client,
     body: Workspace,
-) -> Response[Workspace]:
-    """Create worspace
+) -> Response[Union[Error, Workspace]]:
+    """Create workspace tenant
 
-     Creates a workspace.
+     Creates a new workspace tenant. The authenticated user becomes the workspace admin. Requires a
+    linked billing account.
 
     Args:
-        body (Workspace): Workspace
+        body (Workspace): Tenant container that groups all Blaxel resources (agents, functions,
+            models, etc.) with shared team access control and billing.
 
     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[Workspace]
+        Response[Union[Error, Workspace]]
     """
 
     kwargs = _get_kwargs(
@@ -142,20 +171,22 @@ async def asyncio(
     *,
     client: Client,
     body: Workspace,
-) -> Workspace | None:
-    """Create worspace
+) -> Union[Error, Workspace] | None:
+    """Create workspace tenant
 
-     Creates a workspace.
+     Creates a new workspace tenant. The authenticated user becomes the workspace admin. Requires a
+    linked billing account.
 
     Args:
-        body (Workspace): Workspace
+        body (Workspace): Tenant container that groups all Blaxel resources (agents, functions,
+            models, etc.) with shared team access control and billing.
 
     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:
-        Workspace
+        Union[Error, Workspace]
     """
 
     return (