tensorlake 0.5.140 → 0.5.141

package/dist/sandbox-image.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";
@@ -57,6 +57,56 @@ var init_defaults = __esm({
   }
 });
 
+// src/models.ts
+function snakeToCamel(str) {
+  return str.replace(/_([a-z])/g, (_, ch) => ch.toUpperCase());
+}
+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 init_models = __esm({
+  "src/models.ts"() {
+    "use strict";
+    init_esm_shims();
+  }
+});
+
 // src/errors.ts
 function formatErrorDetails(errorDetails) {
   if (errorDetails == null) return void 0;
@@ -79,7 +129,7 @@ function formatErrorDetails(errorDetails) {
   }
   return String(errorDetails);
 }
-var SandboxException, SandboxError, SandboxConnectionError, SandboxNotFoundError, PoolNotFoundError, PoolInUseError, RemoteAPIError;
+var SandboxException, SandboxError, SandboxResizeError, SandboxConnectionError, SandboxNotFoundError, PoolNotFoundError, PoolInUseError, RemoteAPIError;
 var init_errors = __esm({
   "src/errors.ts"() {
     "use strict";
@@ -100,6 +150,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;
+      }
+    };
     SandboxConnectionError = class extends SandboxError {
       constructor(message, options) {
         super(`Connection error: ${message}`);
@@ -421,7 +490,7 @@ 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;
   }
@@ -454,6 +523,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 });
   }
@@ -544,6 +618,7 @@ 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();
@@ -634,56 +709,6 @@ var init_native_cloud = __esm({
   }
 });
 
-// src/models.ts
-function snakeToCamel(str) {
-  return str.replace(/_([a-z])/g, (_, ch) => ch.toUpperCase());
-}
-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 init_models = __esm({
-  "src/models.ts"() {
-    "use strict";
-    init_esm_shims();
-  }
-});
-
 // src/cloud-client.ts
 import { nanoid } from "nanoid";
 function secretFromMetadata(metadata) {
@@ -1850,7 +1875,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;
@@ -1902,7 +1927,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;