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

blaxel/core/sandbox/__init__.py

@@ -12,6 +12,7 @@ from .default import (
     SandboxInstance,
     SandboxPreviews,
     SandboxProcess,
+    SandboxSystem,
 )
 from .sync import (
     SyncCodeInterpreter,
@@ -20,8 +21,11 @@ from .sync import (
     SyncSandboxInstance,
     SyncSandboxPreviews,
     SyncSandboxProcess,
+    SyncSandboxSystem,
 )
 from .types import (
+    AsyncStreamHandle,
+    AsyncWatchHandle,
     CopyResponse,
     ProcessRequestWithLog,
     ProcessResponseWithLog,
@@ -30,7 +34,9 @@ from .types import (
     SandboxFilesystemFile,
     SessionCreateOptions,
     SessionWithToken,
+    StreamHandle,
     WatchEvent,
+    WatchHandle,
 )
 
 __all__ = [
@@ -41,6 +47,10 @@ __all__ = [
     "SandboxConfiguration",
     "SandboxCreateConfiguration",
     "WatchEvent",
+    "WatchHandle",
+    "AsyncWatchHandle",
+    "StreamHandle",
+    "AsyncStreamHandle",
     "SandboxFilesystemFile",
     "CopyResponse",
     "Sandbox",
@@ -48,6 +58,7 @@ __all__ = [
     "SandboxPreviews",
     "SandboxProcess",
     "SandboxCodegen",
+    "SandboxSystem",
     "ProcessRequestWithLog",
     "ProcessResponseWithLog",
     "ApplyEditRequest",
@@ -59,6 +70,7 @@ __all__ = [
     "SyncSandboxInstance",
     "SyncSandboxPreviews",
     "SyncSandboxProcess",
+    "SyncSandboxSystem",
     "SyncCodeInterpreter",
     "CodeInterpreter",
 ]

blaxel/core/{client/api/invitations/list_all_pending_invitations.py → sandbox/client/api/system/get_health.py}

@@ -1,47 +1,35 @@
 from http import HTTPStatus
-from typing import Any, Union, cast
+from typing import Any
 
 import httpx
 
 from ... import errors
 from ...client import Client
-from ...models.pending_invitation_render import PendingInvitationRender
+from ...models.health_response import HealthResponse
 from ...types import Response
 
 
 def _get_kwargs() -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": "/profile/invitations",
+        "url": "/health",
     }
 
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[Any, list["PendingInvitationRender"]] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> HealthResponse | None:
     if response.status_code == 200:
-        response_200 = []
-        _response_200 = response.json()
-        for response_200_item_data in _response_200:
-            response_200_item = PendingInvitationRender.from_dict(response_200_item_data)
-
-            response_200.append(response_200_item)
+        response_200 = HealthResponse.from_dict(response.json())
 
         return response_200
-    if response.status_code == 404:
-        response_404 = cast(Any, None)
-        return response_404
     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[Union[Any, list["PendingInvitationRender"]]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[HealthResponse]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -53,17 +41,18 @@ def _build_response(
 def sync_detailed(
     *,
     client: Client,
-) -> Response[Union[Any, list["PendingInvitationRender"]]]:
-    """List pending invitations
+) -> Response[HealthResponse]:
+    """Health check
 
-     Returns a list of all pending invitations in the workspace.
+     Returns health status and system information including upgrade count and binary details
+    Also includes last upgrade attempt status with detailed error information if available
 
     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[Union[Any, list['PendingInvitationRender']]]
+        Response[HealthResponse]
     """
 
     kwargs = _get_kwargs()
@@ -78,17 +67,18 @@ def sync_detailed(
 def sync(
     *,
     client: Client,
-) -> Union[Any, list["PendingInvitationRender"]] | None:
-    """List pending invitations
+) -> HealthResponse | None:
+    """Health check
 
-     Returns a list of all pending invitations in the workspace.
+     Returns health status and system information including upgrade count and binary details
+    Also includes last upgrade attempt status with detailed error information if available
 
     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:
-        Union[Any, list['PendingInvitationRender']]
+        HealthResponse
     """
 
     return sync_detailed(
@@ -99,17 +89,18 @@ def sync(
 async def asyncio_detailed(
     *,
     client: Client,
-) -> Response[Union[Any, list["PendingInvitationRender"]]]:
-    """List pending invitations
+) -> Response[HealthResponse]:
+    """Health check
 
-     Returns a list of all pending invitations in the workspace.
+     Returns health status and system information including upgrade count and binary details
+    Also includes last upgrade attempt status with detailed error information if available
 
     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[Union[Any, list['PendingInvitationRender']]]
+        Response[HealthResponse]
     """
 
     kwargs = _get_kwargs()
@@ -122,17 +113,18 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: Client,
-) -> Union[Any, list["PendingInvitationRender"]] | None:
-    """List pending invitations
+) -> HealthResponse | None:
+    """Health check
 
-     Returns a list of all pending invitations in the workspace.
+     Returns health status and system information including upgrade count and binary details
+    Also includes last upgrade attempt status with detailed error information if available
 
     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:
-        Union[Any, list['PendingInvitationRender']]
+        HealthResponse
     """
 
     return (

blaxel/core/sandbox/client/api/system/post_upgrade.py

@@ -0,0 +1,196 @@
+from http import HTTPStatus
+from typing import Any, Union
+
+import httpx
+
+from ... import errors
+from ...client import Client
+from ...models.error_response import ErrorResponse
+from ...models.success_response import SuccessResponse
+from ...models.upgrade_request import UpgradeRequest
+from ...types import Response
+
+
+def _get_kwargs(
+    *,
+    body: UpgradeRequest,
+) -> dict[str, Any]:
+    headers: dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/upgrade",
+    }
+
+    if type(body) is dict:
+        _body = body
+    else:
+        _body = body.to_dict()
+
+    _kwargs["json"] = _body
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[ErrorResponse, SuccessResponse] | None:
+    if response.status_code == 200:
+        response_200 = SuccessResponse.from_dict(response.json())
+
+        return response_200
+    if response.status_code == 500:
+        response_500 = ErrorResponse.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[Union[ErrorResponse, SuccessResponse]]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: Client,
+    body: UpgradeRequest,
+) -> Response[Union[ErrorResponse, SuccessResponse]]:
+    r"""Upgrade the sandbox-api
+
+     Triggers an upgrade of the sandbox-api process. Returns 200 immediately before upgrading.
+    The upgrade will: download the specified binary from GitHub releases, validate it, and restart.
+    All running processes will be preserved across the upgrade.
+    Available versions: \"develop\" (default), \"main\", \"latest\", or specific tag like \"v1.0.0\"
+    You can also specify a custom baseUrl for forks (defaults to https://github.com/blaxel-
+    ai/sandbox/releases)
+
+    Args:
+        body (UpgradeRequest):
+
+    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[Union[ErrorResponse, SuccessResponse]]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: Client,
+    body: UpgradeRequest,
+) -> Union[ErrorResponse, SuccessResponse] | None:
+    r"""Upgrade the sandbox-api
+
+     Triggers an upgrade of the sandbox-api process. Returns 200 immediately before upgrading.
+    The upgrade will: download the specified binary from GitHub releases, validate it, and restart.
+    All running processes will be preserved across the upgrade.
+    Available versions: \"develop\" (default), \"main\", \"latest\", or specific tag like \"v1.0.0\"
+    You can also specify a custom baseUrl for forks (defaults to https://github.com/blaxel-
+    ai/sandbox/releases)
+
+    Args:
+        body (UpgradeRequest):
+
+    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:
+        Union[ErrorResponse, SuccessResponse]
+    """
+
+    return sync_detailed(
+        client=client,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: Client,
+    body: UpgradeRequest,
+) -> Response[Union[ErrorResponse, SuccessResponse]]:
+    r"""Upgrade the sandbox-api
+
+     Triggers an upgrade of the sandbox-api process. Returns 200 immediately before upgrading.
+    The upgrade will: download the specified binary from GitHub releases, validate it, and restart.
+    All running processes will be preserved across the upgrade.
+    Available versions: \"develop\" (default), \"main\", \"latest\", or specific tag like \"v1.0.0\"
+    You can also specify a custom baseUrl for forks (defaults to https://github.com/blaxel-
+    ai/sandbox/releases)
+
+    Args:
+        body (UpgradeRequest):
+
+    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[Union[ErrorResponse, SuccessResponse]]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: Client,
+    body: UpgradeRequest,
+) -> Union[ErrorResponse, SuccessResponse] | None:
+    r"""Upgrade the sandbox-api
+
+     Triggers an upgrade of the sandbox-api process. Returns 200 immediately before upgrading.
+    The upgrade will: download the specified binary from GitHub releases, validate it, and restart.
+    All running processes will be preserved across the upgrade.
+    Available versions: \"develop\" (default), \"main\", \"latest\", or specific tag like \"v1.0.0\"
+    You can also specify a custom baseUrl for forks (defaults to https://github.com/blaxel-
+    ai/sandbox/releases)
+
+    Args:
+        body (UpgradeRequest):
+
+    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:
+        Union[ErrorResponse, SuccessResponse]
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+            body=body,
+        )
+    ).parsed

blaxel/core/sandbox/client/models/__init__.py

@@ -20,6 +20,7 @@ from .find_response import FindResponse
 from .fuzzy_search_match import FuzzySearchMatch
 from .fuzzy_search_response import FuzzySearchResponse
 from .get_network_process_pid_ports_response_200 import GetNetworkProcessPidPortsResponse200
+from .health_response import HealthResponse
 from .multipart_complete_request import MultipartCompleteRequest
 from .multipart_initiate_request import MultipartInitiateRequest
 from .multipart_initiate_response import MultipartInitiateResponse
@@ -34,6 +35,7 @@ from .process_request import ProcessRequest
 from .process_request_env import ProcessRequestEnv
 from .process_response import ProcessResponse
 from .process_response_status import ProcessResponseStatus
+from .process_upgrade_state import ProcessUpgradeState
 from .put_filesystem_multipart_upload_id_part_body import PutFilesystemMultipartUploadIdPartBody
 from .ranked_file import RankedFile
 from .reranking_response import RerankingResponse
@@ -41,6 +43,8 @@ from .subdirectory import Subdirectory
 from .success_response import SuccessResponse
 from .tree_request import TreeRequest
 from .tree_request_files import TreeRequestFiles
+from .upgrade_request import UpgradeRequest
+from .upgrade_status import UpgradeStatus
 
 __all__ = (
     "ApplyEditRequest",
@@ -61,6 +65,7 @@ __all__ = (
     "FuzzySearchMatch",
     "FuzzySearchResponse",
     "GetNetworkProcessPidPortsResponse200",
+    "HealthResponse",
     "MultipartCompleteRequest",
     "MultipartInitiateRequest",
     "MultipartInitiateResponse",
@@ -75,6 +80,7 @@ __all__ = (
     "ProcessRequestEnv",
     "ProcessResponse",
     "ProcessResponseStatus",
+    "ProcessUpgradeState",
     "PutFilesystemMultipartUploadIdPartBody",
     "RankedFile",
     "RerankingResponse",
@@ -82,4 +88,6 @@ __all__ = (
     "SuccessResponse",
     "TreeRequest",
     "TreeRequestFiles",
+    "UpgradeRequest",
+    "UpgradeStatus",
 )

blaxel/core/sandbox/client/models/content_search_match.py

@@ -12,46 +12,45 @@ T = TypeVar("T", bound="ContentSearchMatch")
 class ContentSearchMatch:
     """
     Attributes:
-        column (Union[Unset, int]):  Example: 10.
+        column (int):  Example: 10.
+        line (int):  Example: 42.
+        path (str):  Example: src/main.go.
+        text (str):  Example: const searchText = 'example'.
         context (Union[Unset, str]):  Example: previous line
             current line
             next line.
-        line (Union[Unset, int]):  Example: 42.
-        path (Union[Unset, str]):  Example: src/main.go.
-        text (Union[Unset, str]):  Example: const searchText = 'example'.
     """
 
-    column: Union[Unset, int] = UNSET
+    column: int
+    line: int
+    path: str
+    text: str
     context: Union[Unset, str] = UNSET
-    line: Union[Unset, int] = UNSET
-    path: Union[Unset, str] = UNSET
-    text: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
         column = self.column
 
-        context = self.context
-
         line = self.line
 
         path = self.path
 
         text = self.text
 
+        context = self.context
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if column is not UNSET:
-            field_dict["column"] = column
+        field_dict.update(
+            {
+                "column": column,
+                "line": line,
+                "path": path,
+                "text": text,
+            }
+        )
         if context is not UNSET:
             field_dict["context"] = context
-        if line is not UNSET:
-            field_dict["line"] = line
-        if path is not UNSET:
-            field_dict["path"] = path
-        if text is not UNSET:
-            field_dict["text"] = text
 
         return field_dict
 
@@ -60,22 +59,22 @@ class ContentSearchMatch:
         if not src_dict:
             return None
         d = src_dict.copy()
-        column = d.pop("column", UNSET)
+        column = d.pop("column")
 
-        context = d.pop("context", UNSET)
+        line = d.pop("line")
 
-        line = d.pop("line", UNSET)
+        path = d.pop("path")
 
-        path = d.pop("path", UNSET)
+        text = d.pop("text")
 
-        text = d.pop("text", UNSET)
+        context = d.pop("context", UNSET)
 
         content_search_match = cls(
             column=column,
-            context=context,
             line=line,
             path=path,
             text=text,
+            context=context,
         )
 
         content_search_match.additional_properties = d

blaxel/core/sandbox/client/models/content_search_response.py

@@ -1,10 +1,8 @@
-from typing import TYPE_CHECKING, Any, TypeVar, Union
+from typing import TYPE_CHECKING, Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 if TYPE_CHECKING:
     from ..models.content_search_match import ContentSearchMatch
 
@@ -16,26 +14,24 @@ T = TypeVar("T", bound="ContentSearchResponse")
 class ContentSearchResponse:
     """
     Attributes:
-        matches (Union[Unset, list['ContentSearchMatch']]):
-        query (Union[Unset, str]):  Example: searchText.
-        total (Union[Unset, int]):  Example: 5.
+        matches (list['ContentSearchMatch']):
+        query (str):  Example: searchText.
+        total (int):  Example: 5.
     """
 
-    matches: Union[Unset, list["ContentSearchMatch"]] = UNSET
-    query: Union[Unset, str] = UNSET
-    total: Union[Unset, int] = UNSET
+    matches: list["ContentSearchMatch"]
+    query: str
+    total: int
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        matches: Union[Unset, list[dict[str, Any]]] = UNSET
-        if not isinstance(self.matches, Unset):
-            matches = []
-            for matches_item_data in self.matches:
-                if type(matches_item_data) is dict:
-                    matches_item = matches_item_data
-                else:
-                    matches_item = matches_item_data.to_dict()
-                matches.append(matches_item)
+        matches = []
+        for matches_item_data in self.matches:
+            if type(matches_item_data) is dict:
+                matches_item = matches_item_data
+            else:
+                matches_item = matches_item_data.to_dict()
+            matches.append(matches_item)
 
         query = self.query
 
@@ -43,13 +39,13 @@ class ContentSearchResponse:
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if matches is not UNSET:
-            field_dict["matches"] = matches
-        if query is not UNSET:
-            field_dict["query"] = query
-        if total is not UNSET:
-            field_dict["total"] = total
+        field_dict.update(
+            {
+                "matches": matches,
+                "query": query,
+                "total": total,
+            }
+        )
 
         return field_dict
 
@@ -61,15 +57,15 @@ class ContentSearchResponse:
             return None
         d = src_dict.copy()
         matches = []
-        _matches = d.pop("matches", UNSET)
-        for matches_item_data in _matches or []:
+        _matches = d.pop("matches")
+        for matches_item_data in _matches:
             matches_item = ContentSearchMatch.from_dict(matches_item_data)
 
             matches.append(matches_item)
 
-        query = d.pop("query", UNSET)
+        query = d.pop("query")
 
-        total = d.pop("total", UNSET)
+        total = d.pop("total")
 
         content_search_response = cls(
             matches=matches,

blaxel/core/sandbox/client/models/find_match.py

@@ -1,10 +1,8 @@
-from typing import Any, TypeVar, Union
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 T = TypeVar("T", bound="FindMatch")
 
 
@@ -12,12 +10,12 @@ T = TypeVar("T", bound="FindMatch")
 class FindMatch:
     """
     Attributes:
-        path (Union[Unset, str]):  Example: src/main.go.
-        type_ (Union[Unset, str]): "file" or "directory" Example: file.
+        path (str):  Example: src/main.go.
+        type_ (str): "file" or "directory" Example: file.
     """
 
-    path: Union[Unset, str] = UNSET
-    type_: Union[Unset, str] = UNSET
+    path: str
+    type_: str
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -27,11 +25,12 @@ class FindMatch:
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if path is not UNSET:
-            field_dict["path"] = path
-        if type_ is not UNSET:
-            field_dict["type"] = type_
+        field_dict.update(
+            {
+                "path": path,
+                "type": type_,
+            }
+        )
 
         return field_dict
 
@@ -40,9 +39,9 @@ class FindMatch:
         if not src_dict:
             return None
         d = src_dict.copy()
-        path = d.pop("path", UNSET)
+        path = d.pop("path")
 
-        type_ = d.pop("type", d.pop("type_", UNSET))
+        type_ = d.pop("type") if "type" in d else d.pop("type_")
 
         find_match = cls(
             path=path,