opencomputer-sdk 0.5.2__tar.gz → 0.5.3__tar.gz

{opencomputer_sdk-0.5.2 → opencomputer_sdk-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opencomputer-sdk
-Version: 0.5.2
+Version: 0.5.3
 Summary: Python SDK for OpenComputer - cloud sandbox platform
 Project-URL: Homepage, https://github.com/diggerhq/opensandbox
 Project-URL: Repository, https://github.com/diggerhq/opensandbox

opencomputer_sdk-0.5.3/examples/test_disk_size.py

@@ -0,0 +1,102 @@
+#!/usr/bin/env python3
+"""
+Disk Size Test
+
+Verifies the `disk_mb` sandbox creation parameter:
+  1. Default (no arg) → 20GB workspace
+  2. Explicit 20GB    → 20GB workspace
+  3. Explicit 30GB    → 30GB workspace (requires org `max_disk_mb` >= 30720)
+  4. Below minimum    → rejected with 400
+
+Larger disk sizes are in closed beta — contact us to raise your org's
+max_disk_mb ceiling: https://cal.com/team/digger/opencomputer-founder-chat
+
+Usage:
+  python examples/test_disk_size.py
+
+Environment:
+  OPENCOMPUTER_API_URL  (default: http://localhost:8080)
+  OPENCOMPUTER_API_KEY  (default: test-key)
+"""
+
+import asyncio
+import os
+import sys
+
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
+from opencomputer import Sandbox
+
+API_URL = os.environ.get("OPENCOMPUTER_API_URL", "http://localhost:8080")
+API_KEY = os.environ.get("OPENCOMPUTER_API_KEY", "test-key")
+
+GREEN = "\033[32m"
+RED = "\033[31m"
+BOLD = "\033[1m"
+DIM = "\033[2m"
+RESET = "\033[0m"
+
+passed = 0
+failed = 0
+
+
+def green(msg: str) -> None:
+    print(f"{GREEN}✓ {msg}{RESET}")
+
+
+def red(msg: str) -> None:
+    print(f"{RED}✗ {msg}{RESET}")
+
+
+def bold(msg: str) -> None:
+    print(f"{BOLD}{msg}{RESET}")
+
+
+def dim(msg: str) -> None:
+    print(f"{DIM}  {msg}{RESET}")
+
+
+async def run_case(label: str, disk_mb: int | None, expect_size: str | None, expect_reject: bool = False) -> None:
+    global passed, failed
+    bold(f"\n{label}")
+    kwargs = dict(api_url=API_URL, api_key=API_KEY, timeout=120)
+    if disk_mb is not None:
+        kwargs["disk_mb"] = disk_mb
+    try:
+        async with await Sandbox.create(**kwargs) as sb:
+            dim(f"created {sb.sandbox_id}")
+            r = await sb.exec.run("df -h /home/sandbox | tail -1")
+            dim(f"df: {r.stdout.strip()}")
+            if expect_reject:
+                red("expected rejection but sandbox was created")
+                failed += 1
+            elif expect_size and expect_size in r.stdout:
+                green(f"workspace reports {expect_size}")
+                passed += 1
+            else:
+                red(f"expected {expect_size} in df output")
+                failed += 1
+    except Exception as e:
+        if expect_reject:
+            green(f"rejected as expected ({str(e).splitlines()[0]})")
+            passed += 1
+        else:
+            red(f"unexpected error: {e}")
+            failed += 1
+
+
+async def main() -> None:
+    bold("OpenComputer SDK — disk size test")
+    dim(f"API: {API_URL}")
+
+    await run_case("default (no disk_mb)", None, "20G")
+    await run_case("explicit 20GB", 20480, "20G")
+    await run_case("30GB", 30720, "30G")
+    await run_case("below minimum (8GB)", 8192, None, expect_reject=True)
+
+    print()
+    bold(f"Results: {passed} passed, {failed} failed")
+    sys.exit(1 if failed > 0 else 0)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

{opencomputer_sdk-0.5.2 → opencomputer_sdk-0.5.3}/opencomputer/sandbox.py

@@ -39,6 +39,7 @@ class Sandbox:
         api_url: str | None = None,
         envs: dict[str, str] | None = None,
         metadata: dict[str, str] | None = None,
+        disk_mb: int | None = None,
         secret_store: str | None = None,
         image: Image | None = None,
         snapshot: str | None = None,
@@ -53,6 +54,11 @@ class Sandbox:
             api_url: API URL (or OPENCOMPUTER_API_URL env var).
             envs: Environment variables to inject. Overrides store secrets.
             metadata: Custom metadata key-value pairs.
+            disk_mb: Workspace disk size in MB (default 20480 = 20GB). Any
+                additional GB above 20GB is metered at a per-second rate
+                comparable to EBS gp3. Closed beta: requests above 20GB
+                require the org's ``max_disk_mb`` to be raised. Contact us:
+                https://cal.com/team/digger/opencomputer-founder-chat
             secret_store: Secret store name — resolves encrypted secrets
                 and egress allowlist.
             image: Declarative Image definition. The server builds and caches it as a checkpoint.
@@ -88,6 +94,8 @@ class Sandbox:
             body["envs"] = envs
         if metadata:
             body["metadata"] = metadata
+        if disk_mb is not None:
+            body["diskMB"] = disk_mb
         if secret_store:
             body["secretStore"] = secret_store
         if image is not None:

{opencomputer_sdk-0.5.2 → opencomputer_sdk-0.5.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "opencomputer-sdk"
-version = "0.5.2"
+version = "0.5.3"
 description = "Python SDK for OpenComputer - cloud sandbox platform"
 readme = "README.md"
 requires-python = ">=3.10"