From 0298a504744e00d9b3309d220d887d6003626c97 Mon Sep 17 00:00:00 2001 From: Nathan Flurry Date: Mon, 3 Nov 2025 14:17:55 -0800 Subject: [PATCH] fix(rivetkit): ack ws messages --- .../packages/rivetkit/src/drivers/engine/actor-driver.ts | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/rivetkit-typescript/packages/rivetkit/src/drivers/engine/actor-driver.ts b/rivetkit-typescript/packages/rivetkit/src/drivers/engine/actor-driver.ts index ca8740ee28..6cfb556d9a 100644 --- a/rivetkit-typescript/packages/rivetkit/src/drivers/engine/actor-driver.ts +++ b/rivetkit-typescript/packages/rivetkit/src/drivers/engine/actor-driver.ts @@ -545,8 +545,15 @@ export class EngineActorDriver implements ActorDriver { }); } - websocket.addEventListener("message", (event) => { + websocket.addEventListener("message", (event: RivetMessageEvent) => { wsHandlerPromise.then((x) => x.onMessage?.(event, wsContext)); + + invariant(event.rivetRequestId, "missing rivetRequestId"); + invariant(event.rivetMessageIndex, "missing rivetMessageIndex"); + this.#runner.sendWebsocketMessageAck( + event.rivetRequestId, + event.rivetMessageIndex, + ); }); websocket.addEventListener("close", (event) => {