use-computer 0.0.44__tar.gz → 0.0.45__tar.gz

{use_computer-0.0.44 → use_computer-0.0.45}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: use-computer
-Version: 0.0.44
+Version: 0.0.45
 Summary: Python SDK for use.computer sandboxes across macOS, iOS simulators, Windows, and Ubuntu
 Project-URL: Homepage, https://use.computer
 Project-URL: Documentation, https://api.use.computer/docs
@@ -99,14 +99,19 @@ with Computer().create(type="macos", reservation_id="...") as mac:
     print(mac.exec_ssh("uname -a").stdout)
 ```
 
+Each reserved Mac Mini supports up to 2 parallel macOS VMs. Check
+`Computer().platforms(reservation_id=...)["macos"]["capacity"]` for the selected
+reservation's `max` and `used` counts. iOS capacity is reported separately under
+`["ios"]["capacity"]`.
+
 ### Windows (Beta)
 
 ```python
 with Computer().create(type="windows") as win:
-    print(win.run("$env:COMPUTERNAME").stdout)   # PowerShell exec (no SSH)
+    print(win.run("$env:COMPUTERNAME").stdout)  # PowerShell exec (no SSH)
     win.keyboard.type("hello")
     win.screenshot.take_full_screen()
-    win.ui_tree()                                 # native Windows UIAutomation tree
+    win.ui_tree()  # native Windows UIAutomation tree
 ```
 
 Windows sandboxes are **Beta** (admin-only). Same mouse/keyboard/screenshot/
@@ -132,6 +137,10 @@ with Computer().create(type="ubuntu", version="ubuntu-24.04") as ubuntu:
 
 Ubuntu sandboxes are **Beta** (admin-only). They use the same KVM/noVNC backend
 as Windows; pass `version="ubuntu-24.04"` plus `resources` to select CPU/RAM/disk.
+Ubuntu and Windows share a per-account KVM concurrency limit. The default is 8
+running desktops; when the limit is hit, `create()` raises
+`SandboxLimitReachedError` with `limit` and `running` fields from the gateway's
+HTTP 429 response.
 
 Selectable Ubuntu images are exposed by `Computer().platforms()["ubuntu"]`.
 Current examples include `ubuntu-24.04`, `ubuntu-24.04-4c4g80g`,
@@ -153,27 +162,32 @@ with client.create(
 
 ### Snapshots (Beta)
 
-Windows and Ubuntu snapshots preserve disk + RAM state, including open apps and
-running processes. Seed a VM once, snapshot it, then create new sandboxes from
-that snapshot version.
+macOS, Windows, and Ubuntu snapshots let you seed a VM once, snapshot it, then
+create new sandboxes from that snapshot version. macOS snapshots preserve disk
+state. Windows and Ubuntu snapshots preserve disk + RAM state, including open
+apps and running processes.
+
+macOS snapshot launch is slower because the gateway stores a portable disk-delta
+artifact and transfers it to the Mac Mini that claims the new VM. Windows and
+Ubuntu snapshots restore disk + RAM on their desktop host.
 
 ```python
 client = Computer()
 
 # Configure the desktop by hand over VNC, then snapshot it.
-with client.create(type="ubuntu", version="ubuntu-24.04") as ubuntu:
-    print("Set up the desktop here:", ubuntu.vnc_url)
-    # install apps, sign into accounts, pre-open the windows your agent starts from
+with client.create(type="macos") as mac:
+    print("Set up the desktop here:", mac.vnc_url)
+    # install apps, sign into accounts, create files your agent starts from
     input("Press Enter once the desktop is ready to snapshot...")
-    snapshot = ubuntu.snapshot("chrome-seeded-ubuntu")
+    snapshot = mac.snapshot("chrome-seeded-macos")
 
-# New sandboxes boot from that exact state, no setup needed.
-with client.create(type="ubuntu", snapshot=snapshot.version) as seeded:
-    print(seeded.vnc_url)  # apps, logins, and open windows restored
+# New sandboxes boot from that saved state, no setup needed.
+with client.create(type="macos", snapshot=snapshot.version) as seeded:
+    print(seeded.vnc_url)  # installed apps, files, and logins restored
 ```
 
-Use `client.snapshots("ubuntu")` or `client.snapshots("windows")` to list saved
-snapshot versions.
+Use `client.snapshots("macos")`, `client.snapshots("ubuntu")`, or
+`client.snapshots("windows")` to list saved snapshot versions.
 
 Full DSL reference: [docs.use.computer/docs/sdk](https://docs.use.computer/docs/sdk)
 
@@ -212,7 +226,7 @@ visionOS yet. Use it for read-only flows for now.
 | [`examples/_7_hello_windows.py`](examples/_7_hello_windows.py) | create Windows → run PowerShell → screenshot   |
 | [`examples/_8_hello_ubuntu.py`](examples/_8_hello_ubuntu.py)   | create Ubuntu → run bash → screenshot          |
 | [`examples/_9_seeding.py`](examples/_9_seeding.py)             | typed setup: files, hosts, open URLs/apps      |
-| [`examples/_10_snapshots.py`](examples/_10_snapshots.py)       | snapshot seeded Ubuntu/Windows desktop state   |
+| [`examples/_10_snapshots.py`](examples/_10_snapshots.py)       | snapshot seeded macOS/Ubuntu/Windows state     |
 
 For agent loops and evals: [use-computer-cookbook](https://github.com/josancamon19/use-computer-cookbook). Harbor job YAMLs should use Harbor's `type: use-computer` environment plus `use_computer.harbor.agents:*` agent wrappers.
 

{use_computer-0.0.44 → use_computer-0.0.45}/README.md

@@ -65,14 +65,19 @@ with Computer().create(type="macos", reservation_id="...") as mac:
     print(mac.exec_ssh("uname -a").stdout)
 ```
 
+Each reserved Mac Mini supports up to 2 parallel macOS VMs. Check
+`Computer().platforms(reservation_id=...)["macos"]["capacity"]` for the selected
+reservation's `max` and `used` counts. iOS capacity is reported separately under
+`["ios"]["capacity"]`.
+
 ### Windows (Beta)
 
 ```python
 with Computer().create(type="windows") as win:
-    print(win.run("$env:COMPUTERNAME").stdout)   # PowerShell exec (no SSH)
+    print(win.run("$env:COMPUTERNAME").stdout)  # PowerShell exec (no SSH)
     win.keyboard.type("hello")
     win.screenshot.take_full_screen()
-    win.ui_tree()                                 # native Windows UIAutomation tree
+    win.ui_tree()  # native Windows UIAutomation tree
 ```
 
 Windows sandboxes are **Beta** (admin-only). Same mouse/keyboard/screenshot/
@@ -98,6 +103,10 @@ with Computer().create(type="ubuntu", version="ubuntu-24.04") as ubuntu:
 
 Ubuntu sandboxes are **Beta** (admin-only). They use the same KVM/noVNC backend
 as Windows; pass `version="ubuntu-24.04"` plus `resources` to select CPU/RAM/disk.
+Ubuntu and Windows share a per-account KVM concurrency limit. The default is 8
+running desktops; when the limit is hit, `create()` raises
+`SandboxLimitReachedError` with `limit` and `running` fields from the gateway's
+HTTP 429 response.
 
 Selectable Ubuntu images are exposed by `Computer().platforms()["ubuntu"]`.
 Current examples include `ubuntu-24.04`, `ubuntu-24.04-4c4g80g`,
@@ -119,27 +128,32 @@ with client.create(
 
 ### Snapshots (Beta)
 
-Windows and Ubuntu snapshots preserve disk + RAM state, including open apps and
-running processes. Seed a VM once, snapshot it, then create new sandboxes from
-that snapshot version.
+macOS, Windows, and Ubuntu snapshots let you seed a VM once, snapshot it, then
+create new sandboxes from that snapshot version. macOS snapshots preserve disk
+state. Windows and Ubuntu snapshots preserve disk + RAM state, including open
+apps and running processes.
+
+macOS snapshot launch is slower because the gateway stores a portable disk-delta
+artifact and transfers it to the Mac Mini that claims the new VM. Windows and
+Ubuntu snapshots restore disk + RAM on their desktop host.
 
 ```python
 client = Computer()
 
 # Configure the desktop by hand over VNC, then snapshot it.
-with client.create(type="ubuntu", version="ubuntu-24.04") as ubuntu:
-    print("Set up the desktop here:", ubuntu.vnc_url)
-    # install apps, sign into accounts, pre-open the windows your agent starts from
+with client.create(type="macos") as mac:
+    print("Set up the desktop here:", mac.vnc_url)
+    # install apps, sign into accounts, create files your agent starts from
     input("Press Enter once the desktop is ready to snapshot...")
-    snapshot = ubuntu.snapshot("chrome-seeded-ubuntu")
+    snapshot = mac.snapshot("chrome-seeded-macos")
 
-# New sandboxes boot from that exact state, no setup needed.
-with client.create(type="ubuntu", snapshot=snapshot.version) as seeded:
-    print(seeded.vnc_url)  # apps, logins, and open windows restored
+# New sandboxes boot from that saved state, no setup needed.
+with client.create(type="macos", snapshot=snapshot.version) as seeded:
+    print(seeded.vnc_url)  # installed apps, files, and logins restored
 ```
 
-Use `client.snapshots("ubuntu")` or `client.snapshots("windows")` to list saved
-snapshot versions.
+Use `client.snapshots("macos")`, `client.snapshots("ubuntu")`, or
+`client.snapshots("windows")` to list saved snapshot versions.
 
 Full DSL reference: [docs.use.computer/docs/sdk](https://docs.use.computer/docs/sdk)
 
@@ -178,7 +192,7 @@ visionOS yet. Use it for read-only flows for now.
 | [`examples/_7_hello_windows.py`](examples/_7_hello_windows.py) | create Windows → run PowerShell → screenshot   |
 | [`examples/_8_hello_ubuntu.py`](examples/_8_hello_ubuntu.py)   | create Ubuntu → run bash → screenshot          |
 | [`examples/_9_seeding.py`](examples/_9_seeding.py)             | typed setup: files, hosts, open URLs/apps      |
-| [`examples/_10_snapshots.py`](examples/_10_snapshots.py)       | snapshot seeded Ubuntu/Windows desktop state   |
+| [`examples/_10_snapshots.py`](examples/_10_snapshots.py)       | snapshot seeded macOS/Ubuntu/Windows state     |
 
 For agent loops and evals: [use-computer-cookbook](https://github.com/josancamon19/use-computer-cookbook). Harbor job YAMLs should use Harbor's `type: use-computer` environment plus `use_computer.harbor.agents:*` agent wrappers.
 

{use_computer-0.0.44 → use_computer-0.0.45}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "use-computer"
-version = "0.0.44"
+version = "0.0.45"
 description = "Python SDK for use.computer sandboxes across macOS, iOS simulators, Windows, and Ubuntu"
 readme = "README.md"
 requires-python = ">=3.10"

{use_computer-0.0.44 → use_computer-0.0.45}/use_computer/__init__.py

@@ -5,6 +5,7 @@ from use_computer.core.errors import (
     BillingRequiredError,
     PlatformNotSupportedError,
     ReservationRequiredError,
+    SandboxLimitReachedError,
     UseComputerError,
 )
 from use_computer.core.models import (
@@ -73,6 +74,7 @@ __all__ = [
     "RemoteButton",
     "RunStatus",
     "Sandbox",
+    "SandboxLimitReachedError",
     "SandboxType",
     "Snapshot",
     "SimulatorChoice",

{use_computer-0.0.44 → use_computer-0.0.45}/use_computer/agents/base/agent.py

@@ -16,7 +16,7 @@ from use_computer.agents.base.desktop import (
 from use_computer.agents.base.final_state import FinalStateMixin
 from use_computer.agents.base.recordings import RecordingMixin
 from use_computer.agents.base.sandbox import ComputerSandbox
-from use_computer.agents.base.task_setup import run_task_setup
+from use_computer.agents.base.task_setup import run_task_setup, task_dir_from_env
 from use_computer.agents.base.trajectory import write_trajectory
 from use_computer.agents.core import (
     AgentRunOptions,
@@ -31,13 +31,13 @@ from use_computer.core.sandbox import AsyncSandbox
 
 def _task_dir_from_env(environment: Any) -> Path:
     """Pull the task dir from the environment."""
-    p = getattr(environment, "_task_dir", None) or getattr(environment, "task_dir", None)
+    p = task_dir_from_env(environment)
     if p is None:
         raise RuntimeError(
-            "environment exposes neither _task_dir nor task_dir; "
+            "environment exposes neither task_dir nor task environment_dir; "
             "agent can't locate the task directory"
         )
-    return Path(p).resolve()
+    return p
 
 
 class BaseCUAAgent(RecordingMixin, FinalStateMixin, ArtifactCaptureMixin):

{use_computer-0.0.44 → use_computer-0.0.45}/use_computer/agents/base/task_setup.py

@@ -3,10 +3,13 @@
 from __future__ import annotations
 
 import asyncio
+import importlib.util
+import inspect
 import json
 import logging
 import re
 import shlex
+import sys
 from pathlib import Path, PureWindowsPath
 from typing import Any
 
@@ -35,7 +38,11 @@ async def run_task_setup(environment: Any, logger: logging.Logger) -> None:
     platform = _platform_from_env(environment)
 
     _load_in_process_config(environment, setup_dir)
+    await _run_task_local_setup(environment, setup_dir, logger)
+
     if platform is Platform.IOS:
+        if setup_dir.exists():
+            logger.info("CUA task setup complete")
         return
 
     await _upload_setup_files(environment, setup_dir, platform, logger)
@@ -46,8 +53,25 @@ async def run_task_setup(environment: Any, logger: logging.Logger) -> None:
 
 
 def _task_dir_from_env(environment: Any) -> Path | None:
+    return task_dir_from_env(environment)
+
+
+def task_dir_from_env(environment: Any) -> Path | None:
     raw = getattr(environment, "_task_dir", None) or getattr(environment, "task_dir", None)
-    return Path(raw).resolve() if raw else None
+    if raw:
+        return Path(raw).resolve()
+
+    environment_dir = getattr(environment, "environment_dir", None)
+    if environment_dir:
+        path = Path(environment_dir).resolve()
+        if path.name == "environment" and _looks_like_task_dir(path.parent):
+            return path.parent
+
+    return None
+
+
+def _looks_like_task_dir(path: Path) -> bool:
+    return (path / "task.toml").exists() and (path / "instruction.md").exists()
 
 
 def _platform_from_env(environment: Any) -> Platform:
@@ -143,6 +167,45 @@ async def _run_pre_command(
         await _run_macos_pre_command(environment, setup_dir, logger)
 
 
+async def _run_task_local_setup(
+    environment: Any,
+    setup_dir: Path,
+    logger: logging.Logger,
+) -> None:
+    setup_py = setup_dir / "setup.py"
+    if not setup_py.exists():
+        return
+
+    logger.info("running setup.py")
+    spec = importlib.util.spec_from_file_location(
+        f"_harbor_task_setup_{abs(hash(setup_py.resolve()))}",
+        setup_py,
+    )
+    if spec is None or spec.loader is None:
+        raise RuntimeError(f"could not load task setup script: {setup_py}")
+
+    module = importlib.util.module_from_spec(spec)
+    setup_path = str(setup_dir)
+    sys.path.insert(0, setup_path)
+    try:
+        spec.loader.exec_module(module)
+        setup_task = getattr(module, "setup_task", None)
+        if not callable(setup_task):
+            raise RuntimeError(f"task setup script lacks setup_task(): {setup_py}")
+
+        result = setup_task(environment, logger, setup_dir)
+        if inspect.isawaitable(result):
+            await result
+    finally:
+        if sys.path and sys.path[0] == setup_path:
+            sys.path.pop(0)
+        else:
+            try:
+                sys.path.remove(setup_path)
+            except ValueError:
+                pass
+
+
 async def _run_windows_pre_command(
     environment: Any,
     setup_dir: Path,

{use_computer-0.0.44 → use_computer-0.0.45}/use_computer/agents/ios/agent.py

@@ -32,6 +32,7 @@ from use_computer.agents.base import (
     load_prompt,
     scale_screenshot_for_model,
 )
+from use_computer.agents.base.task_setup import task_dir_from_env
 from use_computer.agents.ios.tools import (
     _build_ios_system_prompt,
     _coerce_coord_args,
@@ -69,12 +70,10 @@ class IOSAgent(BaseCUAAgent):
 
         self.sandbox = sandbox
         if self.task_dir is None:
-            task_dir = getattr(environment, "_task_dir", None) or getattr(
-                environment, "task_dir", None
-            )
+            task_dir = task_dir_from_env(environment)
             if task_dir is None:
                 raise RuntimeError(
-                    "environment exposes neither _task_dir nor task_dir; "
+                    "environment exposes neither task_dir nor task environment_dir; "
                     "agent can't locate the task directory"
                 )
             self.task_dir = task_dir

{use_computer-0.0.44 → use_computer-0.0.45}/use_computer/core/__init__.py

@@ -5,6 +5,7 @@ from use_computer.core.errors import (
     BillingRequiredError,
     PlatformNotSupportedError,
     ReservationRequiredError,
+    SandboxLimitReachedError,
     UseComputerError,
 )
 from use_computer.core.models import (
@@ -56,6 +57,7 @@ __all__ = [
     "ReservationRequiredError",
     "RunStatus",
     "Sandbox",
+    "SandboxLimitReachedError",
     "SandboxType",
     "Snapshot",
     "UbuntuSandbox",

{use_computer-0.0.44 → use_computer-0.0.45}/use_computer/core/client.py

@@ -4,6 +4,7 @@ import asyncio
 import os
 import time
 from dataclasses import dataclass, field
+from importlib.metadata import PackageNotFoundError, version
 from typing import Any, Literal, overload
 
 import httpx
@@ -27,6 +28,11 @@ from use_computer.ios.simulators import SimulatorFamily, select_simulator
 from use_computer.tasks import TasksClient
 
 DEFAULT_BASE_URL = "https://api.use.computer"
+SDK_VERSION = "0.0.0"
+try:
+    SDK_VERSION = version("use-computer")
+except PackageNotFoundError:
+    pass
 
 SandboxTypeInput = SandboxType | Literal["macos", "ios", "windows", "ubuntu"]
 
@@ -42,7 +48,7 @@ def _resolve_base_url(base_url: str | None) -> str:
 
 
 def _resolve_headers(api_key: str | None, host_header: str | None = None) -> dict[str, str]:
-    headers: dict[str, str] = {}
+    headers: dict[str, str] = {"X-Use-Computer-SDK": f"python/{SDK_VERSION}"}
     if api_key:
         headers["Authorization"] = f"Bearer {api_key}"
     host = host_header if host_header is not None else os.getenv("USE_COMPUTER_HOST_HEADER")
@@ -172,7 +178,7 @@ class Computer:
         return resp.json()
 
     def snapshots(self, platform: str = "") -> list[dict[str, Any]]:
-        """List Windur desktop snapshots visible to this API key."""
+        """List desktop snapshots visible to this API key."""
         params = {"platform": platform} if platform else None
         resp = self._http.get("/v1/snapshots", params=params)
         resp.raise_for_status()
@@ -195,6 +201,7 @@ class Computer:
         type: Literal["macos", SandboxType.MACOS] = ...,
         reservation_id: str = ...,
         host: str = ...,
+        snapshot: str = ...,
         ephemeral: bool = ...,
     ) -> MacOSSandbox: ...
 
@@ -260,7 +267,7 @@ class Computer:
             runtime: Simulator runtime identifier, for exact pinning.
             version: KVM desktop image version, for example "windows-11",
                 "windows-10", "ubuntu-24.04", or "osworld".
-            snapshot: Windur snapshot version returned by sandbox.snapshot().
+            snapshot: Snapshot version returned by sandbox.snapshot().
             resources: KVM CPU/RAM/disk request used to select the baked image
                 profile, for example {"cpus": 4, "memory_mb": 4096, "disk_gb": 40}.
             ephemeral: If true, delete the sandbox after the idle timeout.
@@ -280,6 +287,8 @@ class Computer:
                 body["version"] = version
             if resources:
                 body["resources"] = resources
+        elif sandbox_type == SandboxType.MACOS and snapshot:
+            body["snapshot"] = snapshot
         if sandbox_type == SandboxType.IOS:
             if host:
                 body["mini_ip"] = host
@@ -378,17 +387,7 @@ class Computer:
         poll: bool = True,
         poll_interval: float = 3.0,
     ) -> RunStatus:
-        """Run an ad-hoc task: spin up a sandbox and drive a model against a free-form instruction.
-
-        Args:
-            instruction: What the agent should do.
-            platform: "macos", "ios", "windows", or "ubuntu".
-            model: Model to use.
-            max_steps: Max agent steps.
-            files: Optional list of {remote_path, content_b64} to pre-upload.
-            poll: If True, block until the run finishes.
-            poll_interval: Seconds between status polls.
-        """
+        """Run an ad-hoc task (requires admin base URL and key)."""
         body: dict[str, Any] = {
             "instruction": instruction,
             "platform": platform,
@@ -405,7 +404,7 @@ class Computer:
         return self._poll_run(status.id, poll_interval)
 
     def get_run(self, run_id: str) -> RunStatus:
-        """Get current status of a model run."""
+        """Get current status of a model run (requires admin base URL and key)."""
         resp = self._http.get(f"/admin/tasks/model-runs/{run_id}")
         resp.raise_for_status()
         return RunStatus._from_dict(resp.json())
@@ -454,7 +453,7 @@ class AsyncComputer:
         return resp.json()
 
     async def snapshots(self, platform: str = "") -> list[dict[str, Any]]:
-        """List Windur desktop snapshots visible to this API key."""
+        """List desktop snapshots visible to this API key."""
         params = {"platform": platform} if platform else None
         resp = await self._http.get("/v1/snapshots", params=params)
         resp.raise_for_status()
@@ -477,6 +476,7 @@ class AsyncComputer:
         type: Literal["macos", SandboxType.MACOS] = ...,
         reservation_id: str = ...,
         host: str = ...,
+        snapshot: str = ...,
         ephemeral: bool = ...,
     ) -> AsyncMacOSSandbox: ...
 
@@ -542,7 +542,7 @@ class AsyncComputer:
             runtime: Simulator runtime identifier, for exact pinning.
             version: KVM desktop image version, for example "windows-11",
                 "windows-10", "ubuntu-24.04", or "osworld".
-            snapshot: Windur snapshot version returned by sandbox.snapshot().
+            snapshot: Snapshot version returned by sandbox.snapshot().
             resources: KVM CPU/RAM/disk request used to select the baked image
                 profile, for example {"cpus": 4, "memory_mb": 4096, "disk_gb": 40}.
             ephemeral: If true, delete the sandbox after the idle timeout.
@@ -562,6 +562,8 @@ class AsyncComputer:
                 body["version"] = version
             if resources:
                 body["resources"] = resources
+        elif sandbox_type == SandboxType.MACOS and snapshot:
+            body["snapshot"] = snapshot
         if sandbox_type == SandboxType.IOS:
             if host:
                 body["mini_ip"] = host
@@ -662,7 +664,7 @@ class AsyncComputer:
         poll: bool = True,
         poll_interval: float = 3.0,
     ) -> RunStatus:
-        """Run an ad-hoc task and drive a model against a free-form instruction."""
+        """Run an ad-hoc task (requires admin base URL and key)."""
         body: dict[str, Any] = {
             "instruction": instruction,
             "platform": platform,
@@ -679,7 +681,7 @@ class AsyncComputer:
         return await self._poll_run(status.id, poll_interval)
 
     async def get_run(self, run_id: str) -> RunStatus:
-        """Get current status of a model run."""
+        """Get current status of a model run (requires admin base URL and key)."""
         resp = await self._http.get(f"/admin/tasks/model-runs/{run_id}")
         resp.raise_for_status()
         return RunStatus._from_dict(resp.json())

{use_computer-0.0.44 → use_computer-0.0.45}/use_computer/core/errors.py

@@ -50,6 +50,26 @@ class ReservationRequiredError(UseComputerError):
         super().__init__(message or "macOS requires an active Mac Mini reservation")
 
 
+class SandboxLimitReachedError(UseComputerError):
+    """Raised when the account has too many running Ubuntu/Windows sandboxes."""
+
+    def __init__(
+        self,
+        message: str,
+        *,
+        limit: int | None = None,
+        running: int | None = None,
+        response: httpx.Response | None = None,
+    ):
+        self.message = message
+        self.limit = limit
+        self.running = running
+        self.response = response
+        super().__init__(
+            message or "Ubuntu/Windows sandbox limit reached. Delete one sandbox to launch another."
+        )
+
+
 def _typed_error_from_response(resp: httpx.Response) -> UseComputerError | None:
     try:
         data = resp.json()
@@ -70,6 +90,15 @@ def _typed_error_from_response(resp: httpx.Response) -> UseComputerError | None:
         return BillingRequiredError(message=message, response=resp)
     if resp.status_code == 409 and error_code == "macos_reservation_required":
         return ReservationRequiredError(message=message, response=resp)
+    if resp.status_code == 429 and error_code == "sandbox_limit_reached":
+        limit = data.get("limit")
+        running = data.get("running")
+        return SandboxLimitReachedError(
+            message=message,
+            limit=limit if isinstance(limit, int) else None,
+            running=running if isinstance(running, int) else None,
+            response=resp,
+        )
     return None
 
 
@@ -80,7 +109,7 @@ def _raise_if_unsupported(resp: httpx.Response) -> None:
     only touch the body for statuses we own — anything else is left
     untouched for raise_for_status() to handle normally.
     """
-    if resp.status_code not in (402, 409, 501):
+    if resp.status_code not in (402, 409, 429, 501):
         return
     try:
         resp.read()
@@ -92,7 +121,7 @@ def _raise_if_unsupported(resp: httpx.Response) -> None:
 
 
 async def _araise_if_unsupported(resp: httpx.Response) -> None:
-    if resp.status_code not in (402, 409, 501):
+    if resp.status_code not in (402, 409, 429, 501):
         return
     try:
         await resp.aread()

{use_computer-0.0.44 → use_computer-0.0.45}/use_computer/core/sandbox.py

@@ -4,6 +4,7 @@ import base64
 import io
 import tarfile
 import threading
+import urllib.error
 import urllib.request
 from dataclasses import dataclass
 from enum import Enum
@@ -41,7 +42,7 @@ class SandboxType(str, Enum):
 
 @dataclass
 class Snapshot:
-    """Windur-backed desktop snapshot."""
+    """Desktop snapshot."""
 
     name: str
     version: str
@@ -83,6 +84,10 @@ def _start_keepalive_thread(
             try:
                 req = urllib.request.Request(keepalive_url, method="POST", headers=headers)
                 urllib.request.urlopen(req, timeout=10).close()
+            except urllib.error.HTTPError as exc:
+                if exc.code in (404, 410):
+                    stop.set()
+                    break
             except Exception:
                 pass
 
@@ -167,12 +172,12 @@ class Sandbox:
         local_path.write_bytes(resp.content)
 
     def snapshot(self, name: str, *, version: str = "", timeout: int = 900) -> Snapshot:
-        """Capture a Windur desktop snapshot and return its image version."""
-        if self.type not in (SandboxType.WINDOWS, SandboxType.UBUNTU):
+        """Capture a desktop snapshot and return its version."""
+        if self.type not in (SandboxType.MACOS, SandboxType.WINDOWS, SandboxType.UBUNTU):
             raise PlatformNotSupportedError(
                 action="snapshot",
                 sandbox_type=self.type.value,
-                hint="snapshots are supported for Windows and Ubuntu sandboxes",
+                hint="snapshots are supported for macOS, Windows, and Ubuntu sandboxes",
             )
         body = {"name": name}
         if version:
@@ -601,12 +606,12 @@ class AsyncSandbox:
         local_path.write_bytes(resp.content)
 
     async def snapshot(self, name: str, *, version: str = "", timeout: int = 900) -> Snapshot:
-        """Capture a Windur desktop snapshot and return its image version."""
-        if self.type not in (SandboxType.WINDOWS, SandboxType.UBUNTU):
+        """Capture a desktop snapshot and return its version."""
+        if self.type not in (SandboxType.MACOS, SandboxType.WINDOWS, SandboxType.UBUNTU):
             raise PlatformNotSupportedError(
                 action="snapshot",
                 sandbox_type=self.type.value,
-                hint="snapshots are supported for Windows and Ubuntu sandboxes",
+                hint="snapshots are supported for macOS, Windows, and Ubuntu sandboxes",
             )
         body = {"name": name}
         if version: