@ryan_nookpi/pi-extension-subagent 0.4.1 → 0.4.3

package/commands.ts

@@ -63,7 +63,12 @@ import { buildMainContextText, makeSubagentSessionFile, wrapTaskWithMainContext
 import { formatStarterPackNotice, offerStarterPackIfEmpty } from "./starter-pack.js";
 import { type SubagentStore, truncateText, updateRunFromResult } from "./store.js";
 import { createSubagentToolExecute } from "./tool-execute.js";
-import { renderSubagentToolCall, renderSubagentToolResult } from "./tool-render.js";
+import {
+	renderListAgentsCall,
+	renderListAgentsResult,
+	renderSubagentToolCall,
+	renderSubagentToolResult,
+} from "./tool-render.js";
 import type { CommandRunState, SingleResult, SubagentDetails } from "./types.js";
 import { ListAgentsParams, SubagentParams } from "./types.js";
 import { updateCommandRunsWidget, type WidgetRenderCtx } from "./widget.js";
@@ -927,6 +932,8 @@ export function registerAll(pi: ExtensionAPI, store: SubagentStore): SubagentReg
 		description:
 			"List available subagent definitions (name, source, model, thinking, tools, description). Useful before planning delegation.",
 		parameters: ListAgentsParams,
+		renderCall: renderListAgentsCall,
+		renderResult: renderListAgentsResult,
 		execute: async (_toolCallId, _params: Record<string, any>, _signal, _onUpdate, ctx) => {
 			const starterPack = await offerStarterPackIfEmpty(ctx);
 			const discovery = starterPack.discovery;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ryan_nookpi/pi-extension-subagent",
-  "version": "0.4.1",
+  "version": "0.4.3",
   "description": "Asynchronous subagent delegation for pi with run, batch, chain, and continuation workflows.",
   "license": "MIT",
   "repository": {

package/tool-render.ts

@@ -7,6 +7,7 @@
 import { getMarkdownTheme, type ThemeColor } from "@earendil-works/pi-coding-agent";
 import { Container, Markdown, Spacer, Text, visibleWidth } from "@earendil-works/pi-tui";
 import { parseSubagentToolCommand } from "./cli.js";
+import { SUBAGENT_STRONG_WAIT_MESSAGE } from "./constants.js";
 import { formatToolCall, formatUsageStats } from "./format.js";
 import { getDisplayItems, getFinalOutput } from "./runner.js";
 import { COLLAPSED_ITEM_COUNT, truncateText } from "./store.js";
@@ -27,10 +28,12 @@ type ToolRenderResult = {
 type ToolRenderArgs = { command?: unknown };
 type ToolCallRenderContext = { expanded: boolean };
 type CompactBlock = { agent: string; task: string };
+type ListAgentsDetails = { agents?: Array<{ name?: unknown }> };
 
 const COMPACT_TASK_PREVIEW_WIDTH = 72;
 const COMPACT_AGENT_WIDTH = 20;
 const COMPACT_MAX_ITEMS = 3;
+const COMPACT_AGENT_LIST_COUNT = 3;
 
 function normalizeTaskPreview(task: string): string {
 	return truncateText(task.replace(/\s+/g, " ").trim(), COMPACT_TASK_PREVIEW_WIDTH);
@@ -110,6 +113,25 @@ function renderCompactLaunch(command: unknown): string | null {
 	return `  run · ${params.agent} · ${contextLabel}\n  ${task}`;
 }
 
+function getToolResultText(result: ToolRenderResult): string {
+	const raw = result.content[0];
+	return (raw?.type === "text" ? raw.text : undefined) ?? "(no output)";
+}
+
+function renderPlainToolResult(fullText: string, expanded: boolean, theme: RenderTheme): Text {
+	if (expanded) return new Text(fullText, 0, 0);
+
+	const waitMessageIndex = fullText.indexOf(SUBAGENT_STRONG_WAIT_MESSAGE);
+	if (waitMessageIndex >= 0) {
+		return new Text(fullText.slice(0, waitMessageIndex).trimEnd(), 0, 0);
+	}
+
+	const firstLine = fullText.split("\n")[0] ?? "";
+	const lineCount = fullText.split("\n").length;
+	const suffix = lineCount > 1 ? theme.fg("muted", ` (+${lineCount - 1} lines)`) : "";
+	return new Text(firstLine + suffix, 0, 0);
+}
+
 function renderDisplayItems(items: DisplayItem[], expanded: boolean, theme: RenderTheme, limit?: number): string {
 	const toShow = limit ? items.slice(-limit) : items;
 	const skipped = limit && items.length > limit ? items.length - limit : 0;
@@ -128,6 +150,38 @@ function renderDisplayItems(items: DisplayItem[], expanded: boolean, theme: Rend
 
 // ─── renderCall ──────────────────────────────────────────────────────────────
 
+export function renderListAgentsCall(_args: unknown, theme: RenderTheme) {
+	return new Text(theme.fg("toolTitle", theme.bold("list-agents")), 0, 0);
+}
+
+export function renderListAgentsResult(
+	result: ToolRenderResult,
+	{ expanded }: { expanded: boolean },
+	theme: RenderTheme,
+) {
+	const fullText = getToolResultText(result);
+	if (expanded) return new Text(theme.fg("toolOutput", fullText), 0, 0);
+
+	const details = result.details as ListAgentsDetails | undefined;
+	const names = details?.agents
+		?.map((agent) => (typeof agent.name === "string" ? truncateText(agent.name, COMPACT_AGENT_WIDTH) : ""))
+		.filter(Boolean);
+	if (!names) return new Text(theme.fg("toolOutput", fullText.split("\n")[0] ?? fullText), 0, 0);
+	if (names.length === 0) return new Text(theme.fg("muted", "○ no agents"), 0, 0);
+
+	const visibleNames = names.slice(0, COMPACT_AGENT_LIST_COUNT);
+	const hiddenCount = names.length - visibleNames.length;
+	const hiddenSuffix = hiddenCount > 0 ? `, +${hiddenCount}` : "";
+	return new Text(
+		theme.fg(
+			"toolOutput",
+			`✓ ${names.length} agent${names.length === 1 ? "" : "s"} · ${visibleNames.join(", ")}${hiddenSuffix}`,
+		),
+		0,
+		0,
+	);
+}
+
 export function renderSubagentToolCall(
 	args: ToolRenderArgs,
 	theme: RenderTheme,
@@ -161,30 +215,12 @@ export function renderSubagentToolResult(
 ) {
 	const details = result.details as SubagentDetails | undefined;
 	if (!details || details.results.length === 0) {
-		const raw = result.content[0];
-		const fullText = (raw?.type === "text" ? raw.text : undefined) ?? "(no output)";
-		if (!expanded) {
-			const firstLine = fullText.split("\n")[0] ?? "";
-			const lineCount = fullText.split("\n").length;
-			const suffix = lineCount > 1 ? theme.fg("muted", ` (+${lineCount - 1} lines)`) : "";
-			return new Text(firstLine + suffix, 0, 0);
-		}
-		return new Text(fullText, 0, 0);
+		return renderPlainToolResult(getToolResultText(result), expanded, theme);
 	}
 
 	const mdTheme = getMarkdownTheme();
 	const r = details.results[0];
-	if (!r) {
-		const raw2 = result.content[0];
-		const fullText2 = (raw2?.type === "text" ? raw2.text : undefined) ?? "(no output)";
-		if (!expanded) {
-			const firstLine = fullText2.split("\n")[0] ?? "";
-			const lineCount = fullText2.split("\n").length;
-			const suffix = lineCount > 1 ? theme.fg("muted", ` (+${lineCount - 1} lines)`) : "";
-			return new Text(firstLine + suffix, 0, 0);
-		}
-		return new Text(fullText2, 0, 0);
-	}
+	if (!r) return renderPlainToolResult(getToolResultText(result), expanded, theme);
 
 	const isError = r.exitCode !== 0 || r.stopReason === "error" || r.stopReason === "aborted";
 	const icon = isError ? theme.fg("error", "✗") : theme.fg("success", "✓");