pi-agent-flow 0.4.0 → 0.4.3

package/README.md

@@ -1,9 +1,5 @@
 <p align="center"><code>pi install npm:pi-agent-flow</code></p>
 <p align="center"><strong>Pi Agent Flow</strong> is a flow-state delegation extension for the <a href="https://pi.dev">Pi coding agent</a> that runs locally in your terminal.</p>
-<p align="center">
-  <img src="https://github.com/user-attachments/assets/pi-agent-flow-demo.png" alt="Pi Agent Flow demo" width="80%" />
-</p>
-<br/>
 
 ---
 
@@ -162,7 +158,6 @@ flow [explore] accomplished
 ## Docs
 
 - [**Pi Documentation**](https://pi.dev)
-- [**Contributing**](./docs/contributing.md) *(if available)*
 - [**License**](./LICENSE)
 
 This repository is licensed under the [MIT License](LICENSE).

package/index.ts

@@ -284,10 +284,6 @@ function makeFlowDetailsFactory(projectFlowsDir: string | null) {
 	});
 }
 
-function formatFlowNames(flows: FlowConfig[]): string {
-	return flows.map((f) => `${f.name} (${f.source})`).join(", ") || "none";
-}
-
 function getFlowCycleViolations(
 	requestedNames: Set<string>,
 	ancestorFlowStack: string[],

package/package.json

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

package/render-utils.ts

@@ -4,27 +4,6 @@
 
 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:${model}`);
-	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.
@@ -58,30 +37,35 @@ export function formatFlowTypeName(type: string): string {
 	return lower + ".".repeat(targetWidth - lower.length);
 }
 
-export function formatCompactStats(usage: Partial<UsageStats>, model?: string): string {
+export function formatCompactStats(usage: Partial<UsageStats>, model?: string, maxWidth?: number): string {
 	const parts: string[] = [];
 	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)}`);
+	let result = `[ ${parts.join(" ")} ]${model ? ` ─ ${model}` : ""}`;
+
+	if (maxWidth && visibleLength(result) > maxWidth) {
+		// Drop model first
+		let narrow = `[ ${parts.join(" ")} ]`;
+		if (visibleLength(narrow) <= maxWidth) return narrow;
+
+		// Drop context tokens
+		const narrowParts = parts.slice();
+		const ctxIndex = narrowParts.findIndex((p) => p.startsWith("ctx:"));
+		if (ctxIndex !== -1) narrowParts.splice(ctxIndex, 1);
+		narrow = `[ ${narrowParts.join(" ")} ]`;
+		if (visibleLength(narrow) <= maxWidth) return narrow;
 
-	const stats = `[${parts.join(" ")}]${model ? ` ─ ${model}` : ""}`;
-	const contextTokens = usage.contextTokens && usage.contextTokens > 0 ? `ctx:${formatFixedTokens(usage.contextTokens)}` : null;
+		// Bare minimum (just input/output)
+		narrow = `[ ${parts[0]} ${parts[1]} ]`;
+		if (visibleLength(narrow) <= maxWidth) return narrow;
 
-	return { stats, contextTokens };
+		return truncateChars(result, maxWidth);
+	}
+
+	return result;
 }
 
 /** Regex matching ANSI escape sequences. */
@@ -105,12 +89,18 @@ export function getTruncationBudget(prefixLength: number): number {
 
 /**
  * Truncate an ANSI-colored string to at most `max` visible characters,
- * preserving ANSI codes in the kept portions and appending a reset before
- * the ellipsis so colors don't bleed.
+ * preserving ANSI codes in the kept portions. Does not inject reset codes
+ * — the caller is responsible for closing any open styles.
  */
 function truncateAnsi(text: string, max: number): string {
 	if (visibleLength(text) <= max) return text;
 
+	if (max < 6) {
+		// Not enough room for " ... " — just truncate without ellipsis
+		const { raw } = takeVisible(text, max);
+		return raw;
+	}
+
 	const head = Math.ceil(max * 0.6);
 	const tail = max - head - 5; // 5 = " ... ".length
 

package/render.ts

@@ -7,7 +7,7 @@
 
 import * as os from "node:os";
 import { getMarkdownTheme } from "@mariozechner/pi-coding-agent";
-import { Container, Markdown, Spacer, Text } from "@mariozechner/pi-tui";
+import { Container, Markdown, Spacer, Text, TruncatedText } from "@mariozechner/pi-tui";
 import { getFlowSummaryText } from "./runner-events.js";
 import {
 	type DisplayItem,
@@ -22,7 +22,7 @@ import {
 	isFlowError,
 	isFlowSuccess,
 } from "./types.js";
-import { formatTokens, formatFixedTokens, formatFlowUsage, formatCompactStats, formatExpandedStats, formatFlowTypeName, truncateChars, tailText, getTruncationBudget } from "./render-utils.js";
+import { formatFixedTokens, formatCompactStats, formatFlowTypeName } from "./render-utils.js";
 
 function shortenPath(p: string): string {
 	const home = os.homedir();
@@ -217,8 +217,6 @@ function renderFlowExpanded(
 	return container;
 }
 
-
-
 function renderFlowCollapsed(
 	r: SingleResult,
 	icon: string,
@@ -226,36 +224,39 @@ function renderFlowCollapsed(
 	flowOutput: string,
 	theme: FlowTheme,
 	streamingText?: string,
-): Text {
-	const stats = formatCompactStats(r.usage, r.model);
+): Container {
+	const container = new Container();
+	const maxWidth = process.stdout.columns ?? 80;
+	const stats = formatCompactStats(r.usage, r.model, maxWidth);
 	const typeName = formatFlowTypeName(r.type);
-	let text = `${theme.fg("accent", theme.bold(typeName))} ${theme.fg("dim", "─")} ${theme.fg("dim", stats)}`;
-	if (error && r.stopReason) text += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
+	let header = `${theme.fg("accent", theme.bold(typeName))} ${theme.fg("dim", "─")} ${theme.fg("dim", stats)}`;
+	if (error && r.stopReason) header += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
+	container.addChild(new TruncatedText(header, 0, 0));
 
 	// dir: line (intent/objective)
 	if (r.intent) {
-		text += `\n${theme.fg("dim", "├─ dir:")} ${theme.fg("dim", truncateChars(r.intent, getTruncationBudget(8)))}`;
+		container.addChild(new TruncatedText(`${theme.fg("dim", "├─ dir:")} ${theme.fg("dim", r.intent)}`, 0, 0));
 	}
 
 	// 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, getTruncationBudget(8))}`;
+		container.addChild(new TruncatedText(`${theme.fg("dim", "├─ exe:")} ${exeStr}`, 0, 0));
 	}
 
 	// log: line (last assistant text or streaming)
 	if (flowOutput) {
-		text += `\n${theme.fg("dim", "└─ log:")} ${theme.fg("dim", truncateChars(flowOutput, getTruncationBudget(8)))}`;
+		container.addChild(new TruncatedText(`${theme.fg("dim", "└─ log:")} ${theme.fg("dim", flowOutput)}`, 0, 0));
 	} else if (streamingText) {
-		text += `\n${theme.fg("dim", "└─ log:")} ${theme.fg("dim", tailText(streamingText, getTruncationBudget(8)))}`;
+		container.addChild(new TruncatedText(`${theme.fg("dim", "└─ log:")} ${theme.fg("dim", streamingText)}`, 0, 0));
 	} else if (error && r.errorMessage) {
-		text += `\n${theme.fg("dim", "└─ log:")} ${theme.fg("error", truncateChars(r.errorMessage, getTruncationBudget(8)))}`;
+		container.addChild(new TruncatedText(`${theme.fg("dim", "└─ log:")} ${theme.fg("error", r.errorMessage)}`, 0, 0));
 	} else {
-		text += `\n${theme.fg("dim", "└─ log:")} ${theme.fg("dim", "[n/a]")}`;
+		container.addChild(new TruncatedText(`${theme.fg("dim", "└─ log:")} ${theme.fg("dim", "[n/a]")}`, 0, 0));
 	}
 
-	return new Text(text, 0, 0);
+	return container;
 }
 
 // ---------------------------------------------------------------------------
@@ -348,11 +349,12 @@ function renderActivityPanel(
 	theme: FlowTheme,
 ): Container {
 	const container = new Container();
+	const maxWidth = process.stdout.columns ?? 80;
 
 	for (let i = 0; i < results.length; i++) {
 		const r = results[i];
 		const isLast = i === results.length - 1;
-		const stats = formatCompactStats(r.usage, r.model);
+		const stats = formatCompactStats(r.usage, r.model, maxWidth);
 		const error = isFlowError(r);
 		const typeName = formatFlowTypeName(r.type);
 
@@ -362,40 +364,40 @@ function renderActivityPanel(
 		if (error && r.stopReason) {
 			headerLine += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
 		}
-		container.addChild(new Text(headerLine, 0, 0));
+		container.addChild(new TruncatedText(headerLine, 0, 0));
 
 		// Continuation indent for sub-lines
 		const indent = isLast ? "   " : "│  ";
 
 		// dir: line (intent/objective)
 		if (r.intent) {
-			container.addChild(new Text(`${theme.fg("dim", indent + "├─ dir:")} ${theme.fg("dim", truncateChars(r.intent, getTruncationBudget(11)))}`, 0, 0));
+			container.addChild(new TruncatedText(`${theme.fg("dim", indent + "├─ dir:")} ${theme.fg("dim", r.intent)}`, 0, 0));
 		}
 
 		// 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, getTruncationBudget(11))}`, 0, 0));
+			container.addChild(new TruncatedText(`${theme.fg("dim", indent + "├─ exe:")} ${exeStr}`, 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, getTruncationBudget(11)))}`, 0, 0));
+			container.addChild(new TruncatedText(`${theme.fg("dim", indent + "└─ log:")} ${theme.fg("dim", lastText)}`, 0, 0));
 		} else if (error && r.errorMessage) {
-			container.addChild(new Text(`${theme.fg("dim", indent + "└─ log:")} ${theme.fg("error", truncateChars(r.errorMessage, getTruncationBudget(11)))}`, 0, 0));
+			container.addChild(new TruncatedText(`${theme.fg("dim", indent + "└─ log:")} ${theme.fg("error", r.errorMessage)}`, 0, 0));
 		} else {
-			container.addChild(new Text(`${theme.fg("dim", indent + "└─ log:")} ${theme.fg("dim", "[n/a]")}`, 0, 0));
+			container.addChild(new TruncatedText(`${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));
+			container.addChild(new TruncatedText(theme.fg("dim", "│"), 0, 0));
 		}
 	}
 
-	container.addChild(new Text(theme.fg("muted", "(Ctrl+O to expand tool traces)"), 0, 0));
+	container.addChild(new TruncatedText(theme.fg("muted", "(Ctrl+O to expand tool traces)"), 0, 0));
 
 	return container;
 }

package/runner-events.js

@@ -174,7 +174,7 @@ function addFlowAssistantMessages(result, messages) {
   return changed;
 }
 
-export function processFlowEvent(event, result) {
+function processFlowEvent(event, result) {
   if (!event || typeof event !== "object") return false;
 
   switch (event.type) {