@ai-sdk/harness-codex 1.0.118 → 1.0.120

package/dist/index.js

@@ -531,7 +531,7 @@ var inboundMessageSchema = z.discriminatedUnion("type", [
 var CLI_SHIM_FILENAME = "harness-tool.mjs";
 
 // src/version.ts
-var VERSION = true ? "1.0.118" : "0.0.0-test";
+var VERSION = true ? "1.0.120" : "0.0.0-test";
 
 // src/codex-harness.ts
 var DEFAULT_CODEX_MODEL = "gpt-5.5";
@@ -682,11 +682,12 @@ function createCodex(settings = {}) {
             token: coords.token
           });
           const attachChannel = new SandboxChannel({
-            connect: () => openWebSocket(attachEndpoint),
+            connect: ({ abortSignal }) => openWebSocket({ ...attachEndpoint, abortSignal }),
             outboundSchema: outboundMessageSchema,
             initialLastSeenEventId: coords.lastSeenEventId,
             onDiagnostic,
-            onBridgeError
+            onBridgeError,
+            reconnect: settings.reconnect
           });
           await attachChannel.open(isContinue ? { resume: true } : void 0);
           return createSession({
@@ -808,10 +809,11 @@ function createCodex(settings = {}) {
       });
       const bridgeEndpoint = withBridgeToken({ endpoint, token });
       const channel = new SandboxChannel({
-        connect: () => openWebSocket(bridgeEndpoint),
+        connect: ({ abortSignal }) => openWebSocket({ ...bridgeEndpoint, abortSignal }),
         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`).
@@ -896,22 +898,54 @@ async function resolveBridgeEndpoint({
 }
 function openWebSocket({
   url,
-  headers
+  headers,
+  abortSignal
 }) {
   return new Promise((resolve2, reject) => {
+    if (abortSignal.aborted) {
+      reject(abortSignal.reason ?? new Error("WebSocket connection aborted"));
+      return;
+    }
     const ws = new WebSocket(url, {
       headers: headers == null ? void 0 : { ...headers }
     });
-    const onOpen = () => {
+    let settled = false;
+    const cleanup = () => {
+      abortSignal.removeEventListener("abort", onAbort);
+      ws.off("open", onOpen);
       ws.off("error", onError);
+    };
+    const rejectWithCleanup = (error) => {
+      if (settled) return;
+      settled = true;
+      cleanup();
+      ws.once("error", () => {
+      });
+      try {
+        ws.terminate();
+      } catch {
+        try {
+          ws.close();
+        } catch {
+        }
+      }
+      reject(error);
+    };
+    const onOpen = () => {
+      if (settled) return;
+      settled = true;
+      cleanup();
       resolve2(ws);
     };
     const onError = (err) => {
-      ws.off("open", onOpen);
-      reject(err);
+      rejectWithCleanup(err);
     };
+    const onAbort = () => rejectWithCleanup(
+      abortSignal.reason ?? new Error("WebSocket connection aborted")
+    );
     ws.once("open", onOpen);
     ws.once("error", onError);
+    abortSignal.addEventListener("abort", onAbort, { once: true });
   });
 }
 function createSession({