tensorlake 0.5.140 → 0.5.141

package/dist/index.js

@@ -46,7 +46,7 @@ var init_defaults = __esm({
   "src/defaults.ts"() {
     "use strict";
     init_esm_shims();
-    SDK_VERSION = "0.5.140";
+    SDK_VERSION = "0.5.141";
     API_URL = process.env.TENSORLAKE_API_URL ?? "https://api.tensorlake.ai";
     API_KEY = process.env.TENSORLAKE_API_KEY ?? void 0;
     NAMESPACE = process.env.INDEXIFY_NAMESPACE ?? "default";
@@ -79,7 +79,7 @@ function formatErrorDetails(errorDetails) {
   }
   return String(errorDetails);
 }
-var SandboxException, SandboxError, SandboxPending, SandboxConnectionError, SandboxNotFoundError, PoolNotFoundError, PoolInUseError, RemoteAPIError, RequestNotFinishedError, RequestFailedError, RequestExecutionError;
+var SandboxException, SandboxError, SandboxResizeError, SandboxPending, SandboxConnectionError, SandboxNotFoundError, PoolNotFoundError, PoolInUseError, RemoteAPIError, RequestNotFinishedError, RequestFailedError, RequestExecutionError;
 var init_errors = __esm({
   "src/errors.ts"() {
     "use strict";
@@ -100,6 +100,25 @@ var init_errors = __esm({
         if (options?.sandboxId !== void 0) this.sandboxId = options.sandboxId;
       }
     };
+    SandboxResizeError = class extends SandboxError {
+      generation;
+      info;
+      constructor(payload) {
+        const resources = payload.info?.resources;
+        const allocation = resources ? `${resources.cpus} CPUs, ${resources.memoryMb} MiB memory, ${resources.diskMb} MiB disk` : "unavailable";
+        super(
+          `Sandbox ${payload.sandboxId} resize generation ${payload.generation}: ${payload.message}; last confirmed allocation: ${allocation}`,
+          { reason: payload.reason, sandboxId: payload.sandboxId }
+        );
+        this.name = "SandboxResizeError";
+        this.generation = payload.generation;
+        this.info = payload.info ?? void 0;
+      }
+      /** Last confirmed allocation, if an observation was available. */
+      get confirmedResources() {
+        return this.info?.resources;
+      }
+    };
     SandboxPending = class extends SandboxError {
       pendingReason;
       timeout;
@@ -471,12 +490,148 @@ var init_native_worker_client = __esm({
       }
     };
     runtimeKey = /* @__PURE__ */ Symbol.for(
-      `tensorlake.native.worker.${false ? "source" : "0.5.140"}.v1`
+      `tensorlake.native.worker.${false ? "source" : "0.5.141"}.v1`
     );
     runtimes = globalThis;
   }
 });
 
+// src/models.ts
+function isSandboxPending(status) {
+  return String(status) === "pending" /* PENDING */;
+}
+function camelToSnake(str) {
+  return str.replace(CAMEL_TO_SNAKE_RE, (ch) => "_" + ch.toLowerCase());
+}
+function snakeToCamel(str) {
+  return str.replace(/_([a-z])/g, (_, ch) => ch.toUpperCase());
+}
+function toSnakeKeys(obj) {
+  if (Array.isArray(obj)) return obj.map(toSnakeKeys);
+  if (obj !== null && typeof obj === "object" && !(obj instanceof Date)) {
+    const result = {};
+    for (const [k, v] of Object.entries(obj)) {
+      result[camelToSnake(k)] = toSnakeKeys(v);
+    }
+    return result;
+  }
+  return obj;
+}
+function parseTimestamp(v) {
+  if (v == null) return void 0;
+  if (v instanceof Date) return v;
+  if (typeof v === "string") {
+    const parsed = Date.parse(v);
+    return Number.isNaN(parsed) ? void 0 : new Date(parsed);
+  }
+  const ts = Number(v);
+  if (isNaN(ts)) return void 0;
+  if (ts > 1e15) return new Date(ts / 1e3);
+  if (ts > 1e12) return new Date(ts);
+  return new Date(ts * 1e3);
+}
+function fromSnakeKeys(obj, idField) {
+  if (Array.isArray(obj))
+    return obj.map((item) => fromSnakeKeys(item, idField));
+  if (obj !== null && typeof obj === "object" && !(obj instanceof Date)) {
+    const result = {};
+    for (const [k, v] of Object.entries(obj)) {
+      let key;
+      if (k === "id" && idField) {
+        key = idField;
+      } else {
+        key = snakeToCamel(k);
+      }
+      if (key.endsWith("At") || key === "timestamp" || key === "startedAt" || key === "endedAt") {
+        result[key] = parseTimestamp(v);
+      } else if (typeof v === "object" && v !== null && !Array.isArray(v)) {
+        result[key] = fromSnakeKeys(v);
+      } else if (Array.isArray(v)) {
+        result[key] = v.map((item) => fromSnakeKeys(item));
+      } else {
+        result[key] = v === null ? void 0 : v;
+      }
+    }
+    return result;
+  }
+  return obj;
+}
+var SandboxStatus, SandboxPendingReason, TERMINATION_REASON_NO_CAPACITY, TERMINATION_REASON_CANCELLED, SnapshotStatus, ProcessStatus, StdinMode, OutputMode, ContainerState, ResizeStatus, ResizeErrorReason, CAMEL_TO_SNAKE_RE;
+var init_models = __esm({
+  "src/models.ts"() {
+    "use strict";
+    init_esm_shims();
+    SandboxStatus = /* @__PURE__ */ ((SandboxStatus2) => {
+      SandboxStatus2["PENDING"] = "pending";
+      SandboxStatus2["RUNNING"] = "running";
+      SandboxStatus2["SNAPSHOTTING"] = "snapshotting";
+      SandboxStatus2["SUSPENDING"] = "suspending";
+      SandboxStatus2["SUSPENDED"] = "suspended";
+      SandboxStatus2["TERMINATED"] = "terminated";
+      SandboxStatus2["TIMEOUT"] = "timeout";
+      SandboxStatus2["FAILED"] = "failed";
+      return SandboxStatus2;
+    })(SandboxStatus || {});
+    SandboxPendingReason = /* @__PURE__ */ ((SandboxPendingReason2) => {
+      SandboxPendingReason2["SCHEDULING"] = "scheduling";
+      SandboxPendingReason2["WAITING_FOR_CONTAINER"] = "waiting_for_container";
+      SandboxPendingReason2["NO_EXECUTORS_AVAILABLE"] = "no_executors_available";
+      SandboxPendingReason2["NO_RESOURCES_AVAILABLE"] = "no_resources_available";
+      SandboxPendingReason2["POOL_AT_CAPACITY"] = "pool_at_capacity";
+      SandboxPendingReason2["MAX_CAPACITY_REACHED"] = "max_capacity_reached";
+      SandboxPendingReason2["NO_EXECUTOR_FOR_SNAPSHOT_LOCATION"] = "no_executor_for_snapshot_location";
+      SandboxPendingReason2["UNPLACEABLE"] = "unplaceable";
+      return SandboxPendingReason2;
+    })(SandboxPendingReason || {});
+    TERMINATION_REASON_NO_CAPACITY = "no_capacity";
+    TERMINATION_REASON_CANCELLED = "cancelled";
+    SnapshotStatus = /* @__PURE__ */ ((SnapshotStatus2) => {
+      SnapshotStatus2["IN_PROGRESS"] = "in_progress";
+      SnapshotStatus2["LOCAL_READY"] = "local_ready";
+      SnapshotStatus2["COMPLETED"] = "completed";
+      SnapshotStatus2["FAILED"] = "failed";
+      return SnapshotStatus2;
+    })(SnapshotStatus || {});
+    ProcessStatus = /* @__PURE__ */ ((ProcessStatus2) => {
+      ProcessStatus2["RUNNING"] = "running";
+      ProcessStatus2["EXITED"] = "exited";
+      ProcessStatus2["SIGNALED"] = "signaled";
+      ProcessStatus2["OOM_KILLED"] = "oom_killed";
+      return ProcessStatus2;
+    })(ProcessStatus || {});
+    StdinMode = /* @__PURE__ */ ((StdinMode2) => {
+      StdinMode2["CLOSED"] = "closed";
+      StdinMode2["PIPE"] = "pipe";
+      return StdinMode2;
+    })(StdinMode || {});
+    OutputMode = /* @__PURE__ */ ((OutputMode2) => {
+      OutputMode2["CAPTURE"] = "capture";
+      OutputMode2["DISCARD"] = "discard";
+      return OutputMode2;
+    })(OutputMode || {});
+    ContainerState = /* @__PURE__ */ ((ContainerState2) => {
+      ContainerState2["IDLE"] = "Idle";
+      ContainerState2["RUNNING"] = "Running";
+      return ContainerState2;
+    })(ContainerState || {});
+    ResizeStatus = /* @__PURE__ */ ((ResizeStatus2) => {
+      ResizeStatus2["PENDING"] = "pending";
+      ResizeStatus2["SUCCEEDED"] = "succeeded";
+      ResizeStatus2["FAILED"] = "failed";
+      return ResizeStatus2;
+    })(ResizeStatus || {});
+    ResizeErrorReason = /* @__PURE__ */ ((ResizeErrorReason2) => {
+      ResizeErrorReason2["FAILED"] = "failed";
+      ResizeErrorReason2["TIMEOUT"] = "timeout";
+      ResizeErrorReason2["SUPERSEDED"] = "superseded";
+      ResizeErrorReason2["INTERRUPTED"] = "interrupted";
+      ResizeErrorReason2["INCOMPATIBLE_RESPONSE"] = "incompatible_response";
+      return ResizeErrorReason2;
+    })(ResizeErrorReason || {});
+    CAMEL_TO_SNAKE_RE = /[A-Z]/g;
+  }
+});
+
 // src/native-sandbox.ts
 function loadNativeSandboxBinding() {
   return cachedBinding ??= workerBinding();
@@ -504,6 +659,11 @@ function translateNativeError(error, context) {
     return new SandboxError(String(error));
   }
   const { category, status, message } = payload;
+  if (category === "resize") {
+    return new SandboxResizeError(
+      fromSnakeKeys(JSON.parse(message), "sandboxId")
+    );
+  }
   if (category === "connection") {
     return new SandboxConnectionError(message, { cause: error });
   }
@@ -616,134 +776,13 @@ var init_native_sandbox = __esm({
   "src/native-sandbox.ts"() {
     "use strict";
     init_esm_shims();
+    init_models();
     init_errors();
     init_native_worker_client();
     init_native_worker_protocol();
   }
 });
 
-// src/models.ts
-function isSandboxPending(status) {
-  return String(status) === "pending" /* PENDING */;
-}
-function camelToSnake(str) {
-  return str.replace(CAMEL_TO_SNAKE_RE, (ch) => "_" + ch.toLowerCase());
-}
-function snakeToCamel(str) {
-  return str.replace(/_([a-z])/g, (_, ch) => ch.toUpperCase());
-}
-function toSnakeKeys(obj) {
-  if (Array.isArray(obj)) return obj.map(toSnakeKeys);
-  if (obj !== null && typeof obj === "object" && !(obj instanceof Date)) {
-    const result = {};
-    for (const [k, v] of Object.entries(obj)) {
-      result[camelToSnake(k)] = toSnakeKeys(v);
-    }
-    return result;
-  }
-  return obj;
-}
-function parseTimestamp(v) {
-  if (v == null) return void 0;
-  if (v instanceof Date) return v;
-  if (typeof v === "string") {
-    const parsed = Date.parse(v);
-    return Number.isNaN(parsed) ? void 0 : new Date(parsed);
-  }
-  const ts = Number(v);
-  if (isNaN(ts)) return void 0;
-  if (ts > 1e15) return new Date(ts / 1e3);
-  if (ts > 1e12) return new Date(ts);
-  return new Date(ts * 1e3);
-}
-function fromSnakeKeys(obj, idField) {
-  if (Array.isArray(obj))
-    return obj.map((item) => fromSnakeKeys(item, idField));
-  if (obj !== null && typeof obj === "object" && !(obj instanceof Date)) {
-    const result = {};
-    for (const [k, v] of Object.entries(obj)) {
-      let key;
-      if (k === "id" && idField) {
-        key = idField;
-      } else {
-        key = snakeToCamel(k);
-      }
-      if (key.endsWith("At") || key === "timestamp" || key === "startedAt" || key === "endedAt") {
-        result[key] = parseTimestamp(v);
-      } else if (typeof v === "object" && v !== null && !Array.isArray(v)) {
-        result[key] = fromSnakeKeys(v);
-      } else if (Array.isArray(v)) {
-        result[key] = v.map((item) => fromSnakeKeys(item));
-      } else {
-        result[key] = v === null ? void 0 : v;
-      }
-    }
-    return result;
-  }
-  return obj;
-}
-var SandboxStatus, SandboxPendingReason, TERMINATION_REASON_NO_CAPACITY, TERMINATION_REASON_CANCELLED, SnapshotStatus, ProcessStatus, StdinMode, OutputMode, ContainerState, CAMEL_TO_SNAKE_RE;
-var init_models = __esm({
-  "src/models.ts"() {
-    "use strict";
-    init_esm_shims();
-    SandboxStatus = /* @__PURE__ */ ((SandboxStatus2) => {
-      SandboxStatus2["PENDING"] = "pending";
-      SandboxStatus2["RUNNING"] = "running";
-      SandboxStatus2["SNAPSHOTTING"] = "snapshotting";
-      SandboxStatus2["SUSPENDING"] = "suspending";
-      SandboxStatus2["SUSPENDED"] = "suspended";
-      SandboxStatus2["TERMINATED"] = "terminated";
-      SandboxStatus2["TIMEOUT"] = "timeout";
-      SandboxStatus2["FAILED"] = "failed";
-      return SandboxStatus2;
-    })(SandboxStatus || {});
-    SandboxPendingReason = /* @__PURE__ */ ((SandboxPendingReason2) => {
-      SandboxPendingReason2["SCHEDULING"] = "scheduling";
-      SandboxPendingReason2["WAITING_FOR_CONTAINER"] = "waiting_for_container";
-      SandboxPendingReason2["NO_EXECUTORS_AVAILABLE"] = "no_executors_available";
-      SandboxPendingReason2["NO_RESOURCES_AVAILABLE"] = "no_resources_available";
-      SandboxPendingReason2["POOL_AT_CAPACITY"] = "pool_at_capacity";
-      SandboxPendingReason2["MAX_CAPACITY_REACHED"] = "max_capacity_reached";
-      SandboxPendingReason2["NO_EXECUTOR_FOR_SNAPSHOT_LOCATION"] = "no_executor_for_snapshot_location";
-      SandboxPendingReason2["UNPLACEABLE"] = "unplaceable";
-      return SandboxPendingReason2;
-    })(SandboxPendingReason || {});
-    TERMINATION_REASON_NO_CAPACITY = "no_capacity";
-    TERMINATION_REASON_CANCELLED = "cancelled";
-    SnapshotStatus = /* @__PURE__ */ ((SnapshotStatus2) => {
-      SnapshotStatus2["IN_PROGRESS"] = "in_progress";
-      SnapshotStatus2["LOCAL_READY"] = "local_ready";
-      SnapshotStatus2["COMPLETED"] = "completed";
-      SnapshotStatus2["FAILED"] = "failed";
-      return SnapshotStatus2;
-    })(SnapshotStatus || {});
-    ProcessStatus = /* @__PURE__ */ ((ProcessStatus2) => {
-      ProcessStatus2["RUNNING"] = "running";
-      ProcessStatus2["EXITED"] = "exited";
-      ProcessStatus2["SIGNALED"] = "signaled";
-      ProcessStatus2["OOM_KILLED"] = "oom_killed";
-      return ProcessStatus2;
-    })(ProcessStatus || {});
-    StdinMode = /* @__PURE__ */ ((StdinMode2) => {
-      StdinMode2["CLOSED"] = "closed";
-      StdinMode2["PIPE"] = "pipe";
-      return StdinMode2;
-    })(StdinMode || {});
-    OutputMode = /* @__PURE__ */ ((OutputMode2) => {
-      OutputMode2["CAPTURE"] = "capture";
-      OutputMode2["DISCARD"] = "discard";
-      return OutputMode2;
-    })(OutputMode || {});
-    ContainerState = /* @__PURE__ */ ((ContainerState2) => {
-      ContainerState2["IDLE"] = "Idle";
-      ContainerState2["RUNNING"] = "Running";
-      return ContainerState2;
-    })(ContainerState || {});
-    CAMEL_TO_SNAKE_RE = /[A-Z]/g;
-  }
-});
-
 // src/pending-sandbox.ts
 var PendingSandbox;
 var init_pending_sandbox = __esm({
@@ -3963,8 +4002,12 @@ var init_sandbox = __esm({
         return info.status;
       }
       /**
-       * Update this sandbox's properties (name, exposed ports, proxy auth, and
-       * egress network policy).
+       * Update this sandbox's properties or resize its resources.
+       *
+       * Resource targets use create's cpus, memoryMb, and diskMb names. Resize
+       * waits by default; wait=false returns admission. Timeout does not cancel it.
+       * Wait, timeout, and pollInterval are ignored on non-resource updates.
+       * A no-op returns current resources with absent or earlier resize metadata.
        *
        * Naming an ephemeral sandbox makes it non-ephemeral and enables
        * suspend/resume. For `network`, omit to keep the current policy, pass an
@@ -3977,6 +4020,14 @@ var init_sandbox = __esm({
         this._setName(info.name ?? null);
         return info;
       }
+      /** Wait for this exact resize generation without submitting another update. */
+      async waitForResourceResize(generation, options = {}) {
+        const client = this.requireLifecycleClient("waitForResourceResize");
+        const info = await client.waitForResourceResize(this.lifecycleIdentifier, generation, options);
+        this._setLifecycleIdentifier(info.sandboxId);
+        this._setName(info.name ?? null);
+        return info;
+      }
       /**
        * Suspend this sandbox.
        *
@@ -4661,6 +4712,16 @@ function validateUserPort(port) {
 function dedupeAndSortPorts(ports) {
   return [...new Set(ports)].sort((a, b) => a - b);
 }
+function validateResizeWait(options) {
+  const timeout = options.timeout ?? 300;
+  const poll = options.pollInterval ?? 1;
+  if (typeof timeout !== "number" || !Number.isFinite(timeout) || timeout < 0) {
+    throw new SandboxError("timeout must be finite and non-negative");
+  }
+  if (typeof poll !== "number" || !Number.isFinite(poll) || poll <= 0) {
+    throw new SandboxError("pollInterval must be finite and positive");
+  }
+}
 var DEFAULT_WAIT_POLL_INTERVAL_SEC, GPU_MODELS, SandboxClient, RESERVED_SANDBOX_MANAGEMENT_PORT;
 var init_client = __esm({
   "src/client.ts"() {
@@ -5044,9 +5105,39 @@ var init_client = __esm({
           traceId
         });
       }
-      /** Update sandbox properties such as name, exposed ports, and proxy auth settings. */
+      /**
+       * Update properties or resize a Running Cloud Hypervisor sandbox.
+       * Resource names match create. Resize waits by default; wait=false returns
+       * admission. Timeout leaves the resize running and reports its generation.
+       * Wait, timeout, and pollInterval are ignored on non-resource updates.
+       * A no-op returns current resources with absent or earlier resize metadata.
+       */
       async update(sandboxId, options) {
         const body = {};
+        const resources = {};
+        for (const [field, wire, value] of [
+          ["cpus", "cpus", options.cpus],
+          ["memoryMb", "memory_mb", options.memoryMb],
+          ["diskMb", "disk_mb", options.diskMb]
+        ]) {
+          if (value === void 0) continue;
+          if (typeof value !== "number" || !Number.isSafeInteger(value) || value <= 0) {
+            throw new SandboxError(
+              `${field} ${String(value)} must be a finite positive ${field === "cpus" ? "whole-vCPU count" : "integer number of MiB"}`
+            );
+          }
+          resources[wire] = value;
+        }
+        const resizing = Object.keys(resources).length > 0;
+        if (resizing) {
+          validateResizeWait(options);
+          if (options.name != null || options.allowUnauthenticatedAccess != null || options.exposedPorts != null || options.network !== void 0) {
+            throw new SandboxError(
+              "resources cannot be combined with other sandbox update fields"
+            );
+          }
+          body.resources = resources;
+        }
         if (options.name != null) body.name = options.name;
         if (options.allowUnauthenticatedAccess != null) {
           body.allow_unauthenticated_access = options.allowUnauthenticatedAccess;
@@ -5063,7 +5154,30 @@ var init_client = __esm({
           );
         }
         return this.tracedJson(
-          () => this.native.updateSandbox(sandboxId, JSON.stringify(body)),
+          () => resizing ? this.native.updateSandbox(
+            sandboxId,
+            JSON.stringify(body),
+            options.wait ?? true,
+            options.timeout ?? 300,
+            options.pollInterval ?? 1
+          ) : this.native.updateSandbox(sandboxId, JSON.stringify(body)),
+          "sandboxId",
+          { sandboxId, notFoundKind: "sandbox" }
+        );
+      }
+      /** Wait for this exact generation. Timeout leaves the resize running. */
+      async waitForResourceResize(sandboxId, generation, options = {}) {
+        validateResizeWait(options);
+        if (!Number.isSafeInteger(generation) || generation < 1) {
+          throw new SandboxError("generation must be a positive safe integer");
+        }
+        return this.tracedJson(
+          () => this.native.waitForResourceResize(
+            sandboxId,
+            generation,
+            options.timeout ?? 300,
+            options.pollInterval ?? 1
+          ),
           "sandboxId",
           { sandboxId, notFoundKind: "sandbox" }
         );
@@ -6814,7 +6928,7 @@ async function runCreateSandboxImageCli(argv = process.argv.slice(2)) {
   const dockerfilePath = parsed.positionals[0];
   if (!dockerfilePath) {
     throw new Error(
-      "Usage: tensorlake-create-sandbox-image <dockerfile_path> [--name NAME] [--cpus N] [--memory MB] [--disk_mb MB] [--builder_disk_mb MB] [--docker_compat] [--public]"
+      "Usage: tensorlake-create-sandbox-image <dockerfile_path> [--name NAME] [--cpus N] [--memory MiB] [--disk_mb MiB] [--builder_disk_mb MiB] [--docker_compat] [--public]"
     );
   }
   const cpus = parsed.values.cpus != null ? Number(parsed.values.cpus) : void 0;
@@ -6866,7 +6980,7 @@ async function runImportSandboxImageCli(argv = process.argv.slice(2)) {
   const imageReference = parsed.positionals[0];
   if (!imageReference) {
     throw new Error(
-      "Usage: tensorlake-import-sandbox-image <image_reference> [--name NAME] [--cpus N] [--memory MB] [--disk_mb MB] [--builder_disk_mb MB] [--docker_compat] [--public]"
+      "Usage: tensorlake-import-sandbox-image <image_reference> [--name NAME] [--cpus N] [--memory MiB] [--disk_mb MiB] [--builder_disk_mb MiB] [--docker_compat] [--public]"
     );
   }
   const cpus = parsed.values.cpus != null ? Number(parsed.values.cpus) : void 0;
@@ -9946,6 +10060,8 @@ export {
   RequestExecutionError,
   RequestFailedError,
   RequestNotFinishedError,
+  ResizeErrorReason,
+  ResizeStatus,
   Sandbox,
   SandboxClient,
   SandboxConnectionError,
@@ -9954,6 +10070,7 @@ export {
   SandboxNotFoundError,
   SandboxPending,
   SandboxPendingReason,
+  SandboxResizeError,
   SandboxStatus,
   SnapshotStatus,
   StdinMode,