tensorlake 0.4.46 → 0.4.48

package/dist/sandbox-image.d.cts

@@ -1 +1 @@
-export { v as CreateSandboxImageOptions, x as DockerfileBuildPlan, y as DockerfileInstruction, J as SandboxImageSource, V as createSandboxImage, X as defaultRegisteredName, Y as loadDockerfilePlan, Z as loadImagePlan, _ as logicalDockerfileLines, $ as runCreateSandboxImageCli } from './sandbox-image-B5Fa0xqb.cjs';
+export { v as CreateSandboxImageOptions, x as DockerfileBuildPlan, y as DockerfileInstruction, J as SandboxImageSource, V as createSandboxImage, X as defaultRegisteredName, Y as loadDockerfilePlan, Z as loadImagePlan, _ as logicalDockerfileLines, $ as runCreateSandboxImageCli } from './sandbox-image-BQ6fJT92.cjs';

package/dist/sandbox-image.d.ts

@@ -1 +1 @@
-export { v as CreateSandboxImageOptions, x as DockerfileBuildPlan, y as DockerfileInstruction, J as SandboxImageSource, V as createSandboxImage, X as defaultRegisteredName, Y as loadDockerfilePlan, Z as loadImagePlan, _ as logicalDockerfileLines, $ as runCreateSandboxImageCli } from './sandbox-image-B5Fa0xqb.js';
+export { v as CreateSandboxImageOptions, x as DockerfileBuildPlan, y as DockerfileInstruction, J as SandboxImageSource, V as createSandboxImage, X as defaultRegisteredName, Y as loadDockerfilePlan, Z as loadImagePlan, _ as logicalDockerfileLines, $ as runCreateSandboxImageCli } from './sandbox-image-BQ6fJT92.js';

package/dist/sandbox-image.js

@@ -112,6 +112,17 @@ var RemoteAPIError = class extends SandboxError {
 };
 
 // src/http.ts
+import {
+  Agent,
+  fetch as undiciFetch,
+  setGlobalDispatcher
+} from "undici";
+setGlobalDispatcher(
+  new Agent({
+    keepAliveTimeout: 6e4,
+    allowH2: true
+  })
+);
 var HttpClient = class {
   baseUrl;
   headers;
@@ -138,6 +149,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();
@@ -160,31 +174,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;
   }
@@ -223,7 +232,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,
@@ -3032,7 +3041,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. */
@@ -3627,14 +3637,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) {
@@ -3645,11 +3656,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;
       }