@rivetkit/engine-runner 2.0.22-rc.1 → 2.0.22-rc.2

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @rivetkit/engine-runner@2.0.22-rc.1 build /home/runner/work/rivet/rivet/engine/sdks/typescript/runner
+> @rivetkit/engine-runner@2.0.22-rc.2 build /home/runner/work/rivet/rivet/engine/sdks/typescript/runner
 > tsup src/mod.ts
 
 CLI Building entry: src/mod.ts
@@ -10,13 +10,13 @@
 CLI Cleaning output folder
 CJS Build start
 ESM Build start
-ESM dist/mod.js     58.58 KB
-ESM dist/mod.js.map 111.69 KB
-ESM ⚡️ Build success in 259ms
-CJS dist/mod.cjs     59.46 KB
-CJS dist/mod.cjs.map 95.00 KB
-CJS ⚡️ Build success in 312ms
+ESM dist/mod.js     61.25 KB
+ESM dist/mod.js.map 117.35 KB
+ESM ⚡️ Build success in 256ms
+CJS dist/mod.cjs     62.25 KB
+CJS dist/mod.cjs.map 99.90 KB
+CJS ⚡️ Build success in 257ms
 DTS Build start
-DTS ⚡️ Build success in 1571ms
-DTS dist/mod.d.cts 2.85 KB
-DTS dist/mod.d.ts  2.85 KB
+DTS ⚡️ Build success in 1525ms
+DTS dist/mod.d.cts 3.26 KB
+DTS dist/mod.d.ts  3.26 KB

package/dist/mod.cjs

@@ -1,4 +1,4 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } var _class;// src/mod.ts
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } var _class;// src/mod.ts
 var _enginerunnerprotocol = require('@rivetkit/engine-runner-protocol'); var protocol = _interopRequireWildcard(_enginerunnerprotocol);
 
 // src/log.ts
@@ -171,12 +171,18 @@ var WebSocketTunnelAdapter = (_class = class {
     this.#sendCallback(messageData, isBinary);
   }
   close(code, reason) {
+    this.closeInner(code, reason);
+  }
+  __closeWithRetry(code, reason) {
+    this.closeInner(code, reason, true);
+  }
+  closeInner(code, reason, retry = false) {
     if (this.#readyState === 2 || // CLOSING
     this.#readyState === 3) {
       return;
     }
     this.#readyState = 2;
-    this.#closeCallback(code, reason);
+    this.#closeCallback(code, reason, retry);
     this.#readyState = 3;
     const closeEvent = {
       wasClean: true,
@@ -361,20 +367,22 @@ var WebSocketTunnelAdapter = (_class = class {
     }
   }
   // Internal methods called by the Tunnel class
-  _handleOpen() {
+  _handleOpen(requestId) {
     if (this.#readyState !== 0) {
       return;
     }
     this.#readyState = 1;
     const event = {
       type: "open",
+      rivetRequestId: requestId,
       target: this
     };
     this.#fireEvent("open", event);
   }
-  _handleMessage(data, isBinary) {
+  /// Returns false if the message was sent off.
+  _handleMessage(requestId, data, index, isBinary) {
     if (this.#readyState !== 1) {
-      return;
+      return true;
     }
     let messageData;
     if (isBinary) {
@@ -398,22 +406,26 @@ var WebSocketTunnelAdapter = (_class = class {
       messageData = data;
     }
     const event = {
-      data: messageData,
       type: "message",
+      data: messageData,
+      rivetRequestId: requestId,
+      rivetMessageIndex: index,
       target: this
     };
     this.#fireEvent("message", event);
+    return false;
   }
-  _handleClose(code, reason) {
+  _handleClose(requestId, code, reason) {
     if (this.#readyState === 3) {
       return;
     }
     this.#readyState = 3;
     const event = {
+      type: "close",
       wasClean: true,
       code: code || 1e3,
       reason: reason || "",
-      type: "close",
+      rivetRequestId: requestId,
       target: this
     };
     this.#fireEvent("close", event);
@@ -485,12 +497,12 @@ var Tunnel = class {
     }
     this.#actorPendingRequests.clear();
     for (const [_, ws] of this.#actorWebSockets) {
-      ws.close();
+      ws.__closeWithRetry();
     }
     this.#actorWebSockets.clear();
   }
   #sendMessage(requestId, messageKind) {
-    var _a;
+    var _a, _b;
     if (!this.#runner.__webSocketReady()) {
       (_a = logger()) == null ? void 0 : _a.warn(
         "cannot send tunnel message, socket not connected to engine"
@@ -498,11 +510,18 @@ var Tunnel = class {
       return;
     }
     const messageId = generateUuidBuffer();
-    const requestIdStr = bufferToString(requestId);
-    this.#pendingTunnelMessages.set(bufferToString(messageId), {
+    const requestIdStr = idToStr(requestId);
+    const messageIdStr = idToStr(messageId);
+    this.#pendingTunnelMessages.set(messageIdStr, {
       sentAt: Date.now(),
       requestIdStr
     });
+    (_b = logger()) == null ? void 0 : _b.debug({
+      msg: "send tunnel msg",
+      requestId: requestIdStr,
+      messageId: messageIdStr,
+      message: messageKind
+    });
     const message = {
       tag: "ToServerTunnelMessage",
       val: {
@@ -514,6 +533,7 @@ var Tunnel = class {
     this.#runner.__sendToServer(message);
   }
   #sendAck(requestId, messageId) {
+    var _a;
     if (!this.#runner.__webSocketReady()) {
       return;
     }
@@ -525,6 +545,11 @@ var Tunnel = class {
         messageKind: { tag: "TunnelAck", val: null }
       }
     };
+    (_a = logger()) == null ? void 0 : _a.debug({
+      msg: "ack tunnel msg",
+      requestId: idToStr(requestId),
+      messageId: idToStr(messageId)
+    });
     this.#runner.__sendToServer(message);
   }
   #startGarbageCollector() {
@@ -557,7 +582,10 @@ var Tunnel = class {
         }
         const webSocket = this.#actorWebSockets.get(requestIdStr);
         if (webSocket) {
-          webSocket.close(1e3, "Message acknowledgment timeout");
+          webSocket.__closeWithRetry(
+            1e3,
+            "Message acknowledgment timeout"
+          );
           this.#actorWebSockets.delete(requestIdStr);
         }
       }
@@ -572,7 +600,7 @@ var Tunnel = class {
       }
     }
   }
-  unregisterActor(actor) {
+  closeActiveRequests(actor) {
     const actorId = actor.actorId;
     for (const requestId of actor.requests) {
       const pending = this.#actorPendingRequests.get(requestId);
@@ -585,24 +613,28 @@ var Tunnel = class {
     for (const webSocketId of actor.webSockets) {
       const ws = this.#actorWebSockets.get(webSocketId);
       if (ws) {
-        ws.close(1e3, "Actor stopped");
+        ws.__closeWithRetry(1e3, "Actor stopped");
         this.#actorWebSockets.delete(webSocketId);
       }
     }
     actor.webSockets.clear();
   }
-  async #fetch(actorId, request) {
+  async #fetch(actorId, requestId, request) {
     var _a;
     if (!this.#runner.hasActor(actorId)) {
       (_a = logger()) == null ? void 0 : _a.warn({
         msg: "ignoring request for unknown actor",
         actorId
       });
-      return new Response("Actor not found", { status: 404 });
+      return new Response("Actor not found", {
+        status: 503,
+        headers: { "x-rivet-error": "runner.actor_not_found" }
+      });
     }
     const fetchHandler = this.#runner.config.fetch(
       this.#runner,
       actorId,
+      requestId,
       request
     );
     if (!fetchHandler) {
@@ -611,43 +643,64 @@ var Tunnel = class {
     return fetchHandler;
   }
   async handleTunnelMessage(message) {
+    var _a, _b;
+    const requestIdStr = idToStr(message.requestId);
+    const messageIdStr = idToStr(message.messageId);
+    (_a = logger()) == null ? void 0 : _a.debug({
+      msg: "receive tunnel msg",
+      requestId: requestIdStr,
+      messageId: messageIdStr,
+      message: message.messageKind
+    });
     if (message.messageKind.tag === "TunnelAck") {
-      const msgIdStr = bufferToString(message.messageId);
-      const pending = this.#pendingTunnelMessages.get(msgIdStr);
+      const pending = this.#pendingTunnelMessages.get(messageIdStr);
       if (pending) {
-        this.#pendingTunnelMessages.delete(msgIdStr);
+        const didDelete = this.#pendingTunnelMessages.delete(messageIdStr);
+        if (!didDelete) {
+          (_b = logger()) == null ? void 0 : _b.warn({
+            msg: "received tunnel ack for nonexistent message",
+            requestId: requestIdStr,
+            messageId: messageIdStr
+          });
+        }
       }
     } else {
-      this.#sendAck(message.requestId, message.messageId);
       switch (message.messageKind.tag) {
         case "ToClientRequestStart":
+          this.#sendAck(message.requestId, message.messageId);
           await this.#handleRequestStart(
             message.requestId,
             message.messageKind.val
           );
           break;
         case "ToClientRequestChunk":
+          this.#sendAck(message.requestId, message.messageId);
           await this.#handleRequestChunk(
             message.requestId,
             message.messageKind.val
           );
           break;
         case "ToClientRequestAbort":
+          this.#sendAck(message.requestId, message.messageId);
           await this.#handleRequestAbort(message.requestId);
           break;
         case "ToClientWebSocketOpen":
+          this.#sendAck(message.requestId, message.messageId);
           await this.#handleWebSocketOpen(
             message.requestId,
             message.messageKind.val
           );
           break;
-        case "ToClientWebSocketMessage":
-          await this.#handleWebSocketMessage(
+        case "ToClientWebSocketMessage": {
+          this.#sendAck(message.requestId, message.messageId);
+          const _unhandled = await this.#handleWebSocketMessage(
             message.requestId,
             message.messageKind.val
           );
           break;
+        }
         case "ToClientWebSocketClose":
+          this.#sendAck(message.requestId, message.messageId);
           await this.#handleWebSocketClose(
             message.requestId,
             message.messageKind.val
@@ -660,7 +713,7 @@ var Tunnel = class {
   }
   async #handleRequestStart(requestId, req) {
     var _a;
-    const requestIdStr = bufferToString(requestId);
+    const requestIdStr = idToStr(requestId);
     const actor = this.#runner.getActor(req.actorId);
     if (actor) {
       actor.requests.add(requestIdStr);
@@ -700,11 +753,16 @@ var Tunnel = class {
         });
         const response = await this.#fetch(
           req.actorId,
+          requestId,
           streamingRequest
         );
         await this.#sendResponse(requestId, response);
       } else {
-        const response = await this.#fetch(req.actorId, request);
+        const response = await this.#fetch(
+          req.actorId,
+          requestId,
+          request
+        );
         await this.#sendResponse(requestId, response);
       }
     } catch (error) {
@@ -718,7 +776,7 @@ var Tunnel = class {
     }
   }
   async #handleRequestChunk(requestId, chunk) {
-    const requestIdStr = bufferToString(requestId);
+    const requestIdStr = idToStr(requestId);
     const pending = this.#actorPendingRequests.get(requestIdStr);
     if (pending == null ? void 0 : pending.streamController) {
       pending.streamController.enqueue(new Uint8Array(chunk.body));
@@ -729,7 +787,7 @@ var Tunnel = class {
     }
   }
   async #handleRequestAbort(requestId) {
-    const requestIdStr = bufferToString(requestId);
+    const requestIdStr = idToStr(requestId);
     const pending = this.#actorPendingRequests.get(requestIdStr);
     if (pending == null ? void 0 : pending.streamController) {
       pending.streamController.error(new Error("Request aborted"));
@@ -770,7 +828,7 @@ var Tunnel = class {
   }
   async #handleWebSocketOpen(requestId, open) {
     var _a, _b, _c;
-    const webSocketId = bufferToString(requestId);
+    const webSocketId = idToStr(requestId);
     const actor = this.#runner.getActor(open.actorId);
     if (!actor) {
       (_a = logger()) == null ? void 0 : _a.warn({
@@ -781,7 +839,8 @@ var Tunnel = class {
         tag: "ToServerWebSocketClose",
         val: {
           code: 1011,
-          reason: "Actor not found"
+          reason: "Actor not found",
+          retry: false
         }
       });
       return;
@@ -795,7 +854,8 @@ var Tunnel = class {
         tag: "ToServerWebSocketClose",
         val: {
           code: 1011,
-          reason: "Not Implemented"
+          reason: "Not Implemented",
+          retry: false
         }
       });
       return;
@@ -816,12 +876,13 @@ var Tunnel = class {
             }
           });
         },
-        (code, reason) => {
+        (code, reason, retry = false) => {
           this.#sendMessage(requestId, {
             tag: "ToServerWebSocketClose",
             val: {
               code: code || null,
-              reason: reason || null
+              reason: reason || null,
+              retry
             }
           });
           this.#actorWebSockets.delete(webSocketId);
@@ -831,11 +892,6 @@ var Tunnel = class {
         }
       );
       this.#actorWebSockets.set(webSocketId, adapter);
-      this.#sendMessage(requestId, {
-        tag: "ToServerWebSocketOpen",
-        val: null
-      });
-      adapter._handleOpen();
       const headerInit = {};
       if (open.headers) {
         for (const [k, v] of open.headers) {
@@ -848,10 +904,24 @@ var Tunnel = class {
         method: "GET",
         headers: headerInit
       });
+      const hibernationConfig = this.#runner.config.getActorHibernationConfig(
+        actor.actorId,
+        requestId,
+        request
+      );
+      this.#sendMessage(requestId, {
+        tag: "ToServerWebSocketOpen",
+        val: {
+          canHibernate: hibernationConfig.enabled,
+          lastMsgIndex: BigInt(_nullishCoalesce(hibernationConfig.lastMsgIndex, () => ( -1)))
+        }
+      });
+      adapter._handleOpen(requestId);
       await websocketHandler(
         this.#runner,
         open.actorId,
         adapter,
+        requestId,
         request
       );
     } catch (error) {
@@ -860,7 +930,8 @@ var Tunnel = class {
         tag: "ToServerWebSocketClose",
         val: {
           code: 1011,
-          reason: "Server Error"
+          reason: "Server Error",
+          retry: false
         }
       });
       this.#actorWebSockets.delete(webSocketId);
@@ -869,34 +940,59 @@ var Tunnel = class {
       }
     }
   }
+  /// Returns false if the message was sent off
   async #handleWebSocketMessage(requestId, msg) {
-    const webSocketId = bufferToString(requestId);
+    const webSocketId = idToStr(requestId);
     const adapter = this.#actorWebSockets.get(webSocketId);
     if (adapter) {
       const data = msg.binary ? new Uint8Array(msg.data) : new TextDecoder().decode(new Uint8Array(msg.data));
-      adapter._handleMessage(data, msg.binary);
+      return adapter._handleMessage(
+        requestId,
+        data,
+        msg.index,
+        msg.binary
+      );
+    } else {
+      return true;
     }
   }
+  __ackWebsocketMessage(requestId, index) {
+    var _a;
+    (_a = logger()) == null ? void 0 : _a.debug({
+      msg: "ack ws msg",
+      requestId: idToStr(requestId),
+      index
+    });
+    if (index < 0 || index > 65535)
+      throw new Error("invalid websocket ack index");
+    this.#sendMessage(requestId, {
+      tag: "ToServerWebSocketMessageAck",
+      val: {
+        index
+      }
+    });
+  }
   async #handleWebSocketClose(requestId, close) {
-    const webSocketId = bufferToString(requestId);
-    const adapter = this.#actorWebSockets.get(webSocketId);
+    const requestIdStr = idToStr(requestId);
+    const adapter = this.#actorWebSockets.get(requestIdStr);
     if (adapter) {
       adapter._handleClose(
+        requestId,
         close.code || void 0,
         close.reason || void 0
       );
-      this.#actorWebSockets.delete(webSocketId);
+      this.#actorWebSockets.delete(requestIdStr);
     }
   }
 };
-function bufferToString(buffer) {
-  return Buffer.from(buffer).toString("base64");
-}
 function generateUuidBuffer() {
   const buffer = new Uint8Array(16);
   _uuid.v4.call(void 0, void 0, buffer);
   return buffer.buffer;
 }
+function idToStr(id) {
+  return _uuid.stringify.call(void 0, new Uint8Array(id));
+}
 
 // src/websocket.ts
 var webSocketPromise = null;
@@ -933,7 +1029,8 @@ async function importWebSocket() {
 
 // src/mod.ts
 var KV_EXPIRE = 3e4;
-var PROTOCOL_VERSION = 1;
+var PROTOCOL_VERSION = 2;
+var RUNNER_PING_INTERVAL = 3e3;
 var EVENT_BACKLOG_WARN_THRESHOLD = 1e4;
 var SIGNAL_HANDLERS = [];
 var Runner = class {
@@ -989,22 +1086,13 @@ var Runner = class {
     var _a;
     const actor = this.#removeActor(actorId, generation);
     if (!actor) return;
-    (_a = this.#tunnel) == null ? void 0 : _a.unregisterActor(actor);
     try {
       await this.#config.onActorStop(actorId, actor.generation);
     } catch (err) {
       console.error(`Error in onActorStop for actor ${actorId}:`, err);
     }
+    (_a = this.#tunnel) == null ? void 0 : _a.closeActiveRequests(actor);
     this.#sendActorStateUpdate(actorId, actor.generation, "stopped");
-    this.#config.onActorStop(actorId, actor.generation).catch((err) => {
-      var _a2;
-      (_a2 = logger()) == null ? void 0 : _a2.error({
-        msg: "error in onactorstop for actor",
-        runnerId: this.runnerId,
-        actorId,
-        err
-      });
-    });
   }
   #stopAllActors() {
     var _a;
@@ -1043,8 +1131,9 @@ var Runner = class {
     const actor = this.#actors.get(actorId);
     return !!actor && (generation === void 0 || actor.generation === generation);
   }
+  // IMPORTANT: Make sure to call stopActiveRequests if calling #removeActor
   #removeActor(actorId, generation) {
-    var _a, _b, _c;
+    var _a, _b;
     const actor = this.#actors.get(actorId);
     if (!actor) {
       (_a = logger()) == null ? void 0 : _a.error({
@@ -1064,22 +1153,6 @@ var Runner = class {
       return void 0;
     }
     this.#actors.delete(actorId);
-    const actorWebSockets = this.#actorWebSockets.get(actorId);
-    if (actorWebSockets) {
-      for (const ws of actorWebSockets) {
-        try {
-          ws.close(1e3, "Actor stopped");
-        } catch (err) {
-          (_c = logger()) == null ? void 0 : _c.error({
-            msg: "error closing websocket for actor",
-            runnerId: this.runnerId,
-            actorId,
-            err
-          });
-        }
-      }
-      this.#actorWebSockets.delete(actorId);
-    }
     return actor;
   }
   // MARK: Start
@@ -1231,22 +1304,33 @@ var Runner = class {
     this.#config.onShutdown();
   }
   // MARK: Networking
+  get pegboardEndpoint() {
+    return this.#config.pegboardEndpoint || this.#config.endpoint;
+  }
   get pegboardUrl() {
-    const endpoint = this.#config.pegboardEndpoint || this.#config.endpoint;
-    const wsEndpoint = endpoint.replace("http://", "ws://").replace("https://", "wss://");
-    return `${wsEndpoint}?protocol_version=${PROTOCOL_VERSION}&namespace=${encodeURIComponent(this.#config.namespace)}&runner_key=${encodeURIComponent(this.#config.runnerKey)}`;
+    const wsEndpoint = this.pegboardEndpoint.replace("http://", "ws://").replace("https://", "wss://");
+    const baseUrl = wsEndpoint.endsWith("/") ? wsEndpoint.slice(0, -1) : wsEndpoint;
+    return `${baseUrl}/runners/connect?protocol_version=${PROTOCOL_VERSION}&namespace=${encodeURIComponent(this.#config.namespace)}&runner_key=${encodeURIComponent(this.#config.runnerKey)}`;
   }
   // MARK: Runner protocol
   async #openPegboardWebSocket() {
-    const protocols = ["rivet", `rivet_target.runner`];
+    var _a;
+    const protocols = ["rivet"];
     if (this.config.token)
       protocols.push(`rivet_token.${this.config.token}`);
     const WS = await importWebSocket();
     const ws = new WS(this.pegboardUrl, protocols);
     this.#pegboardWebSocket = ws;
+    (_a = logger()) == null ? void 0 : _a.info({
+      msg: "connecting",
+      endpoint: this.pegboardEndpoint,
+      namespace: this.#config.namespace,
+      runnerKey: this.#config.runnerKey,
+      hasToken: !!this.config.token
+    });
     ws.addEventListener("open", () => {
-      var _a;
-      (_a = logger()) == null ? void 0 : _a.info({ msg: "Connected" });
+      var _a2;
+      (_a2 = logger()) == null ? void 0 : _a2.info({ msg: "connected" });
       this.#reconnectAttempt = 0;
       if (this.#reconnectTimeout) {
         clearTimeout(this.#reconnectTimeout);
@@ -1276,9 +1360,8 @@ var Runner = class {
         val: init
       });
       this.#processUnsentKvRequests();
-      const pingInterval = 1e3;
       const pingLoop = setInterval(() => {
-        var _a2;
+        var _a3;
         if (ws.readyState === 1) {
           this.__sendToServer({
             tag: "ToServerPing",
@@ -1288,21 +1371,21 @@ var Runner = class {
           });
         } else {
           clearInterval(pingLoop);
-          (_a2 = logger()) == null ? void 0 : _a2.info({
+          (_a3 = logger()) == null ? void 0 : _a3.info({
             msg: "WebSocket not open, stopping ping loop",
             runnerId: this.runnerId
           });
         }
-      }, pingInterval);
+      }, RUNNER_PING_INTERVAL);
       this.#pingLoop = pingLoop;
       const ackInterval = 5 * 60 * 1e3;
       const ackLoop = setInterval(() => {
-        var _a2;
+        var _a3;
         if (ws.readyState === 1) {
           this.#sendCommandAcknowledgment();
         } else {
           clearInterval(ackLoop);
-          (_a2 = logger()) == null ? void 0 : _a2.info({
+          (_a3 = logger()) == null ? void 0 : _a3.info({
             msg: "WebSocket not open, stopping ack loop",
             runnerId: this.runnerId
           });
@@ -1311,7 +1394,7 @@ var Runner = class {
       this.#ackInterval = ackLoop;
     });
     ws.addEventListener("message", async (ev) => {
-      var _a, _b, _c, _d;
+      var _a2, _b, _c, _d;
       let buf;
       if (ev.data instanceof Blob) {
         buf = new Uint8Array(await ev.data.arrayBuffer());
@@ -1327,7 +1410,7 @@ var Runner = class {
           this.runnerId = init.runnerId;
           this.#eventHistory.length = 0;
         }
-        this.#runnerLostThreshold = ((_a = init.metadata) == null ? void 0 : _a.runnerLostThreshold) ? Number(init.metadata.runnerLostThreshold) : void 0;
+        this.#runnerLostThreshold = ((_a2 = init.metadata) == null ? void 0 : _a2.runnerLostThreshold) ? Number(init.metadata.runnerLostThreshold) : void 0;
         (_b = logger()) == null ? void 0 : _b.info({
           msg: "received init",
           runnerId: init.runnerId,
@@ -1354,8 +1437,8 @@ var Runner = class {
       }
     });
     ws.addEventListener("error", (ev) => {
-      var _a, _b;
-      (_a = logger()) == null ? void 0 : _a.error({
+      var _a2, _b;
+      (_a2 = logger()) == null ? void 0 : _a2.error({
         msg: `WebSocket error: ${ev.error}`,
         runnerId: this.runnerId
       });
@@ -1374,14 +1457,14 @@ var Runner = class {
       }
     });
     ws.addEventListener("close", async (ev) => {
-      var _a, _b, _c;
-      (_a = logger()) == null ? void 0 : _a.info({
+      var _a2, _b, _c;
+      (_a2 = logger()) == null ? void 0 : _a2.info({
         msg: "connection closed",
         runnerId: this.runnerId,
         code: ev.code,
         reason: ev.reason.toString()
       });
-      this.#config.onDisconnected();
+      this.#config.onDisconnected(ev.code, ev.reason);
       if (ev.reason.toString().startsWith("ws.eviction")) {
         (_b = logger()) == null ? void 0 : _b.info({
           msg: "runner evicted",
@@ -1947,6 +2030,10 @@ var Runner = class {
       });
     }
   }
+  sendWebsocketMessageAck(requestId, index) {
+    var _a;
+    (_a = this.#tunnel) == null ? void 0 : _a.__ackWebsocketMessage(requestId, index);
+  }
   getServerlessInitPacket() {
     if (!this.runnerId) return void 0;
     const data = protocol.encodeToServerlessServer({