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

blaxel/__init__.py

@@ -4,8 +4,8 @@ from .core.common.autoload import autoload
 from .core.common.env import env
 from .core.common.settings import settings
 
-__version__ = "0.2.36"
-__commit__ = "70f38e31dd50b26cbb54665420e165b89f3fa4b9"
+__version__ = "0.2.37"
+__commit__ = "a09d7ea6c786074ceaa4639ce6415958524a7a5f"
 __sentry_dsn__ = "https://9711de13cd02b285ca4378c01de8dc30@o4508714045276160.ingest.us.sentry.io/4510461121462272"
 __all__ = ["autoload", "settings", "env"]
 

blaxel/core/client/models/create_job_execution_request_env.py

@@ -15,7 +15,7 @@ class CreateJobExecutionRequestEnv:
 
     """
 
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+    additional_properties: dict[str, str] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
         field_dict: dict[str, Any] = {}
@@ -37,10 +37,10 @@ class CreateJobExecutionRequestEnv:
     def additional_keys(self) -> list[str]:
         return list(self.additional_properties.keys())
 
-    def __getitem__(self, key: str) -> Any:
+    def __getitem__(self, key: str) -> str:
         return self.additional_properties[key]
 
-    def __setitem__(self, key: str, value: Any) -> None:
+    def __setitem__(self, key: str, value: str) -> None:
         self.additional_properties[key] = value
 
     def __delitem__(self, key: str) -> None:

blaxel/core/client/models/preview.py

@@ -1,9 +1,13 @@
-from typing import TYPE_CHECKING, Any, TypeVar
+from typing import TYPE_CHECKING, Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
+from ..models.status import Status
+from ..types import UNSET, Unset
+
 if TYPE_CHECKING:
+    from ..models.core_event import CoreEvent
     from ..models.preview_metadata import PreviewMetadata
     from ..models.preview_spec import PreviewSpec
 
@@ -18,10 +22,14 @@ class Preview:
     Attributes:
         metadata (PreviewMetadata): PreviewMetadata
         spec (PreviewSpec): Preview of a Resource
+        events (Union[Unset, list['CoreEvent']]): Events happening on a resource deployed on Blaxel
+        status (Union[Unset, Status]): Deployment status of a resource deployed on Blaxel
     """
 
     metadata: "PreviewMetadata"
     spec: "PreviewSpec"
+    events: Union[Unset, list["CoreEvent"]] = UNSET
+    status: Union[Unset, Status] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -35,6 +43,22 @@ class Preview:
         else:
             spec = self.spec.to_dict()
 
+        events: Union[Unset, list[dict[str, Any]]] = UNSET
+        if not isinstance(self.events, Unset):
+            events = []
+            for componentsschemas_core_events_item_data in self.events:
+                if type(componentsschemas_core_events_item_data) is dict:
+                    componentsschemas_core_events_item = componentsschemas_core_events_item_data
+                else:
+                    componentsschemas_core_events_item = (
+                        componentsschemas_core_events_item_data.to_dict()
+                    )
+                events.append(componentsschemas_core_events_item)
+
+        status: Union[Unset, str] = UNSET
+        if not isinstance(self.status, Unset):
+            status = self.status.value
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -43,11 +67,16 @@ class Preview:
                 "spec": spec,
             }
         )
+        if events is not UNSET:
+            field_dict["events"] = events
+        if status is not UNSET:
+            field_dict["status"] = status
 
         return field_dict
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
+        from ..models.core_event import CoreEvent
         from ..models.preview_metadata import PreviewMetadata
         from ..models.preview_spec import PreviewSpec
 
@@ -58,9 +87,27 @@ class Preview:
 
         spec = PreviewSpec.from_dict(d.pop("spec"))
 
+        events = []
+        _events = d.pop("events", UNSET)
+        for componentsschemas_core_events_item_data in _events or []:
+            componentsschemas_core_events_item = CoreEvent.from_dict(
+                componentsschemas_core_events_item_data
+            )
+
+            events.append(componentsschemas_core_events_item)
+
+        _status = d.pop("status", UNSET)
+        status: Union[Unset, Status]
+        if isinstance(_status, Unset):
+            status = UNSET
+        else:
+            status = Status(_status)
+
         preview = cls(
             metadata=metadata,
             spec=spec,
+            events=events,
+            status=status,
         )
 
         preview.additional_properties = d

blaxel/core/client/models/sandbox.py

@@ -26,6 +26,8 @@ class Sandbox:
             spec (SandboxSpec): Configuration for a sandbox including its image, memory, ports, region, and lifecycle
                 policies
             events (Union[Unset, list['CoreEvent']]): Events happening on a resource deployed on Blaxel
+            expires_in (Union[Unset, int]): Time in seconds until the sandbox is automatically deleted based on TTL and
+                lifecycle policies. Only present for sandboxes with lifecycle configured.
             last_used_at (Union[Unset, str]): Last time the sandbox was used (read-only, managed by the system)
             status (Union[Unset, Status]): Deployment status of a resource deployed on Blaxel
     """
@@ -33,6 +35,7 @@ class Sandbox:
     metadata: "Metadata"
     spec: "SandboxSpec"
     events: Union[Unset, list["CoreEvent"]] = UNSET
+    expires_in: Union[Unset, int] = UNSET
     last_used_at: Union[Unset, str] = UNSET
     status: Union[Unset, Status] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -60,6 +63,8 @@ class Sandbox:
                     )
                 events.append(componentsschemas_core_events_item)
 
+        expires_in = self.expires_in
+
         last_used_at = self.last_used_at
 
         status: Union[Unset, str] = UNSET
@@ -76,6 +81,8 @@ class Sandbox:
         )
         if events is not UNSET:
             field_dict["events"] = events
+        if expires_in is not UNSET:
+            field_dict["expiresIn"] = expires_in
         if last_used_at is not UNSET:
             field_dict["lastUsedAt"] = last_used_at
         if status is not UNSET:
@@ -105,6 +112,8 @@ class Sandbox:
 
             events.append(componentsschemas_core_events_item)
 
+        expires_in = d.pop("expiresIn", d.pop("expires_in", UNSET))
+
         last_used_at = d.pop("lastUsedAt", d.pop("last_used_at", UNSET))
 
         _status = d.pop("status", UNSET)
@@ -118,6 +127,7 @@ class Sandbox:
             metadata=metadata,
             spec=spec,
             events=events,
+            expires_in=expires_in,
             last_used_at=last_used_at,
             status=status,
         )

blaxel/core/jobs/__init__.py

@@ -75,7 +75,7 @@ class BlJob:
     def run(
         self,
         tasks: List[Dict[str, Any]],
-        env: Dict[str, Any] | None = None,
+        env: Dict[str, str] | None = None,
         memory: int | None = None,
         execution_id: str | None = None,
     ) -> str:
@@ -109,7 +109,7 @@ class BlJob:
     async def arun(
         self,
         tasks: List[Dict[str, Any]],
-        env: Dict[str, Any] | None = None,
+        env: Dict[str, str] | None = None,
         memory: int | None = None,
         execution_id: str | None = None,
     ) -> str:

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/sandbox/client/api/system/get_health.py

@@ -0,0 +1,134 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import Client
+from ...models.health_response import HealthResponse
+from ...types import Response
+
+
+def _get_kwargs() -> dict[str, Any]:
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": "/health",
+    }
+
+    return _kwargs
+
+
+def _parse_response(*, client: Client, response: httpx.Response) -> HealthResponse | None:
+    if response.status_code == 200:
+        response_200 = HealthResponse.from_dict(response.json())
+
+        return response_200
+    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[HealthResponse]:
+    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,
+) -> Response[HealthResponse]:
+    """Health check
+
+     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[HealthResponse]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: Client,
+) -> HealthResponse | None:
+    """Health check
+
+     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:
+        HealthResponse
+    """
+
+    return sync_detailed(
+        client=client,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: Client,
+) -> Response[HealthResponse]:
+    """Health check
+
+     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[HealthResponse]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: Client,
+) -> HealthResponse | None:
+    """Health check
+
+     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:
+        HealthResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+        )
+    ).parsed

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",
 )