@ryan_nookpi/pi-extension-subagent 0.4.0 → 0.4.2

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.0",
+  "version": "0.4.2",
   "description": "Asynchronous subagent delegation for pi with run, batch, chain, and continuation workflows.",
   "license": "MIT",
   "repository": {

package/tool-render.ts

@@ -5,10 +5,11 @@
  */
 
 import { getMarkdownTheme, type ThemeColor } from "@earendil-works/pi-coding-agent";
-import { Container, Markdown, Spacer, Text } from "@earendil-works/pi-tui";
+import { Container, Markdown, Spacer, Text, visibleWidth } from "@earendil-works/pi-tui";
+import { parseSubagentToolCommand } from "./cli.js";
 import { formatToolCall, formatUsageStats } from "./format.js";
 import { getDisplayItems, getFinalOutput } from "./runner.js";
-import { COLLAPSED_ITEM_COUNT } from "./store.js";
+import { COLLAPSED_ITEM_COUNT, truncateText } from "./store.js";
 import type { DisplayItem, SubagentDetails } from "./types.js";
 
 // ─── Helpers (internal) ──────────────────────────────────────────────────────
@@ -24,6 +25,97 @@ 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);
+}
+
+function getContextLabel(params: Record<string, unknown>): "main" | "isolated" {
+	return params.contextMode === "main" ? "main" : "isolated";
+}
+
+function getCompactBlocks(value: unknown): CompactBlock[] | null {
+	if (!Array.isArray(value)) return null;
+	const blocks: CompactBlock[] = [];
+	for (const item of value) {
+		if (!item || typeof item !== "object") return null;
+		const agent = "agent" in item ? item.agent : undefined;
+		const task = "task" in item ? item.task : undefined;
+		if (typeof agent !== "string" || typeof task !== "string") return null;
+		blocks.push({ agent, task });
+	}
+	return blocks;
+}
+
+function formatCompactAgent(agent: string, width: number): string {
+	const label = truncateText(agent, COMPACT_AGENT_WIDTH);
+	return `${label}${" ".repeat(Math.max(0, width - visibleWidth(label)))}`;
+}
+
+function renderCompactBlocks(kind: "batch" | "chain", params: Record<string, unknown>): string | null {
+	const blocks = getCompactBlocks(kind === "batch" ? params.runs : params.steps);
+	if (!blocks || blocks.length === 0) return null;
+
+	const visibleBlocks = blocks.slice(0, COMPACT_MAX_ITEMS);
+	const agentWidth = Math.max(
+		...visibleBlocks.map((block) => visibleWidth(truncateText(block.agent, COMPACT_AGENT_WIDTH))),
+	);
+	const contextLabel = getContextLabel(params);
+	const executionLabel = kind === "batch" ? "parallel" : "sequential";
+	const lines = [`  ${kind} · ${contextLabel} · ${blocks.length} ${executionLabel}`];
+
+	for (const [index, block] of visibleBlocks.entries()) {
+		const agent = formatCompactAgent(block.agent, agentWidth);
+		const task = normalizeTaskPreview(block.task);
+		if (kind === "batch") {
+			lines.push(`  ∥ ${agent}  ${task}`);
+			continue;
+		}
+
+		const connector = index === 0 ? "" : `${"   ".repeat(index - 1)}└─ `;
+		lines.push(`  ${connector}${index + 1} ${agent}  ${task}`);
+	}
+
+	const hiddenCount = blocks.length - visibleBlocks.length;
+	if (hiddenCount > 0) {
+		if (kind === "batch") lines.push(`  ∥ +${hiddenCount} jobs`);
+		else lines.push(`  ${"   ".repeat(visibleBlocks.length - 1)}└─ +${hiddenCount} steps`);
+	}
+
+	return lines.join("\n");
+}
+
+function renderCompactLaunch(command: unknown): string | null {
+	const parsed = parseSubagentToolCommand(command);
+	if (parsed.type !== "params") return null;
+
+	const { params } = parsed;
+	if (params.asyncAction === "batch") return renderCompactBlocks("batch", params);
+	if (params.asyncAction === "chain") return renderCompactBlocks("chain", params);
+	if (typeof params.task !== "string") return null;
+
+	const contextLabel = getContextLabel(params);
+	const task = normalizeTaskPreview(params.task);
+	if (typeof params.runId === "number") {
+		const agent = typeof params.agent === "string" ? ` · ${params.agent}` : "";
+		return `  continue · #${params.runId}${agent} · ${contextLabel}\n  ${task}`;
+	}
+	if (typeof params.agent !== "string") return 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 renderDisplayItems(items: DisplayItem[], expanded: boolean, theme: RenderTheme, limit?: number): string {
 	const toShow = limit ? items.slice(-limit) : items;
@@ -43,15 +135,57 @@ function renderDisplayItems(items: DisplayItem[], expanded: boolean, theme: Rend
 
 // ─── renderCall ──────────────────────────────────────────────────────────────
 
-export function renderSubagentToolCall(args: ToolRenderArgs, theme: RenderTheme) {
+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,
+	context: ToolCallRenderContext = { expanded: false },
+) {
 	const raw = typeof args.command === "string" ? args.command.trim() : "";
 	const command = raw || "subagent help";
+	let text = theme.fg("toolTitle", theme.bold("subagent ")) + theme.fg("accent", "cli");
+
+	if (!context.expanded) {
+		const compactLaunch = renderCompactLaunch(args.command);
+		if (compactLaunch) return new Text(`${text}\n${theme.fg("dim", compactLaunch)}`, 0, 0);
+	}
+
 	const MAX_CALL_LINES = 5;
 	const commandLines = command.split("\n");
 	const truncated = commandLines.length > MAX_CALL_LINES;
 	const preview = truncated ? commandLines.slice(0, MAX_CALL_LINES).join("\n") : command;
 
-	let text = theme.fg("toolTitle", theme.bold("subagent ")) + theme.fg("accent", "cli");
 	text += `\n  ${theme.fg("dim", preview)}`;
 	if (truncated) text += `\n  ${theme.fg("muted", `... +${commandLines.length - MAX_CALL_LINES} more lines`)}`;
 	return new Text(text, 0, 0);