tensorlake 0.4.45 → 0.4.47

package/dist/index.cjs

@@ -176,6 +176,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();
@@ -210,12 +213,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
       }
@@ -3148,7 +3152,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. */
@@ -3169,37 +3174,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 ---
@@ -3448,9 +3461,6 @@ var Sandbox = class {
     return fromSnakeKeys(raw);
   }
 };
-function sleep2(ms) {
-  return new Promise((resolve) => setTimeout(resolve, ms));
-}
 
 // src/client.ts
 var SandboxClient = class _SandboxClient {
@@ -3666,7 +3676,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`
@@ -3738,14 +3748,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) {
@@ -3756,6 +3767,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);
@@ -3769,7 +3785,7 @@ var SandboxClient = class _SandboxClient {
           `Sandbox ${result.sandboxId} terminated during startup`
         );
       }
-      await sleep3(500);
+      await sleep2(500);
     }
     try {
       await this.delete(result.sandboxId);
@@ -3780,7 +3796,7 @@ var SandboxClient = class _SandboxClient {
     );
   }
 };
-function sleep3(ms) {
+function sleep2(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
 var RESERVED_SANDBOX_MANAGEMENT_PORT = 9501;
@@ -4662,7 +4678,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,
@@ -4687,13 +4703,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;
@@ -4791,7 +4807,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 = import_node_path.default.posix.basename(parsedUrl.pathname.replace(/\/$/, "")) || "downloaded";
@@ -4807,7 +4823,7 @@ async function addUrlToSandbox(sandbox, emit, url, destination, workingDir, proc
   await runStreaming(
     sandbox,
     emit,
-    sleep4,
+    sleep3,
     "sh",
     [
       "-c",
@@ -4817,7 +4833,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) {
@@ -4827,7 +4843,7 @@ async function executeDockerfilePlan(sandbox, plan, emit, sleep4) {
       await runStreaming(
         sandbox,
         emit,
-        sleep4,
+        sleep3,
         "sh",
         ["-c", value],
         processEnv,
@@ -4884,7 +4900,7 @@ async function executeDockerfilePlan(sandbox, plan, emit, sleep4) {
           destination,
           workingDir,
           processEnv,
-          sleep4
+          sleep3
         );
       } else {
         await copyFromContext(
@@ -4952,7 +4968,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));
@@ -4975,7 +4991,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"