tensorlake 0.4.46 → 0.4.48

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-BQ6fJT92.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-BQ6fJT92.cjs';
 
 type MouseButton = "left" | "middle" | "right";
 interface ConnectDesktopOptions {
@@ -249,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-BQ6fJT92.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-BQ6fJT92.js';
 
 type MouseButton = "left" | "middle" | "right";
 interface ConnectDesktopOptions {
@@ -249,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

@@ -88,6 +88,17 @@ var RequestExecutionError = class extends Error {
 };
 
 // src/http.ts
+import {
+  Agent,
+  fetch as undiciFetch,
+  setGlobalDispatcher
+} from "undici";
+setGlobalDispatcher(
+  new Agent({
+    keepAliveTimeout: 6e4,
+    allowH2: true
+  })
+);
 var HttpClient = class {
   baseUrl;
   headers;
@@ -114,6 +125,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();
@@ -136,31 +150,26 @@ var HttpClient = class {
     if (options?.contentType) {
       headers["Content-Type"] = options.contentType;
     }
-    const response = await this.requestResponse(
-      method,
-      path2,
-      {
-        body: options?.body,
-        headers,
-        signal: options?.signal
-      }
-    );
+    const response = await this.requestResponse(method, path2, {
+      body: options?.body,
+      headers,
+      signal: options?.signal
+    });
     const buffer = await response.arrayBuffer();
     return new Uint8Array(buffer);
   }
   /** 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
-      }
-    );
+    const response = await this.requestResponse(method, path2, {
+      json: options?.json,
+      headers: { Accept: "text/event-stream" },
+      signal: options?.signal
+    });
     if (!response.body) {
-      throw new RemoteAPIError(response.status, "No response body for SSE stream");
+      throw new RemoteAPIError(
+        response.status,
+        "No response body for SSE stream"
+      );
     }
     return response.body;
   }
@@ -199,7 +208,7 @@ var HttpClient = class {
       );
       const combinedSignal = signal ? anySignal([signal, this.abortController.signal]) : this.abortController.signal;
       try {
-        const response = await fetch(url, {
+        const response = await undiciFetch(url, {
           method,
           headers,
           body,
@@ -3087,7 +3096,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. */
@@ -3682,14 +3692,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) {
@@ -3700,11 +3711,16 @@ 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);
       if (info.status === "running" /* RUNNING */) {
-        const sandbox = this.connect(result.sandboxId, options?.proxyUrl);
+        const sandbox = this.connect(result.sandboxId, options?.proxyUrl, info.routingHint);
         sandbox._setOwner(this);
         return sandbox;
       }