pi-agent-flow 0.1.0-alpha.4 → 0.1.0-alpha.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-agent-flow",
-  "version": "0.1.0-alpha.4",
+  "version": "0.1.0-alpha.6",
   "description": "Flow-state delegation extension for Pi coding agent.",
   "type": "module",
   "main": "index.ts",
@@ -11,6 +11,7 @@
     "runner-cli.js",
     "runner-events.js",
     "render.ts",
+    "render-utils.ts",
     "types.ts",
     "agents/",
     "README.md"

package/render-utils.ts

@@ -0,0 +1,53 @@
+/**
+ * Pure utility functions for rendering — extracted for testability.
+ */
+
+import type { UsageStats } from "./types.js";
+
+export 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`;
+}
+
+export function formatFlowUsage(usage: Partial<UsageStats>, model?: string): string {
+	const parts: string[] = [];
+	if (usage.toolCalls && usage.toolCalls > 0) parts.push(`${usage.toolCalls} calls`);
+	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(`CR:${formatTokens(usage.cacheRead)}`);
+	if (usage.cacheWrite) parts.push(`CW:${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(" ");
+}
+
+export function formatCompactStats(usage: Partial<UsageStats>, model?: string): string {
+	const io: string[] = [];
+	if (usage.input) io.push(`↑${formatTokens(usage.input)}`);
+	if (usage.output) io.push(`↓${formatTokens(usage.output)}`);
+	const meta: string[] = [];
+	if (usage.cacheRead) meta.push(`CR:${formatTokens(usage.cacheRead)}`);
+	if (usage.contextTokens && usage.contextTokens > 0) meta.push(`ctx:${formatTokens(usage.contextTokens)}`);
+	if (model) meta.push(model);
+	const parts: string[] = [];
+	if (io.length) parts.push(io.join(" "));
+	if (meta.length) parts.push(meta.join(" "));
+	return parts.join(" │ ");
+}
+
+export function truncateChars(text: string, max: number): string {
+	if (text.length <= max) return text;
+	const head = Math.ceil(max * 0.6);
+	const tail = max - head - 3;
+	return text.slice(0, head) + " ... " + text.slice(-tail);
+}
+
+export function tailText(text: string, max: number): string {
+	const flat = text.replace(/[\n\r\t]+/g, " ").replace(/ +/g, " ").trim();
+	if (flat.length <= max) return flat;
+	return flat.slice(-max);
+}

package/render.ts

@@ -235,10 +235,8 @@ function renderFlowCollapsed(
 	if (error && r.stopReason) text += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
 
 	// Intent line
-	const hasOutput = !!(flowOutput || streamingText || (error && r.errorMessage));
 	if (r.intent) {
-		const prefix = hasOutput ? "├" : "└";
-		text += `\n${theme.fg("dim", `${prefix}─ int:`)} ${theme.fg("dim", truncateChars(r.intent, 40))}`;
+		text += `\n${theme.fg("dim", `├─ int:`)} ${theme.fg("dim", truncateChars(r.intent, 40))}`;
 	}
 
 	// Output line
@@ -248,6 +246,8 @@ function renderFlowCollapsed(
 		text += `\n${theme.fg("dim", "└─ msg:")} ${theme.fg("dim", tailText(streamingText, 40))}`;
 	} else if (error && r.errorMessage) {
 		text += `\n${theme.fg("dim", "└─ msg:")} ${theme.fg("error", truncateChars(r.errorMessage, 25))}`;
+	} else {
+		text += `\n${theme.fg("dim", "└─ msg:")} ${theme.fg("dim", "[n/a]")}`;
 	}
 
 	return new Text(text, 0, 0);
@@ -270,7 +270,7 @@ function renderMultiFlowResult(
 	if (expanded) {
 		return renderMultiFlowExpanded(results, successCount, icon, theme);
 	}
-	return renderMultiFlowCollapsed(results, successCount, icon, theme);
+	return renderMultiFlowCollapsed(results, theme);
 }
 
 function renderMultiFlowExpanded(
@@ -324,11 +324,9 @@ function renderMultiFlowExpanded(
 
 function renderMultiFlowCollapsed(
 	results: SingleResult[],
-	successCount: number,
-	icon: string,
 	theme: FlowTheme,
 ): Text {
-	let text = `${icon} ${theme.fg("toolTitle", theme.bold("flow "))}${theme.fg("accent", `${successCount}/${results.length} flows`)}`;
+	let text = "";
 
 	for (let i = 0; i < results.length; i++) {
 		const r = results[i];
@@ -339,22 +337,22 @@ function renderMultiFlowCollapsed(
 
 		// Header line
 		const headerPrefix = isLast ? "└─" : "├─";
-		let line = `\n${theme.fg("dim", headerPrefix)} ${theme.bg("selectedBg", theme.fg("accent", theme.bold(r.type)))}`;
+		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)}`;
 		}
 		if (error && r.stopReason) {
 			line += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
 		}
+		if (i > 0) text += "\n";
 		text += line;
 
 		// Continuation indent for sub-lines
 		const indent = isLast ? "   " : "│  ";
 
 		// Intent line
-		const hasOutput = !!(flowOutput || (error && r.errorMessage));
 		if (r.intent) {
-			const intentPrefix = hasOutput ? "├─" : "└─";
+			const intentPrefix = hasMsg ? "├─" : "└─";
 			text += `\n${theme.fg("dim", indent + intentPrefix + " int:")} ${theme.fg("dim", truncateChars(r.intent, 40))}`;
 		}
 
@@ -363,6 +361,8 @@ function renderMultiFlowCollapsed(
 			text += `\n${theme.fg("dim", indent + "└─ msg:")} ${renderFlowReport(truncateChars(flowOutput, 25), theme)}`;
 		} else if (error && r.errorMessage) {
 			text += `\n${theme.fg("dim", indent + "└─ msg:")} ${theme.fg("error", truncateChars(r.errorMessage, 25))}`;
+		} else {
+			text += `\n${theme.fg("dim", indent + "└─ msg:")} ${theme.fg("dim", "[n/a]")}`;
 		}
 	}