@arhen/pi-core-subagent 1.0.4 → 1.0.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arhen/pi-core-subagent",
-  "version": "1.0.4",
+  "version": "1.0.6",
   "type": "module",
   "description": "pi extension: fast in-process subagents with single/parallel/chain, background runs, intercom and agent-to-agent mailbox. Leader defines agents inline.",
   "license": "MIT",

package/src/index.ts

@@ -253,7 +253,7 @@ function compactLines(run: RunSnapshot): string[] {
  */
 class SubagentsWidget implements Component {
 	constructor(
-		private readonly getRun: () => RunSnapshot | undefined,
+		private readonly getRuns: () => RunSnapshot[],
 		private readonly theme: Theme,
 	) {}
 
@@ -262,28 +262,33 @@ class SubagentsWidget implements Component {
 	}
 
 	render(width: number): string[] {
-		const run = this.getRun();
-		if (!run || run.tasks.length === 0) return [];
-		const done = run.tasks.filter((t) => TERMINAL.includes(t.status)).length;
-		const active = !TERMINAL.includes(run.status);
-		const head = active ? "accent" : "dim";
-		const lines = [truncateToWidth(`${this.theme.fg(head, active ? "●" : "○")} ${this.theme.fg(head, `Subagents (${done}/${run.tasks.length})`)}`, width, "…")];
-		const allDone = TERMINAL.includes(run.status);
-		const visible = run.tasks.slice(0, MAX_TASKS);
-		visible.forEach((task, i) => {
-			const last = i === visible.length - 1 && run.tasks.length <= MAX_TASKS;
-			const conn = this.theme.fg("dim", last ? "└─" : "├─");
-			const activity =
-				!TERMINAL.includes(task.status) && task.lastActivity
-					? `${this.theme.fg("dim", `→ ${task.lastActivity}`)} · `
-					: "";
-			// Completed run: dim everything except the agent name.
-			const line = allDone
-				? `${this.theme.fg("dim", `${statusIcon(task.status)} `)}${task.agent} ${this.theme.fg("dim", `· ${taskStatsWithUsage(task)} · ${taskTimer(task)}`)}`
-				: `${statusIcon(task.status)} ${task.agent} · ${activity}${taskStatsWithUsage(task)} · ${taskTimer(task)}`;
-			lines.push(truncateToWidth(`${conn} ${line}`, width, "…"));
+		const runs = this.getRuns();
+		if (runs.length === 0) return [];
+		const lines: string[] = [];
+		runs.forEach((run, ri) => {
+			if (run.tasks.length === 0) return;
+			if (ri > 0) lines.push("");
+			const done = run.tasks.filter((t) => TERMINAL.includes(t.status)).length;
+			const active = !TERMINAL.includes(run.status);
+			const head = active ? "accent" : "dim";
+			lines.push(truncateToWidth(`${this.theme.fg(head, active ? "●" : "○")} ${this.theme.fg(head, `Subagents (${done}/${run.tasks.length})`)}`, width, "…"));
+			const allDone = TERMINAL.includes(run.status);
+			const visible = run.tasks.slice(0, MAX_TASKS);
+			visible.forEach((task, i) => {
+				const last = i === visible.length - 1 && run.tasks.length <= MAX_TASKS;
+				const conn = this.theme.fg("dim", last ? "└─" : "├─");
+				const activity =
+					!TERMINAL.includes(task.status) && task.lastActivity
+						? `${this.theme.fg("dim", `→ ${task.lastActivity}`)} · `
+						: "";
+				// Finished run: dim everything except the agent name.
+				const line = allDone
+					? `${this.theme.fg("dim", `${statusIcon(task.status)} `)}${task.agent} ${this.theme.fg("dim", `· ${taskStatsWithUsage(task)} · ${taskTimer(task)}`)}`
+					: `${statusIcon(task.status)} ${task.agent} · ${activity}${taskStatsWithUsage(task)} · ${taskTimer(task)}`;
+				lines.push(truncateToWidth(`${conn} ${line}`, width, "…"));
+			});
+			if (run.tasks.length > MAX_TASKS) lines.push(`${this.theme.fg("dim", "└─")} ${this.theme.fg("dim", `+${run.tasks.length - MAX_TASKS} more`)}`);
 		});
-		if (run.tasks.length > MAX_TASKS) lines.push(`${this.theme.fg("dim", "└─")} ${this.theme.fg("dim", `+${run.tasks.length - MAX_TASKS} more`)}`);
 		return lines;
 	}
 }
@@ -383,7 +388,7 @@ class SubagentManager {
 	private mailboxes: Mailbox = createMailbox();
 	private runControllers = new Map<string, AbortController>();
 	private widgetTimer: ReturnType<typeof setTimeout> | undefined;
-	private widgetRun: RunSnapshot | undefined;
+	private widgetRuns: RunSnapshot[] = [];
 
 	turnActivity = false;
 
@@ -399,7 +404,7 @@ class SubagentManager {
 
 	/** Hide the widget. */
 	clearWidget(ctx: ExtensionContext): void {
-		this.widgetRun = undefined;
+		this.widgetRuns = [];
 		this.widgetTui = null;
 		if (ctx.hasUI) {
 			try {
@@ -425,7 +430,7 @@ class SubagentManager {
 		this.widgetTui = null; // force re-registration on the next session
 		if (this.widgetTimer) clearTimeout(this.widgetTimer);
 		this.widgetTimer = undefined;
-		this.widgetRun = undefined;
+		this.widgetRuns = [];
 	}
 
 	// ── persistence (sidecar per parent session) ────────────────────────
@@ -508,19 +513,27 @@ class SubagentManager {
 	// The component self-animates the spinner via its own 100ms interval;
 	// scheduleWidget just throttles status changes into requestRender calls.
 	private widgetTui: TUI | null = null;
+	/** Upsert a run into the widget's visible set (all runs, not just the latest). */
+	private upsertWidgetRun(run: RunSnapshot | undefined): void {
+		if (!run) return;
+		const idx = this.widgetRuns.findIndex((r) => r.id === run.id);
+		if (idx >= 0) this.widgetRuns[idx] = run;
+		else this.widgetRuns.push(run);
+	}
 	private scheduleWidget(run: RunSnapshot | undefined, ctx?: ExtensionContext, onUpdate?: (partial: any) => void): void {
-		if (run) this.widgetRun = run;
-		if (this.widgetTimer || !this.widgetRun) return;
+		this.upsertWidgetRun(run);
+		if (this.widgetTimer || this.widgetRuns.length === 0) return;
 		this.widgetTimer = setTimeout(() => {
 			this.widgetTimer = undefined;
-			const target = this.widgetRun; // read at fire: never render a stale run
-			if (!target) return;
+			if (this.widgetRuns.length === 0) return;
+			const targets = [...this.widgetRuns]; // read at fire: never render stale runs
 			if (ctx?.hasUI) {
-				ctx.ui.setStatus("subagents", `subagents: ${target.tasks.filter((t) => !TERMINAL.includes(t.status)).length} running`);
+				const live = targets.reduce((n, r) => n + r.tasks.filter((t) => !TERMINAL.includes(t.status)).length, 0);
+				ctx.ui.setStatus("subagents", `subagents: ${live} running`);
 				this.ensureWidget(ctx);
 				this.widgetTui?.requestRender();
 			}
-			onUpdate?.({ content: [{ type: "text", text: compactLines(target).join("\n") }] });
+			onUpdate?.({ content: [{ type: "text", text: targets.flatMap(compactLines).join("\n") }] });
 		}, WIDGET_THROTTLE_MS);
 	}
 	private flushWidget(ctx?: ExtensionContext, onUpdate?: (partial: any) => void): void {
@@ -528,14 +541,12 @@ class SubagentManager {
 			clearTimeout(this.widgetTimer);
 			this.widgetTimer = undefined;
 		}
-		const run = this.widgetRun;
-		if (!run) return;
+		if (this.widgetRuns.length === 0) return;
 		if (ctx?.hasUI) {
-			ctx.ui.setStatus("subagents", `subagents: ${run.status}`);
 			this.ensureWidget(ctx);
 			this.widgetTui?.requestRender();
 		}
-		onUpdate?.({ content: [{ type: "text", text: compactLines(run).join("\n") }] });
+		onUpdate?.({ content: [{ type: "text", text: this.widgetRuns.flatMap(compactLines).join("\n") }] });
 	}
 	private ensureWidget(ctx: ExtensionContext): void {
 		if (this.widgetTui !== null || !ctx.hasUI) return;
@@ -543,7 +554,7 @@ class SubagentManager {
 			"subagents",
 			(tui, theme) => {
 				this.widgetTui = tui;
-				return new SubagentsWidget(() => this.widgetRun, theme);
+				return new SubagentsWidget(() => [...this.widgetRuns], theme);
 			},
 			{ placement: "aboveEditor" },
 		);