blaxel 0.2.35__tar.gz → 0.2.36__tar.gz

{blaxel-0.2.35 → blaxel-0.2.36}/.github/workflows/test.yaml

@@ -2,21 +2,19 @@ name: Test blaxel package
 
 on:
   pull_request:
-    branches: ["main", "develop"]
-  push:
-    branches: ["main", "develop"]
+    branches:
+      - "main"
 
 permissions:
   contents: read
 
-env:
-  BL_API_KEY: fake_api_key
-  BL_WORKSPACE: fake_workspace
-
 jobs:
   test:
-    name: Test blaxel package
+    name: Unit Tests
     runs-on: ubuntu-latest
+    env:
+      BL_API_KEY: fake_api_key
+      BL_WORKSPACE: fake_workspace
     strategy:
       matrix:
         python-version: ["3.10", "3.11", "3.12"]
@@ -56,4 +54,32 @@ jobs:
       - name: Run tests
         run: |
           source .venv/bin/activate
-          make test
+          make test
+
+  integration-test:
+    name: Integration Tests
+    runs-on: ubuntu-latest
+    env:
+      BL_API_KEY: ${{ secrets.BL_API_KEY }}
+      BL_WORKSPACE: ${{ secrets.BL_WORKSPACE }}
+    # Only run if secrets are available (not on forks)
+    if: github.event.pull_request.head.repo.full_name == github.repository || github.event_name == 'push'
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: "Install uv"
+        uses: astral-sh/setup-uv@v4
+        with:
+          python-version: "3.12"
+
+      - name: Install dependencies
+        run: |
+          uv venv
+          source .venv/bin/activate
+          make install
+
+      - name: Run integration tests
+        run: |
+          source .venv/bin/activate
+          make test-integration

{blaxel-0.2.35 → blaxel-0.2.36}/Makefile

@@ -88,7 +88,7 @@ test:
 	uv run pytest tests/ -v --ignore=tests/integration/ --ignore=tests/sandbox/integration/
 
 test-integration:
-	uv run pytest tests/integration/ -n auto
+	uv run pytest tests/integration/
 
 install-dev:
 	uv sync --group test

{blaxel-0.2.35 → blaxel-0.2.36}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.2.35
+Version: 0.2.36
 Summary: Blaxel - AI development platform SDK
 Project-URL: Homepage, https://blaxel.ai
 Project-URL: Documentation, https://docs.blaxel.ai

{blaxel-0.2.35 → blaxel-0.2.36}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "blaxel"
-version = "0.2.35"
+version = "0.2.36"
 description = "Blaxel - AI development platform SDK"
 readme = "README.md"
 authors = [{ name = "cploujoux", email = "cploujoux@blaxel.ai" }]
@@ -147,4 +147,5 @@ test = [
   "aiohttp>=3.11.16",
   "flask>=3.0.0",
   "requests>=2.32.0",
+  "pydantic-ai>=0.0.48",
 ]

{blaxel-0.2.35 → blaxel-0.2.36}/src/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.35"
-__commit__ = "4cbf852e4b6e8e7ec6b054ac2a79d2959a4e2fe6"
+__version__ = "0.2.36"
+__commit__ = "70f38e31dd50b26cbb54665420e165b89f3fa4b9"
 __sentry_dsn__ = "https://9711de13cd02b285ca4378c01de8dc30@o4508714045276160.ingest.us.sentry.io/4510461121462272"
 __all__ = ["autoload", "settings", "env"]
 

{blaxel-0.2.35 → blaxel-0.2.36}/src/blaxel/core/client/api/compute/create_sandbox.py

@@ -7,18 +7,26 @@ from ... import errors
 from ...client import Client
 from ...models.sandbox import Sandbox
 from ...models.sandbox_error import SandboxError
-from ...types import Response
+from ...types import UNSET, Response, Unset
 
 
 def _get_kwargs(
     *,
     body: Sandbox,
+    create_if_not_exist: Union[Unset, bool] = False,
 ) -> dict[str, Any]:
     headers: dict[str, Any] = {}
 
+    params: dict[str, Any] = {}
+
+    params["createIfNotExist"] = create_if_not_exist
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
     _kwargs: dict[str, Any] = {
         "method": "post",
         "url": "/sandboxes",
+        "params": params,
     }
 
     if type(body) is dict:
@@ -81,6 +89,7 @@ def sync_detailed(
     *,
     client: Client,
     body: Sandbox,
+    create_if_not_exist: Union[Unset, bool] = False,
 ) -> Response[Union[Sandbox, SandboxError]]:
     """Create sandbox
 
@@ -88,6 +97,7 @@ def sync_detailed(
     idle and resume instantly, preserving memory state including running processes and filesystem.
 
     Args:
+        create_if_not_exist (Union[Unset, bool]):  Default: False.
         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.
@@ -102,6 +112,7 @@ def sync_detailed(
 
     kwargs = _get_kwargs(
         body=body,
+        create_if_not_exist=create_if_not_exist,
     )
 
     response = client.get_httpx_client().request(
@@ -115,6 +126,7 @@ def sync(
     *,
     client: Client,
     body: Sandbox,
+    create_if_not_exist: Union[Unset, bool] = False,
 ) -> Union[Sandbox, SandboxError] | None:
     """Create sandbox
 
@@ -122,6 +134,7 @@ def sync(
     idle and resume instantly, preserving memory state including running processes and filesystem.
 
     Args:
+        create_if_not_exist (Union[Unset, bool]):  Default: False.
         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.
@@ -137,6 +150,7 @@ def sync(
     return sync_detailed(
         client=client,
         body=body,
+        create_if_not_exist=create_if_not_exist,
     ).parsed
 
 
@@ -144,6 +158,7 @@ async def asyncio_detailed(
     *,
     client: Client,
     body: Sandbox,
+    create_if_not_exist: Union[Unset, bool] = False,
 ) -> Response[Union[Sandbox, SandboxError]]:
     """Create sandbox
 
@@ -151,6 +166,7 @@ async def asyncio_detailed(
     idle and resume instantly, preserving memory state including running processes and filesystem.
 
     Args:
+        create_if_not_exist (Union[Unset, bool]):  Default: False.
         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.
@@ -165,6 +181,7 @@ async def asyncio_detailed(
 
     kwargs = _get_kwargs(
         body=body,
+        create_if_not_exist=create_if_not_exist,
     )
 
     response = await client.get_async_httpx_client().request(**kwargs)
@@ -176,6 +193,7 @@ async def asyncio(
     *,
     client: Client,
     body: Sandbox,
+    create_if_not_exist: Union[Unset, bool] = False,
 ) -> Union[Sandbox, SandboxError] | None:
     """Create sandbox
 
@@ -183,6 +201,7 @@ async def asyncio(
     idle and resume instantly, preserving memory state including running processes and filesystem.
 
     Args:
+        create_if_not_exist (Union[Unset, bool]):  Default: False.
         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.
@@ -199,5 +218,6 @@ async def asyncio(
         await asyncio_detailed(
             client=client,
             body=body,
+            create_if_not_exist=create_if_not_exist,
         )
     ).parsed

{blaxel-0.2.35 → blaxel-0.2.36}/src/blaxel/core/client/api/jobs/create_job_execution.py

@@ -5,8 +5,8 @@ import httpx
 
 from ... import errors
 from ...client import Client
+from ...models.create_job_execution_output import CreateJobExecutionOutput
 from ...models.create_job_execution_request import CreateJobExecutionRequest
-from ...models.create_job_execution_response import CreateJobExecutionResponse
 from ...types import Response
 
 
@@ -36,9 +36,9 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Client, response: httpx.Response
-) -> Union[Any, CreateJobExecutionResponse] | None:
+) -> Union[Any, CreateJobExecutionOutput] | None:
     if response.status_code == 200:
-        response_200 = CreateJobExecutionResponse.from_dict(response.json())
+        response_200 = CreateJobExecutionOutput.from_dict(response.json())
 
         return response_200
     if response.status_code == 400:
@@ -55,7 +55,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Client, response: httpx.Response
-) -> Response[Union[Any, CreateJobExecutionResponse]]:
+) -> Response[Union[Any, CreateJobExecutionOutput]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -69,7 +69,7 @@ def sync_detailed(
     *,
     client: Client,
     body: CreateJobExecutionRequest,
-) -> Response[Union[Any, CreateJobExecutionResponse]]:
+) -> Response[Union[Any, CreateJobExecutionOutput]]:
     """Create job execution
 
      Triggers a new execution of the batch job. Each execution runs multiple tasks in parallel according
@@ -84,7 +84,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Union[Any, CreateJobExecutionResponse]]
+        Response[Union[Any, CreateJobExecutionOutput]]
     """
 
     kwargs = _get_kwargs(
@@ -104,7 +104,7 @@ def sync(
     *,
     client: Client,
     body: CreateJobExecutionRequest,
-) -> Union[Any, CreateJobExecutionResponse] | None:
+) -> Union[Any, CreateJobExecutionOutput] | None:
     """Create job execution
 
      Triggers a new execution of the batch job. Each execution runs multiple tasks in parallel according
@@ -119,7 +119,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Union[Any, CreateJobExecutionResponse]
+        Union[Any, CreateJobExecutionOutput]
     """
 
     return sync_detailed(
@@ -134,7 +134,7 @@ async def asyncio_detailed(
     *,
     client: Client,
     body: CreateJobExecutionRequest,
-) -> Response[Union[Any, CreateJobExecutionResponse]]:
+) -> Response[Union[Any, CreateJobExecutionOutput]]:
     """Create job execution
 
      Triggers a new execution of the batch job. Each execution runs multiple tasks in parallel according
@@ -149,7 +149,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Union[Any, CreateJobExecutionResponse]]
+        Response[Union[Any, CreateJobExecutionOutput]]
     """
 
     kwargs = _get_kwargs(
@@ -167,7 +167,7 @@ async def asyncio(
     *,
     client: Client,
     body: CreateJobExecutionRequest,
-) -> Union[Any, CreateJobExecutionResponse] | None:
+) -> Union[Any, CreateJobExecutionOutput] | None:
     """Create job execution
 
      Triggers a new execution of the batch job. Each execution runs multiple tasks in parallel according
@@ -182,7 +182,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Union[Any, CreateJobExecutionResponse]
+        Union[Any, CreateJobExecutionOutput]
     """
 
     return (

blaxel-0.2.36/src/blaxel/core/client/api/volumes/update_volume.py

@@ -0,0 +1,187 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import Client
+from ...models.volume import Volume
+from ...types import Response
+
+
+def _get_kwargs(
+    volume_name: str,
+    *,
+    body: Volume,
+) -> dict[str, Any]:
+    headers: dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "put",
+        "url": f"/volumes/{volume_name}",
+    }
+
+    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) -> Volume | None:
+    if response.status_code == 200:
+        response_200 = Volume.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[Volume]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    volume_name: str,
+    *,
+    client: Client,
+    body: Volume,
+) -> Response[Volume]:
+    """Update volume
+
+     Updates a volume.
+
+    Args:
+        volume_name (str):
+        body (Volume): Persistent storage volume that can be attached to sandboxes for durable
+            file storage across sessions. Volumes survive sandbox deletion and can be reattached to
+            new sandboxes.
+
+    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[Volume]
+    """
+
+    kwargs = _get_kwargs(
+        volume_name=volume_name,
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    volume_name: str,
+    *,
+    client: Client,
+    body: Volume,
+) -> Volume | None:
+    """Update volume
+
+     Updates a volume.
+
+    Args:
+        volume_name (str):
+        body (Volume): Persistent storage volume that can be attached to sandboxes for durable
+            file storage across sessions. Volumes survive sandbox deletion and can be reattached to
+            new sandboxes.
+
+    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:
+        Volume
+    """
+
+    return sync_detailed(
+        volume_name=volume_name,
+        client=client,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    volume_name: str,
+    *,
+    client: Client,
+    body: Volume,
+) -> Response[Volume]:
+    """Update volume
+
+     Updates a volume.
+
+    Args:
+        volume_name (str):
+        body (Volume): Persistent storage volume that can be attached to sandboxes for durable
+            file storage across sessions. Volumes survive sandbox deletion and can be reattached to
+            new sandboxes.
+
+    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[Volume]
+    """
+
+    kwargs = _get_kwargs(
+        volume_name=volume_name,
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    volume_name: str,
+    *,
+    client: Client,
+    body: Volume,
+) -> Volume | None:
+    """Update volume
+
+     Updates a volume.
+
+    Args:
+        volume_name (str):
+        body (Volume): Persistent storage volume that can be attached to sandboxes for durable
+            file storage across sessions. Volumes survive sandbox deletion and can be reattached to
+            new sandboxes.
+
+    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:
+        Volume
+    """
+
+    return (
+        await asyncio_detailed(
+            volume_name=volume_name,
+            client=client,
+            body=body,
+        )
+    ).parsed

{blaxel-0.2.35 → blaxel-0.2.36}/src/blaxel/core/client/models/__init__.py

@@ -12,10 +12,11 @@ from .continent import Continent
 from .core_event import CoreEvent
 from .country import Country
 from .create_api_key_for_service_account_body import CreateApiKeyForServiceAccountBody
+from .create_job_execution_output import CreateJobExecutionOutput
+from .create_job_execution_output_tasks_item import CreateJobExecutionOutputTasksItem
 from .create_job_execution_request import CreateJobExecutionRequest
+from .create_job_execution_request_env import CreateJobExecutionRequestEnv
 from .create_job_execution_request_tasks_item import CreateJobExecutionRequestTasksItem
-from .create_job_execution_response import CreateJobExecutionResponse
-from .create_job_execution_response_tasks_item import CreateJobExecutionResponseTasksItem
 from .create_workspace_service_account_body import CreateWorkspaceServiceAccountBody
 from .create_workspace_service_account_response_200 import CreateWorkspaceServiceAccountResponse200
 from .custom_domain import CustomDomain
@@ -43,8 +44,8 @@ from .form_secrets import FormSecrets
 from .function import Function
 from .function_runtime import FunctionRuntime
 from .function_runtime_generation import FunctionRuntimeGeneration
+from .function_runtime_transport import FunctionRuntimeTransport
 from .function_spec import FunctionSpec
-from .function_spec_transport import FunctionSpecTransport
 from .get_workspace_service_accounts_response_200_item import (
     GetWorkspaceServiceAccountsResponse200Item,
 )
@@ -72,6 +73,7 @@ from .job import Job
 from .job_execution import JobExecution
 from .job_execution_metadata import JobExecutionMetadata
 from .job_execution_spec import JobExecutionSpec
+from .job_execution_spec_env_override import JobExecutionSpecEnvOverride
 from .job_execution_stats import JobExecutionStats
 from .job_execution_status import JobExecutionStatus
 from .job_execution_task import JobExecutionTask
@@ -174,10 +176,11 @@ __all__ = (
     "CoreEvent",
     "Country",
     "CreateApiKeyForServiceAccountBody",
+    "CreateJobExecutionOutput",
+    "CreateJobExecutionOutputTasksItem",
     "CreateJobExecutionRequest",
+    "CreateJobExecutionRequestEnv",
     "CreateJobExecutionRequestTasksItem",
-    "CreateJobExecutionResponse",
-    "CreateJobExecutionResponseTasksItem",
     "CreateWorkspaceServiceAccountBody",
     "CreateWorkspaceServiceAccountResponse200",
     "CustomDomain",
@@ -205,8 +208,8 @@ __all__ = (
     "Function",
     "FunctionRuntime",
     "FunctionRuntimeGeneration",
+    "FunctionRuntimeTransport",
     "FunctionSpec",
-    "FunctionSpecTransport",
     "GetWorkspaceServiceAccountsResponse200Item",
     "Image",
     "ImageMetadata",
@@ -232,6 +235,7 @@ __all__ = (
     "JobExecution",
     "JobExecutionMetadata",
     "JobExecutionSpec",
+    "JobExecutionSpecEnvOverride",
     "JobExecutionStats",
     "JobExecutionStatus",
     "JobExecutionTask",

blaxel-0.2.35/src/blaxel/core/client/models/create_job_execution_response.py → blaxel-0.2.36/src/blaxel/core/client/models/create_job_execution_output.py

@@ -6,16 +6,14 @@ from attrs import field as _attrs_field
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
-    from ..models.create_job_execution_response_tasks_item import (
-        CreateJobExecutionResponseTasksItem,
-    )
+    from ..models.create_job_execution_output_tasks_item import CreateJobExecutionOutputTasksItem
 
 
-T = TypeVar("T", bound="CreateJobExecutionResponse")
+T = TypeVar("T", bound="CreateJobExecutionOutput")
 
 
 @_attrs_define
-class CreateJobExecutionResponse:
+class CreateJobExecutionOutput:
     """Response returned when a job execution is successfully created. Contains identifiers and the tasks that will be
     executed.
 
@@ -25,7 +23,7 @@ class CreateJobExecutionResponse:
             id (Union[Unset, str]): Unique identifier for this request, used for idempotency and tracking. Auto-generated if
                 not provided in the request. Example: 550e8400-e29b-41d4-a716-446655440000.
             job_id (Union[Unset, str]): Name of the job that this execution belongs to Example: data-processing-job.
-            tasks (Union[Unset, list['CreateJobExecutionResponseTasksItem']]): Array of task configurations that will be
+            tasks (Union[Unset, list['CreateJobExecutionOutputTasksItem']]): Array of task configurations that will be
                 executed in parallel according to the job's concurrency settings. Each task can have custom parameters.
             workspace_id (Union[Unset, str]): Name of the workspace where the job execution was created Example: my-
                 workspace.
@@ -34,7 +32,7 @@ class CreateJobExecutionResponse:
     execution_id: Union[Unset, str] = UNSET
     id: Union[Unset, str] = UNSET
     job_id: Union[Unset, str] = UNSET
-    tasks: Union[Unset, list["CreateJobExecutionResponseTasksItem"]] = UNSET
+    tasks: Union[Unset, list["CreateJobExecutionOutputTasksItem"]] = UNSET
     workspace_id: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -75,8 +73,8 @@ class CreateJobExecutionResponse:
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
-        from ..models.create_job_execution_response_tasks_item import (
-            CreateJobExecutionResponseTasksItem,
+        from ..models.create_job_execution_output_tasks_item import (
+            CreateJobExecutionOutputTasksItem,
         )
 
         if not src_dict:
@@ -91,13 +89,13 @@ class CreateJobExecutionResponse:
         tasks = []
         _tasks = d.pop("tasks", UNSET)
         for tasks_item_data in _tasks or []:
-            tasks_item = CreateJobExecutionResponseTasksItem.from_dict(tasks_item_data)
+            tasks_item = CreateJobExecutionOutputTasksItem.from_dict(tasks_item_data)
 
             tasks.append(tasks_item)
 
         workspace_id = d.pop("workspaceId", d.pop("workspace_id", UNSET))
 
-        create_job_execution_response = cls(
+        create_job_execution_output = cls(
             execution_id=execution_id,
             id=id,
             job_id=job_id,
@@ -105,8 +103,8 @@ class CreateJobExecutionResponse:
             workspace_id=workspace_id,
         )
 
-        create_job_execution_response.additional_properties = d
-        return create_job_execution_response
+        create_job_execution_output.additional_properties = d
+        return create_job_execution_output
 
     @property
     def additional_keys(self) -> list[str]: