fix the new configurable waitUntil timeout

Author: ericallam

packages/cli-v3/src/entryPoints/dev-run-worker.ts

@@ -336,7 +336,7 @@ function resetExecutionEnvironment() {
   // Wait for all streams to finish before completing the run
   waitUntil.register({
     requiresResolving: () => standardRealtimeStreamsManager.hasActiveStreams(),
-    promise: () => standardRealtimeStreamsManager.waitForAllStreams(),
+    promise: (timeoutInMs) => standardRealtimeStreamsManager.waitForAllStreams(timeoutInMs),
   });
 
   log(`[${new Date().toISOString()}] Reset execution environment`);

packages/cli-v3/src/entryPoints/managed-run-worker.ts

@@ -310,7 +310,7 @@ function resetExecutionEnvironment() {
   // Wait for all streams to finish before completing the run
   waitUntil.register({
     requiresResolving: () => standardRealtimeStreamsManager.hasActiveStreams(),
-    promise: () => standardRealtimeStreamsManager.waitForAllStreams(),
+    promise: (timeoutInMs) => standardRealtimeStreamsManager.waitForAllStreams(timeoutInMs),
   });
 
   console.log(`[${new Date().toISOString()}] Reset execution environment`);

packages/core/src/v3/waitUntil/manager.ts

@@ -19,7 +19,7 @@ export class StandardWaitUntilManager implements WaitUntilManager {
     }
 
     const promises = this.promisesRequringResolving.map((p) =>
-      typeof p.promise === "function" ? p.promise() : p.promise
+      typeof p.promise === "function" ? p.promise(this.timeoutInMs) : p.promise
     );
 
     await Promise.race([

packages/core/src/v3/waitUntil/types.ts

@@ -1,6 +1,6 @@
 export type MaybeDeferredPromise = {
   requiresResolving(): boolean;
-  promise: Promise<any> | (() => Promise<any>);
+  promise: Promise<any> | ((timeoutInMs: number) => Promise<any>);
 };
 
 export interface WaitUntilManager {