tensorlake 0.4.46 → 0.4.48

package/dist/index.cjs

@@ -150,6 +150,13 @@ var RequestExecutionError = class extends Error {
 };
 
 // src/http.ts
+var import_undici = require("undici");
+(0, import_undici.setGlobalDispatcher)(
+  new import_undici.Agent({
+    keepAliveTimeout: 6e4,
+    allowH2: true
+  })
+);
 var HttpClient = class {
   baseUrl;
   headers;
@@ -176,6 +183,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();
@@ -198,31 +208,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;
   }
@@ -261,7 +266,7 @@ var HttpClient = class {
       );
       const combinedSignal = signal ? anySignal([signal, this.abortController.signal]) : this.abortController.signal;
       try {
-        const response = await fetch(url, {
+        const response = await (0, import_undici.fetch)(url, {
           method,
           headers,
           body,
@@ -3149,7 +3154,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. */
@@ -3744,14 +3750,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) {
@@ -3762,11 +3769,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;
       }