tensorlake 0.5.9 → 0.5.10

package/dist/index.cjs

@@ -35,7 +35,7 @@ var SDK_VERSION, API_URL, API_KEY, NAMESPACE, SANDBOX_PROXY_URL, DEFAULT_HTTP_TI
 var init_defaults = __esm({
   "src/defaults.ts"() {
     "use strict";
-    SDK_VERSION = "0.5.9";
+    SDK_VERSION = "0.5.10";
     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";
@@ -426,11 +426,12 @@ var init_models = __esm({
       SandboxStatus3["TERMINATED"] = "terminated";
       return SandboxStatus3;
     })(SandboxStatus || {});
-    SnapshotStatus = /* @__PURE__ */ ((SnapshotStatus3) => {
-      SnapshotStatus3["IN_PROGRESS"] = "in_progress";
-      SnapshotStatus3["COMPLETED"] = "completed";
-      SnapshotStatus3["FAILED"] = "failed";
-      return SnapshotStatus3;
+    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";
@@ -3395,11 +3396,10 @@ var init_sandbox = __esm({
         await client.resume(this.lifecycleIdentifier, options);
       }
       /**
-       * Create a snapshot of this sandbox's filesystem and wait for it to
-       * be committed.
+       * Create a checkpoint of this sandbox and wait for it to be locally ready.
        *
-       * By default blocks until the snapshot artifact is ready and returns
-       * the completed `SnapshotInfo`. Pass `{ wait: false }` to fire-and-return
+       * By default blocks until the checkpoint is resumable and returns
+       * `SnapshotInfo`. Pass `{ wait: false }` to fire-and-return
        * (returns `undefined`).
        */
       async checkpoint(options) {
@@ -3411,7 +3411,8 @@ var init_sandbox = __esm({
         return client.snapshotAndWait(this.lifecycleIdentifier, {
           timeout: options?.timeout,
           pollInterval: options?.pollInterval,
-          snapshotType: options?.checkpointType
+          snapshotType: options?.checkpointType,
+          waitUntil: options?.waitUntil
         });
       }
       /**
@@ -3806,6 +3807,12 @@ __export(client_exports, {
 function sleep2(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
+function snapshotStatusSatisfiesWaitCondition(status, waitUntil) {
+  if (waitUntil === "local_ready") {
+    return status === "local_ready" /* LOCAL_READY */ || status === "completed" /* COMPLETED */;
+  }
+  return status === "completed" /* COMPLETED */;
+}
 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);
@@ -4106,7 +4113,8 @@ var init_client = __esm({
        *
        * This call **returns immediately** with a `snapshotId` and `in_progress`
        * status — the snapshot is created asynchronously. Poll `getSnapshot()` until
-       * `completed` or `failed`, or use `snapshotAndWait()` to block automatically.
+       * `local_ready`, `completed`, or `failed`, or use `snapshotAndWait()` to
+       * block automatically.
        *
        * @param options.snapshotType - `"filesystem"` for cold-boot snapshots (e.g. image builds).
        *   Omit to use the server default (`filesystem`).
@@ -4147,9 +4155,11 @@ var init_client = __esm({
         );
       }
       /**
-       * Create a snapshot and block until it is committed.
+       * Create a snapshot and block until it is locally ready.
        *
-       * Combines `snapshot()` with polling `getSnapshot()` until `completed`.
+       * Combines `snapshot()` with polling `getSnapshot()` until `local_ready`
+       * or `completed`. Pass `{ waitUntil: "completed" }` when durable
+       * `snapshotUri` metadata is required.
        * Prefer `sandbox.checkpoint()` on a `Sandbox` handle for the same behavior
        * without managing the client separately.
        *
@@ -4162,13 +4172,14 @@ var init_client = __esm({
       async snapshotAndWait(sandboxId, options) {
         const timeout = options?.timeout ?? 300;
         const pollInterval = options?.pollInterval ?? 1;
+        const waitUntil = options?.waitUntil ?? "local_ready";
         const result = await this.snapshot(sandboxId, {
           snapshotType: options?.snapshotType
         });
         const deadline = Date.now() + timeout * 1e3;
         while (Date.now() < deadline) {
           const info = await this.getSnapshot(result.snapshotId);
-          if (info.status === "completed" /* COMPLETED */) return info;
+          if (snapshotStatusSatisfiesWaitCondition(info.status, waitUntil)) return info;
           if (info.status === "failed" /* FAILED */) {
             throw new SandboxError(
               `Snapshot ${result.snapshotId} failed: ${info.error}`
@@ -4177,7 +4188,7 @@ var init_client = __esm({
           await sleep2(pollInterval * 1e3);
         }
         throw new SandboxError(
-          `Snapshot ${result.snapshotId} did not complete within ${timeout}s`
+          `Snapshot ${result.snapshotId} did not reach ${waitUntil} within ${timeout}s`
         );
       }
       // --- Pools ---
@@ -4462,6 +4473,7 @@ __export(sandbox_image_exports, {
   loadDockerfilePlan: () => loadDockerfilePlan,
   loadImagePlan: () => loadImagePlan,
   logicalDockerfileLines: () => logicalDockerfileLines,
+  registerImage: () => registerImage,
   runCreateSandboxImageCli: () => runCreateSandboxImageCli
 });
 function defaultRegisteredName(dockerfilePath) {
@@ -5084,23 +5096,26 @@ async function executeDockerfilePlan(sandbox, plan, emit, sleep3) {
     );
   }
 }
-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'."
-    );
-  }
+async function registerImage(context, name, dockerfile, snapshotId, snapshotSandboxId, snapshotUri, snapshotSizeBytes, rootfsDiskBytes, isPublic, snapshotFormatVersion) {
   const bearerToken = context.apiKey ?? context.personalAccessToken;
   if (!bearerToken) {
     throw new Error("Missing TENSORLAKE_API_KEY or TENSORLAKE_PAT.");
   }
   const baseUrl = context.apiUrl.replace(/\/+$/, "");
-  const url = `${baseUrl}/platform/v1/organizations/${encodeURIComponent(context.organizationId)}/projects/${encodeURIComponent(context.projectId)}/sandbox-templates`;
   const headers = {
     Authorization: `Bearer ${bearerToken}`,
     "Content-Type": "application/json"
   };
-  if (context.personalAccessToken && !context.apiKey) {
+  let url;
+  if (context.apiKey) {
+    url = `${baseUrl}/platform/v1/sandbox-templates`;
+  } else {
+    if (!context.organizationId || !context.projectId) {
+      throw new Error(
+        "Personal Access Token authentication requires TENSORLAKE_ORGANIZATION_ID and TENSORLAKE_PROJECT_ID to be set (e.g. via 'tl login && tl init'). To skip this requirement, authenticate with TENSORLAKE_API_KEY instead \u2014 API keys are bound to a single project at creation."
+      );
+    }
+    url = `${baseUrl}/platform/v1/organizations/${encodeURIComponent(context.organizationId)}/projects/${encodeURIComponent(context.projectId)}/sandbox-templates`;
     headers["X-Forwarded-Organization-Id"] = context.organizationId;
     headers["X-Forwarded-Project-Id"] = context.projectId;
   }
@@ -5113,6 +5128,7 @@ async function registerImage(context, name, dockerfile, snapshotId, snapshotSand
       snapshotId,
       snapshotSandboxId,
       snapshotUri,
+      ...snapshotFormatVersion ? { snapshotFormatVersion } : {},
       snapshotSizeBytes,
       rootfsDiskBytes,
       public: isPublic
@@ -5155,7 +5171,8 @@ async function createSandboxImage(source, options = {}, deps = {}) {
     await executeDockerfilePlan(sandbox, plan, emit, sleep3);
     emit({ type: "status", message: "Creating snapshot..." });
     const snapshot = await client.snapshotAndWait(sandbox.sandboxId, {
-      snapshotType: "filesystem"
+      snapshotType: "filesystem",
+      waitUntil: "completed"
     });
     emit({
       type: "snapshot_created",
@@ -5189,7 +5206,8 @@ async function createSandboxImage(source, options = {}, deps = {}) {
       snapshot.snapshotUri,
       snapshot.sizeBytes,
       snapshot.rootfsDiskBytes,
-      options.isPublic ?? false
+      options.isPublic ?? false,
+      snapshot.snapshotFormatVersion
     );
     emit({
       type: "image_registered",