@selesai/code 0.8.2 → 0.8.4

package/dist/extensions/pi-subagents/src/runs/background/async-job-tracker.ts

@@ -15,14 +15,15 @@ import {
 	SUBAGENT_CONTROL_INTERCOM_EVENT,
 	SUBAGENT_STEERING_NOTICE_EVENT,
 } from "../../shared/types.ts";
-import { readStatus } from "../../shared/utils.ts";
+import { readStatus, resolveWatchPath } from "../../shared/utils.ts";
 import { normalizeParallelGroups } from "./parallel-groups.ts";
 import { reconcileAsyncRun, reconcileNestedAsyncDescendants } from "./stale-run-reconciler.ts";
-import { hasLiveNestedDescendants, updateAsyncJobNestedProjection } from "../shared/nested-events.ts";
+import { findNestedRouteForRootId, hasLiveNestedDescendants, updateAsyncJobNestedProjection } from "../shared/nested-events.ts";
 import { listAsyncRuns, type AsyncRunSummary } from "./async-status.ts";
 
 interface AsyncJobTrackerOptions {
 	completionRetentionMs?: number;
+	/** Slow safety sweep for liveness repair when filesystem events are missed. */
 	pollIntervalMs?: number;
 	resultsDir?: string;
 	widgetEnabled?: boolean;
@@ -34,6 +35,9 @@ const CONTROL_EVENT_READ_CHUNK_BYTES = 64 * 1024;
 const MAX_CONTROL_EVENT_LINE_BYTES = 1024 * 1024;
 const CONTROL_EVENT_SCAN_WINDOW_BYTES = 2 * 1024 * 1024;
 const MAX_RECENT_FLEET_JOBS = 20;
+const DEFAULT_LIVENESS_INTERVAL_MS = 5000;
+const EVENT_REFRESH_DEBOUNCE_MS = 25;
+const WATCH_ATTACHMENT_RETRY_MS = 100;
 
 function rememberFleetJob(state: SubagentState, job: AsyncJobState): void {
 	state.fleetJobs ??= new Map();
@@ -51,14 +55,21 @@ export function createAsyncJobTracker(pi: Pick<ExtensionAPI, "events">, state: S
 	handleComplete: (data: unknown) => void;
 	resetJobs: (ctx?: ExtensionContext) => void;
 	restoreActiveJobs: (ctx?: ExtensionContext) => void;
+	dispose: () => void;
 } {
 	const completionRetentionMs = options.completionRetentionMs ?? 10000;
-	const pollIntervalMs = options.pollIntervalMs ?? POLL_INTERVAL_MS;
+	const livenessIntervalMs = options.pollIntervalMs ?? DEFAULT_LIVENESS_INTERVAL_MS;
 	const resultsDir = options.resultsDir ?? DIRS.results;
 	const steeringNoticeSeen = new Map<string, number>();
+	const jobWatchers = new Map<string, { watchers: Map<string, fs.FSWatcher>; retryTimer?: ReturnType<typeof setTimeout> }>();
+	const refreshTimers = new Map<string, ReturnType<typeof setTimeout>>();
+	const runningJobIds = new Set<string>();
+	let rootWatcher: fs.FSWatcher | undefined;
+	let nextLivenessAt = Date.now() + livenessIntervalMs;
 	const rerenderWidget = (ctx: ExtensionContext, jobs = Array.from(state.asyncJobs.values())) => {
+		if (state.widgetsSuspended) return;
 		renderWidget(ctx, options.widgetEnabled === false ? [] : jobs);
-		ctx.ui.requestRender?.();
+		(ctx.ui as { requestRender?: () => void }).requestRender?.();
 	};
 	const rerenderLastWidget = (jobs = Array.from(state.asyncJobs.values())) => {
 		const ctx = state.lastUiContext;
@@ -73,6 +84,19 @@ export function createAsyncJobTracker(pi: Pick<ExtensionAPI, "events">, state: S
 			throw error;
 		}
 	};
+	const requestLastWidgetRender = () => {
+		const ctx = state.lastUiContext;
+		if (!ctx || state.widgetsSuspended || options.widgetEnabled === false) return;
+		try {
+			if (ctx.hasUI) (ctx.ui as { requestRender?: () => void }).requestRender?.();
+		} catch (error) {
+			if (error instanceof Error && error.message.includes("extension ctx is stale")) {
+				state.lastUiContext = null;
+				return;
+			}
+			throw error;
+		}
+	};
 	const refreshWidget = (ctx: ExtensionContext) => rerenderWidget(ctx);
 	const restoredControlEventCursor = (asyncDir: string) => {
 		try {
@@ -82,6 +106,14 @@ export function createAsyncJobTracker(pi: Pick<ExtensionAPI, "events">, state: S
 			throw error;
 		}
 	};
+	const findNestedRouteForJob = (runId: string) => {
+		try {
+			return findNestedRouteForRootId(runId);
+		} catch (error) {
+			console.error(`Failed to resolve nested event route for '${runId}':`, error);
+			return undefined;
+		}
+	};
 	const summaryToJob = (run: AsyncRunSummary): AsyncJobState => {
 		const groups = normalizeParallelGroups(run.parallelGroups, run.steps.length, run.chainStepCount ?? run.steps.length);
 		const activeGroup = run.currentStep !== undefined
@@ -130,6 +162,7 @@ export function createAsyncJobTracker(pi: Pick<ExtensionAPI, "events">, state: S
 			totalTokens: run.totalTokens,
 			sessionFile: run.sessionFile,
 			controlEventCursor: restoredControlEventCursor(run.asyncDir),
+			nestedRoute: findNestedRouteForJob(run.id),
 			nestedChildren: run.nestedChildren,
 			parentWorkflowRunId: run.parentWorkflowRunId,
 			workflowKey: run.workflowKey,
@@ -146,6 +179,7 @@ export function createAsyncJobTracker(pi: Pick<ExtensionAPI, "events">, state: S
 		cancelCleanup(asyncId);
 		const timer = setTimeout(() => {
 			state.cleanupTimers.delete(asyncId);
+			closeJobWatcher(asyncId);
 			state.asyncJobs.delete(asyncId);
 			rerenderLastWidget();
 		}, completionRetentionMs);
@@ -268,139 +302,245 @@ export function createAsyncJobTracker(pi: Pick<ExtensionAPI, "events">, state: S
 		}
 	};
 
+	const closeJobWatcher = (asyncId: string) => {
+		const watched = jobWatchers.get(asyncId);
+		if (watched?.retryTimer) clearTimeout(watched.retryTimer);
+		for (const watcher of watched?.watchers.values() ?? []) watcher.close();
+		jobWatchers.delete(asyncId);
+		const timer = refreshTimers.get(asyncId);
+		if (timer) clearTimeout(timer);
+		refreshTimers.delete(asyncId);
+		runningJobIds.delete(asyncId);
+	};
+
+	const refreshJob = (job: AsyncJobState): boolean => {
+		const widgetStateBefore = widgetRenderKey(job);
+		let nestedRefreshFailed = false;
+		const refreshNestedProjection = () => {
+			try {
+				updateAsyncJobNestedProjection(job);
+			} catch (error) {
+				nestedRefreshFailed = true;
+				console.error(`Failed to refresh nested async descendants for '${job.asyncDir}':`, error);
+			}
+		};
+		try {
+			emitNewControlEvents(job);
+			try {
+				if (job.nestedRoute) reconcileNestedAsyncDescendants(job.nestedRoute, { resultsDir, kill: options.kill, now: options.now });
+			} catch (error) {
+				nestedRefreshFailed = true;
+				console.error(`Failed to refresh nested async descendants for '${job.asyncDir}':`, error);
+			}
+			refreshNestedProjection();
+			const reconciliation = reconcileAsyncRun(job.asyncDir, {
+				resultsDir,
+				kill: options.kill,
+				now: options.now,
+				startedRun: {
+					runId: job.asyncId,
+					pid: job.pid,
+					sessionId: job.sessionId,
+					mode: job.mode,
+					agents: job.agents,
+					chainStepCount: job.chainStepCount,
+					parallelGroups: job.parallelGroups,
+					startedAt: job.startedAt,
+					sessionFile: job.sessionFile,
+				},
+			});
+			const status = reconciliation.status ?? readStatus(job.asyncDir);
+			if (status) {
+				const previousStatus = job.status;
+				job.status = status.state;
+				if (job.status === "running") runningJobIds.add(job.asyncId);
+				else runningJobIds.delete(job.asyncId);
+				if (job.status !== "complete" && job.status !== "failed" && job.status !== "paused" && job.status !== "stopped") cancelCleanup(job.asyncId);
+				job.sessionId = status.sessionId ?? job.sessionId;
+				job.activityState = status.activityState;
+				job.lastActivityAt = status.lastActivityAt ?? job.lastActivityAt;
+				job.currentTool = status.currentTool;
+				job.currentToolStartedAt = status.currentToolStartedAt;
+				job.currentPath = status.currentPath;
+				job.turnCount = status.turnCount ?? job.turnCount;
+				job.toolCount = status.toolCount ?? job.toolCount;
+				job.steering = status.steering ?? job.steering;
+				job.mode = status.mode;
+				job.parentWorkflowRunId = status.parentWorkflowRunId ?? job.parentWorkflowRunId;
+				job.workflowKey = status.workflowKey ?? job.workflowKey;
+				job.workflow = status.workflow ?? job.workflow;
+				job.currentStep = status.currentStep ?? job.currentStep;
+				job.chainStepCount = status.chainStepCount ?? job.chainStepCount;
+				job.startedAt = status.startedAt ?? job.startedAt;
+				if (status.lastUpdate !== undefined) job.updatedAt = status.lastUpdate;
+				if (status.steps?.length) {
+					const groups = normalizeParallelGroups(status.parallelGroups, status.steps.length, status.chainStepCount ?? status.steps.length);
+					job.parallelGroups = groups.length ? groups : job.parallelGroups;
+					job.hasParallelGroups = groups.length > 0 || job.hasParallelGroups;
+					const activeGroup = status.currentStep !== undefined
+						? groups.find((group) => status.currentStep! >= group.start && status.currentStep! < group.start + group.count)
+						: undefined;
+					const visibleSteps = activeGroup
+						? status.steps.slice(activeGroup.start, activeGroup.start + activeGroup.count).map((step, index) => ({ ...step, index: activeGroup.start + index }))
+						: status.steps.map((step, index) => ({ ...step, index }));
+					job.activeParallelGroup = Boolean(activeGroup);
+					job.agents = visibleSteps.map((step) => step.agent);
+					job.steps = visibleSteps;
+					refreshNestedProjection();
+					job.stepsTotal = visibleSteps.length;
+					job.runningSteps = visibleSteps.filter((step) => step.status === "running").length;
+					job.completedSteps = visibleSteps.filter((step) => step.status === "complete" || step.status === "completed").length;
+					if (status.state === "complete") job.completedSteps = visibleSteps.length;
+				}
+				job.sessionDir = status.sessionDir ?? job.sessionDir;
+				job.outputFile = status.outputFile ?? job.outputFile;
+				job.totalTokens = status.totalTokens ?? job.totalTokens;
+				job.timeoutMs = status.timeoutMs ?? job.timeoutMs;
+				job.deadlineAt = status.deadlineAt ?? job.deadlineAt;
+				job.timedOut = status.timedOut ?? job.timedOut;
+				job.stopped = status.stopped ?? job.stopped;
+				job.turnBudget = status.turnBudget ?? job.turnBudget;
+				job.turnBudgetExceeded = status.turnBudgetExceeded ?? job.turnBudgetExceeded;
+				job.wrapUpRequested = status.wrapUpRequested ?? job.wrapUpRequested;
+				job.sessionFile = status.sessionFile ?? job.sessionFile;
+				if (job.status === "complete" || job.status === "failed" || job.status === "paused" || job.status === "stopped") {
+					rememberFleetJob(state, job);
+					if (!nestedRefreshFailed && !hasLiveNestedDescendants(job.nestedChildren) && (previousStatus !== job.status || !state.cleanupTimers.has(job.asyncId))) {
+						scheduleCleanup(job.asyncId);
+					}
+				}
+				return widgetRenderKey(job) !== widgetStateBefore;
+			}
+			if (job.status === "queued") {
+				job.status = "running";
+				job.updatedAt = Date.now();
+				runningJobIds.add(job.asyncId);
+			}
+		} catch (error) {
+			if (job.status !== "failed") {
+				console.error(`Failed to read async status for '${job.asyncDir}':`, error);
+				job.status = "failed";
+				job.updatedAt = Date.now();
+			}
+			runningJobIds.delete(job.asyncId);
+			rememberFleetJob(state, job);
+			if (!hasLiveNestedDescendants(job.nestedChildren) && !state.cleanupTimers.has(job.asyncId)) scheduleCleanup(job.asyncId);
+		}
+		return widgetRenderKey(job) !== widgetStateBefore;
+	};
+
+	const scheduleJobRefresh = (asyncId: string, delayMs = EVENT_REFRESH_DEBOUNCE_MS) => {
+		if (refreshTimers.has(asyncId)) return;
+		const timer = setTimeout(() => {
+			refreshTimers.delete(asyncId);
+			const job = state.asyncJobs.get(asyncId);
+			if (job && refreshJob(job)) rerenderLastWidget();
+		}, delayMs);
+		timer.unref?.();
+		refreshTimers.set(asyncId, timer);
+	};
+
+	const watchJob = (job: AsyncJobState) => {
+		const watched = jobWatchers.get(job.asyncId) ?? { watchers: new Map<string, fs.FSWatcher>() };
+		const active = job.status === "queued" || job.status === "running";
+		const watchPath = (watchPath: string): boolean => {
+			if (watched.watchers.has(watchPath)) return true;
+			const nestedEventSink = job.nestedRoute?.eventSink === watchPath;
+			let watcher: fs.FSWatcher;
+			try {
+				watcher = fs.watch(resolveWatchPath(watchPath), (_event, file) => {
+					const rawFileName = file?.toString();
+					const fileName = rawFileName ?? path.basename(watchPath);
+					const nestedEventFile = nestedEventSink && (!rawFileName || rawFileName.endsWith(".json") || rawFileName.endsWith(".jsonl"));
+					if (fileName === "status.json" || fileName === "events.jsonl" || nestedEventFile) {
+						scheduleJobRefresh(job.asyncId);
+					}
+					if (watchPath !== job.asyncDir && !nestedEventSink) {
+						watcher.close();
+						watched.watchers.delete(watchPath);
+						watchJob(job);
+					}
+				});
+			} catch {
+				return false;
+			}
+			watcher.on("error", () => {
+				watcher.close();
+				watched.watchers.delete(watchPath);
+				if (watchPath.endsWith("status.json") || nestedEventSink) watchJob(job);
+			});
+			watcher.unref?.();
+			watched.watchers.set(watchPath, watcher);
+			return true;
+		};
+		watchPath(job.asyncDir);
+		const statusPath = path.join(job.asyncDir, "status.json");
+		const hadStatusWatcher = watched.watchers.has(statusPath);
+		const statusWatched = watchPath(statusPath);
+		if (statusWatched && !hadStatusWatcher) scheduleJobRefresh(job.asyncId);
+		watchPath(path.join(job.asyncDir, "events.jsonl"));
+		if (job.nestedRoute) watchPath(job.nestedRoute.eventSink);
+		if (!statusWatched && active && !watched.retryTimer) {
+			watched.retryTimer = setTimeout(() => {
+				watched.retryTimer = undefined;
+				const current = state.asyncJobs.get(job.asyncId);
+				if (current) watchJob(current);
+			}, WATCH_ATTACHMENT_RETRY_MS);
+			watched.retryTimer.unref?.();
+		} else if (statusWatched && watched.retryTimer) {
+			clearTimeout(watched.retryTimer);
+			watched.retryTimer = undefined;
+		}
+		if (watched.watchers.size > 0 || watched.retryTimer) jobWatchers.set(job.asyncId, watched);
+		else jobWatchers.delete(job.asyncId);
+	};
+
+	const watchAsyncRoot = () => {
+		if (rootWatcher) return;
+		try {
+			rootWatcher = fs.watch(resolveWatchPath(asyncDirRoot), (_event, file) => {
+				const runDirName = file?.toString();
+				for (const job of state.asyncJobs.values()) {
+					if (jobWatchers.has(job.asyncId) || (runDirName && path.basename(job.asyncDir) !== runDirName)) continue;
+					watchJob(job);
+					if (jobWatchers.has(job.asyncId)) scheduleJobRefresh(job.asyncId);
+				}
+			});
+			rootWatcher.on("error", () => {
+				rootWatcher?.close();
+				rootWatcher = undefined;
+			});
+			rootWatcher.unref?.();
+		} catch {
+			// The liveness sweep retries if the async root is not available yet.
+		}
+	};
+
 	const ensurePoller = () => {
+		watchAsyncRoot();
 		if (state.poller) return;
+		nextLivenessAt = Date.now() + livenessIntervalMs;
 		state.poller = setInterval(() => {
 			if (state.asyncJobs.size === 0) {
 				rerenderLastWidget([]);
-				if (state.poller) {
-					clearInterval(state.poller);
-					state.poller = null;
-				}
+				if (state.poller) clearInterval(state.poller);
+				state.poller = null;
+				rootWatcher?.close();
+				rootWatcher = undefined;
 				return;
 			}
-
-			let widgetChanged = false;
-			for (const job of state.asyncJobs.values()) {
-				const widgetStateBefore = widgetRenderKey(job);
-				let nestedRefreshFailed = false;
-				const refreshNestedProjection = () => {
-					try {
-						updateAsyncJobNestedProjection(job);
-					} catch (error) {
-						nestedRefreshFailed = true;
-						console.error(`Failed to refresh nested async descendants for '${job.asyncDir}':`, error);
-					}
-				};
-				const reconcileNestedDescendants = () => {
-					try {
-						if (job.nestedRoute) reconcileNestedAsyncDescendants(job.nestedRoute, { resultsDir, kill: options.kill, now: options.now });
-					} catch (error) {
-						nestedRefreshFailed = true;
-						console.error(`Failed to refresh nested async descendants for '${job.asyncDir}':`, error);
-					}
-					refreshNestedProjection();
-				};
-				try {
-					emitNewControlEvents(job);
-					reconcileNestedDescendants();
-					const reconciliation = reconcileAsyncRun(job.asyncDir, {
-						resultsDir,
-						kill: options.kill,
-						now: options.now,
-						startedRun: {
-							runId: job.asyncId,
-							pid: job.pid,
-							sessionId: job.sessionId,
-							mode: job.mode,
-							agents: job.agents,
-							chainStepCount: job.chainStepCount,
-							parallelGroups: job.parallelGroups,
-							startedAt: job.startedAt,
-							sessionFile: job.sessionFile,
-						},
-					});
-					const status = reconciliation.status ?? readStatus(job.asyncDir);
-					if (status) {
-						const previousStatus = job.status;
-						job.status = status.state;
-						if (job.status !== "complete" && job.status !== "failed" && job.status !== "paused" && job.status !== "stopped") cancelCleanup(job.asyncId);
-						job.sessionId = status.sessionId ?? job.sessionId;
-						job.activityState = status.activityState;
-						job.lastActivityAt = status.lastActivityAt ?? job.lastActivityAt;
-						job.currentTool = status.currentTool;
-						job.currentToolStartedAt = status.currentToolStartedAt;
-						job.currentPath = status.currentPath;
-						job.turnCount = status.turnCount ?? job.turnCount;
-						job.toolCount = status.toolCount ?? job.toolCount;
-						job.steering = status.steering ?? job.steering;
-						job.mode = status.mode;
-						job.parentWorkflowRunId = status.parentWorkflowRunId ?? job.parentWorkflowRunId;
-						job.workflowKey = status.workflowKey ?? job.workflowKey;
-						job.workflow = status.workflow ?? job.workflow;
-						job.currentStep = status.currentStep ?? job.currentStep;
-						job.chainStepCount = status.chainStepCount ?? job.chainStepCount;
-						job.startedAt = status.startedAt ?? job.startedAt;
-						if (status.lastUpdate !== undefined) job.updatedAt = status.lastUpdate;
-						if (status.steps?.length) {
-							const groups = normalizeParallelGroups(status.parallelGroups, status.steps.length, status.chainStepCount ?? status.steps.length);
-							job.parallelGroups = groups.length ? groups : job.parallelGroups;
-							job.hasParallelGroups = groups.length > 0 || job.hasParallelGroups;
-							const activeGroup = status.currentStep !== undefined
-								? groups.find((group) => status.currentStep! >= group.start && status.currentStep! < group.start + group.count)
-								: undefined;
-							const visibleSteps = activeGroup
-								? status.steps.slice(activeGroup.start, activeGroup.start + activeGroup.count).map((step, index) => ({ ...step, index: activeGroup.start + index }))
-								: status.steps.map((step, index) => ({ ...step, index }));
-							job.activeParallelGroup = Boolean(activeGroup);
-							job.agents = visibleSteps.map((step) => step.agent);
-							job.steps = visibleSteps;
-							refreshNestedProjection();
-							job.stepsTotal = visibleSteps.length;
-							job.runningSteps = visibleSteps.filter((step) => step.status === "running").length;
-							job.completedSteps = visibleSteps.filter((step) => step.status === "complete" || step.status === "completed").length;
-							if (status.state === "complete") job.completedSteps = visibleSteps.length;
-						}
-						job.sessionDir = status.sessionDir ?? job.sessionDir;
-						job.outputFile = status.outputFile ?? job.outputFile;
-						job.totalTokens = status.totalTokens ?? job.totalTokens;
-						job.timeoutMs = status.timeoutMs ?? job.timeoutMs;
-						job.deadlineAt = status.deadlineAt ?? job.deadlineAt;
-						job.timedOut = status.timedOut ?? job.timedOut;
-						job.stopped = status.stopped ?? job.stopped;
-						job.turnBudget = status.turnBudget ?? job.turnBudget;
-						job.turnBudgetExceeded = status.turnBudgetExceeded ?? job.turnBudgetExceeded;
-						job.wrapUpRequested = status.wrapUpRequested ?? job.wrapUpRequested;
-						job.sessionFile = status.sessionFile ?? job.sessionFile;
-						if (job.status === "complete" || job.status === "failed" || job.status === "paused" || job.status === "stopped") {
-							rememberFleetJob(state, job);
-							if (!nestedRefreshFailed && !hasLiveNestedDescendants(job.nestedChildren) && (previousStatus !== job.status || !state.cleanupTimers.has(job.asyncId))) {
-								scheduleCleanup(job.asyncId);
-							}
-						}
-						if (widgetRenderKey(job) !== widgetStateBefore) widgetChanged = true;
-						continue;
-					}
-					if (job.status === "queued") {
-						job.status = "running";
-						job.updatedAt = Date.now();
-					}
-				} catch (error) {
-					if (job.status !== "failed") {
-						console.error(`Failed to read async status for '${job.asyncDir}':`, error);
-						job.status = "failed";
-						job.updatedAt = Date.now();
-					}
-					rememberFleetJob(state, job);
-					if (!hasLiveNestedDescendants(job.nestedChildren) && !state.cleanupTimers.has(job.asyncId)) {
-						scheduleCleanup(job.asyncId);
-					}
+			const now = Date.now();
+			if (now >= nextLivenessAt) {
+				nextLivenessAt = now + livenessIntervalMs;
+				let widgetChanged = false;
+				for (const job of state.asyncJobs.values()) {
+					watchJob(job);
+					if (refreshJob(job)) widgetChanged = true;
 				}
-				if (widgetRenderKey(job) !== widgetStateBefore) widgetChanged = true;
+				if (widgetChanged) rerenderLastWidget();
 			}
-
-			if (widgetChanged) rerenderLastWidget();
-		}, pollIntervalMs);
+			if (runningJobIds.size > 0) requestLastWidgetRender();
+		}, Math.min(POLL_INTERVAL_MS, livenessIntervalMs));
 		state.poller.unref?.();
 	};
 
@@ -442,7 +582,10 @@ export function createAsyncJobTracker(pi: Pick<ExtensionAPI, "events">, state: S
 			workflowKey: info.workflowKey,
 			controlEventCursor: 0,
 		});
-		rememberFleetJob(state, state.asyncJobs.get(info.id)!);
+		const job = state.asyncJobs.get(info.id)!;
+		rememberFleetJob(state, job);
+		watchJob(job);
+		scheduleJobRefresh(info.id, 0);
 		ensurePoller();
 		rerenderLastWidget();
 	};
@@ -456,9 +599,14 @@ export function createAsyncJobTracker(pi: Pick<ExtensionAPI, "events">, state: S
 		let nestedRefreshFailed = false;
 		if (job) {
 			job.status = result.state ?? (result.success ? "complete" : "failed");
+			runningJobIds.delete(asyncId);
 			job.stopped = result.stopped ?? job.stopped;
 			job.updatedAt = Date.now();
-			if (result.asyncDir) job.asyncDir = result.asyncDir;
+			if (result.asyncDir && result.asyncDir !== job.asyncDir) {
+				closeJobWatcher(asyncId);
+				job.asyncDir = result.asyncDir;
+				watchJob(job);
+			}
 			try {
 				updateAsyncJobNestedProjection(job);
 			} catch (error) {
@@ -471,10 +619,20 @@ export function createAsyncJobTracker(pi: Pick<ExtensionAPI, "events">, state: S
 		if (!nestedRefreshFailed && !hasLiveNestedDescendants(job?.nestedChildren)) scheduleCleanup(asyncId);
 	};
 
+	const dispose = () => {
+		if (state.poller) clearInterval(state.poller);
+		state.poller = null;
+		rootWatcher?.close();
+		rootWatcher = undefined;
+		for (const asyncId of jobWatchers.keys()) closeJobWatcher(asyncId);
+		for (const timer of refreshTimers.values()) clearTimeout(timer);
+		refreshTimers.clear();
+		runningJobIds.clear();
+	};
+
 	const resetJobs = (ctx?: ExtensionContext) => {
-		for (const timer of state.cleanupTimers.values()) {
-			clearTimeout(timer);
-		}
+		dispose();
+		for (const timer of state.cleanupTimers.values()) clearTimeout(timer);
 		state.cleanupTimers.clear();
 		state.asyncJobs.clear();
 		state.fleetJobs?.clear();
@@ -500,12 +658,14 @@ export function createAsyncJobTracker(pi: Pick<ExtensionAPI, "events">, state: S
 		for (const run of runs) {
 			const job = summaryToJob(run);
 			state.asyncJobs.set(run.id, job);
+			if (job.status === "running") runningJobIds.add(job.asyncId);
 			rememberFleetJob(state, job);
+			watchJob(job);
 		}
 		if (runs.length === 0) return;
 		ensurePoller();
 		rerenderLastWidget();
 	};
 
-	return { ensurePoller, refreshWidget, handleStarted, handleComplete, resetJobs, restoreActiveJobs };
+	return { ensurePoller, refreshWidget, handleStarted, handleComplete, resetJobs, restoreActiveJobs, dispose };
 }

package/dist/extensions/pi-subagents/src/runs/background/async-resume.ts

@@ -1,6 +1,6 @@
 import * as fs from "node:fs";
 import * as path from "node:path";
-import { DIRS, type AcceptanceInput, type AsyncStatus, type SteeringRecoveryDescriptor } from "../../shared/types.ts";
+import { DIRS, type AcceptanceInput, type AsyncStatus, type ResolvedTurnBudget, type SteeringRecoveryDescriptor } from "../../shared/types.ts";
 import type { AgentConfig } from "../../agents/agents.ts";
 import { validateAcceptanceInput } from "../shared/acceptance.ts";
 import { validateToolBudgetConfig } from "../shared/tool-budget.ts";
@@ -273,6 +273,23 @@ function normalizeRecoveryAcceptance(value: unknown, descriptorPath: string): Ac
 	return value as AcceptanceInput;
 }
 
+function normalizeRecoveryTurnBudget(value: unknown, descriptorPath: string): ResolvedTurnBudget {
+	if (value && typeof value === "object" && !Array.isArray(value)) {
+		const {
+			outcome: _outcome,
+			turnCount: _turnCount,
+			wrapUpRequestedAtTurn: _wrapUpRequestedAtTurn,
+			terminationDeferredAtTurn: _terminationDeferredAtTurn,
+			exceededAtTurn: _exceededAtTurn,
+			...publicTurnBudget
+		} = value as Record<string, unknown>;
+		value = publicTurnBudget;
+	}
+	const result = resolveTurnBudgetConfig(value, "recoveryDescriptor.initialTurnBudget");
+	if (result.error || !result.turnBudget) throw new Error(`Invalid async recovery descriptor '${descriptorPath}': ${result.error ?? "recoveryDescriptor.initialTurnBudget is invalid."}`);
+	return result.turnBudget;
+}
+
 export function readAsyncRecoveryDescriptor(asyncDir: string | undefined): SteeringRecoveryDescriptor | undefined {
 	if (!asyncDir) return undefined;
 	const descriptorPath = path.join(asyncDir, "recovery-descriptor.json");
@@ -329,8 +346,7 @@ export function readAsyncRecoveryDescriptor(asyncDir: string | undefined): Steer
 	}
 	if (parsed.absoluteDeadlineAt !== undefined && (!Number.isFinite(parsed.absoluteDeadlineAt) || (parsed.absoluteDeadlineAt as number) <= 0)) throw new Error(`Invalid async recovery descriptor '${descriptorPath}': absoluteDeadlineAt must be a positive timestamp.`);
 	if (parsed.initialTurnBudget !== undefined) {
-		const result = resolveTurnBudgetConfig(parsed.initialTurnBudget, "recoveryDescriptor.initialTurnBudget");
-		if (result.error) throw new Error(`Invalid async recovery descriptor '${descriptorPath}': ${result.error}`);
+		parsed.initialTurnBudget = normalizeRecoveryTurnBudget(parsed.initialTurnBudget, descriptorPath);
 	}
 	if (parsed.initialToolBudget !== undefined) {
 		const result = validateToolBudgetConfig(parsed.initialToolBudget, "recoveryDescriptor.initialToolBudget");