pi-agent-flow 0.1.0-alpha.8 → 0.2.4

package/agents/architect.md

@@ -1,15 +1,15 @@
 ---
 name: architect
 description: Plan structure, break down requirements, design solutions
-tools: read, bash
+tools: read, bash, find, grep, ls
 maxDepth: 2
 ---
 
-You are the architect flow — your mission is to design. Stay focused on your intent at all times. The conversation history above provides background context; treat it as reference only and do not let it distract from your objective.
+You are the architect flow — your mission is to design. Be conservative: prefer existing patterns and proven conventions over novelty. The conversation history above provides background context; treat it as reference only and do not let it distract from your objective.
 
 Workflow:
 1. Understand — what problem, what constraints, what exists
-2. Explore — find patterns, map dependencies
+2. Explore — find patterns, map dependencies (delegate to [explore] if you need to survey a large codebase)
 3. Design — simplest solution that works, prefer existing patterns
 4. Plan — concrete ordered tasks, identify parallel vs sequential
 

package/agents/brainstorm.md

@@ -8,6 +8,11 @@ maxDepth: 0
 
 You are the brainstorm flow — your mission is to generate ideas and think creatively. You start with a clean slate and only have the intent to guide you. Stay focused on your intent at all times.
 
+Approach:
+1. Diverge — explore many possibilities without judgment
+2. Evaluate — weigh trade-offs, risks, and feasibility
+3. Recommend — present the best options with reasoning
+
 When your mission is accomplished, end your response with:
 
 flow [brainstorm] accomplished

package/agents/code.md

@@ -1,11 +1,11 @@
 ---
 name: code
 description: Implement features, fix bugs, write tests
-tools: read, write, edit, bash
+tools: read, write, edit, bash, find, grep, ls
 maxDepth: 2
 ---
 
-You are the code flow — your mission is to build. Stay focused on your intent at all times. The conversation history above provides background context; treat it as reference only and do not let it distract from your objective.
+You are the code flow — your mission is to build. Be a craftsman: verify first, then ship. The conversation history above provides background context; treat it as reference only and do not let it distract from your objective.
 
 Core Principles:
 - SOLID: Single Responsibility, Open/Closed
@@ -15,9 +15,13 @@ Core Principles:
 Workflow:
 1. Analyze — read existing code for context
 2. Plan — step-by-step approach before modifying
-3. Execute — implement changes following core principles
-4. Verify — write and run tests before considering done
-5. Finalize — all tests pass, implementation verified
+3. Test — write a failing test that proves the bug or validates the feature (red)
+4. Execute — implement changes following core principles (green)
+5. Refactor — clean up only if the change is working (optional)
+6. Verify — run tests and any relevant checks before considering done
+7. Finalize — all tests pass, implementation verified
+
+If you hit an unexpected error or need to trace execution, delegate to [debug] rather than guessing.
 
 When your mission is accomplished, end your response with:
 

package/agents/debug.md

@@ -1,11 +1,11 @@
 ---
 name: debug
 description: Investigate logs, errors, stack traces, root causes
-tools: read, bash
+tools: read, bash, find, grep, ls
 maxDepth: 0
 ---
 
-You are the debug flow — your mission is investigation. Stay focused on your intent at all times. The conversation history above provides background context; treat it as reference only and do not let it distract from your objective.
+You are the debug flow — your mission is investigation. Be forensic: every claim must be backed by evidence. The conversation history above provides background context; treat it as reference only and do not let it distract from your objective.
 
 Workflow:
 1. Collect evidence — logs, error messages, stack traces

package/agents/explore.md

@@ -1,14 +1,14 @@
 ---
 name: explore
 description: Discover files, trace code paths, map architecture
-tools: read, bash
+tools: read, bash, find, grep, ls
 maxDepth: 0
 ---
 
-You are the explore flow — your mission is discovery. Stay focused on your intent at all times. The conversation history above provides background context; treat it as reference only and do not let it distract from your objective.
+You are the explore flow — your mission is discovery. Move fast and stay surgical. The conversation history above provides background context; treat it as reference only and do not let it distract from your objective.
 
 Rules:
-- Use grep/find before reading entire files — be efficient.
+- Use grep/find/ls before reading entire files — be efficient.
 - Report findings with file paths and line numbers.
 - Show actual code/data, not excessive summaries.
 - If not found, say so immediately — don't guess.

package/agents/review.md

@@ -1,11 +1,11 @@
 ---
 name: review
 description: Audit security, quality, correctness
-tools: read, write, edit, bash
+tools: read, bash, find, grep, ls
 maxDepth: 2
 ---
 
-You are the review flow — your mission is to audit. Stay focused on your intent at all times. The conversation history above provides background context; treat it as reference only and do not let it distract from your objective.
+You are the review flow — your mission is to audit. Be adversarial: look for what others miss, but stay honest. The conversation history above provides background context; treat it as reference only and do not let it distract from your objective.
 
 Focus Areas:
 - Security — injection, auth bypass, exposed secrets
@@ -16,7 +16,8 @@ Focus Areas:
 Rules:
 - Be specific — cite exact file paths and line numbers
 - If code is clean, say so — don't invent issues
-- Apply fixes, don't just suggest them
+- Do not modify source files — report findings with severity only
+- If the audit reveals structural issues requiring redesign, recommend [architect] in [Next Steps]
 
 When your mission is accomplished, end your response with:
 
@@ -26,7 +27,6 @@ flow [review] accomplished
 
 [Done]
 - issues found with file:line and severity
-- fixes applied
 
 [Not Done]
 - areas not covered and why

package/flow.ts

@@ -12,7 +12,7 @@ import * as path from "node:path";
 import type { AgentToolResult } from "@mariozechner/pi-agent-core";
 import type { FlowConfig } from "./agents.js";
 import { parseFlowCliArgs } from "./runner-cli.js";
-import { processFlowJsonLine, drainStreamingText } from "./runner-events.js";
+import { processFlowJsonLine, drainStreamingText, drainStreamingEstimate } from "./runner-events.js";
 import {
 	type SingleResult,
 	type FlowDetails,
@@ -32,6 +32,18 @@ const PI_OFFLINE_ENV = "PI_OFFLINE";
 
 type FlowUpdateCallback = (partial: AgentToolResult<FlowDetails>) => void;
 
+/**
+ * Merge actual usage with streaming estimate.
+ * Uses Math.max for output to avoid double-counting.
+ */
+function mergeStreamingUsage(actual: SingleResult["usage"], estimatedTokens: number): SingleResult["usage"] {
+	if (estimatedTokens <= 0) return actual;
+	return {
+		...actual,
+		output: Math.max(actual.output, estimatedTokens),
+	};
+}
+
 // ---------------------------------------------------------------------------
 // Process helpers
 // ---------------------------------------------------------------------------
@@ -116,12 +128,12 @@ function buildFlowArgs(
 	// Flow instructions go in the intent message instead.
 
 	const flowDirectives =
-		`<flow_directives>\n` +
+		`=== flow directive ===\n` +
 		`You are a flow state executing a mission. The conversation history above is background context — use it for reference, but your sole focus is the intent below.\n` +
-		`</flow_directives>`;
+		`=== end flow directive ===`;
 
 	const flowInstructions = flow.systemPrompt.trim()
-		? `\n\n<flow_instructions>\n${flow.systemPrompt.trim()}\n</flow_instructions>`
+		? `\n\n=== system directive ===\n${flow.systemPrompt.trim()}\n=== end system directive ===`
 		: "";
 
 	args.push(`${flowDirectives}${flowInstructions}\n\nIntent: ${intent}`);
@@ -210,6 +222,8 @@ export async function runFlow(opts: RunFlowOptions): Promise<SingleResult> {
 
 	const emitUpdate = () => {
 		const streaming = drainStreamingText(result);
+		const estimatedTokens = drainStreamingEstimate(result);
+		const mergedUsage = mergeStreamingUsage(result.usage, estimatedTokens);
 		onUpdate?.({
 			content: [
 				{
@@ -217,7 +231,7 @@ export async function runFlow(opts: RunFlowOptions): Promise<SingleResult> {
 					text: streaming || getFlowOutput(result.messages) || "(running...)",
 				},
 			],
-			details: makeDetails([result]),
+			details: makeDetails([{ ...result, usage: mergedUsage }]),
 		});
 	};
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-agent-flow",
-  "version": "0.1.0-alpha.8",
+  "version": "0.2.4",
   "description": "Flow-state delegation extension for Pi coding agent.",
   "type": "module",
   "main": "index.ts",

package/render-utils.ts

@@ -25,14 +25,63 @@ export function formatFlowUsage(usage: Partial<UsageStats>, model?: string): str
 	return parts.join(" ");
 }
 
+/**
+ * Format a token count to exactly 5 characters with leading spaces.
+ * Shifts from k to M when value would exceed 5 chars.
+ * Examples: 500 → "  500", 1300 → " 1.3k", 32000 → "32.0k", 950500 → "0.95M"
+ */
+export function formatFixedTokens(count: number): string {
+	if (count < 1000) {
+		return count.toString().padStart(5);
+	}
+
+	const k = count / 1000;
+	if (k < 100) {
+		return (k.toFixed(1) + "k").padStart(5);
+	} else if (k < 1000) {
+		const m = count / 1000000;
+		return (m.toFixed(2) + "M").padStart(5);
+	} else {
+		const m = count / 1000000;
+		return (m.toFixed(2) + "M").padStart(5);
+	}
+}
+
+/**
+ * Format flow type name to fixed width (10 chars) in uppercase with dot padding.
+ * Examples: "debug" → "DEBUG.....", "architect" → "ARCHITECT.", "brainstorm" → "BRAINSTORM"
+ */
+export function formatFlowTypeName(type: string): string {
+	const upper = type.toUpperCase();
+	const targetWidth = 10;
+	if (upper.length >= targetWidth) return upper.slice(0, targetWidth);
+	return upper + ".".repeat(targetWidth - upper.length);
+}
+
 export function formatCompactStats(usage: Partial<UsageStats>, model?: string): string {
 	const parts: string[] = [];
-	parts.push(`↑${formatTokens(usage.input || 0)}`);
-	parts.push(`↓${formatTokens(usage.output || 0)}`);
-	if (usage.cacheRead) parts.push(`cr:${formatTokens(usage.cacheRead)}`);
-	if (usage.contextTokens && usage.contextTokens > 0) parts.push(`ctx:${formatTokens(usage.contextTokens)}`);
-	if (model) parts.push(`model:${model}`);
-	return parts.join(" ");
+	parts.push(`${formatFixedTokens(usage.input || 0)}↑`);
+	parts.push(`${formatFixedTokens(usage.output || 0)}↓`);
+	if (usage.cacheRead) parts.push(`cr:${formatFixedTokens(usage.cacheRead)}`);
+	if (usage.contextTokens && usage.contextTokens > 0) parts.push(`ctx:${formatFixedTokens(usage.contextTokens)}`);
+	return `[ ${parts.join(" ")} ]${model ? ` ─ ${model}` : ""}`;
+}
+
+/**
+ * Format usage stats for expanded view with context tokens on separate line.
+ * Returns stats line and optional context tokens line.
+ * Example: { stats: "[12.4k↑ 2.1k↓ cr:85.0k] ─ mimo-v2.5-pro", contextTokens: "ctx: 32.0k" }
+ */
+export function formatExpandedStats(usage: Partial<UsageStats>, model?: string): { stats: string; contextTokens: string | null } {
+	const parts: string[] = [];
+	parts.push(`${formatFixedTokens(usage.input || 0)}↑`);
+	parts.push(`${formatFixedTokens(usage.output || 0)}↓`);
+	if (usage.cacheRead) parts.push(`cr:${formatFixedTokens(usage.cacheRead)}`);
+
+	const stats = `[${parts.join(" ")}]${model ? ` ─ ${model}` : ""}`;
+	const contextTokens = usage.contextTokens && usage.contextTokens > 0 ? `ctx:${formatFixedTokens(usage.contextTokens)}` : null;
+
+	return { stats, contextTokens };
 }
 
 export function truncateChars(text: string, max: number): string {

package/render.ts

@@ -17,10 +17,12 @@ import {
 	aggregateFlowUsage,
 	getFlowDisplayItems,
 	getFlowOutput,
+	getLastToolCall,
+	getLastAssistantText,
 	isFlowError,
 	isFlowSuccess,
 } from "./types.js";
-import { formatTokens, formatFlowUsage, formatCompactStats, truncateChars, tailText } from "./render-utils.js";
+import { formatTokens, formatFixedTokens, formatFlowUsage, formatCompactStats, formatExpandedStats, formatFlowTypeName, truncateChars, tailText } from "./render-utils.js";
 
 function shortenPath(p: string): string {
 	const home = os.homedir();
@@ -171,14 +173,24 @@ function renderFlowExpanded(
 	const mdTheme = getMarkdownTheme();
 	const container = new Container();
 
-	// Header
-	let header = `${icon} ${theme.fg("toolTitle", theme.bold(`[${r.type}]`))}${theme.fg("muted", ` (${r.agentSource})`)}`;
+	// Header: uppercase type name with dots, no icon, no source
+	const typeName = formatFlowTypeName(r.type);
+	let header = theme.fg("toolTitle", theme.bold(typeName));
 	if (error && r.stopReason) header += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
 	container.addChild(new Text(header, 0, 0));
 	if (error && r.errorMessage) {
 		container.addChild(new Text(theme.fg("error", `Error: ${r.errorMessage}`), 0, 0));
 	}
 
+	// Stats: all-in-one bracket format with context inline
+	const statsParts: string[] = [];
+	statsParts.push(`${formatFixedTokens(r.usage.input || 0)}↑`);
+	statsParts.push(`${formatFixedTokens(r.usage.output || 0)}↓`);
+	if (r.usage.cacheRead) statsParts.push(`cr:${formatFixedTokens(r.usage.cacheRead)}`);
+	if (r.usage.contextTokens > 0) statsParts.push(`ctx:${formatFixedTokens(r.usage.contextTokens)}`);
+	const inlineStats = `[ ${statsParts.join(" ")} ]${r.model ? ` ─ ${r.model}` : ""}`;
+	container.addChild(new Text(theme.fg("dim", inlineStats), 0, 0));
+
 	// Intent
 	container.addChild(new Spacer(1));
 	container.addChild(new Text(theme.fg("muted", "─── Intent ───"), 0, 0));
@@ -202,13 +214,6 @@ function renderFlowExpanded(
 		container.addChild(new Text(toolTraces, 0, 0));
 	}
 
-	// Usage
-	const usageStr = formatFlowUsage(r.usage, r.model);
-	if (usageStr) {
-		container.addChild(new Spacer(1));
-		container.addChild(new Text(theme.fg("dim", usageStr), 0, 0));
-	}
-
 	return container;
 }
 
@@ -223,23 +228,31 @@ function renderFlowCollapsed(
 	streamingText?: string,
 ): Text {
 	const stats = formatCompactStats(r.usage, r.model);
-	let text = `${theme.bg("selectedBg", theme.fg("accent", theme.bold(r.type)))} ${theme.fg("dim", "─")} ${theme.fg("dim", stats)}`;
+	const typeName = formatFlowTypeName(r.type);
+	let text = `${theme.bg("selectedBg", theme.fg("accent", theme.bold(typeName)))} ${theme.fg("dim", "─")} ${theme.fg("dim", stats)}`;
 	if (error && r.stopReason) text += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
 
-	// Intent line
+	// DIR: line (intent/objective)
 	if (r.intent) {
-		text += `\n${theme.fg("dim", `├─ int:`)} ${theme.fg("dim", truncateChars(r.intent, 40))}`;
+		text += `\n${theme.fg("dim", "├─ DIR:")} ${theme.fg("dim", truncateChars(r.intent, 50))}`;
 	}
 
-	// Output line
+	// EXE: line (last tool call)
+	const lastTool = getLastToolCall(r.messages);
+	if (lastTool) {
+		const exeStr = formatFlowToolCall(lastTool.name, lastTool.args, theme.fg.bind(theme));
+		text += `\n${theme.fg("dim", "├─ EXE:")} ${truncateChars(exeStr, 50)}`;
+	}
+
+	// LOG: line (last assistant text or streaming)
 	if (flowOutput) {
-		text += `\n${theme.fg("dim", "└─ msg:")} ${renderFlowReport(truncateChars(flowOutput, 25), theme)}`;
+		text += `\n${theme.fg("dim", "└─ LOG:")} ${theme.fg("dim", truncateChars(flowOutput, 50))}`;
 	} else if (streamingText) {
-		text += `\n${theme.fg("dim", "└─ msg:")} ${theme.fg("dim", tailText(streamingText, 40))}`;
+		text += `\n${theme.fg("dim", "└─ LOG:")} ${theme.fg("dim", tailText(streamingText, 50))}`;
 	} else if (error && r.errorMessage) {
-		text += `\n${theme.fg("dim", "└─ msg:")} ${theme.fg("error", truncateChars(r.errorMessage, 25))}`;
+		text += `\n${theme.fg("dim", "└─ LOG:")} ${theme.fg("error", truncateChars(r.errorMessage, 50))}`;
 	} else {
-		text += `\n${theme.fg("dim", "└─ msg:")} ${theme.fg("dim", "[n/a]")}`;
+		text += `\n${theme.fg("dim", "└─ LOG:")} ${theme.fg("dim", "[n/a]")}`;
 	}
 
 	return new Text(text, 0, 0);
@@ -274,19 +287,32 @@ function renderMultiFlowExpanded(
 	const mdTheme = getMarkdownTheme();
 	const container = new Container();
 
+	// Summary: just show count, no icon
 	container.addChild(new Text(
-		`${icon} ${theme.fg("toolTitle", theme.bold("flow "))}${theme.fg("accent", `${successCount}/${results.length} flows`)}`,
+		theme.fg("accent", `${results.length} flows`),
 		0, 0,
 	));
 
 	for (const r of results) {
-		const rIcon = flowStatusIcon(r, theme);
 		const displayItems = getFlowDisplayItems(r.messages);
 		const flowOutput = getFlowOutput(r.messages);
+		const typeName = formatFlowTypeName(r.type);
 
 		container.addChild(new Spacer(1));
-		container.addChild(new Text(`${theme.fg("muted", "─── ")}${theme.fg("accent", `[${r.type}]`)} ${rIcon}`, 0, 0));
-		container.addChild(new Text(theme.fg("muted", "Intent: ") + theme.fg("dim", r.intent), 0, 0));
+		// Per-flow header: ─── EXPLORER (no icon)
+		container.addChild(new Text(`${theme.fg("muted", "─── ")}${theme.fg("accent", typeName)}`, 0, 0));
+
+		// Stats: all-in-one bracket format with context inline
+		const flowParts: string[] = [];
+		flowParts.push(`${formatFixedTokens(r.usage.input || 0)}↑`);
+		flowParts.push(`${formatFixedTokens(r.usage.output || 0)}↓`);
+		if (r.usage.cacheRead) flowParts.push(`cr:${formatFixedTokens(r.usage.cacheRead)}`);
+		if (r.usage.contextTokens > 0) flowParts.push(`ctx:${formatFixedTokens(r.usage.contextTokens)}`);
+		const flowStats = `[ ${flowParts.join(" ")} ]${r.model ? ` ─ ${r.model}` : ""}`;
+		container.addChild(new Text(theme.fg("dim", flowStats), 0, 0));
+
+		// Intent: just show text, no prefix
+		container.addChild(new Text(theme.fg("dim", r.intent), 0, 0));
 
 		if (flowOutput) {
 			container.addChild(new Spacer(1));
@@ -300,65 +326,83 @@ function renderMultiFlowExpanded(
 			container.addChild(new Text(theme.fg("muted", "─── Tool Calls ───"), 0, 0));
 			container.addChild(new Text(toolTraces, 0, 0));
 		}
-
-		const taskUsage = formatFlowUsage(r.usage, r.model);
-		if (taskUsage) container.addChild(new Text(theme.fg("dim", taskUsage), 0, 0));
 	}
 
-	const totalUsage = formatFlowUsage(aggregateFlowUsage(results));
-	if (totalUsage) {
-		container.addChild(new Spacer(1));
-		container.addChild(new Text(theme.fg("dim", `Total: ${totalUsage}`), 0, 0));
-	}
+	// Total stats: all-in-one bracket format with context inline
+	const totalUsage = aggregateFlowUsage(results);
+	const totalModel = results[0]?.model;
+	const totalParts: string[] = [];
+	totalParts.push(`${formatFixedTokens(totalUsage.input || 0)}↑`);
+	totalParts.push(`${formatFixedTokens(totalUsage.output || 0)}↓`);
+	if (totalUsage.cacheRead) totalParts.push(`cr:${formatFixedTokens(totalUsage.cacheRead)}`);
+	if (totalUsage.contextTokens > 0) totalParts.push(`ctx:${formatFixedTokens(totalUsage.contextTokens)}`);
+	const totalStats = `[ ${totalParts.join(" ")} ]${totalModel ? ` ─ ${totalModel}` : ""}`;
+	container.addChild(new Spacer(1));
+	container.addChild(new Text(theme.fg("dim", totalStats), 0, 0));
 
 	return container;
 }
 
-function renderMultiFlowCollapsed(
+function renderActivityPanel(
 	results: SingleResult[],
 	theme: FlowTheme,
-): Text {
-	let text = "";
+): Container {
+	const container = new Container();
 
 	for (let i = 0; i < results.length; i++) {
 		const r = results[i];
 		const isLast = i === results.length - 1;
-		const flowOutput = getFlowOutput(r.messages);
 		const stats = formatCompactStats(r.usage, r.model);
 		const error = isFlowError(r);
+		const typeName = formatFlowTypeName(r.type);
 
 		// Header line
 		const headerPrefix = isLast ? "└─" : "├─";
-		let line = `${theme.fg("dim", headerPrefix)} ${theme.bg("selectedBg", theme.fg("accent", theme.bold(r.type)))}`;
-		if (stats) {
-			line += ` ${theme.fg("dim", "─")} ${theme.fg("dim", stats)}`;
-		}
+		let headerLine = `${theme.fg("dim", headerPrefix)} ${theme.bg("selectedBg", theme.fg("accent", theme.bold(typeName)))} ${theme.fg("dim", "─")} ${theme.fg("dim", stats)}`;
 		if (error && r.stopReason) {
-			line += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
+			headerLine += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
 		}
-		if (i > 0) text += "\n";
-		text += line;
+		container.addChild(new Text(headerLine, 0, 0));
 
 		// Continuation indent for sub-lines
 		const indent = isLast ? "   " : "│  ";
 
-		// Intent line
+		// DIR: line (intent/objective)
 		if (r.intent) {
-			const intentPrefix = "├─";
-			text += `\n${theme.fg("dim", indent + intentPrefix + " int:")} ${theme.fg("dim", truncateChars(r.intent, 40))}`;
+			container.addChild(new Text(`${theme.fg("dim", indent + "├─ DIR:")} ${theme.fg("dim", truncateChars(r.intent, 50))}`, 0, 0));
 		}
 
-		// Output line
-		if (flowOutput) {
-			text += `\n${theme.fg("dim", indent + "└─ msg:")} ${renderFlowReport(truncateChars(flowOutput, 25), theme)}`;
+		// EXE: line (last tool call)
+		const lastTool = getLastToolCall(r.messages);
+		if (lastTool) {
+			const exeStr = formatFlowToolCall(lastTool.name, lastTool.args, theme.fg.bind(theme));
+			container.addChild(new Text(`${theme.fg("dim", indent + "├─ EXE:")} ${truncateChars(exeStr, 50)}`, 0, 0));
+		}
+
+		// LOG: line (last assistant text)
+		const lastText = getLastAssistantText(r.messages);
+		if (lastText) {
+			container.addChild(new Text(`${theme.fg("dim", indent + "└─ LOG:")} ${theme.fg("dim", truncateChars(lastText, 50))}`, 0, 0));
 		} else if (error && r.errorMessage) {
-			text += `\n${theme.fg("dim", indent + "└─ msg:")} ${theme.fg("error", truncateChars(r.errorMessage, 25))}`;
+			container.addChild(new Text(`${theme.fg("dim", indent + "└─ LOG:")} ${theme.fg("error", truncateChars(r.errorMessage, 50))}`, 0, 0));
 		} else {
-			text += `\n${theme.fg("dim", indent + "└─ msg:")} ${theme.fg("dim", "[n/a]")}`;
+			container.addChild(new Text(`${theme.fg("dim", indent + "└─ LOG:")} ${theme.fg("dim", "[n/a]")}`, 0, 0));
+		}
+
+		// Add blank line separator between flows (with continuation pipe)
+		if (!isLast) {
+			container.addChild(new Text(theme.fg("dim", "│"), 0, 0));
 		}
 	}
 
-	text += `\n${theme.fg("muted", "(Ctrl+O to expand tool traces)")}`;
+	container.addChild(new Text(theme.fg("muted", "(Ctrl+O to expand tool traces)"), 0, 0));
 
-	return new Text(text, 0, 0);
+	return container;
+}
+
+function renderMultiFlowCollapsed(
+	results: SingleResult[],
+	theme: FlowTheme,
+): Container {
+	return renderActivityPanel(results, theme);
 }

package/runner-events.js

@@ -44,6 +44,46 @@ export function drainStreamingText(result) {
   return buf;
 }
 
+// ---------------------------------------------------------------------------
+// Streaming token estimate
+// ---------------------------------------------------------------------------
+
+/** Chars per token heuristic for output estimation. */
+const CHARS_PER_TOKEN = 4;
+
+function getStreamingEstimate(result) {
+  if (!Object.prototype.hasOwnProperty.call(result, "__streamingEstimate")) {
+    Object.defineProperty(result, "__streamingEstimate", {
+      value: { chars: 0 },
+      enumerable: false,
+      configurable: false,
+      writable: true,
+    });
+  }
+  return result.__streamingEstimate;
+}
+
+/**
+ * Track streaming characters and estimate output tokens.
+ * Called on every streaming delta.
+ */
+function updateStreamingEstimate(result, deltaLength) {
+  if (deltaLength <= 0) return;
+  const est = getStreamingEstimate(result);
+  est.chars += deltaLength;
+}
+
+/**
+ * Drain the current streaming estimate and return estimated output tokens.
+ * Returns 0 when no streaming has occurred.
+ */
+export function drainStreamingEstimate(result) {
+  const est = getStreamingEstimate(result);
+  const tokens = Math.floor(est.chars / CHARS_PER_TOKEN);
+  est.chars = 0;
+  return tokens;
+}
+
 /**
  * Accumulate a text or thinking delta into the streaming buffer.
  * Returns true if the caller should emit an update.
@@ -52,6 +92,7 @@ function accumulateStreamingDelta(result, delta) {
   if (!delta) return false;
   const buf = getStreamingTextBuffer(result);
   result.__streamingTextBuffer = buf + delta;
+  updateStreamingEstimate(result, delta.length);
   if (result.__streamingTextBuffer.length - result.__lastEmittedWordCount >= 40) {
     result.__lastEmittedWordCount = result.__streamingTextBuffer.length;
     return true;
@@ -97,6 +138,10 @@ function addFlowAssistantMessage(result, message) {
 
   result.messages.push(message);
 
+  // Reset streaming estimate when actual usage arrives
+  const est = getStreamingEstimate(result);
+  est.chars = 0;
+
   result.usage.turns++;
   const usage = message.usage;
   if (usage) {

package/types.ts

@@ -149,3 +149,37 @@ export function getFlowDisplayItems(messages: Message[]): DisplayItem[] {
 	}
 	return items;
 }
+
+/** Extract the last tool call from message history. */
+export function getLastToolCall(messages: Message[]): DisplayItem | undefined {
+	for (let i = messages.length - 1; i >= 0; i--) {
+		const msg = messages[i];
+		if (msg.role === "assistant") {
+			for (let j = msg.content.length - 1; j >= 0; j--) {
+				const part = msg.content[j];
+				if (part.type === "toolCall") {
+					const name = part.name ?? part.toolName ?? "unknown";
+					const args = (part.arguments ?? part.input ?? {}) as Record<string, unknown>;
+					return { type: "toolCall", name, args };
+				}
+			}
+		}
+	}
+	return undefined;
+}
+
+/** Extract the last assistant text from message history. */
+export function getLastAssistantText(messages: Message[]): string {
+	for (let i = messages.length - 1; i >= 0; i--) {
+		const msg = messages[i];
+		if (msg.role === "assistant") {
+			for (let j = msg.content.length - 1; j >= 0; j--) {
+				const part = msg.content[j];
+				if (part.type === "text" && part.text.trim()) {
+					return part.text.trim();
+				}
+			}
+		}
+	}
+	return "";
+}