tensorlake 0.5.1 → 0.5.3

package/dist/index.cjs

@@ -3721,6 +3721,38 @@ __export(client_exports, {
 function sleep2(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
+function formatStartupFailureMessage(sandboxId, status, options) {
+  const prefix = status === "terminated" /* TERMINATED */ ? `Sandbox ${sandboxId} terminated during startup` : `Sandbox ${sandboxId} became ${status} during startup`;
+  const detail = formatErrorDetails(options.errorDetails);
+  if (detail) {
+    return `${prefix}: ${detail}`;
+  }
+  if (options.terminationReason) {
+    return `${prefix}: termination reason: ${options.terminationReason}`;
+  }
+  return prefix;
+}
+function formatErrorDetails(errorDetails) {
+  if (errorDetails == null) return void 0;
+  if (typeof errorDetails === "string") {
+    const detail = errorDetails.trim();
+    return detail || void 0;
+  }
+  if (Array.isArray(errorDetails)) {
+    const parts = errorDetails.map((item) => formatErrorDetails(item)).filter((item) => Boolean(item));
+    return parts.length > 0 ? parts.join("; ") : JSON.stringify(errorDetails);
+  }
+  if (typeof errorDetails === "object") {
+    for (const key of ["message", "detail", "error", "reason"]) {
+      const value = errorDetails[key];
+      if (typeof value === "string" && value.trim()) {
+        return value.trim();
+      }
+    }
+    return JSON.stringify(errorDetails);
+  }
+  return String(errorDetails);
+}
 function normalizeUserPorts(ports) {
   return dedupeAndSortPorts(ports.map(validateUserPort));
 }
@@ -4171,15 +4203,26 @@ var init_client = __esm({
         if (result.status === "running" /* RUNNING */) {
           return finishConnect(result.routingHint, result.name);
         }
+        if (result.status === "suspended" /* SUSPENDED */ || result.status === "terminated" /* TERMINATED */) {
+          throw new SandboxError(
+            formatStartupFailureMessage(result.sandboxId, result.status, {
+              errorDetails: result.errorDetails,
+              terminationReason: result.terminationReason
+            })
+          );
+        }
         const deadline = Date.now() + startupTimeout * 1e3;
         while (Date.now() < deadline) {
           const info = await this.get(result.sandboxId);
           if (info.status === "running" /* RUNNING */) {
             return finishConnect(info.routingHint, info.name);
           }
-          if (info.status === "terminated" /* TERMINATED */) {
+          if (info.status === "suspended" /* SUSPENDED */ || info.status === "terminated" /* TERMINATED */) {
             throw new SandboxError(
-              `Sandbox ${result.sandboxId} terminated during startup`
+              formatStartupFailureMessage(result.sandboxId, info.status, {
+                errorDetails: info.errorDetails,
+                terminationReason: info.terminationReason
+              })
             );
           }
           await sleep2(500);
@@ -4956,7 +4999,7 @@ async function executeDockerfilePlan(sandbox, plan, emit, sleep3) {
     );
   }
 }
-async function registerImage(context, name, dockerfile, snapshotId, snapshotUri, isPublic) {
+async function registerImage(context, name, dockerfile, snapshotId, snapshotSandboxId, snapshotUri, snapshotSizeBytes, rootfsDiskBytes, isPublic) {
   if (!context.organizationId || !context.projectId) {
     throw new Error(
       "Organization ID and Project ID are required. Run 'tl login' and 'tl init'."
@@ -4983,8 +5026,11 @@ async function registerImage(context, name, dockerfile, snapshotId, snapshotUri,
       name,
       dockerfile,
       snapshotId,
+      snapshotSandboxId,
       snapshotUri,
-      isPublic
+      snapshotSizeBytes,
+      rootfsDiskBytes,
+      public: isPublic
     })
   });
   if (!response.ok) {
@@ -5035,6 +5081,16 @@ async function createSandboxImage(source, options = {}, deps = {}) {
         `Snapshot ${snapshot.snapshotId} is missing snapshotUri and cannot be registered as a sandbox image.`
       );
     }
+    if (snapshot.sizeBytes == null) {
+      throw new Error(
+        `Snapshot ${snapshot.snapshotId} is missing sizeBytes and cannot be registered as a sandbox image.`
+      );
+    }
+    if (snapshot.rootfsDiskBytes == null) {
+      throw new Error(
+        `Snapshot ${snapshot.snapshotId} is missing rootfsDiskBytes and cannot be registered as a sandbox image.`
+      );
+    }
     emit({
       type: "status",
       message: `Registering image '${plan.registeredName}'...`
@@ -5044,7 +5100,10 @@ async function createSandboxImage(source, options = {}, deps = {}) {
       plan.registeredName,
       plan.dockerfileText,
       snapshot.snapshotId,
+      snapshot.sandboxId,
       snapshot.snapshotUri,
+      snapshot.sizeBytes,
+      snapshot.rootfsDiskBytes,
       options.isPublic ?? false
     );
     emit({
@@ -5072,25 +5131,25 @@ async function runCreateSandboxImageCli(argv = process.argv.slice(2)) {
       name: { type: "string", short: "n" },
       cpus: { type: "string" },
       memory: { type: "string" },
-      disk: { type: "string" },
+      disk_mb: { type: "string" },
       public: { type: "boolean", default: false }
     }
   });
   const dockerfilePath = parsed.positionals[0];
   if (!dockerfilePath) {
-    throw new Error("Usage: tensorlake-create-sandbox-image <dockerfile_path> [--name NAME] [--cpus N] [--memory MB] [--disk GB] [--public]");
+    throw new Error("Usage: tensorlake-create-sandbox-image <dockerfile_path> [--name NAME] [--cpus N] [--memory MB] [--disk_mb MB] [--public]");
   }
   const cpus = parsed.values.cpus != null ? Number(parsed.values.cpus) : void 0;
   const memoryMb = parsed.values.memory != null ? Number(parsed.values.memory) : void 0;
-  const diskGb = parsed.values.disk != null ? Number(parsed.values.disk) : void 0;
+  const diskMb = parsed.values.disk_mb != null ? Number(parsed.values.disk_mb) : void 0;
   if (cpus != null && !Number.isFinite(cpus)) {
     throw new Error(`Invalid --cpus value: ${parsed.values.cpus}`);
   }
   if (memoryMb != null && !Number.isInteger(memoryMb)) {
     throw new Error(`Invalid --memory value: ${parsed.values.memory}`);
   }
-  if (diskGb != null && !Number.isInteger(diskGb)) {
-    throw new Error(`Invalid --disk value: ${parsed.values.disk}`);
+  if (diskMb != null && !Number.isInteger(diskMb)) {
+    throw new Error(`Invalid --disk_mb value: ${parsed.values.disk_mb}`);
   }
   await createSandboxImage(
     dockerfilePath,
@@ -5098,7 +5157,7 @@ async function runCreateSandboxImageCli(argv = process.argv.slice(2)) {
       registeredName: parsed.values.name,
       cpus,
       memoryMb,
-      diskMb: diskGb != null ? diskGb * 1024 : void 0,
+      diskMb,
       isPublic: parsed.values.public
     },
     { emit: ndjsonStdoutEmit }