opencomputer-sdk 0.6.2__tar.gz → 0.6.3__tar.gz

{opencomputer_sdk-0.6.2 → opencomputer_sdk-0.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opencomputer-sdk
-Version: 0.6.2
+Version: 0.6.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.6.2 → opencomputer_sdk-0.6.3}/opencomputer/__init__.py

@@ -65,4 +65,4 @@ __all__ = [
     "TagKeyInfo",
 ]
 
-__version__ = "0.6.0"
+__version__ = "0.6.3"

{opencomputer_sdk-0.6.2 → opencomputer_sdk-0.6.3}/opencomputer/image.py

@@ -6,7 +6,7 @@ import base64
 import hashlib
 import json
 import os
-from dataclasses import dataclass, field
+from dataclasses import dataclass, field, replace
 from typing import Any
 
 
@@ -47,6 +47,10 @@ class Image:
 
     _base: str = "base"
     _steps: tuple[ImageStep, ...] = field(default_factory=tuple)
+    # RAM for the build phase (apt/pip). 0 = server default. Does not pin the
+    # output image — the server re-snapshots at the default 1 GB floor, and you
+    # size the actual sandbox at create time via memory_mb.
+    _builder_memory_mb: int = 0
 
     @classmethod
     def base(cls) -> Image:
@@ -60,36 +64,36 @@ class Image:
 
     def apt_install(self, packages: list[str]) -> Image:
         """Install system packages via apt-get."""
-        return Image(
-            _base=self._base,
+        return replace(
+            self,
             _steps=(*self._steps, ImageStep("apt_install", {"packages": packages})),
         )
 
     def pip_install(self, packages: list[str]) -> Image:
         """Install Python packages via pip."""
-        return Image(
-            _base=self._base,
+        return replace(
+            self,
             _steps=(*self._steps, ImageStep("pip_install", {"packages": packages})),
         )
 
     def run_commands(self, *commands: str) -> Image:
         """Run one or more shell commands."""
-        return Image(
-            _base=self._base,
+        return replace(
+            self,
             _steps=(*self._steps, ImageStep("run", {"commands": list(commands)})),
         )
 
     def env(self, vars: dict[str, str]) -> Image:
         """Set environment variables (written to /etc/environment)."""
-        return Image(
-            _base=self._base,
+        return replace(
+            self,
             _steps=(*self._steps, ImageStep("env", {"vars": vars})),
         )
 
     def workdir(self, path: str) -> Image:
         """Set the default working directory."""
-        return Image(
-            _base=self._base,
+        return replace(
+            self,
             _steps=(*self._steps, ImageStep("workdir", {"path": path})),
         )
 
@@ -101,8 +105,8 @@ class Image:
             content: String content of the file.
         """
         encoded = base64.b64encode(content.encode()).decode()
-        return Image(
-            _base=self._base,
+        return replace(
+            self,
             _steps=(*self._steps, ImageStep("add_file", {
                 "path": remote_path,
                 "content": encoded,
@@ -121,8 +125,8 @@ class Image:
         """
         with open(local_path, "rb") as f:
             encoded = base64.b64encode(f.read()).decode()
-        return Image(
-            _base=self._base,
+        return replace(
+            self,
             _steps=(*self._steps, ImageStep("add_file", {
                 "path": remote_path,
                 "content": encoded,
@@ -147,22 +151,33 @@ class Image:
                 with open(full, "rb") as f:
                     encoded = base64.b64encode(f.read()).decode()
                 files.append({"relativePath": rel, "content": encoded})
-        return Image(
-            _base=self._base,
+        return replace(
+            self,
             _steps=(*self._steps, ImageStep("add_dir", {
                 "path": remote_path,
                 "files": files,
             })),
         )
 
+    def builder_memory(self, mb: int) -> Image:
+        """Set the RAM (MB) for the build phase. Use this when a build OOMs at the
+        default 4 GB (e.g. heavy ``apt``/``pip``/``npm``). Does not affect the
+        resulting image's memory — size the sandbox at create time via ``memory_mb``."""
+        return replace(self, _builder_memory_mb=mb)
+
     def to_dict(self) -> dict[str, Any]:
         """Returns the manifest as a plain dict (for JSON serialization)."""
-        return {
+        d: dict[str, Any] = {
             "base": self._base,
             "steps": [s.to_dict() for s in self._steps],
         }
+        if self._builder_memory_mb > 0:
+            d["builderMemoryMB"] = self._builder_memory_mb
+        return d
 
     def cache_key(self) -> str:
-        """Compute a deterministic content hash for caching."""
-        canonical = json.dumps(self.to_dict(), sort_keys=True, separators=(",", ":"))
+        """Deterministic content hash for caching. Memory knobs are resource
+        params, not image content, so they're excluded (matches the server)."""
+        content = {"base": self._base, "steps": [s.to_dict() for s in self._steps]}
+        canonical = json.dumps(content, sort_keys=True, separators=(",", ":"))
         return hashlib.sha256(canonical.encode()).hexdigest()

{opencomputer_sdk-0.6.2 → opencomputer_sdk-0.6.3}/opencomputer/sandbox.py

@@ -83,6 +83,10 @@ class Sandbox:
     _client: httpx.AsyncClient = field(default=None, repr=False)
     _data_client: httpx.AsyncClient = field(default=None, repr=False)
 
+    @property
+    def id(self) -> str:
+        return self.sandbox_id
+
     @classmethod
     async def create(
         cls,

{opencomputer_sdk-0.6.2 → opencomputer_sdk-0.6.3}/pyproject.toml

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