@tianhai/pi-workflow-kit 0.5.1 → 0.6.0

package/extensions/subagent/index.ts

@@ -1,1181 +0,0 @@
-/**
- * Subagent Tool - Delegate tasks to specialized agents
- *
- * Spawns a separate `pi` process for each subagent invocation,
- * giving it an isolated context window.
- *
- * Supports three modes:
- *   - Single: { agent: "name", task: "..." }
- *   - Parallel: { tasks: [{ agent: "name", task: "..." }, ...] }
- *   - Chain: { chain: [{ agent: "name", task: "... {previous} ..." }, ...] }
- *
- * Uses JSON mode to capture structured output from subagents.
- */
-
-import { spawn } from "node:child_process";
-import * as fs from "node:fs";
-import * as os from "node:os";
-import * as path from "node:path";
-import type { Message } from "@mariozechner/pi-ai";
-import { StringEnum } from "@mariozechner/pi-ai";
-import type { AgentToolResult } from "@mariozechner/pi-coding-agent";
-import { type ExtensionAPI, getMarkdownTheme } from "@mariozechner/pi-coding-agent";
-import { Container, Markdown, Spacer, Text } from "@mariozechner/pi-tui";
-import { Type } from "@sinclair/typebox";
-import { DEFAULT_MODEL } from "../../agents/config.js";
-import { log } from "../lib/logging.js";
-import { type AgentConfig, type AgentScope, discoverAgents } from "./agents.js";
-import { getSubagentConcurrency, Semaphore } from "./concurrency.js";
-import { buildSubagentEnv } from "./env.js";
-import { ProcessTracker } from "./lifecycle.js";
-import { getSubagentTimeoutMs } from "./timeout.js";
-
-const MAX_PARALLEL_TASKS = 8;
-const COLLAPSED_ITEM_COUNT = 10;
-export const INACTIVITY_TIMEOUT_MS = 120_000;
-
-function formatTokens(count: number): string {
-  if (count < 1000) return count.toString();
-  if (count < 10000) return `${(count / 1000).toFixed(1)}k`;
-  if (count < 1000000) return `${Math.round(count / 1000)}k`;
-  return `${(count / 1000000).toFixed(1)}M`;
-}
-
-function formatUsageStats(
-  usage: {
-    input: number;
-    output: number;
-    cacheRead: number;
-    cacheWrite: number;
-    cost: number;
-    contextTokens?: number;
-    turns?: number;
-  },
-  model?: string,
-): string {
-  const parts: string[] = [];
-  if (usage.turns) parts.push(`${usage.turns} turn${usage.turns > 1 ? "s" : ""}`);
-  if (usage.input) parts.push(`↑${formatTokens(usage.input)}`);
-  if (usage.output) parts.push(`↓${formatTokens(usage.output)}`);
-  if (usage.cacheRead) parts.push(`R${formatTokens(usage.cacheRead)}`);
-  if (usage.cacheWrite) parts.push(`W${formatTokens(usage.cacheWrite)}`);
-  if (usage.cost) parts.push(`$${usage.cost.toFixed(4)}`);
-  if (usage.contextTokens && usage.contextTokens > 0) {
-    parts.push(`ctx:${formatTokens(usage.contextTokens)}`);
-  }
-  if (model) parts.push(model);
-  return parts.join(" ");
-}
-
-function formatToolCall(
-  toolName: string,
-  args: Record<string, unknown>,
-  // biome-ignore lint/suspicious/noExplicitAny: pi SDK theme callback type
-  themeFg: (color: any, text: string) => string,
-): string {
-  const shortenPath = (p: string) => {
-    const home = os.homedir();
-    return p.startsWith(home) ? `~${p.slice(home.length)}` : p;
-  };
-
-  switch (toolName) {
-    case "bash": {
-      const command = (args.command as string) || "...";
-      const preview = command.length > 60 ? `${command.slice(0, 60)}...` : command;
-      return themeFg("muted", "$ ") + themeFg("toolOutput", preview);
-    }
-    case "read": {
-      const rawPath = (args.file_path || args.path || "...") as string;
-      const filePath = shortenPath(rawPath);
-      const offset = args.offset as number | undefined;
-      const limit = args.limit as number | undefined;
-      let text = themeFg("accent", filePath);
-      if (offset !== undefined || limit !== undefined) {
-        const startLine = offset ?? 1;
-        const endLine = limit !== undefined ? startLine + limit - 1 : "";
-        text += themeFg("warning", `:${startLine}${endLine ? `-${endLine}` : ""}`);
-      }
-      return themeFg("muted", "read ") + text;
-    }
-    case "write": {
-      const rawPath = (args.file_path || args.path || "...") as string;
-      const filePath = shortenPath(rawPath);
-      const content = (args.content || "") as string;
-      const lines = content.split("\n").length;
-      let text = themeFg("muted", "write ") + themeFg("accent", filePath);
-      if (lines > 1) text += themeFg("dim", ` (${lines} lines)`);
-      return text;
-    }
-    case "edit": {
-      const rawPath = (args.file_path || args.path || "...") as string;
-      return themeFg("muted", "edit ") + themeFg("accent", shortenPath(rawPath));
-    }
-    case "ls": {
-      const rawPath = (args.path || ".") as string;
-      return themeFg("muted", "ls ") + themeFg("accent", shortenPath(rawPath));
-    }
-    case "find": {
-      const pattern = (args.pattern || "*") as string;
-      const rawPath = (args.path || ".") as string;
-      return themeFg("muted", "find ") + themeFg("accent", pattern) + themeFg("dim", ` in ${shortenPath(rawPath)}`);
-    }
-    case "grep": {
-      const pattern = (args.pattern || "") as string;
-      const rawPath = (args.path || ".") as string;
-      return (
-        themeFg("muted", "grep ") + themeFg("accent", `/${pattern}/`) + themeFg("dim", ` in ${shortenPath(rawPath)}`)
-      );
-    }
-    default: {
-      const argsStr = JSON.stringify(args);
-      const preview = argsStr.length > 50 ? `${argsStr.slice(0, 50)}...` : argsStr;
-      return themeFg("accent", toolName) + themeFg("dim", ` ${preview}`);
-    }
-  }
-}
-
-interface UsageStats {
-  input: number;
-  output: number;
-  cacheRead: number;
-  cacheWrite: number;
-  cost: number;
-  contextTokens: number;
-  turns: number;
-}
-
-interface SingleResult {
-  agent: string;
-  agentSource: "user" | "project" | "bundled" | "unknown";
-  task: string;
-  exitCode: number;
-  messages: Message[];
-  stderr: string;
-  usage: UsageStats;
-  model?: string;
-  modelProvider?: string;
-  modelSource?: "agent" | "parent" | "default";
-  stopReason?: string;
-  errorMessage?: string;
-  step?: number;
-}
-
-interface SubagentDetails {
-  mode: "single" | "parallel" | "chain";
-  agentScope: AgentScope;
-  projectAgentsDir: string | null;
-  results: SingleResult[];
-}
-
-interface ParentModelInfo {
-  id: string;
-  provider: string;
-}
-
-interface ResolvedModelSelection {
-  model: string;
-  provider?: string;
-  source: "agent" | "parent" | "default";
-}
-
-function resolveModelSelection(
-  agentModel: string | undefined,
-  parentModel: ParentModelInfo | undefined,
-): ResolvedModelSelection {
-  if (agentModel) {
-    return { model: agentModel, provider: undefined, source: "agent" };
-  }
-
-  if (parentModel?.id) {
-    return { model: parentModel.id, provider: parentModel.provider, source: "parent" };
-  }
-
-  return { model: DEFAULT_MODEL, provider: undefined, source: "default" };
-}
-
-function getFinalOutput(messages: Message[]): string {
-  for (let i = messages.length - 1; i >= 0; i--) {
-    const msg = messages[i];
-    if (msg.role === "assistant") {
-      for (const part of msg.content) {
-        if (part.type === "text") return part.text;
-      }
-    }
-  }
-  return "";
-}
-
-function buildModelArgs(selection: ResolvedModelSelection): string[] {
-  const args: string[] = [];
-  if (selection.provider) args.push("--provider", selection.provider);
-  args.push("--model", selection.model);
-  return args;
-}
-
-function formatModelSelection(
-  result: Pick<SingleResult, "model" | "modelProvider" | "modelSource">,
-): string | undefined {
-  if (!result.model) return undefined;
-  const modelLabel = result.modelProvider ? `${result.modelProvider}/${result.model}` : result.model;
-  switch (result.modelSource) {
-    case "parent":
-      return `${modelLabel} (inherited from parent session)`;
-    case "agent":
-      return `${modelLabel} (pinned by agent config)`;
-    case "default":
-      return `${modelLabel} (default fallback)`;
-    default:
-      return modelLabel;
-  }
-}
-
-function buildFailureMessage(prefix: string, result: SingleResult): string {
-  const errorMsg = result.errorMessage || result.stderr || getFinalOutput(result.messages) || "(no output)";
-  const modelSelection = formatModelSelection(result);
-  return modelSelection ? `${prefix}: ${errorMsg}\nModel: ${modelSelection}` : `${prefix}: ${errorMsg}`;
-}
-
-// biome-ignore lint/suspicious/noExplicitAny: pi SDK message content type
-type DisplayItem = { type: "text"; text: string } | { type: "toolCall"; name: string; args: Record<string, any> };
-
-function getDisplayItems(messages: Message[]): DisplayItem[] {
-  const items: DisplayItem[] = [];
-  for (const msg of messages) {
-    if (msg.role === "assistant") {
-      for (const part of msg.content) {
-        if (part.type === "text") items.push({ type: "text", text: part.text });
-        else if (part.type === "toolCall") items.push({ type: "toolCall", name: part.name, args: part.arguments });
-      }
-    }
-  }
-  return items;
-}
-
-function isTestCommand(cmd: string): boolean {
-  return (
-    /\bvitest\b/.test(cmd) ||
-    /\bpytest\b/.test(cmd) ||
-    /\bnpm\s+test\b/.test(cmd) ||
-    /\bpnpm\s+test\b/.test(cmd) ||
-    /\byarn\s+test\b/.test(cmd)
-  );
-}
-
-function collectSummary(messages: Message[]): { filesChanged: string[]; testsRan: boolean } {
-  const files = new Set<string>();
-  let testsRan = false;
-
-  for (const msg of messages) {
-    if (msg.role !== "assistant") continue;
-    for (const part of msg.content) {
-      if (part.type !== "toolCall") continue;
-      // biome-ignore lint/suspicious/noExplicitAny: pi SDK message content type
-      if ((part.name === "write" || part.name === "edit") && typeof (part.arguments as any)?.path === "string") {
-        // biome-ignore lint/suspicious/noExplicitAny: pi SDK message content type
-        files.add((part.arguments as any).path);
-      }
-      if (part.name === "bash") {
-        // biome-ignore lint/suspicious/noExplicitAny: pi SDK message content type
-        const cmd = (part.arguments as any)?.command;
-        if (typeof cmd === "string" && isTestCommand(cmd)) testsRan = true;
-      }
-    }
-  }
-
-  return { filesChanged: Array.from(files), testsRan };
-}
-
-export const __internal = { collectSummary, resolveModelSelection };
-
-async function mapWithConcurrencyLimit<TIn, TOut>(
-  items: TIn[],
-  concurrency: number,
-  fn: (item: TIn, index: number) => Promise<TOut>,
-): Promise<TOut[]> {
-  if (items.length === 0) return [];
-  const limit = Math.max(1, Math.min(concurrency, items.length));
-  const results: TOut[] = new Array(items.length);
-  let nextIndex = 0;
-  const workers = new Array(limit).fill(null).map(async () => {
-    while (true) {
-      const current = nextIndex++;
-      if (current >= items.length) return;
-      results[current] = await fn(items[current], current);
-    }
-  });
-  await Promise.all(workers);
-  return results;
-}
-
-function writePromptToTempFile(tmpDir: string, agentName: string, prompt: string): { filePath: string } {
-  const safeName = agentName.replace(/[^\w.-]+/g, "_");
-  const filePath = path.join(tmpDir, `prompt-${safeName}.md`);
-  fs.writeFileSync(filePath, prompt, { encoding: "utf-8", mode: 0o600 });
-  return { filePath };
-}
-
-type OnUpdateCallback = (partial: AgentToolResult<SubagentDetails>) => void;
-
-async function runSingleAgent(
-  defaultCwd: string,
-  agents: AgentConfig[],
-  agentName: string,
-  task: string,
-  cwd: string | undefined,
-  step: number | undefined,
-  parentModel: ParentModelInfo | undefined,
-  signal: AbortSignal | undefined,
-  onUpdate: OnUpdateCallback | undefined,
-  makeDetails: (results: SingleResult[]) => SubagentDetails,
-  processTracker: ProcessTracker,
-  semaphore: Semaphore,
-): Promise<SingleResult> {
-  const agent = agents.find((a) => a.name === agentName);
-
-  if (!agent) {
-    const available = agents.map((a) => `"${a.name}"`).join(", ") || "none";
-    return {
-      agent: agentName,
-      agentSource: "unknown",
-      task,
-      exitCode: 1,
-      messages: [],
-      stderr: `Unknown agent: "${agentName}". Available agents: ${available}.`,
-      usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0, contextTokens: 0, turns: 0 },
-      step,
-    };
-  }
-
-  const args: string[] = ["--mode", "json", "-p", "--no-session"];
-  const selectedModel = resolveModelSelection(agent.model, parentModel);
-  args.push(...buildModelArgs(selectedModel));
-  if (agent.tools && agent.tools.length > 0) args.push("--tools", agent.tools.join(","));
-  if (agent.extensions) {
-    for (const ext of agent.extensions) {
-      args.push("--extension", path.resolve(path.dirname(agent.filePath), ext));
-    }
-  }
-
-  let tmpDir: string | null = null;
-  let tmpPromptPath: string | null = null;
-  const currentResult: SingleResult = {
-    agent: agentName,
-    agentSource: agent.source,
-    task,
-    exitCode: 0,
-    messages: [],
-    stderr: "",
-    usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0, contextTokens: 0, turns: 0 },
-    model: selectedModel.model,
-    modelProvider: selectedModel.provider,
-    modelSource: selectedModel.source,
-    step,
-  };
-
-  const emitUpdate = () => {
-    if (onUpdate) {
-      onUpdate({
-        content: [{ type: "text", text: getFinalOutput(currentResult.messages) || "(running...)" }],
-        details: makeDetails([currentResult]),
-      });
-    }
-  };
-
-  if (semaphore.active >= semaphore.limit) {
-    log.debug(`Subagent queued — ${semaphore.active}/${semaphore.limit} slots in use`);
-  }
-  const release = await semaphore.acquire();
-  try {
-    tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "pi-subagent-"));
-    if (agent.systemPrompt.trim()) {
-      const tmp = writePromptToTempFile(tmpDir, agent.name, agent.systemPrompt);
-      tmpPromptPath = tmp.filePath;
-      args.push("--append-system-prompt", tmpPromptPath);
-    }
-
-    args.push(`Task: ${task}`);
-
-    const resolvedCwd = path.resolve(cwd ?? defaultCwd);
-    let cwdError: string | undefined;
-    try {
-      const stat = fs.statSync(resolvedCwd);
-      if (!stat.isDirectory()) cwdError = `Subagent cwd is not a directory: ${resolvedCwd}`;
-    } catch {
-      cwdError = `Subagent cwd does not exist: ${resolvedCwd}`;
-    }
-    if (cwdError) {
-      return {
-        agent: agentName,
-        agentSource: agent.source,
-        task,
-        exitCode: 1,
-        messages: [],
-        stderr: cwdError,
-        usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0, contextTokens: 0, turns: 0 },
-        step,
-        errorMessage: cwdError,
-      };
-    }
-
-    let wasAborted = false;
-
-    const exitCode = await new Promise<number>((resolve) => {
-      const proc = spawn("pi", args, {
-        cwd: resolvedCwd,
-        shell: false,
-        stdio: ["ignore", "pipe", "pipe"],
-        env: buildSubagentEnv(),
-      });
-      processTracker.add(proc);
-      let buffer = "";
-      let inactivityTimer: ReturnType<typeof setTimeout> | null = null;
-      let absoluteTimer: ReturnType<typeof setTimeout> | null = null;
-      let exitResolved = false;
-
-      const resolveOnce = (code: number) => {
-        if (exitResolved) return;
-        exitResolved = true;
-        if (inactivityTimer) clearTimeout(inactivityTimer);
-        if (absoluteTimer) clearTimeout(absoluteTimer);
-        resolve(code);
-      };
-
-      let resetInactivityTimer = () => {};
-
-      const processLine = (line: string) => {
-        if (!line.trim()) return;
-        // biome-ignore lint/suspicious/noExplicitAny: pi SDK JSON event type
-        let event: any;
-        try {
-          event = JSON.parse(line);
-        } catch (_err) {
-          log.debug(`Ignoring non-JSON line from subagent stdout: ${line.slice(0, 120)}`);
-          return;
-        }
-
-        if (event.type === "message_end" && event.message) {
-          const msg = event.message as Message;
-          currentResult.messages.push(msg);
-
-          if (msg.role === "assistant") {
-            currentResult.usage.turns++;
-            const usage = msg.usage;
-            if (usage) {
-              currentResult.usage.input += usage.input || 0;
-              currentResult.usage.output += usage.output || 0;
-              currentResult.usage.cacheRead += usage.cacheRead || 0;
-              currentResult.usage.cacheWrite += usage.cacheWrite || 0;
-              currentResult.usage.cost += usage.cost?.total || 0;
-              currentResult.usage.contextTokens = usage.totalTokens || 0;
-            }
-            if (!currentResult.model && msg.model) currentResult.model = msg.model;
-            if (msg.stopReason) currentResult.stopReason = msg.stopReason;
-            if (msg.errorMessage) currentResult.errorMessage = msg.errorMessage;
-          }
-          emitUpdate();
-          resetInactivityTimer();
-        }
-
-        if (event.type === "tool_result_end" && event.message) {
-          currentResult.messages.push(event.message as Message);
-          emitUpdate();
-        }
-      };
-
-      resetInactivityTimer = () => {
-        if (inactivityTimer) clearTimeout(inactivityTimer);
-        inactivityTimer = setTimeout(() => {
-          if (exitResolved) return;
-          log.debug(`Subagent killed after ${INACTIVITY_TIMEOUT_MS}ms of inactivity`);
-          currentResult.errorMessage = `Subagent killed after ${INACTIVITY_TIMEOUT_MS / 1000}s of inactivity`;
-          if (buffer.trim()) processLine(buffer);
-          proc.kill("SIGTERM");
-          setTimeout(() => {
-            try {
-              proc.kill("SIGKILL");
-            } catch {
-              /* already exited */
-            }
-          }, 5000);
-          resolveOnce(1);
-        }, INACTIVITY_TIMEOUT_MS);
-      };
-
-      resetInactivityTimer();
-
-      // Absolute timeout — kills regardless of activity
-      const absoluteTimeoutMs = getSubagentTimeoutMs(agent.timeout);
-      absoluteTimer = setTimeout(() => {
-        if (exitResolved) return;
-        const seconds = Math.round(absoluteTimeoutMs / 1000);
-        log.debug(`Subagent killed after ${seconds}s absolute timeout`);
-        currentResult.errorMessage = `Subagent timed out after ${seconds}s`;
-        if (buffer.trim()) processLine(buffer);
-        proc.kill("SIGTERM");
-        setTimeout(() => {
-          try {
-            proc.kill("SIGKILL");
-          } catch {
-            /* already exited */
-          }
-        }, 5000);
-        resolveOnce(1);
-      }, absoluteTimeoutMs);
-
-      proc.stdout.on("data", (data) => {
-        buffer += data.toString();
-        const lines = buffer.split("\n");
-        buffer = lines.pop() || "";
-        for (const line of lines) processLine(line);
-      });
-
-      proc.stderr.on("data", (data) => {
-        currentResult.stderr += data.toString();
-      });
-
-      proc.on("exit", (code) => {
-        processTracker.remove(proc);
-        if (exitResolved) return;
-        if (inactivityTimer) clearTimeout(inactivityTimer);
-        // Wait for any remaining stdout data to arrive after process exit, then drain.
-        // 2s is generous for local I/O; the process has already exited at this point.
-        setTimeout(() => {
-          if (buffer.trim()) processLine(buffer);
-          resolveOnce(code ?? 0);
-        }, 2000);
-      });
-
-      proc.on("error", () => {
-        processTracker.remove(proc);
-        if (inactivityTimer) clearTimeout(inactivityTimer);
-        resolveOnce(1);
-      });
-
-      if (signal) {
-        const killProc = () => {
-          wasAborted = true;
-          proc.kill("SIGTERM");
-          setTimeout(() => {
-            try {
-              proc.kill("SIGKILL");
-            } catch {
-              /* already exited */
-            }
-          }, 5000);
-        };
-        if (signal.aborted) killProc();
-        else signal.addEventListener("abort", killProc, { once: true });
-      }
-    });
-
-    currentResult.exitCode = exitCode;
-    if (wasAborted) throw new Error("Subagent was aborted");
-    return currentResult;
-  } finally {
-    release();
-    if (tmpPromptPath)
-      try {
-        fs.unlinkSync(tmpPromptPath);
-      } catch (err) {
-        log.debug(
-          `Failed to clean up temp prompt file: ${tmpPromptPath} — ${err instanceof Error ? err.message : err}`,
-        );
-      }
-    if (tmpDir)
-      try {
-        fs.rmSync(tmpDir, { recursive: true, force: true });
-      } catch (err) {
-        log.debug(`Failed to clean up temp directory: ${tmpDir} — ${err instanceof Error ? err.message : err}`);
-      }
-  }
-}
-
-const TaskItem = Type.Object({
-  agent: Type.String({ description: "Name of the agent to invoke" }),
-  task: Type.String({ description: "Task to delegate to the agent" }),
-  cwd: Type.Optional(Type.String({ description: "Working directory for the agent process" })),
-});
-
-const ChainItem = Type.Object({
-  agent: Type.String({ description: "Name of the agent to invoke" }),
-  task: Type.String({ description: "Task with optional {previous} placeholder for prior output" }),
-  cwd: Type.Optional(Type.String({ description: "Working directory for the agent process" })),
-});
-
-const AgentScopeSchema = StringEnum(["user", "project", "both"] as const, {
-  description:
-    'Which agent directories to use. Default: "user". Use "both" to include bundled and project-local agents.',
-  default: "user",
-});
-
-const SubagentParams = Type.Object({
-  agent: Type.Optional(Type.String({ description: "Name of the agent to invoke (for single mode)" })),
-  task: Type.Optional(Type.String({ description: "Task to delegate (for single mode)" })),
-  tasks: Type.Optional(Type.Array(TaskItem, { description: "Array of {agent, task} for parallel execution" })),
-  chain: Type.Optional(Type.Array(ChainItem, { description: "Array of {agent, task} for sequential execution" })),
-  agentScope: Type.Optional(AgentScopeSchema),
-  confirmProjectAgents: Type.Optional(
-    Type.Boolean({ description: "Prompt before running project-local agents. Default: true.", default: true }),
-  ),
-  cwd: Type.Optional(Type.String({ description: "Working directory for the agent process (single mode)" })),
-});
-
-export default function (pi: ExtensionAPI) {
-  const processTracker = new ProcessTracker();
-  const semaphore = new Semaphore(getSubagentConcurrency());
-
-  process.on("exit", () => processTracker.killAll());
-
-  pi.registerTool({
-    name: "subagent",
-    label: "Subagent",
-    description: [
-      "Delegate tasks to specialized subagents with isolated context.",
-      "Modes: single (agent + task), parallel (tasks array), chain (sequential with {previous} placeholder).",
-      'Default agent scope is "user" (from ~/.pi/agent/agents).',
-      'Bundled agents (worker, implementer, code-reviewer, spec-reviewer) require agentScope: "both" or "project".',
-      'To also include project-local agents in .pi/agents, set agentScope: "both".',
-    ].join(" "),
-    parameters: SubagentParams,
-
-    async execute(_toolCallId, params, signal, onUpdate, ctx) {
-      const agentScope: AgentScope = params.agentScope ?? "user";
-      const discovery = discoverAgents(ctx.cwd, agentScope);
-      const agents = discovery.agents;
-      const confirmProjectAgents = params.confirmProjectAgents ?? true;
-
-      const hasChain = (params.chain?.length ?? 0) > 0;
-      const hasTasks = (params.tasks?.length ?? 0) > 0;
-      const hasSingle = Boolean(params.agent && params.task);
-      const modeCount = Number(hasChain) + Number(hasTasks) + Number(hasSingle);
-
-      const makeDetails =
-        (mode: "single" | "parallel" | "chain") =>
-        (results: SingleResult[]): SubagentDetails => ({
-          mode,
-          agentScope,
-          projectAgentsDir: discovery.projectAgentsDir,
-          results,
-        });
-      const parentModel = ctx.model ? { id: ctx.model.id, provider: ctx.model.provider } : undefined;
-
-      if (modeCount !== 1) {
-        const available = agents.map((a) => `${a.name} (${a.source})`).join(", ") || "none";
-        return {
-          content: [
-            {
-              type: "text",
-              text: `Invalid parameters. Provide exactly one mode.\nAvailable agents: ${available}`,
-            },
-          ],
-          details: makeDetails("single")([]),
-        };
-      }
-
-      if ((agentScope === "project" || agentScope === "both") && confirmProjectAgents && ctx.hasUI) {
-        const requestedAgentNames = new Set<string>();
-        if (params.chain) for (const step of params.chain) requestedAgentNames.add(step.agent);
-        if (params.tasks) for (const t of params.tasks) requestedAgentNames.add(t.agent);
-        if (params.agent) requestedAgentNames.add(params.agent);
-
-        const projectAgentsRequested = Array.from(requestedAgentNames)
-          .map((name) => agents.find((a) => a.name === name))
-          .filter((a): a is AgentConfig => a?.source === "project");
-
-        if (projectAgentsRequested.length > 0) {
-          const names = projectAgentsRequested.map((a) => a.name).join(", ");
-          const dir = discovery.projectAgentsDir ?? "(unknown)";
-          const ok = await ctx.ui.confirm(
-            "Run project-local agents?",
-            `Agents: ${names}\nSource: ${dir}\n\nProject agents are repo-controlled. Only continue for trusted repositories.`,
-          );
-          if (!ok)
-            return {
-              content: [{ type: "text", text: "Canceled: project-local agents not approved." }],
-              details: makeDetails(hasChain ? "chain" : hasTasks ? "parallel" : "single")([]),
-            };
-        }
-      }
-
-      if (params.chain && params.chain.length > 0) {
-        const results: SingleResult[] = [];
-        let previousOutput = "";
-
-        for (let i = 0; i < params.chain.length; i++) {
-          const step = params.chain[i];
-          const taskWithContext = step.task.replace(/\{previous\}/g, previousOutput);
-
-          // Create update callback that includes all previous results
-          const chainUpdate: OnUpdateCallback | undefined = onUpdate
-            ? (partial) => {
-                // Combine completed results with current streaming result
-                const currentResult = partial.details?.results[0];
-                if (currentResult) {
-                  const allResults = [...results, currentResult];
-                  onUpdate({
-                    content: partial.content,
-                    details: makeDetails("chain")(allResults),
-                  });
-                }
-              }
-            : undefined;
-
-          const result = await runSingleAgent(
-            ctx.cwd,
-            agents,
-            step.agent,
-            taskWithContext,
-            step.cwd,
-            i + 1,
-            parentModel,
-            signal,
-            chainUpdate,
-            makeDetails("chain"),
-            processTracker,
-            semaphore,
-          );
-          results.push(result);
-
-          const isError = result.exitCode !== 0 || result.stopReason === "error" || result.stopReason === "aborted";
-          if (isError) {
-            return {
-              content: [
-                { type: "text", text: buildFailureMessage(`Chain stopped at step ${i + 1} (${step.agent})`, result) },
-              ],
-              details: makeDetails("chain")(results),
-              isError: true,
-            };
-          }
-          previousOutput = getFinalOutput(result.messages);
-        }
-        return {
-          content: [{ type: "text", text: getFinalOutput(results[results.length - 1].messages) || "(no output)" }],
-          details: makeDetails("chain")(results),
-        };
-      }
-
-      if (params.tasks && params.tasks.length > 0) {
-        if (params.tasks.length > MAX_PARALLEL_TASKS)
-          return {
-            content: [
-              {
-                type: "text",
-                text: `Too many parallel tasks (${params.tasks.length}). Max is ${MAX_PARALLEL_TASKS}.`,
-              },
-            ],
-            details: makeDetails("parallel")([]),
-          };
-
-        // Track all results for streaming updates
-        const allResults: SingleResult[] = new Array(params.tasks.length);
-
-        // Initialize placeholder results
-        for (let i = 0; i < params.tasks.length; i++) {
-          allResults[i] = {
-            agent: params.tasks[i].agent,
-            agentSource: "unknown",
-            task: params.tasks[i].task,
-            exitCode: -1, // -1 = still running
-            messages: [],
-            stderr: "",
-            usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0, contextTokens: 0, turns: 0 },
-          };
-        }
-
-        const emitParallelUpdate = () => {
-          if (onUpdate) {
-            const running = allResults.filter((r) => r.exitCode === -1).length;
-            const done = allResults.filter((r) => r.exitCode !== -1).length;
-            onUpdate({
-              content: [{ type: "text", text: `Parallel: ${done}/${allResults.length} done, ${running} running...` }],
-              details: makeDetails("parallel")([...allResults]),
-            });
-          }
-        };
-
-        const results = await mapWithConcurrencyLimit(params.tasks, params.tasks.length, async (t, index) => {
-          const result = await runSingleAgent(
-            ctx.cwd,
-            agents,
-            t.agent,
-            t.task,
-            t.cwd,
-            undefined,
-            parentModel,
-            signal,
-            // Per-task update callback
-            (partial) => {
-              if (partial.details?.results[0]) {
-                allResults[index] = partial.details.results[0];
-                emitParallelUpdate();
-              }
-            },
-            makeDetails("parallel"),
-            processTracker,
-            semaphore,
-          );
-          allResults[index] = result;
-          emitParallelUpdate();
-          return result;
-        });
-
-        const successCount = results.filter((r) => r.exitCode === 0).length;
-        const summaries = results.map((r) => {
-          const output = getFinalOutput(r.messages);
-          const preview = output.slice(0, 100) + (output.length > 100 ? "..." : "");
-          return `[${r.agent}] ${r.exitCode === 0 ? "completed" : "failed"}: ${preview || "(no output)"}`;
-        });
-        return {
-          content: [
-            {
-              type: "text",
-              text: `Parallel: ${successCount}/${results.length} succeeded\n\n${summaries.join("\n\n")}`,
-            },
-          ],
-          details: makeDetails("parallel")(results),
-        };
-      }
-
-      if (params.agent && params.task) {
-        const result = await runSingleAgent(
-          ctx.cwd,
-          agents,
-          params.agent,
-          params.task,
-          params.cwd,
-          undefined,
-          parentModel,
-          signal,
-          onUpdate,
-          makeDetails("single"),
-          processTracker,
-          semaphore,
-        );
-        const summary = collectSummary(result.messages);
-        const stableDetails = {
-          ...makeDetails("single")([result]),
-          status:
-            result.exitCode !== 0 || result.stopReason === "error" || result.stopReason === "aborted"
-              ? ("failed" as const)
-              : ("completed" as const),
-          agent: result.agent,
-          task: result.task,
-          result: getFinalOutput(result.messages),
-          filesChanged: summary.filesChanged,
-          testsRan: summary.testsRan,
-        };
-        const isError = result.exitCode !== 0 || result.stopReason === "error" || result.stopReason === "aborted";
-        if (isError) {
-          return {
-            content: [{ type: "text", text: buildFailureMessage(`Agent ${result.stopReason || "failed"}`, result) }],
-            details: stableDetails,
-            isError: true,
-          };
-        }
-        return {
-          content: [{ type: "text", text: getFinalOutput(result.messages) || "(no output)" }],
-          details: stableDetails,
-        };
-      }
-
-      const available = agents.map((a) => `${a.name} (${a.source})`).join(", ") || "none";
-      return {
-        content: [{ type: "text", text: `Invalid parameters. Available agents: ${available}` }],
-        details: makeDetails("single")([]),
-      };
-    },
-
-    renderCall(args, theme) {
-      const scope: AgentScope = args.agentScope ?? "user";
-      if (args.chain && args.chain.length > 0) {
-        let text =
-          theme.fg("toolTitle", theme.bold("subagent ")) +
-          theme.fg("accent", `chain (${args.chain.length} steps)`) +
-          theme.fg("muted", ` [${scope}]`);
-        for (let i = 0; i < Math.min(args.chain.length, 3); i++) {
-          const step = args.chain[i];
-          // Clean up {previous} placeholder for display
-          const cleanTask = step.task.replace(/\{previous\}/g, "").trim();
-          const preview = cleanTask.length > 40 ? `${cleanTask.slice(0, 40)}...` : cleanTask;
-          text +=
-            "\n  " +
-            theme.fg("muted", `${i + 1}.`) +
-            " " +
-            theme.fg("accent", step.agent) +
-            theme.fg("dim", ` ${preview}`);
-        }
-        if (args.chain.length > 3) text += `\n  ${theme.fg("muted", `... +${args.chain.length - 3} more`)}`;
-        return new Text(text, 0, 0);
-      }
-      if (args.tasks && args.tasks.length > 0) {
-        let text =
-          theme.fg("toolTitle", theme.bold("subagent ")) +
-          theme.fg("accent", `parallel (${args.tasks.length} tasks)`) +
-          theme.fg("muted", ` [${scope}]`);
-        for (const t of args.tasks.slice(0, 3)) {
-          const preview = t.task.length > 40 ? `${t.task.slice(0, 40)}...` : t.task;
-          text += `\n  ${theme.fg("accent", t.agent)}${theme.fg("dim", ` ${preview}`)}`;
-        }
-        if (args.tasks.length > 3) text += `\n  ${theme.fg("muted", `... +${args.tasks.length - 3} more`)}`;
-        return new Text(text, 0, 0);
-      }
-      const agentName = args.agent || "...";
-      const preview = args.task ? (args.task.length > 60 ? `${args.task.slice(0, 60)}...` : args.task) : "...";
-      let text =
-        theme.fg("toolTitle", theme.bold("subagent ")) +
-        theme.fg("accent", agentName) +
-        theme.fg("muted", ` [${scope}]`);
-      text += `\n  ${theme.fg("dim", preview)}`;
-      return new Text(text, 0, 0);
-    },
-
-    renderResult(result, { expanded }, theme) {
-      const details = result.details as SubagentDetails | undefined;
-      if (!details || details.results.length === 0) {
-        const text = result.content[0];
-        return new Text(text?.type === "text" ? text.text : "(no output)", 0, 0);
-      }
-
-      const mdTheme = getMarkdownTheme();
-
-      const renderDisplayItems = (items: DisplayItem[], limit?: number) => {
-        const toShow = limit ? items.slice(-limit) : items;
-        const skipped = limit && items.length > limit ? items.length - limit : 0;
-        let text = "";
-        if (skipped > 0) text += theme.fg("muted", `... ${skipped} earlier items\n`);
-        for (const item of toShow) {
-          if (item.type === "text") {
-            const preview = expanded ? item.text : item.text.split("\n").slice(0, 3).join("\n");
-            text += `${theme.fg("toolOutput", preview)}\n`;
-          } else {
-            text += `${theme.fg("muted", "→ ") + formatToolCall(item.name, item.args, theme.fg.bind(theme))}\n`;
-          }
-        }
-        return text.trimEnd();
-      };
-
-      if (details.mode === "single" && details.results.length === 1) {
-        const r = details.results[0];
-        const isError = r.exitCode !== 0 || r.stopReason === "error" || r.stopReason === "aborted";
-        const icon = isError ? theme.fg("error", "✗") : theme.fg("success", "✓");
-        const displayItems = getDisplayItems(r.messages);
-        const finalOutput = getFinalOutput(r.messages);
-
-        if (expanded) {
-          const container = new Container();
-          let header = `${icon} ${theme.fg("toolTitle", theme.bold(r.agent))}${theme.fg("muted", ` (${r.agentSource})`)}`;
-          if (isError && r.stopReason) header += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
-          container.addChild(new Text(header, 0, 0));
-          if (isError && r.errorMessage)
-            container.addChild(new Text(theme.fg("error", `Error: ${r.errorMessage}`), 0, 0));
-          container.addChild(new Spacer(1));
-          container.addChild(new Text(theme.fg("muted", "─── Task ───"), 0, 0));
-          container.addChild(new Text(theme.fg("dim", r.task), 0, 0));
-          container.addChild(new Spacer(1));
-          container.addChild(new Text(theme.fg("muted", "─── Output ───"), 0, 0));
-          if (displayItems.length === 0 && !finalOutput) {
-            container.addChild(new Text(theme.fg("muted", "(no output)"), 0, 0));
-          } else {
-            for (const item of displayItems) {
-              if (item.type === "toolCall")
-                container.addChild(
-                  new Text(theme.fg("muted", "→ ") + formatToolCall(item.name, item.args, theme.fg.bind(theme)), 0, 0),
-                );
-            }
-            if (finalOutput) {
-              container.addChild(new Spacer(1));
-              container.addChild(new Markdown(finalOutput.trim(), 0, 0, mdTheme));
-            }
-          }
-          const usageStr = formatUsageStats(r.usage, r.model);
-          if (usageStr) {
-            container.addChild(new Spacer(1));
-            container.addChild(new Text(theme.fg("dim", usageStr), 0, 0));
-          }
-          return container;
-        }
-
-        let text = `${icon} ${theme.fg("toolTitle", theme.bold(r.agent))}${theme.fg("muted", ` (${r.agentSource})`)}`;
-        if (isError && r.stopReason) text += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
-        if (isError && r.errorMessage) text += `\n${theme.fg("error", `Error: ${r.errorMessage}`)}`;
-        else if (displayItems.length === 0) text += `\n${theme.fg("muted", "(no output)")}`;
-        else {
-          text += `\n${renderDisplayItems(displayItems, COLLAPSED_ITEM_COUNT)}`;
-          if (displayItems.length > COLLAPSED_ITEM_COUNT) text += `\n${theme.fg("muted", "(Ctrl+O to expand)")}`;
-        }
-        const usageStr = formatUsageStats(r.usage, r.model);
-        if (usageStr) text += `\n${theme.fg("dim", usageStr)}`;
-        return new Text(text, 0, 0);
-      }
-
-      const aggregateUsage = (results: SingleResult[]) => {
-        const total = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0, turns: 0 };
-        for (const r of results) {
-          total.input += r.usage.input;
-          total.output += r.usage.output;
-          total.cacheRead += r.usage.cacheRead;
-          total.cacheWrite += r.usage.cacheWrite;
-          total.cost += r.usage.cost;
-          total.turns += r.usage.turns;
-        }
-        return total;
-      };
-
-      if (details.mode === "chain") {
-        const successCount = details.results.filter((r) => r.exitCode === 0).length;
-        const icon = successCount === details.results.length ? theme.fg("success", "✓") : theme.fg("error", "✗");
-
-        if (expanded) {
-          const container = new Container();
-          container.addChild(
-            new Text(
-              icon +
-                " " +
-                theme.fg("toolTitle", theme.bold("chain ")) +
-                theme.fg("accent", `${successCount}/${details.results.length} steps`),
-              0,
-              0,
-            ),
-          );
-
-          for (const r of details.results) {
-            const rIcon = r.exitCode === 0 ? theme.fg("success", "✓") : theme.fg("error", "✗");
-            const displayItems = getDisplayItems(r.messages);
-            const finalOutput = getFinalOutput(r.messages);
-
-            container.addChild(new Spacer(1));
-            container.addChild(
-              new Text(`${theme.fg("muted", `─── Step ${r.step}: `) + theme.fg("accent", r.agent)} ${rIcon}`, 0, 0),
-            );
-            container.addChild(new Text(theme.fg("muted", "Task: ") + theme.fg("dim", r.task), 0, 0));
-
-            // Show tool calls
-            for (const item of displayItems) {
-              if (item.type === "toolCall") {
-                container.addChild(
-                  new Text(theme.fg("muted", "→ ") + formatToolCall(item.name, item.args, theme.fg.bind(theme)), 0, 0),
-                );
-              }
-            }
-
-            // Show final output as markdown
-            if (finalOutput) {
-              container.addChild(new Spacer(1));
-              container.addChild(new Markdown(finalOutput.trim(), 0, 0, mdTheme));
-            }
-
-            const stepUsage = formatUsageStats(r.usage, r.model);
-            if (stepUsage) container.addChild(new Text(theme.fg("dim", stepUsage), 0, 0));
-          }
-
-          const usageStr = formatUsageStats(aggregateUsage(details.results));
-          if (usageStr) {
-            container.addChild(new Spacer(1));
-            container.addChild(new Text(theme.fg("dim", `Total: ${usageStr}`), 0, 0));
-          }
-          return container;
-        }
-
-        // Collapsed view
-        let text =
-          icon +
-          " " +
-          theme.fg("toolTitle", theme.bold("chain ")) +
-          theme.fg("accent", `${successCount}/${details.results.length} steps`);
-        for (const r of details.results) {
-          const rIcon = r.exitCode === 0 ? theme.fg("success", "✓") : theme.fg("error", "✗");
-          const displayItems = getDisplayItems(r.messages);
-          text += `\n\n${theme.fg("muted", `─── Step ${r.step}: `)}${theme.fg("accent", r.agent)} ${rIcon}`;
-          if (displayItems.length === 0) text += `\n${theme.fg("muted", "(no output)")}`;
-          else text += `\n${renderDisplayItems(displayItems, 5)}`;
-        }
-        const usageStr = formatUsageStats(aggregateUsage(details.results));
-        if (usageStr) text += `\n\n${theme.fg("dim", `Total: ${usageStr}`)}`;
-        text += `\n${theme.fg("muted", "(Ctrl+O to expand)")}`;
-        return new Text(text, 0, 0);
-      }
-
-      if (details.mode === "parallel") {
-        const running = details.results.filter((r) => r.exitCode === -1).length;
-        const successCount = details.results.filter((r) => r.exitCode === 0).length;
-        const failCount = details.results.filter((r) => r.exitCode > 0).length;
-        const isRunning = running > 0;
-        const icon = isRunning
-          ? theme.fg("warning", "⏳")
-          : failCount > 0
-            ? theme.fg("warning", "◐")
-            : theme.fg("success", "✓");
-        const status = isRunning
-          ? `${successCount + failCount}/${details.results.length} done, ${running} running`
-          : `${successCount}/${details.results.length} tasks`;
-
-        if (expanded && !isRunning) {
-          const container = new Container();
-          container.addChild(
-            new Text(`${icon} ${theme.fg("toolTitle", theme.bold("parallel "))}${theme.fg("accent", status)}`, 0, 0),
-          );
-
-          for (const r of details.results) {
-            const rIcon = r.exitCode === 0 ? theme.fg("success", "✓") : theme.fg("error", "✗");
-            const displayItems = getDisplayItems(r.messages);
-            const finalOutput = getFinalOutput(r.messages);
-
-            container.addChild(new Spacer(1));
-            container.addChild(new Text(`${theme.fg("muted", "─── ") + theme.fg("accent", r.agent)} ${rIcon}`, 0, 0));
-            container.addChild(new Text(theme.fg("muted", "Task: ") + theme.fg("dim", r.task), 0, 0));
-
-            // Show tool calls
-            for (const item of displayItems) {
-              if (item.type === "toolCall") {
-                container.addChild(
-                  new Text(theme.fg("muted", "→ ") + formatToolCall(item.name, item.args, theme.fg.bind(theme)), 0, 0),
-                );
-              }
-            }
-
-            // Show final output as markdown
-            if (finalOutput) {
-              container.addChild(new Spacer(1));
-              container.addChild(new Markdown(finalOutput.trim(), 0, 0, mdTheme));
-            }
-
-            const taskUsage = formatUsageStats(r.usage, r.model);
-            if (taskUsage) container.addChild(new Text(theme.fg("dim", taskUsage), 0, 0));
-          }
-
-          const usageStr = formatUsageStats(aggregateUsage(details.results));
-          if (usageStr) {
-            container.addChild(new Spacer(1));
-            container.addChild(new Text(theme.fg("dim", `Total: ${usageStr}`), 0, 0));
-          }
-          return container;
-        }
-
-        // Collapsed view (or still running)
-        let text = `${icon} ${theme.fg("toolTitle", theme.bold("parallel "))}${theme.fg("accent", status)}`;
-        for (const r of details.results) {
-          const rIcon =
-            r.exitCode === -1
-              ? theme.fg("warning", "⏳")
-              : r.exitCode === 0
-                ? theme.fg("success", "✓")
-                : theme.fg("error", "✗");
-          const displayItems = getDisplayItems(r.messages);
-          text += `\n\n${theme.fg("muted", "─── ")}${theme.fg("accent", r.agent)} ${rIcon}`;
-          if (displayItems.length === 0)
-            text += `\n${theme.fg("muted", r.exitCode === -1 ? "(running...)" : "(no output)")}`;
-          else text += `\n${renderDisplayItems(displayItems, 5)}`;
-        }
-        if (!isRunning) {
-          const usageStr = formatUsageStats(aggregateUsage(details.results));
-          if (usageStr) text += `\n\n${theme.fg("dim", `Total: ${usageStr}`)}`;
-        }
-        if (!expanded) text += `\n${theme.fg("muted", "(Ctrl+O to expand)")}`;
-        return new Text(text, 0, 0);
-      }
-
-      const text = result.content[0];
-      return new Text(text?.type === "text" ? text.text : "(no output)", 0, 0);
-    },
-  });
-}