From dcedee1471f79f00bc13d0912f87c3d7139d5ca2 Mon Sep 17 00:00:00 2001 From: ABCxFF <79597906+abcxff@users.noreply.github.com> Date: Wed, 13 May 2026 02:56:25 +0000 Subject: [PATCH] fix(rivetkit): keep actor alive while workflow waits for queue messages --- .../packages/rivetkit/src/workflow/driver.ts | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/rivetkit-typescript/packages/rivetkit/src/workflow/driver.ts b/rivetkit-typescript/packages/rivetkit/src/workflow/driver.ts index 6bad4e4cd1..e1f4bd4f26 100644 --- a/rivetkit-typescript/packages/rivetkit/src/workflow/driver.ts +++ b/rivetkit-typescript/packages/rivetkit/src/workflow/driver.ts @@ -223,9 +223,11 @@ export class ActorWorkflowDriver implements EngineDriver { messageNames: string[], abortSignal: AbortSignal, ): Promise { - return this.#actor.queueManager.waitForNames( - messageNames.length > 0 ? messageNames : undefined, - abortSignal, + return this.#runCtx.internalKeepAwake( + this.#actor.queueManager.waitForNames( + messageNames.length > 0 ? messageNames : undefined, + abortSignal, + ), ); } }