@dougbots/avenor-pi 0.3.4 → 0.3.6

package/dist/index.js

@@ -90,6 +90,14 @@ function buildCompletionText(run, result) {
 	lines.push(`\nCall \`avenor_events\` with \`run_id: "${run.runId}"\` to review the output, or \`avenor_follow_up\` to iterate.`);
 	return lines.join("\n");
 }
+function buildWaitingText(run, status) {
+	const perm = status.pending_permission;
+	const lines = [`Sub-agent "${run.label}" (${run.agent}) is waiting for input.`];
+	if (perm) lines.push(`Permission request: ${perm.description}`, `To answer: call \`avenor_answer_permission\` with run_id "${run.runId}", option_id "allow_once" | "allow_always" | "deny"`);
+	else if (status.phase_label) lines.push(`Question: ${status.phase_label}`);
+	if (!perm) lines.push(`Use \`avenor_follow_up\` with run_id "${run.runId}" to respond, or \`avenor_status\` to re-check.`);
+	return lines.join("\n");
+}
 function summarizeEvent(evt) {
 	const type = String(evt.type ?? evt.event ?? "");
 	if (type === "tool.call" || type === "tool.use") return `called ${String(evt.tool ?? evt.name ?? type)}`;
@@ -100,31 +108,79 @@ function summarizeEvent(evt) {
 async function src_default(pi) {
 	const trackedRuns = /* @__PURE__ */ new Map();
 	let pollingActive = false;
+	let pollingGeneration = 0;
+	let pollInFlight = null;
 	let sessionCtx = null;
 	async function pollRuns() {
-		let raw;
+		const liveMap = /* @__PURE__ */ new Map();
 		try {
-			raw = await statusTool({});
+			const allLive = await statusTool({});
+			const liveList = Array.isArray(allLive) ? allLive : [allLive];
+			for (const result of liveList) liveMap.set(result.run_id, result);
 		} catch (err) {
-			console.error("avenor pollRuns: statusTool failed", err);
-			return [];
+			console.error("avenor pollRuns: failed to list singleton runs", err);
 		}
-		const results = Array.isArray(raw) ? raw : [raw];
 		const entries = [];
-		for (const r of results) {
-			const run = trackedRuns.get(r.run_id);
-			if (run) run.lastStatus = r;
-			entries.push({
-				runId: r.run_id,
-				label: r.label,
-				status: r.status,
-				phase: r.phase,
-				phaseLabel: r.phase_label,
-				agent: trackedRuns.get(r.run_id)?.agent ?? "unknown",
-				pendingPermission: !!r.pending_permission,
-				permissionDescription: r.pending_permission?.description
-			});
+		for (const run of trackedRuns.values()) {
+			const live = liveMap.get(run.runId);
+			if (live) {
+				liveMap.delete(live.run_id);
+				run.lastStatus = live;
+				entries.push({
+					runId: run.runId,
+					label: run.label,
+					status: live.status,
+					phase: live.phase,
+					phaseLabel: live.phase_label,
+					agent: run.agent,
+					pendingPermission: !!live.pending_permission,
+					permissionDescription: live.pending_permission?.description
+				});
+				continue;
+			}
+			try {
+				const raw = await statusTool({
+					runId: run.runId,
+					supervisorId: run.supervisorId
+				});
+				const result = Array.isArray(raw) ? raw[0] : raw;
+				if (!result) throw new Error("status response was empty");
+				run.lastStatus = result;
+				entries.push({
+					runId: run.runId,
+					label: run.label,
+					status: result.status,
+					phase: result.phase,
+					phaseLabel: result.phase_label,
+					agent: run.agent,
+					pendingPermission: !!result.pending_permission,
+					permissionDescription: result.pending_permission?.description
+				});
+			} catch (err) {
+				console.error(`avenor pollRuns: statusTool failed for ${run.runId}`, err);
+				const previous = run.lastStatus;
+				entries.push({
+					runId: run.runId,
+					label: run.label,
+					status: previous?.status ?? "unavailable",
+					phase: previous?.phase,
+					phaseLabel: previous?.phase_label,
+					agent: run.agent,
+					pendingPermission: !!previous?.pending_permission,
+					permissionDescription: previous?.pending_permission?.description
+				});
+			}
 		}
+		for (const live of liveMap.values()) entries.push({
+			runId: live.run_id,
+			label: live.label,
+			status: live.status,
+			phase: live.phase,
+			phaseLabel: live.phase_label,
+			agent: "unknown",
+			pendingPermission: !!live.pending_permission,
+			permissionDescription: live.pending_permission?.description
+		});
 		return entries;
 	}
 	function renderStatusLines(entries) {
@@ -152,32 +208,77 @@ async function src_default(pi) {
 	function startPolling() {
 		if (pollingActive) return;
 		pollingActive = true;
+		const generation = ++pollingGeneration;
 		const tick = async () => {
+			if (!pollingActive || generation !== pollingGeneration) return;
 			const prevStatuses = /* @__PURE__ */ new Map();
 			for (const [id, run] of trackedRuns) prevStatuses.set(id, run.lastStatus?.status);
-			const entries = await pollRuns();
+			const currentPoll = pollRuns();
+			pollInFlight = currentPoll;
+			const entries = await currentPoll.finally(() => {
+				if (pollInFlight === currentPoll) pollInFlight = null;
+			});
+			if (!pollingActive || generation !== pollingGeneration) return;
 			updateWidget(entries);
-			for (const entry of entries) if (TERMINAL_STATUSES.has(entry.status)) {
-				if (trackedRuns.get(entry.runId)) {
+			for (const entry of entries) {
+				const run = trackedRuns.get(entry.runId);
+				if (!run) continue;
+				if (TERMINAL_STATUSES.has(entry.status)) {
 					const prevStatus = prevStatuses.get(entry.runId);
-					if (prevStatus && !TERMINAL_STATUSES.has(prevStatus)) sessionCtx?.ui.notify(`Sub-agent "${entry.label}" finished: ${entry.status}`, entry.status === "done" ? "info" : "warning");
+					if (!prevStatus || !TERMINAL_STATUSES.has(prevStatus)) {
+						sessionCtx?.ui.notify(`Sub-agent "${entry.label}" finished: ${entry.status}`, entry.status === "done" ? "info" : "warning");
+						if (!run.blocking) try {
+							pi.sendUserMessage(buildCompletionText({
+								runId: run.runId,
+								label: run.label
+							}, run.lastStatus), { deliverAs: "followUp" });
+						} catch (err) {
+							console.error("avenor tick: sendUserMessage failed", err);
+						}
+					}
 					trackedRuns.delete(entry.runId);
-				}
+				} else if (entry.pendingPermission && !run.blocking && !run.permissionNotified) {
+					run.permissionNotified = true;
+					try {
+						pi.sendUserMessage(buildWaitingText({
+							runId: run.runId,
+							label: run.label,
+							agent: run.agent
+						}, run.lastStatus), { deliverAs: "followUp" });
+					} catch (err) {
+						console.error("avenor tick: sendUserMessage for permission failed", err);
+					}
+				} else if (!entry.pendingPermission && run.permissionNotified) run.permissionNotified = false;
 			}
-			if (entries.some((e) => !TERMINAL_STATUSES.has(e.status))) setTimeout(tick, POLL_INTERVAL_MS);
+			if (entries.some((e) => !TERMINAL_STATUSES.has(e.status)) && pollingActive && generation === pollingGeneration) setTimeout(() => {
+				tick().catch((err) => {
+					pollingActive = false;
+					console.error("avenor polling loop failed", err);
+				});
+			}, POLL_INTERVAL_MS);
 			else {
 				pollingActive = false;
 				sessionCtx?.ui.setStatus("avenor", "");
 				sessionCtx?.ui.setWidget("avenor-status", void 0);
 			}
 		};
-		tick();
+		tick().catch((err) => {
+			pollingActive = false;
+			console.error("avenor polling loop failed", err);
+		});
+	}
+	async function stopPolling() {
+		pollingActive = false;
+		pollingGeneration++;
+		await pollInFlight?.catch(() => {});
+		sessionCtx?.ui.setStatus("avenor", "");
+		sessionCtx?.ui.setWidget("avenor-status", void 0);
 	}
 	pi.on("session_start", async (_event, ctx) => {
 		sessionCtx = ctx;
 	});
 	pi.on("session_shutdown", async (_event, _ctx) => {
-		pollingActive = false;
+		await stopPolling();
 		sessionCtx = null;
 	});
 	pi.on("before_agent_start", async (event, _ctx) => {
@@ -254,7 +355,9 @@ async function src_default(pi) {
 				agent: params.agent,
 				label,
 				supervisorId: result.supervisor_id,
-				startTime: Date.now()
+				runtimeId: result.runtime_id,
+				startTime: Date.now(),
+				blocking: wait
 			});
 			startPolling();
 			if (!wait) return {
@@ -305,7 +408,7 @@ async function src_default(pi) {
 				});
 				else {
 					let lastAction = "";
-					if (!result.supervisor_id) try {
+					if (!params.supervisor_id) try {
 						const eventsResult = await eventsTool({
 							runId: result.run_id,
 							limit: 3
@@ -327,6 +430,26 @@ async function src_default(pi) {
 						}
 					});
 				}
+				if (status.status === "waiting") {
+					const tracked = trackedRuns.get(result.run_id);
+					if (tracked) tracked.permissionNotified = true;
+					return {
+						content: [{
+							type: "text",
+							text: buildWaitingText({
+								runId: result.run_id,
+								label,
+								agent: params.agent
+							}, status)
+						}],
+						details: {
+							status: "waiting",
+							run_id: result.run_id,
+							session_id: status.session_id,
+							...status.pending_permission && { pending_permission: status.pending_permission }
+						}
+					};
+				}
 				if (TERMINAL_STATUSES.has(status.status)) {
 					trackedRuns.delete(result.run_id);
 					return {
@@ -485,6 +608,8 @@ async function src_default(pi) {
 			force: Type.Optional(Type.Boolean({ description: "Force shutdown instead of graceful" }))
 		}),
 		async execute(_toolCallId, params) {
+			await stopPolling();
+			trackedRuns.clear();
 			const result = await shutdownTool({
 				supervisorId: params.supervisor_id,
 				force: params.force
@@ -606,7 +731,7 @@ async function src_default(pi) {
 			if (!await ctx.ui.confirm("Cancel run", `Cancel "${label}" (${runId.slice(0, 8)})?`)) return;
 			try {
 				const { Supervisor } = await import("@dougbots/avenor-core");
-				await (await Supervisor.get()).getClient().cancel(run?.lastStatus?.runtime_id ?? runId);
+				await (await Supervisor.get()).getClient().cancel(run?.lastStatus?.runtime_id ?? run?.runtimeId ?? runId);
 				ctx.ui.notify(`Cancelled "${label}"`, "info");
 			} catch (err) {
 				ctx.ui.notify(`Failed to cancel: ${err instanceof Error ? err.message : String(err)}`, "error");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dougbots/avenor-pi",
-  "version": "0.3.4",
+  "version": "0.3.6",
   "type": "module",
   "keywords": [
     "pi-package"
@@ -35,7 +35,7 @@
     "dev": "tsdown --watch"
   },
   "dependencies": {
-    "@dougbots/avenor-core": "^0.3.4"
+    "@dougbots/avenor-core": "^0.3.6"
   },
   "peerDependencies": {
     "@earendil-works/pi-coding-agent": "*",