tensorlake 0.5.5 → 0.5.7

package/dist/{sandbox-image-BmhjX2bC.d.cts → sandbox-image-BMDaNpZ2.d.cts}

@@ -443,4 +443,4 @@ declare function loadImagePlan(image: Image, options?: Pick<CreateSandboxImageOp
 declare function createSandboxImage(source: SandboxImageSource, options?: CreateSandboxImageOptions, deps?: CreateSandboxImageDeps): Promise<Record<string, unknown>>;
 declare function runCreateSandboxImageCli(argv?: string[]): Promise<void>;
 
-export { defaultRegisteredName as $, type CreateSandboxImageOptions as A, type DirectoryEntry as B, type CreatePtySessionOptions as C, type DaemonInfo as D, type DockerfileBuildPlan as E, type DockerfileInstruction as F, type ImageBuildOperation as G, type HealthResponse as H, Image as I, ImageBuildOperationType as J, type ImageOptions as K, type ListDirectoryResponse as L, OutputMode as M, type NetworkConfig as N, type OutputResponse as O, type ProcessInfo as P, type PoolContainerInfo as Q, type RunOptions as R, type SandboxOptions as S, ProcessStatus as T, type UpdateSandboxOptions as U, type SandboxImageSource as V, SnapshotStatus as W, type SnapshotType as X, StdinMode as Y, createSandboxImage as Z, dockerfileContent as _, type CreateAndConnectOptions as a, loadDockerfilePlan as a0, loadImagePlan as a1, logicalDockerfileLines as a2, runCreateSandboxImageCli as a3, type SandboxClientOptions as b, type ConnectOptions as c, type SnapshotInfo as d, SandboxStatus as e, type SandboxInfo as f, type SuspendResumeOptions as g, type CheckpointOptions as h, type CommandResult as i, type StartProcessOptions as j, type SendSignalResponse as k, type OutputEvent as l, type PtySessionInfo as m, type CreateSandboxOptions as n, type CreateSandboxResponse as o, type SandboxPortAccess as p, type SnapshotOptions as q, type CreateSnapshotResponse as r, type SnapshotAndWaitOptions as s, type CreatePoolOptions as t, type CreateSandboxPoolResponse as u, type SandboxPoolInfo as v, type UpdatePoolOptions as w, type CheckpointType as x, type ContainerResourcesInfo as y, ContainerState as z };
+export { defaultRegisteredName as $, type CreateSandboxImageOptions as A, type DirectoryEntry as B, type CreatePtySessionOptions as C, type DaemonInfo as D, type DockerfileBuildPlan as E, type DockerfileInstruction as F, type ImageBuildOperation as G, type HealthResponse as H, Image as I, ImageBuildOperationType as J, type ImageOptions as K, type ListDirectoryResponse as L, OutputMode as M, type NetworkConfig as N, type OutputResponse as O, type ProcessInfo as P, type PoolContainerInfo as Q, type RunOptions as R, type SandboxOptions as S, ProcessStatus as T, type UpdateSandboxOptions as U, type SandboxImageSource as V, SnapshotStatus as W, type SnapshotType as X, StdinMode as Y, createSandboxImage as Z, dockerfileContent as _, type CreateAndConnectOptions as a, loadDockerfilePlan as a0, loadImagePlan as a1, logicalDockerfileLines as a2, runCreateSandboxImageCli as a3, type SandboxClientOptions as b, type ConnectOptions as c, type SnapshotInfo as d, type SandboxInfo as e, SandboxStatus as f, type SuspendResumeOptions as g, type CheckpointOptions as h, type CommandResult as i, type StartProcessOptions as j, type SendSignalResponse as k, type OutputEvent as l, type PtySessionInfo as m, type CreateSandboxOptions as n, type CreateSandboxResponse as o, type SandboxPortAccess as p, type SnapshotOptions as q, type CreateSnapshotResponse as r, type SnapshotAndWaitOptions as s, type CreatePoolOptions as t, type CreateSandboxPoolResponse as u, type SandboxPoolInfo as v, type UpdatePoolOptions as w, type CheckpointType as x, type ContainerResourcesInfo as y, ContainerState as z };

package/dist/{sandbox-image-BmhjX2bC.d.ts → sandbox-image-BMDaNpZ2.d.ts}

@@ -443,4 +443,4 @@ declare function loadImagePlan(image: Image, options?: Pick<CreateSandboxImageOp
 declare function createSandboxImage(source: SandboxImageSource, options?: CreateSandboxImageOptions, deps?: CreateSandboxImageDeps): Promise<Record<string, unknown>>;
 declare function runCreateSandboxImageCli(argv?: string[]): Promise<void>;
 
-export { defaultRegisteredName as $, type CreateSandboxImageOptions as A, type DirectoryEntry as B, type CreatePtySessionOptions as C, type DaemonInfo as D, type DockerfileBuildPlan as E, type DockerfileInstruction as F, type ImageBuildOperation as G, type HealthResponse as H, Image as I, ImageBuildOperationType as J, type ImageOptions as K, type ListDirectoryResponse as L, OutputMode as M, type NetworkConfig as N, type OutputResponse as O, type ProcessInfo as P, type PoolContainerInfo as Q, type RunOptions as R, type SandboxOptions as S, ProcessStatus as T, type UpdateSandboxOptions as U, type SandboxImageSource as V, SnapshotStatus as W, type SnapshotType as X, StdinMode as Y, createSandboxImage as Z, dockerfileContent as _, type CreateAndConnectOptions as a, loadDockerfilePlan as a0, loadImagePlan as a1, logicalDockerfileLines as a2, runCreateSandboxImageCli as a3, type SandboxClientOptions as b, type ConnectOptions as c, type SnapshotInfo as d, SandboxStatus as e, type SandboxInfo as f, type SuspendResumeOptions as g, type CheckpointOptions as h, type CommandResult as i, type StartProcessOptions as j, type SendSignalResponse as k, type OutputEvent as l, type PtySessionInfo as m, type CreateSandboxOptions as n, type CreateSandboxResponse as o, type SandboxPortAccess as p, type SnapshotOptions as q, type CreateSnapshotResponse as r, type SnapshotAndWaitOptions as s, type CreatePoolOptions as t, type CreateSandboxPoolResponse as u, type SandboxPoolInfo as v, type UpdatePoolOptions as w, type CheckpointType as x, type ContainerResourcesInfo as y, ContainerState as z };
+export { defaultRegisteredName as $, type CreateSandboxImageOptions as A, type DirectoryEntry as B, type CreatePtySessionOptions as C, type DaemonInfo as D, type DockerfileBuildPlan as E, type DockerfileInstruction as F, type ImageBuildOperation as G, type HealthResponse as H, Image as I, ImageBuildOperationType as J, type ImageOptions as K, type ListDirectoryResponse as L, OutputMode as M, type NetworkConfig as N, type OutputResponse as O, type ProcessInfo as P, type PoolContainerInfo as Q, type RunOptions as R, type SandboxOptions as S, ProcessStatus as T, type UpdateSandboxOptions as U, type SandboxImageSource as V, SnapshotStatus as W, type SnapshotType as X, StdinMode as Y, createSandboxImage as Z, dockerfileContent as _, type CreateAndConnectOptions as a, loadDockerfilePlan as a0, loadImagePlan as a1, logicalDockerfileLines as a2, runCreateSandboxImageCli as a3, type SandboxClientOptions as b, type ConnectOptions as c, type SnapshotInfo as d, type SandboxInfo as e, SandboxStatus as f, type SuspendResumeOptions as g, type CheckpointOptions as h, type CommandResult as i, type StartProcessOptions as j, type SendSignalResponse as k, type OutputEvent as l, type PtySessionInfo as m, type CreateSandboxOptions as n, type CreateSandboxResponse as o, type SandboxPortAccess as p, type SnapshotOptions as q, type CreateSnapshotResponse as r, type SnapshotAndWaitOptions as s, type CreatePoolOptions as t, type CreateSandboxPoolResponse as u, type SandboxPoolInfo as v, type UpdatePoolOptions as w, type CheckpointType as x, type ContainerResourcesInfo as y, ContainerState as z };

package/dist/sandbox-image.cjs

@@ -83,7 +83,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.4.49";
+    SDK_VERSION = "0.5.7";
     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";
@@ -260,6 +260,9 @@ var init_http = __esm({
         if (options.hostHeader) {
           this.headers["Host"] = options.hostHeader;
         }
+        if (options.sandboxIdHeader) {
+          this.headers["X-Tensorlake-Sandbox-Id"] = options.sandboxIdHeader;
+        }
         if (options.routingHint) {
           this.headers["X-Tensorlake-Route-Hint"] = options.routingHint;
         }
@@ -2886,21 +2889,36 @@ function resolveProxyTarget(proxyUrl, sandboxId) {
     if (host === "localhost" || host === "127.0.0.1") {
       return {
         baseUrl: trimTrailingSlashes(proxyUrl),
-        hostHeader: `${sandboxId}.local`
+        hostHeader: `${sandboxId}.local`,
+        sandboxIdHeader: void 0
       };
     }
     const port = parsed.port ? `:${parsed.port}` : "";
     return {
-      baseUrl: `${parsed.protocol}//${sandboxId}.${host}${port}`,
-      hostHeader: void 0
+      baseUrl: `${parsed.protocol}//${host}${port}`,
+      hostHeader: void 0,
+      sandboxIdHeader: sandboxId
     };
   } catch {
     return {
       baseUrl: `${trimTrailingSlashes(proxyUrl)}/${sandboxId}`,
-      hostHeader: void 0
+      hostHeader: void 0,
+      sandboxIdHeader: void 0
     };
   }
 }
+function resolveSandboxLifecycleUrl(apiUrl) {
+  if (isLocalhost(apiUrl)) return apiUrl;
+  try {
+    const parsed = new URL(apiUrl);
+    if (parsed.hostname.startsWith("api.")) {
+      parsed.hostname = "sandbox." + parsed.hostname.slice(4);
+      return parsed.toString().replace(/\/$/, "");
+    }
+  } catch {
+  }
+  return apiUrl;
+}
 function lifecyclePath(path2, isLocal, namespace) {
   if (isLocal) {
     return `/v1/namespaces/${namespace}/${path2}`;
@@ -3127,7 +3145,7 @@ var init_sandbox = __esm({
         this.sandboxId = options.sandboxId;
         this.lifecycleIdentifier = options.sandboxId;
         const proxyUrl = options.proxyUrl ?? SANDBOX_PROXY_URL;
-        const { baseUrl, hostHeader } = resolveProxyTarget(proxyUrl, options.sandboxId);
+        const { baseUrl, hostHeader, sandboxIdHeader } = resolveProxyTarget(proxyUrl, options.sandboxId);
         this.baseUrl = baseUrl;
         this.wsHeaders = {};
         if (options.apiKey) {
@@ -3142,12 +3160,16 @@ var init_sandbox = __esm({
         if (hostHeader) {
           this.wsHeaders.Host = hostHeader;
         }
+        if (sandboxIdHeader) {
+          this.wsHeaders["X-Tensorlake-Sandbox-Id"] = sandboxIdHeader;
+        }
         this.http = new HttpClient({
           baseUrl,
           apiKey: options.apiKey,
           organizationId: options.organizationId,
           projectId: options.projectId,
           hostHeader,
+          sandboxIdHeader,
           routingHint: options.routingHint
         });
       }
@@ -3188,8 +3210,8 @@ var init_sandbox = __esm({
       /**
        * Attach to an existing sandbox and return a connected handle.
        *
-       * Verifies the sandbox exists via a server GET call, then returns a handle
-       * in whatever state the sandbox is in. Does **not** auto-resume a suspended
+       * Returns immediately without contacting the server. Call `sandbox.info()`
+       * to fetch the current state on demand. Does **not** auto-resume a suspended
        * sandbox — call `sandbox.resume()` explicitly.
        */
       static async connect(options) {
@@ -3199,15 +3221,12 @@ var init_sandbox = __esm({
           /* _internal */
           true
         );
-        const info = await client.get(options.sandboxId);
         const sandbox = client.connect(
-          info.sandboxId,
+          options.sandboxId,
           options.proxyUrl,
-          options.routingHint ?? info.routingHint
+          options.routingHint
         );
         sandbox.lifecycleClient = client;
-        sandbox._setLifecycleIdentifier(info.sandboxId);
-        sandbox._setName(info.name ?? null);
         return sandbox;
       }
       // --- Static snapshot management ---
@@ -3231,6 +3250,26 @@ var init_sandbox = __esm({
         );
         await client.deleteSnapshot(snapshotId);
       }
+      /** List all sandboxes. No sandbox handle needed. */
+      static async list(options) {
+        const { SandboxClient: SandboxClient2 } = await Promise.resolve().then(() => (init_client(), client_exports));
+        const client = new SandboxClient2(
+          options,
+          /* _internal */
+          true
+        );
+        return client.list();
+      }
+      /** List all snapshots in the project. No sandbox handle needed. */
+      static async listSnapshots(options) {
+        const { SandboxClient: SandboxClient2 } = await Promise.resolve().then(() => (init_client(), client_exports));
+        const client = new SandboxClient2(
+          options,
+          /* _internal */
+          true
+        );
+        return client.listSnapshots();
+      }
       // --- Instance lifecycle methods ---
       requireLifecycleClient(operation) {
         if (!this.lifecycleClient) {
@@ -3240,6 +3279,14 @@ var init_sandbox = __esm({
         }
         return this.lifecycleClient;
       }
+      /** Fetch the current sandbox information from the server on demand. */
+      async info() {
+        const client = this.requireLifecycleClient("info");
+        const info = await client.get(this.lifecycleIdentifier);
+        this._setLifecycleIdentifier(info.sandboxId);
+        this._setName(info.name ?? null);
+        return info;
+      }
       /**
        * Fetch the current sandbox status from the server.
        *
@@ -3347,6 +3394,7 @@ var init_sandbox = __esm({
           "/api/v1/processes/run",
           { json: body }
         );
+        const traceId = sseStream.traceId;
         const stdoutLines = [];
         const stderrLines = [];
         let exitCode = -1;
@@ -3365,11 +3413,10 @@ var init_sandbox = __esm({
             }
           }
         }
-        return {
-          exitCode,
-          stdout: stdoutLines.join("\n"),
-          stderr: stderrLines.join("\n")
-        };
+        return Object.assign(
+          { exitCode, stdout: stdoutLines.join("\n"), stderr: stderrLines.join("\n") },
+          { traceId }
+        );
       }
       // --- Process management ---
       /**
@@ -3399,7 +3446,7 @@ var init_sandbox = __esm({
           "/api/v1/processes",
           { body: payload }
         );
-        return fromSnakeKeys(raw);
+        return Object.assign(fromSnakeKeys(raw), { traceId: raw.traceId });
       }
       /** List all processes (running and exited) tracked by the sandbox daemon. */
       async listProcesses() {
@@ -3416,7 +3463,7 @@ var init_sandbox = __esm({
           "GET",
           `/api/v1/processes/${pid}`
         );
-        return fromSnakeKeys(raw);
+        return Object.assign(fromSnakeKeys(raw), { traceId: raw.traceId });
       }
       /** Send SIGKILL to a process. */
       async killProcess(pid) {
@@ -3429,7 +3476,7 @@ var init_sandbox = __esm({
           `/api/v1/processes/${pid}/signal`,
           { body: { signal } }
         );
-        return fromSnakeKeys(raw);
+        return Object.assign(fromSnakeKeys(raw), { traceId: raw.traceId });
       }
       // --- Process I/O ---
       /** Write bytes to a process's stdin. The process must have been started with `stdinMode: StdinMode.PIPE`. */
@@ -3449,7 +3496,7 @@ var init_sandbox = __esm({
           "GET",
           `/api/v1/processes/${pid}/stdout`
         );
-        return fromSnakeKeys(raw);
+        return Object.assign(fromSnakeKeys(raw), { traceId: raw.traceId });
       }
       /** Return all captured stderr lines produced so far by a process. */
       async getStderr(pid) {
@@ -3457,7 +3504,7 @@ var init_sandbox = __esm({
           "GET",
           `/api/v1/processes/${pid}/stderr`
         );
-        return fromSnakeKeys(raw);
+        return Object.assign(fromSnakeKeys(raw), { traceId: raw.traceId });
       }
       /** Return all captured stdout+stderr lines produced so far by a process. */
       async getOutput(pid) {
@@ -3465,7 +3512,7 @@ var init_sandbox = __esm({
           "GET",
           `/api/v1/processes/${pid}/output`
         );
-        return fromSnakeKeys(raw);
+        return Object.assign(fromSnakeKeys(raw), { traceId: raw.traceId });
       }
       // --- Streaming (SSE) ---
       /** Stream stdout events from a process until it exits. Yields one `OutputEvent` per line. */
@@ -3539,7 +3586,7 @@ var init_sandbox = __esm({
           "GET",
           `/api/v1/files/list?path=${encodeURIComponent(path2)}`
         );
-        return fromSnakeKeys(raw);
+        return Object.assign(fromSnakeKeys(raw), { traceId: raw.traceId });
       }
       // --- PTY ---
       /** Create an interactive PTY session. Returns a `sessionId` and `token` for WebSocket connection via `connectPty()`. */
@@ -3557,7 +3604,7 @@ var init_sandbox = __esm({
           "/api/v1/pty",
           { body: payload }
         );
-        return fromSnakeKeys(raw);
+        return Object.assign(fromSnakeKeys(raw), { traceId: raw.traceId });
       }
       /** Create a PTY session and connect to it immediately. Cleans up the session if the WebSocket connection fails. */
       async createPty(options) {
@@ -3611,15 +3658,53 @@ var init_sandbox = __esm({
       }
       /** Connect to a sandbox VNC session for programmatic desktop control. */
       async connectDesktop(options) {
+        const port = options?.port ?? 5901;
+        const connectTimeout = options?.connectTimeout ?? 10;
+        const startMs = Date.now();
+        const deadlineMs = startMs + connectTimeout * 1e3;
+        await this.waitForPortReady(port, deadlineMs);
+        const remainingSecs = Math.max(0.1, (deadlineMs - Date.now()) / 1e3);
         return Desktop.connect({
           baseUrl: this.baseUrl,
           wsHeaders: this.wsHeaders,
-          port: options?.port,
+          port,
           password: options?.password,
           shared: options?.shared,
-          connectTimeout: options?.connectTimeout
+          connectTimeout: remainingSecs
         });
       }
+      /**
+       * Poll the in-sandbox daemon until `127.0.0.1:port` accepts a TCP connection.
+       * Uses `bash`'s `/dev/tcp` builtin via `processes/run` — no extra deps in
+       * the sandbox image. `bash` is present on every image we ship.
+       */
+      async waitForPortReady(port, deadlineMs) {
+        const probeIntervalMs = 250;
+        const probeProcessTimeoutSecs = 2;
+        let lastError;
+        while (Date.now() < deadlineMs) {
+          try {
+            const result = await this.run("/bin/bash", {
+              args: ["-c", `exec 3<>/dev/tcp/127.0.0.1/${port}`],
+              timeout: probeProcessTimeoutSecs
+            });
+            if (result.exitCode === 0) {
+              return;
+            }
+          } catch (error) {
+            lastError = error;
+          }
+          const remainingMs = deadlineMs - Date.now();
+          if (remainingMs <= 0) break;
+          await new Promise(
+            (resolve) => setTimeout(resolve, Math.min(probeIntervalMs, remainingMs))
+          );
+        }
+        const detail = lastError instanceof Error ? `: ${lastError.message}` : "";
+        throw new SandboxError(
+          `port ${port} did not become reachable inside sandbox within the connect timeout${detail}`
+        );
+      }
       ptyWsUrl(sessionId, token) {
         let wsBase;
         if (this.baseUrl.startsWith("https://")) {
@@ -3638,15 +3723,15 @@ var init_sandbox = __esm({
           "GET",
           "/api/v1/health"
         );
-        return fromSnakeKeys(raw);
+        return Object.assign(fromSnakeKeys(raw), { traceId: raw.traceId });
       }
       /** Get sandbox daemon info (version, uptime, process counts). */
-      async info() {
+      async daemonInfo() {
         const raw = await this.http.requestJson(
           "GET",
           "/api/v1/info"
         );
-        return fromSnakeKeys(raw);
+        return Object.assign(fromSnakeKeys(raw), { traceId: raw.traceId });
       }
     };
   }
@@ -3739,7 +3824,7 @@ var init_client = __esm({
         this.namespace = options?.namespace ?? NAMESPACE;
         this.local = isLocalhost(this.apiUrl);
         this.http = new HttpClient({
-          baseUrl: this.apiUrl,
+          baseUrl: resolveSandboxLifecycleUrl(this.apiUrl),
           apiKey: this.apiKey,
           organizationId: this.organizationId,
           projectId: this.projectId,
@@ -3807,7 +3892,7 @@ var init_client = __esm({
           "GET",
           this.path(`sandboxes/${sandboxId}`)
         );
-        return fromSnakeKeys(raw, "sandboxId");
+        return Object.assign(fromSnakeKeys(raw, "sandboxId"), { traceId: raw.traceId });
       }
       /** List all sandboxes in the namespace. */
       async list() {
@@ -3838,7 +3923,7 @@ var init_client = __esm({
           this.path(`sandboxes/${sandboxId}`),
           { body }
         );
-        return fromSnakeKeys(raw, "sandboxId");
+        return Object.assign(fromSnakeKeys(raw, "sandboxId"), { traceId: raw.traceId });
       }
       /** Get the current proxy port settings for a sandbox. */
       async getPortAccess(sandboxId) {
@@ -3980,7 +4065,7 @@ var init_client = __esm({
           "GET",
           this.path(`snapshots/${snapshotId}`)
         );
-        return fromSnakeKeys(raw, "snapshotId");
+        return Object.assign(fromSnakeKeys(raw, "snapshotId"), { traceId: raw.traceId });
       }
       /** List all snapshots in the namespace. */
       async listSnapshots() {