@arhen/pi-core-subagent 1.0.1 → 1.0.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arhen/pi-core-subagent",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "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

@@ -385,8 +385,23 @@ class SubagentManager {
 	private widgetTimer: ReturnType<typeof setTimeout> | undefined;
 	private widgetRun: RunSnapshot | undefined;
 
+	turnActivity = false;
+
 	constructor(private readonly pi: ExtensionAPI) {}
 
+	/** Hide the widget. Call on agent_start when the turn made no subagent calls. */
+	clearWidget(ctx: ExtensionContext): void {
+		this.widgetRun = undefined;
+		this.widgetTui = null;
+		if (ctx.hasUI) {
+			try {
+				ctx.ui.setWidget("subagents", undefined);
+			} catch {
+				/* ignore */
+			}
+		}
+	}
+
 	listRuns(): RunSnapshot[] {
 		return Array.from(this.runs.values()).sort((a, b) => b.createdAt - a.createdAt);
 	}
@@ -812,6 +827,7 @@ class SubagentManager {
 			task.roster = roster;
 		}
 		run.tasks.forEach((t) => (t.runId = run.id));
+		this.turnActivity = true;
 		this.runs.set(run.id, run);
 		this.settlers.set(run.id, () => {});
 		this.runControllers.set(run.id, new AbortController());
@@ -859,6 +875,7 @@ class SubagentManager {
 		run.status = aborted ? "aborted" : failed ? "failed" : "completed";
 		run.endedAt = Date.now();
 		this.flushWidget(ctx, onUpdate);
+		this.clearWidget(ctx); // run is done — drop the widget immediately
 		this.emit("subagent:run-completed", { runId: run.id, status: run.status, run: cloneRun(run), aggregateUsage: run.aggregateUsage });
 		this.settleRun(run.id, run);
 		this.runControllers.delete(run.id);
@@ -1036,6 +1053,11 @@ export default function (pi: ExtensionAPI) {
 		},
 	});
 
+	pi.on("agent_start", (_event, ctx) => {
+		if (!manager.turnActivity) manager.clearWidget(ctx);
+		manager.turnActivity = false;
+	});
+
 	pi.on("session_start", async (_event, ctx) => {
 		await manager.restoreFromSidecar(ctx);
 	});