@ai-sdk/harness-claude-code 1.0.121 → 1.0.122

package/dist/index.d.ts

@@ -3,6 +3,7 @@ import * as _ai_sdk_provider_utils from '@ai-sdk/provider-utils';
 import { Experimental_SandboxSession } from '@ai-sdk/provider-utils';
 import * as _ai_sdk_harness from '@ai-sdk/harness';
 import { HarnessV1Authentication, HarnessV1CredentialForwarding, HarnessV1PortEndpoint, HarnessV1MintBridgeTokenCallback, HarnessV1, HarnessV1BuiltinTool, HARNESS_V1_BUILTIN_TOOLS } from '@ai-sdk/harness';
+import { SandboxChannelReconnectOptions } from '@ai-sdk/harness/utils';
 
 type ClaudeCodeAuthenticationMode = HarnessV1Authentication;
 
@@ -60,6 +61,13 @@ type ClaudeCodeHarnessSettings = {
     readonly portEndpoint?: HarnessV1PortEndpoint;
     /** Maximum milliseconds to wait for the bridge to advertise its port. Defaults to 120000. */
     readonly startupTimeoutMs?: number;
+    /**
+     * Configures reconnection attempts after an established bridge connection
+     * drops. The reconnect window includes connection establishment and
+     * backoff delays. Defaults to 30 seconds with exponential backoff from 50
+     * milliseconds up to 2 seconds.
+     */
+    readonly reconnect?: SandboxChannelReconnectOptions;
     /**
      * Creates the authentication token used by the sandbox bridge. Defaults to
      * a random 32-byte hexadecimal token.

package/dist/index.js

@@ -22,6 +22,7 @@ import {
   resolveSandboxHomeDir,
   SandboxChannel,
   shellQuote,
+  sleep,
   warnCredentialBrokeringUnavailable,
   waitForBridgeReady,
   withBridgeToken,
@@ -442,7 +443,7 @@ var inboundMessageSchema = z.discriminatedUnion("type", [
 ]);
 
 // src/version.ts
-var VERSION = true ? "1.0.121" : "0.0.0-test";
+var VERSION = true ? "1.0.122" : "0.0.0-test";
 
 // src/claude-code-harness.ts
 var CLAUDE_CODE_CLIENT_APP = `ai-sdk/harness-claude-code/${VERSION}`;
@@ -1164,10 +1165,13 @@ function createClaudeCode(settings = {}) {
         sessionId: startOpts.sessionId
       });
       let supportsUserMessageResponses = false;
-      const buildConnect = (endpoint2) => async () => {
+      const buildConnect = (endpoint2) => async ({
+        abortSignal
+      }) => {
         return openBridgeWebSocket({
           endpoint: endpoint2,
           timeoutMs,
+          abortSignal,
           onHello: (supportsResponses) => {
             supportsUserMessageResponses = supportsResponses;
           }
@@ -1189,7 +1193,8 @@ function createClaudeCode(settings = {}) {
             outboundSchema: outboundMessageSchema,
             initialLastSeenEventId: coords.lastSeenEventId,
             onDiagnostic,
-            onBridgeError
+            onBridgeError,
+            reconnect: settings.reconnect
           });
           await attachChannel.open(isContinue ? { resume: true } : void 0);
           return createSession({
@@ -1303,6 +1308,7 @@ function createClaudeCode(settings = {}) {
         outboundSchema: outboundMessageSchema,
         onDiagnostic,
         onBridgeError,
+        reconnect: settings.reconnect,
         // In replay mode the respawned bridge reloaded the finished turn from
         // disk; seed the cursor and resume so it streams the tail (incl.
         // `finish`) rather than starting empty.
@@ -1388,9 +1394,15 @@ function openWebSocketAndWaitForBridgeHello({
   endpoint,
   openTimeoutMs,
   getHelloTimeoutMs,
+  abortSignal,
   onHello
 }) {
   return new Promise((resolve2, reject) => {
+    const abortReason = () => abortSignal.reason ?? new Error("WebSocket connection aborted");
+    if (abortSignal.aborted) {
+      reject(abortReason());
+      return;
+    }
     const ws = new WebSocket(endpoint.url, {
       headers: endpoint.headers == null ? void 0 : { ...endpoint.headers }
     });
@@ -1399,6 +1411,7 @@ function openWebSocketAndWaitForBridgeHello({
     let settled = false;
     let openTimer;
     let helloTimer;
+    let onAbort;
     const cleanup = ({
       keepTerminationListeners = false
     } = {}) => {
@@ -1406,6 +1419,9 @@ function openWebSocketAndWaitForBridgeHello({
       if (helloTimer) clearTimeout(helloTimer);
       ws.off("open", onOpen);
       ws.off("message", onMessage);
+      if (onAbort != null) {
+        abortSignal.removeEventListener("abort", onAbort);
+      }
       if (!keepTerminationListeners) {
         ws.off("close", onClose);
         ws.off("error", onError);
@@ -1474,6 +1490,7 @@ function openWebSocketAndWaitForBridgeHello({
       cleanup();
     };
     const onError = (err) => settle(err);
+    onAbort = () => settle(abortReason());
     openTimer = setTimeout(
       () => settle(new Error(`WebSocket open timed out after ${openTimeoutMs}ms`)),
       openTimeoutMs
@@ -1483,17 +1500,19 @@ function openWebSocketAndWaitForBridgeHello({
     ws.on("message", onMessage);
     ws.on("close", onClose);
     ws.on("error", onError);
+    abortSignal.addEventListener("abort", onAbort, { once: true });
   });
 }
 async function openBridgeWebSocket({
   endpoint,
   timeoutMs,
+  abortSignal,
   onHello
 }) {
   const deadline = Date.now() + timeoutMs;
   let attempt = 0;
   let lastError;
-  while (Date.now() < deadline) {
+  while (!abortSignal.aborted && Date.now() < deadline) {
     attempt++;
     try {
       const remaining = Math.max(1, deadline - Date.now());
@@ -1501,15 +1520,25 @@ async function openBridgeWebSocket({
         endpoint,
         openTimeoutMs: Math.min(1e4, remaining),
         getHelloTimeoutMs: () => Math.min(5e3, Math.max(1, deadline - Date.now())),
+        abortSignal,
         onHello
       });
     } catch (err) {
+      if (abortSignal.aborted) {
+        throw abortSignal.reason ?? new Error("WebSocket connection aborted");
+      }
       lastError = err;
       const remaining = deadline - Date.now();
       if (remaining <= 0) break;
-      await sleep(Math.min(250 * attempt, 1e3, remaining));
+      await sleep({
+        ms: Math.min(250 * attempt, 1e3, remaining),
+        abortSignal
+      });
     }
   }
+  if (abortSignal.aborted) {
+    throw abortSignal.reason ?? new Error("WebSocket connection aborted");
+  }
   throw new Error(
     `claude-code bridge did not complete WebSocket handshake within ${timeoutMs}ms after ${attempt} attempt(s). Last error: ${formatUnknownError(lastError)}`
   );
@@ -1526,12 +1555,6 @@ function webSocketMessageToString(raw) {
   }
   return String(raw);
 }
-function sleep(ms) {
-  return new Promise((resolve2) => {
-    const timer = setTimeout(resolve2, ms);
-    timer.unref?.();
-  });
-}
 function formatUnknownError(error) {
   if (error instanceof Error) return error.message;
   return String(error);