@arhen/pi-core-subagent 1.1.14 → 1.1.15

package/package.json

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

@@ -1027,6 +1027,22 @@ class SubagentManager {
 		return { run: cloneRun(run), background: true };
 	}
 
+	/** Abort ONE task; siblings keep running. Returns false when unknown or already finished. */
+	cancelTask(runId: string, taskId: string, ctx?: ExtensionContext): boolean {
+		const run = this.runs.get(runId);
+		const task = run?.tasks.find((t) => t.id === taskId);
+		if (!run || !task || TERMINAL.includes(task.status)) return false;
+		// Mark first: runChild's catch reads task.status to classify the outcome as aborted.
+		task.status = "aborted";
+		task.error = task.error || "Canceled from peek";
+		task.endedAt = Date.now();
+		this.liveChildren.get(`${runId}:${taskId}`)?.abort();
+		this.mailboxes.close(`${runId}:${taskId}`);
+		if (ctx) this.flushWidget(run, ctx);
+		this.emit("subagent:task-aborted", { runId, taskId });
+		return true;
+	}
+
 	cancelRun(runId: string): { aborted: number } {
 		const run = this.runs.get(runId);
 		if (!run) return { aborted: 0 };
@@ -1178,13 +1194,24 @@ export default function (pi: ExtensionAPI) {
 			manager
 				.listRuns()
 				.flatMap((run) => run.tasks)
-				.map((task) => ({ agent: task.agent, status: task.status, sessionFile: task.sessionFile, line: taskLine(task) }));
+				.map((task) => ({
+					runId: task.runId,
+					taskId: task.id,
+					agent: task.agent,
+					status: task.status,
+					running: !TERMINAL.includes(task.status),
+					sessionFile: task.sessionFile,
+					line: taskLine(task),
+				}));
 		if (getTasks().length === 0) {
 			ctx.ui.notify("No subagents in this session.", "info");
 			return;
 		}
 		await ctx.ui.custom<void>(
-			(tui, theme, _keybindings, done) => createPeekPane(getTasks, theme, () => tui.requestRender(), () => done(undefined)),
+			(tui, theme, _keybindings, done) =>
+				createPeekPane(getTasks, theme, () => tui.requestRender(), () => done(undefined), (t) => {
+					if (manager.cancelTask(t.runId, t.taskId, ctx)) ctx.ui.notify(`Aborted subagent ${t.agent}.`, "warning");
+				}),
 			{ overlay: true, overlayOptions: { anchor: "center", width: "80%", maxHeight: "70%" } },
 		);
 	};

package/src/peek.ts

@@ -15,8 +15,11 @@ const TAIL_BYTES = 64 * 1024;
 const POLL_MS = 700;
 
 export interface PeekTask {
+	runId: string;
+	taskId: string;
 	agent: string;
 	status: string;
+	running: boolean;
 	sessionFile?: string;
 	line: string; // pre-rendered stats line from the caller
 }
@@ -82,9 +85,16 @@ export interface PeekPane {
  * Build the peek component. `getTasks` is polled live, so the pane keeps
  * updating while agents run.
  */
-export function createPeekPane(getTasks: () => PeekTask[], theme: Theme, requestRender: () => void, close: () => void): PeekPane {
+export function createPeekPane(
+	getTasks: () => PeekTask[],
+	theme: Theme,
+	requestRender: () => void,
+	close: () => void,
+	abort: (task: PeekTask) => void,
+): PeekPane {
 	let selected = 0;
 	let tailing = false;
+	let confirming = false;
 	const timer = setInterval(requestRender, POLL_MS);
 
 	const clamp = (n: number, len: number) => (len === 0 ? 0 : Math.max(0, Math.min(len - 1, n)));
@@ -95,7 +105,7 @@ export function createPeekPane(getTasks: () => PeekTask[], theme: Theme, request
 			selected = clamp(selected, tasks.length);
 			if (tasks.length === 0) return [theme.fg("dim", "No subagents in this session.")];
 			const task = tasks[selected]!;
-			const hint = tailing ? "esc back" : "↑↓ move · enter tail · esc close";
+			const hint = confirming ? theme.fg("error", `abort ${task.agent}? y / n`) : tailing ? "esc back · x abort" : "↑↓ move · enter tail · x abort · esc close";
 			const head = `${theme.fg("accent", theme.bold(tailing ? task.agent : "Subagents"))} ${theme.fg("dim", `(${selected + 1}/${tasks.length}) · ${hint}`)}`;
 			if (!tailing) {
 				return [head, ...tasks.map((t, i) => truncateToWidth(`${i === selected ? theme.fg("accent", "❯ ") : "  "}${t.line}`, width, "…"))];
@@ -105,8 +115,21 @@ export function createPeekPane(getTasks: () => PeekTask[], theme: Theme, request
 			return [head, ...tailLines(task.sessionFile, 18).map((l) => truncateToWidth(`  ${l}`, width, "…"))];
 		},
 		handleInput(data: string): void {
-			const len = getTasks().length;
-			if (matchesKey(data, Key.escape)) {
+			const tasks = getTasks();
+			const len = tasks.length;
+			if (confirming) {
+				// Abort is irreversible, so it always costs a second keystroke.
+				confirming = false;
+				if (data === "y" || data === "Y") {
+					const task = tasks[selected];
+					if (task) abort(task);
+				}
+				requestRender();
+				return;
+			}
+			if (data === "x" || data === "X") {
+				if (tasks[selected]?.running) confirming = true;
+			} else if (matchesKey(data, Key.escape)) {
 				if (tailing) tailing = false;
 				else close();
 			} else if (matchesKey(data, Key.enter) || matchesKey(data, Key.right)) {