tensorlake 0.5.9 → 0.5.10

package/dist/index.d.cts

@@ -1,5 +1,5 @@
-import { C as CreatePtySessionOptions, S as SandboxOptions, a as CreateAndConnectOptions, b as SandboxClientOptions, c as ConnectOptions, d as SnapshotInfo, e as SandboxInfo, f as SandboxStatus, U as UpdateSandboxOptions, g as SuspendResumeOptions, h as CheckpointOptions, R as RunOptions, i as CommandResult, j as StartProcessOptions, P as ProcessInfo, k as SendSignalResponse, O as OutputResponse, l as OutputEvent, L as ListDirectoryResponse, m as PtySessionInfo, H as HealthResponse, D as DaemonInfo, n as CreateSandboxOptions, o as CreateSandboxResponse, p as SandboxPortAccess, q as SnapshotOptions, r as CreateSnapshotResponse, s as SnapshotAndWaitOptions, t as CreatePoolOptions, u as CreateSandboxPoolResponse, v as SandboxPoolInfo, w as UpdatePoolOptions } from './sandbox-image-BMDaNpZ2.cjs';
-export { x as CheckpointType, y as ContainerResourcesInfo, z as ContainerState, A as CreateSandboxImageOptions, B as DirectoryEntry, E as DockerfileBuildPlan, F as DockerfileInstruction, I as Image, G as ImageBuildOperation, J as ImageBuildOperationType, J as ImageBuildOperationTypeValue, K as ImageOptions, N as NetworkConfig, M as OutputMode, Q as PoolContainerInfo, T as ProcessStatus, V as SandboxImageSource, W as SnapshotStatus, X as SnapshotType, Y as StdinMode, Z as createSandboxImage, _ as dockerfileContent } from './sandbox-image-BMDaNpZ2.cjs';
+import { C as CreatePtySessionOptions, S as SandboxOptions, a as CreateAndConnectOptions, b as SandboxClientOptions, c as ConnectOptions, d as SnapshotInfo, e as SandboxInfo, f as SandboxStatus, U as UpdateSandboxOptions, g as SuspendResumeOptions, h as CheckpointOptions, R as RunOptions, i as CommandResult, j as StartProcessOptions, P as ProcessInfo, k as SendSignalResponse, O as OutputResponse, l as OutputEvent, L as ListDirectoryResponse, m as PtySessionInfo, H as HealthResponse, D as DaemonInfo, n as CreateSandboxOptions, o as CreateSandboxResponse, p as SandboxPortAccess, q as SnapshotOptions, r as CreateSnapshotResponse, s as SnapshotAndWaitOptions, t as CreatePoolOptions, u as CreateSandboxPoolResponse, v as SandboxPoolInfo, w as UpdatePoolOptions } from './sandbox-image-CUEKMhZ1.cjs';
+export { x as CheckpointType, y as ContainerResourcesInfo, z as ContainerState, A as CreateSandboxImageOptions, B as DirectoryEntry, E as DockerfileBuildPlan, F as DockerfileInstruction, I as Image, G as ImageBuildOperation, J as ImageBuildOperationType, J as ImageBuildOperationTypeValue, K as ImageOptions, N as NetworkConfig, M as OutputMode, Q as PoolContainerInfo, T as ProcessStatus, V as SandboxImageSource, W as SnapshotStatus, X as SnapshotType, Y as SnapshotWaitCondition, Z as StdinMode, _ as createSandboxImage, $ as dockerfileContent } from './sandbox-image-CUEKMhZ1.cjs';
 
 /**
  * The return value of every SDK operation. Carries the W3C `trace_id` so callers
@@ -236,11 +236,10 @@ declare class Sandbox {
      */
     resume(options?: SuspendResumeOptions): Promise<void>;
     /**
-     * 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`).
      */
     checkpoint(options?: CheckpointOptions): Promise<Traced<SnapshotInfo> | undefined>;
@@ -413,7 +412,8 @@ declare class SandboxClient {
      *
      * 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`).
@@ -426,9 +426,11 @@ declare class SandboxClient {
     /** Delete a snapshot by ID. */
     deleteSnapshot(snapshotId: string): Promise<void>;
     /**
-     * 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.
      *

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { C as CreatePtySessionOptions, S as SandboxOptions, a as CreateAndConnectOptions, b as SandboxClientOptions, c as ConnectOptions, d as SnapshotInfo, e as SandboxInfo, f as SandboxStatus, U as UpdateSandboxOptions, g as SuspendResumeOptions, h as CheckpointOptions, R as RunOptions, i as CommandResult, j as StartProcessOptions, P as ProcessInfo, k as SendSignalResponse, O as OutputResponse, l as OutputEvent, L as ListDirectoryResponse, m as PtySessionInfo, H as HealthResponse, D as DaemonInfo, n as CreateSandboxOptions, o as CreateSandboxResponse, p as SandboxPortAccess, q as SnapshotOptions, r as CreateSnapshotResponse, s as SnapshotAndWaitOptions, t as CreatePoolOptions, u as CreateSandboxPoolResponse, v as SandboxPoolInfo, w as UpdatePoolOptions } from './sandbox-image-BMDaNpZ2.js';
-export { x as CheckpointType, y as ContainerResourcesInfo, z as ContainerState, A as CreateSandboxImageOptions, B as DirectoryEntry, E as DockerfileBuildPlan, F as DockerfileInstruction, I as Image, G as ImageBuildOperation, J as ImageBuildOperationType, J as ImageBuildOperationTypeValue, K as ImageOptions, N as NetworkConfig, M as OutputMode, Q as PoolContainerInfo, T as ProcessStatus, V as SandboxImageSource, W as SnapshotStatus, X as SnapshotType, Y as StdinMode, Z as createSandboxImage, _ as dockerfileContent } from './sandbox-image-BMDaNpZ2.js';
+import { C as CreatePtySessionOptions, S as SandboxOptions, a as CreateAndConnectOptions, b as SandboxClientOptions, c as ConnectOptions, d as SnapshotInfo, e as SandboxInfo, f as SandboxStatus, U as UpdateSandboxOptions, g as SuspendResumeOptions, h as CheckpointOptions, R as RunOptions, i as CommandResult, j as StartProcessOptions, P as ProcessInfo, k as SendSignalResponse, O as OutputResponse, l as OutputEvent, L as ListDirectoryResponse, m as PtySessionInfo, H as HealthResponse, D as DaemonInfo, n as CreateSandboxOptions, o as CreateSandboxResponse, p as SandboxPortAccess, q as SnapshotOptions, r as CreateSnapshotResponse, s as SnapshotAndWaitOptions, t as CreatePoolOptions, u as CreateSandboxPoolResponse, v as SandboxPoolInfo, w as UpdatePoolOptions } from './sandbox-image-CUEKMhZ1.js';
+export { x as CheckpointType, y as ContainerResourcesInfo, z as ContainerState, A as CreateSandboxImageOptions, B as DirectoryEntry, E as DockerfileBuildPlan, F as DockerfileInstruction, I as Image, G as ImageBuildOperation, J as ImageBuildOperationType, J as ImageBuildOperationTypeValue, K as ImageOptions, N as NetworkConfig, M as OutputMode, Q as PoolContainerInfo, T as ProcessStatus, V as SandboxImageSource, W as SnapshotStatus, X as SnapshotType, Y as SnapshotWaitCondition, Z as StdinMode, _ as createSandboxImage, $ as dockerfileContent } from './sandbox-image-CUEKMhZ1.js';
 
 /**
  * The return value of every SDK operation. Carries the W3C `trace_id` so callers
@@ -236,11 +236,10 @@ declare class Sandbox {
      */
     resume(options?: SuspendResumeOptions): Promise<void>;
     /**
-     * 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`).
      */
     checkpoint(options?: CheckpointOptions): Promise<Traced<SnapshotInfo> | undefined>;
@@ -413,7 +412,8 @@ declare class SandboxClient {
      *
      * 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`).
@@ -426,9 +426,11 @@ declare class SandboxClient {
     /** Delete a snapshot by ID. */
     deleteSnapshot(snapshotId: string): Promise<void>;
     /**
-     * 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.
      *

package/dist/index.js

@@ -13,7 +13,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";
@@ -408,11 +408,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";
@@ -3377,11 +3378,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) {
@@ -3393,7 +3393,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
         });
       }
       /**
@@ -3788,6 +3789,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);
@@ -4088,7 +4095,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`).
@@ -4129,9 +4137,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.
        *
@@ -4144,13 +4154,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}`
@@ -4159,7 +4170,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 ---
@@ -4444,6 +4455,7 @@ __export(sandbox_image_exports, {
   loadDockerfilePlan: () => loadDockerfilePlan,
   loadImagePlan: () => loadImagePlan,
   logicalDockerfileLines: () => logicalDockerfileLines,
+  registerImage: () => registerImage,
   runCreateSandboxImageCli: () => runCreateSandboxImageCli
 });
 import { readFile, readdir, stat } from "fs/promises";
@@ -5069,23 +5081,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;
   }
@@ -5098,6 +5113,7 @@ async function registerImage(context, name, dockerfile, snapshotId, snapshotSand
       snapshotId,
       snapshotSandboxId,
       snapshotUri,
+      ...snapshotFormatVersion ? { snapshotFormatVersion } : {},
       snapshotSizeBytes,
       rootfsDiskBytes,
       public: isPublic
@@ -5140,7 +5156,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",
@@ -5174,7 +5191,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",