diff options
| -rw-r--r-- | pi/agent/extensions/loop-scheduler/index.ts | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/pi/agent/extensions/loop-scheduler/index.ts b/pi/agent/extensions/loop-scheduler/index.ts index 67247b8..cdd30a4 100644 --- a/pi/agent/extensions/loop-scheduler/index.ts +++ b/pi/agent/extensions/loop-scheduler/index.ts @@ -1210,7 +1210,14 @@ export default function loopSchedulerExtension(pi: ExtensionAPI): void { // stuck "Follow-up: ..." in pi's UI. We'd also leak agentBusy=true and // block every subsequent pending job because no further agent_end fires. // Wait for the run to actually finish before draining. - await ctx.waitForIdle(); + // ctx.waitForIdle() is only available on ExtensionCommandContext, not + // ExtensionContext, so we poll isIdle() instead. + let attempts = 0; + const maxAttempts = 600; // ~30s at 50ms each + while (!ctx.isIdle() && attempts < maxAttempts) { + await new Promise((r) => setTimeout(r, 50)); + attempts++; + } agentBusy = false; drainPendingJobs(); }); |