tensorlake 0.4.45 → 0.4.47

package/dist/index.d.cts

@@ -1,5 +1,5 @@
-import { C as CreatePtySessionOptions, S as SandboxOptions, R as RunOptions, a as CommandResult, b as StartProcessOptions, P as ProcessInfo, c as SendSignalResponse, O as OutputResponse, d as OutputEvent, L as ListDirectoryResponse, e as PtySessionInfo, H as HealthResponse, D as DaemonInfo, f as SandboxClientOptions, g as CreateSandboxOptions, h as CreateSandboxResponse, i as SandboxInfo, U as UpdateSandboxOptions, j as SandboxPortAccess, k as SnapshotOptions, l as CreateSnapshotResponse, m as SnapshotInfo, n as SnapshotAndWaitOptions, o as CreatePoolOptions, p as CreateSandboxPoolResponse, q as SandboxPoolInfo, r as UpdatePoolOptions, s as CreateAndConnectOptions } from './sandbox-image-B5Fa0xqb.cjs';
-export { t as ContainerResourcesInfo, u as ContainerState, v as CreateSandboxImageOptions, w as DirectoryEntry, x as DockerfileBuildPlan, y as DockerfileInstruction, I as Image, z as ImageBuildOperation, A as ImageBuildOperationType, A as ImageBuildOperationTypeValue, B as ImageOptions, N as NetworkConfig, E as OutputMode, F as PoolContainerInfo, G as ProcessStatus, J as SandboxImageSource, K as SandboxStatus, M as SnapshotContentMode, Q as SnapshotStatus, T as StdinMode, V as createSandboxImage, W as dockerfileContent } from './sandbox-image-B5Fa0xqb.cjs';
+import { C as CreatePtySessionOptions, S as SandboxOptions, R as RunOptions, a as CommandResult, b as StartProcessOptions, P as ProcessInfo, c as SendSignalResponse, O as OutputResponse, d as OutputEvent, L as ListDirectoryResponse, e as PtySessionInfo, H as HealthResponse, D as DaemonInfo, f as SandboxClientOptions, g as CreateSandboxOptions, h as CreateSandboxResponse, i as SandboxInfo, U as UpdateSandboxOptions, j as SandboxPortAccess, k as SnapshotOptions, l as CreateSnapshotResponse, m as SnapshotInfo, n as SnapshotAndWaitOptions, o as CreatePoolOptions, p as CreateSandboxPoolResponse, q as SandboxPoolInfo, r as UpdatePoolOptions, s as CreateAndConnectOptions } from './sandbox-image-B130uMYt.cjs';
+export { t as ContainerResourcesInfo, u as ContainerState, v as CreateSandboxImageOptions, w as DirectoryEntry, x as DockerfileBuildPlan, y as DockerfileInstruction, I as Image, z as ImageBuildOperation, A as ImageBuildOperationType, A as ImageBuildOperationTypeValue, B as ImageOptions, N as NetworkConfig, E as OutputMode, F as PoolContainerInfo, G as ProcessStatus, J as SandboxImageSource, K as SandboxStatus, M as SnapshotContentMode, Q as SnapshotStatus, T as StdinMode, V as createSandboxImage, W as dockerfileContent } from './sandbox-image-B130uMYt.cjs';
 
 type MouseButton = "left" | "middle" | "right";
 interface ConnectDesktopOptions {
@@ -157,6 +157,12 @@ declare class Sandbox {
     _setOwner(client: SandboxClient): void;
     close(): void;
     terminate(): Promise<void>;
+    /**
+     * Run a command to completion and return its output.
+     *
+     * Uses a single streaming `POST /api/v1/processes/run` request that starts
+     * the process, streams output, and delivers the exit code over one connection.
+     */
     run(command: string, options?: RunOptions): Promise<CommandResult>;
     startProcess(command: string, options?: StartProcessOptions): Promise<ProcessInfo>;
     listProcesses(): Promise<ProcessInfo[]>;
@@ -243,7 +249,7 @@ declare class SandboxClient {
     listPools(): Promise<SandboxPoolInfo[]>;
     updatePool(poolId: string, options: UpdatePoolOptions): Promise<SandboxPoolInfo>;
     deletePool(poolId: string): Promise<void>;
-    connect(identifier: string, proxyUrl?: string): Sandbox;
+    connect(identifier: string, proxyUrl?: string, routingHint?: string): Sandbox;
     createAndConnect(options?: CreateAndConnectOptions): Promise<Sandbox>;
 }
 

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { C as CreatePtySessionOptions, S as SandboxOptions, R as RunOptions, a as CommandResult, b as StartProcessOptions, P as ProcessInfo, c as SendSignalResponse, O as OutputResponse, d as OutputEvent, L as ListDirectoryResponse, e as PtySessionInfo, H as HealthResponse, D as DaemonInfo, f as SandboxClientOptions, g as CreateSandboxOptions, h as CreateSandboxResponse, i as SandboxInfo, U as UpdateSandboxOptions, j as SandboxPortAccess, k as SnapshotOptions, l as CreateSnapshotResponse, m as SnapshotInfo, n as SnapshotAndWaitOptions, o as CreatePoolOptions, p as CreateSandboxPoolResponse, q as SandboxPoolInfo, r as UpdatePoolOptions, s as CreateAndConnectOptions } from './sandbox-image-B5Fa0xqb.js';
-export { t as ContainerResourcesInfo, u as ContainerState, v as CreateSandboxImageOptions, w as DirectoryEntry, x as DockerfileBuildPlan, y as DockerfileInstruction, I as Image, z as ImageBuildOperation, A as ImageBuildOperationType, A as ImageBuildOperationTypeValue, B as ImageOptions, N as NetworkConfig, E as OutputMode, F as PoolContainerInfo, G as ProcessStatus, J as SandboxImageSource, K as SandboxStatus, M as SnapshotContentMode, Q as SnapshotStatus, T as StdinMode, V as createSandboxImage, W as dockerfileContent } from './sandbox-image-B5Fa0xqb.js';
+import { C as CreatePtySessionOptions, S as SandboxOptions, R as RunOptions, a as CommandResult, b as StartProcessOptions, P as ProcessInfo, c as SendSignalResponse, O as OutputResponse, d as OutputEvent, L as ListDirectoryResponse, e as PtySessionInfo, H as HealthResponse, D as DaemonInfo, f as SandboxClientOptions, g as CreateSandboxOptions, h as CreateSandboxResponse, i as SandboxInfo, U as UpdateSandboxOptions, j as SandboxPortAccess, k as SnapshotOptions, l as CreateSnapshotResponse, m as SnapshotInfo, n as SnapshotAndWaitOptions, o as CreatePoolOptions, p as CreateSandboxPoolResponse, q as SandboxPoolInfo, r as UpdatePoolOptions, s as CreateAndConnectOptions } from './sandbox-image-B130uMYt.js';
+export { t as ContainerResourcesInfo, u as ContainerState, v as CreateSandboxImageOptions, w as DirectoryEntry, x as DockerfileBuildPlan, y as DockerfileInstruction, I as Image, z as ImageBuildOperation, A as ImageBuildOperationType, A as ImageBuildOperationTypeValue, B as ImageOptions, N as NetworkConfig, E as OutputMode, F as PoolContainerInfo, G as ProcessStatus, J as SandboxImageSource, K as SandboxStatus, M as SnapshotContentMode, Q as SnapshotStatus, T as StdinMode, V as createSandboxImage, W as dockerfileContent } from './sandbox-image-B130uMYt.js';
 
 type MouseButton = "left" | "middle" | "right";
 interface ConnectDesktopOptions {
@@ -157,6 +157,12 @@ declare class Sandbox {
     _setOwner(client: SandboxClient): void;
     close(): void;
     terminate(): Promise<void>;
+    /**
+     * Run a command to completion and return its output.
+     *
+     * Uses a single streaming `POST /api/v1/processes/run` request that starts
+     * the process, streams output, and delivers the exit code over one connection.
+     */
     run(command: string, options?: RunOptions): Promise<CommandResult>;
     startProcess(command: string, options?: StartProcessOptions): Promise<ProcessInfo>;
     listProcesses(): Promise<ProcessInfo[]>;
@@ -243,7 +249,7 @@ declare class SandboxClient {
     listPools(): Promise<SandboxPoolInfo[]>;
     updatePool(poolId: string, options: UpdatePoolOptions): Promise<SandboxPoolInfo>;
     deletePool(poolId: string): Promise<void>;
-    connect(identifier: string, proxyUrl?: string): Sandbox;
+    connect(identifier: string, proxyUrl?: string, routingHint?: string): Sandbox;
     createAndConnect(options?: CreateAndConnectOptions): Promise<Sandbox>;
 }
 

package/dist/index.js

@@ -114,6 +114,9 @@ var HttpClient = class {
     if (options.hostHeader) {
       this.headers["Host"] = options.hostHeader;
     }
+    if (options.routingHint) {
+      this.headers["X-Tensorlake-Route-Hint"] = options.routingHint;
+    }
   }
   close() {
     this.abortController?.abort();
@@ -148,12 +151,13 @@ var HttpClient = class {
     const buffer = await response.arrayBuffer();
     return new Uint8Array(buffer);
   }
-  /** Make a request and return the raw Response (for SSE streaming). */
+  /** Make a request and return the response body as an SSE stream. */
   async requestStream(method, path2, options) {
     const response = await this.requestResponse(
       method,
       path2,
       {
+        json: options?.json,
         headers: { Accept: "text/event-stream" },
         signal: options?.signal
       }
@@ -3086,7 +3090,8 @@ var Sandbox = class {
       apiKey: options.apiKey,
       organizationId: options.organizationId,
       projectId: options.projectId,
-      hostHeader
+      hostHeader,
+      routingHint: options.routingHint
     });
   }
   /** @internal Used by SandboxClient.createAndConnect to set ownership. */
@@ -3107,37 +3112,45 @@ var Sandbox = class {
     }
   }
   // --- High-level convenience ---
+  /**
+   * Run a command to completion and return its output.
+   *
+   * Uses a single streaming `POST /api/v1/processes/run` request that starts
+   * the process, streams output, and delivers the exit code over one connection.
+   */
   async run(command, options) {
-    const proc = await this.startProcess(command, {
-      args: options?.args,
-      env: options?.env,
-      workingDir: options?.workingDir
-    });
-    const deadline = options?.timeout ? Date.now() + options.timeout * 1e3 : null;
-    let info;
-    while (true) {
-      info = await this.getProcess(proc.pid);
-      if (info.status !== "running" /* RUNNING */) break;
-      if (deadline && Date.now() > deadline) {
-        await this.killProcess(proc.pid);
-        throw new SandboxError(`Command timed out after ${options.timeout}s`);
+    const body = { command };
+    if (options?.args) body.args = options.args;
+    if (options?.env) body.env = options.env;
+    if (options?.workingDir) body.working_dir = options.workingDir;
+    if (options?.timeout != null) body.timeout = options.timeout;
+    const sseStream = await this.http.requestStream(
+      "POST",
+      "/api/v1/processes/run",
+      { json: body }
+    );
+    const stdoutLines = [];
+    const stderrLines = [];
+    let exitCode = -1;
+    for await (const raw of parseSSEStream(sseStream)) {
+      if (typeof raw.line === "string") {
+        if (raw.stream === "stderr") {
+          stderrLines.push(raw.line);
+        } else {
+          stdoutLines.push(raw.line);
+        }
+      } else if ("exit_code" in raw || "signal" in raw) {
+        if (typeof raw.exit_code === "number") {
+          exitCode = raw.exit_code;
+        } else if (typeof raw.signal === "number") {
+          exitCode = -raw.signal;
+        }
       }
-      await sleep2(100);
-    }
-    const stdoutResp = await this.getStdout(proc.pid);
-    const stderrResp = await this.getStderr(proc.pid);
-    let exitCode;
-    if (info.exitCode != null) {
-      exitCode = info.exitCode;
-    } else if (info.signal != null) {
-      exitCode = -info.signal;
-    } else {
-      exitCode = -1;
     }
     return {
       exitCode,
-      stdout: stdoutResp.lines.join("\n"),
-      stderr: stderrResp.lines.join("\n")
+      stdout: stdoutLines.join("\n"),
+      stderr: stderrLines.join("\n")
     };
   }
   // --- Process management ---
@@ -3386,9 +3399,6 @@ var Sandbox = class {
     return fromSnakeKeys(raw);
   }
 };
-function sleep2(ms) {
-  return new Promise((resolve) => setTimeout(resolve, ms));
-}
 
 // src/client.ts
 var SandboxClient = class _SandboxClient {
@@ -3604,7 +3614,7 @@ var SandboxClient = class _SandboxClient {
           `Snapshot ${result.snapshotId} failed: ${info.error}`
         );
       }
-      await sleep3(pollInterval * 1e3);
+      await sleep2(pollInterval * 1e3);
     }
     throw new SandboxError(
       `Snapshot ${result.snapshotId} did not complete within ${timeout}s`
@@ -3676,14 +3686,15 @@ var SandboxClient = class _SandboxClient {
     );
   }
   // --- Connect ---
-  connect(identifier, proxyUrl) {
+  connect(identifier, proxyUrl, routingHint) {
     const resolvedProxy = proxyUrl ?? resolveProxyUrl(this.apiUrl);
     return new Sandbox({
       sandboxId: identifier,
       proxyUrl: resolvedProxy,
       apiKey: this.apiKey,
       organizationId: this.organizationId,
-      projectId: this.projectId
+      projectId: this.projectId,
+      routingHint
     });
   }
   async createAndConnect(options) {
@@ -3694,6 +3705,11 @@ var SandboxClient = class _SandboxClient {
     } else {
       result = await this.create(options);
     }
+    if (result.status === "running" /* RUNNING */) {
+      const sandbox = this.connect(result.sandboxId, options?.proxyUrl, result.routingHint);
+      sandbox._setOwner(this);
+      return sandbox;
+    }
     const deadline = Date.now() + startupTimeout * 1e3;
     while (Date.now() < deadline) {
       const info = await this.get(result.sandboxId);
@@ -3707,7 +3723,7 @@ var SandboxClient = class _SandboxClient {
           `Sandbox ${result.sandboxId} terminated during startup`
         );
       }
-      await sleep3(500);
+      await sleep2(500);
     }
     try {
       await this.delete(result.sandboxId);
@@ -3718,7 +3734,7 @@ var SandboxClient = class _SandboxClient {
     );
   }
 };
-function sleep3(ms) {
+function sleep2(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
 var RESERVED_SANDBOX_MANAGEMENT_PORT = 9501;
@@ -4600,7 +4616,7 @@ async function runChecked(sandbox, command, args, env, workingDir) {
   }
   return result;
 }
-async function runStreaming(sandbox, emit, sleep4, command, args = [], env, workingDir) {
+async function runStreaming(sandbox, emit, sleep3, command, args = [], env, workingDir) {
   const proc = await sandbox.startProcess(command, {
     args,
     env,
@@ -4625,13 +4641,13 @@ async function runStreaming(sandbox, emit, sleep4, command, args = [], env, work
       emitOutputLines(emit, "stderr", finalStderr, stderrSeen);
       break;
     }
-    await sleep4(300);
+    await sleep3(300);
   }
   for (let i = 0; i < 10; i++) {
     if (info.exitCode != null || info.signal != null) {
       break;
     }
-    await sleep4(200);
+    await sleep3(200);
     info = await sandbox.getProcess(proc.pid);
   }
   const exitCode = info.exitCode != null ? info.exitCode : info.signal != null ? -info.signal : 0;
@@ -4729,7 +4745,7 @@ async function copyFromContext(sandbox, emit, contextDir, sources, destination,
     await copyLocalPathToSandbox(sandbox, localSource, remoteDestination);
   }
 }
-async function addUrlToSandbox(sandbox, emit, url, destination, workingDir, processEnv, sleep4) {
+async function addUrlToSandbox(sandbox, emit, url, destination, workingDir, processEnv, sleep3) {
   let destinationPath = resolveContainerPath(destination, workingDir);
   const parsedUrl = new URL(url);
   const fileName = path.posix.basename(parsedUrl.pathname.replace(/\/$/, "")) || "downloaded";
@@ -4745,7 +4761,7 @@ async function addUrlToSandbox(sandbox, emit, url, destination, workingDir, proc
   await runStreaming(
     sandbox,
     emit,
-    sleep4,
+    sleep3,
     "sh",
     [
       "-c",
@@ -4755,7 +4771,7 @@ async function addUrlToSandbox(sandbox, emit, url, destination, workingDir, proc
     workingDir
   );
 }
-async function executeDockerfilePlan(sandbox, plan, emit, sleep4) {
+async function executeDockerfilePlan(sandbox, plan, emit, sleep3) {
   const processEnv = { ...BUILD_SANDBOX_PIP_ENV };
   let workingDir = "/";
   for (const instruction of plan.instructions) {
@@ -4765,7 +4781,7 @@ async function executeDockerfilePlan(sandbox, plan, emit, sleep4) {
       await runStreaming(
         sandbox,
         emit,
-        sleep4,
+        sleep3,
         "sh",
         ["-c", value],
         processEnv,
@@ -4822,7 +4838,7 @@ async function executeDockerfilePlan(sandbox, plan, emit, sleep4) {
           destination,
           workingDir,
           processEnv,
-          sleep4
+          sleep3
         );
       } else {
         await copyFromContext(
@@ -4890,7 +4906,7 @@ async function registerImage(context, name, dockerfile, snapshotId, snapshotUri,
 }
 async function createSandboxImage(source, options = {}, deps = {}) {
   const emit = deps.emit ?? defaultEmit;
-  const sleep4 = deps.sleep ?? ((ms) => new Promise((r) => setTimeout(r, ms)));
+  const sleep3 = deps.sleep ?? ((ms) => new Promise((r) => setTimeout(r, ms)));
   const context = buildContextFromEnv();
   const clientFactory = deps.createClient ?? createDefaultClient;
   const register = deps.registerImage ?? ((...args) => registerImage(...args));
@@ -4913,7 +4929,7 @@ async function createSandboxImage(source, options = {}, deps = {}) {
       type: "status",
       message: `Materializing image in sandbox ${sandbox.sandboxId}...`
     });
-    await executeDockerfilePlan(sandbox, plan, emit, sleep4);
+    await executeDockerfilePlan(sandbox, plan, emit, sleep3);
     emit({ type: "status", message: "Creating snapshot..." });
     const snapshot = await client.snapshotAndWait(sandbox.sandboxId, {
       contentMode: "filesystem_only"