pi-agent-flow 0.3.0 → 0.4.2

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/>
 
 ---
 
@@ -82,9 +78,9 @@ description: Discover files, trace code paths, map architecture
 tools: read, bash
 ---
 
-You are the explore flow — your mission is discovery. Stay focused on your intent at all times.
+During this explore flow — your mission is discovery. Stay focused on your intent at all times.
 
-When your mission is accomplished, end your response with:
+When accomplished, end your response with:
 
 flow [explore] accomplished
 
@@ -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/agents/architect.md

@@ -5,7 +5,7 @@ tools: read, bash, find, grep, ls
 maxDepth: 2
 ---
 
-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.
+During this 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
@@ -18,7 +18,7 @@ Principles:
 - Design for 10x, build for 1x
 - No tech debt — do it right or don't
 
-When your mission is accomplished, end your response with:
+When accomplished, end your response with:
 
 flow [architect] accomplished
 

package/agents/brainstorm.md

@@ -6,14 +6,14 @@ inheritContext: false
 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.
+During this 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:
+When accomplished, end your response with:
 
 flow [brainstorm] accomplished
 

package/agents/code.md

@@ -5,7 +5,7 @@ tools: read, write, edit, bash, find, grep, ls
 maxDepth: 2
 ---
 
-You are the code flow — your mission is to build and ship. 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.
+During this code flow — your mission is to build and ship. 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
@@ -32,7 +32,7 @@ Shipping Guidelines:
 
 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:
+When accomplished, end your response with:
 
 flow [code] accomplished
 

package/agents/debug.md

@@ -5,7 +5,7 @@ tools: read, bash, find, grep, ls
 maxDepth: 0
 ---
 
-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.
+During this 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
@@ -18,7 +18,7 @@ Rules:
 - Read logs before reading code — symptoms point to cause.
 - Don't suggest fixes until root cause is confirmed.
 
-When your mission is accomplished, end your response with:
+When accomplished, end your response with:
 
 flow [debug] accomplished
 

package/agents/explore.md

@@ -5,7 +5,7 @@ tools: read, bash, find, grep, ls
 maxDepth: 0
 ---
 
-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.
+During this 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/ls before reading entire files — be efficient.
@@ -13,7 +13,7 @@ Rules:
 - Show actual code/data, not excessive summaries.
 - If not found, say so immediately — don't guess.
 
-When your mission is accomplished, end your response with:
+When accomplished, end your response with:
 
 flow [explore] accomplished
 

package/agents/review.md

@@ -5,7 +5,7 @@ tools: read, bash, find, grep, ls
 maxDepth: 2
 ---
 
-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.
+During this 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
@@ -19,7 +19,7 @@ Rules:
 - 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:
+When accomplished, end your response with:
 
 flow [review] accomplished
 

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.3.0",
+  "version": "0.4.2",
   "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. */
@@ -92,14 +76,31 @@ export function visibleLength(text: string): number {
 	return text.replace(ANSI_RE, "").length;
 }
 
+/**
+ * Compute the remaining visible-character budget for a line,
+ * given the length of its prefix (indent + label + space).
+ * Respects `process.stdout.columns` with a floor of 40 and default of 80.
+ */
+export function getTruncationBudget(prefixLength: number): number {
+	const cols = process.stdout.columns ?? 80;
+	const width = Math.max(cols, 40);
+	return Math.max(width - prefixLength, 1);
+}
+
 /**
  * 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
 
@@ -115,9 +116,12 @@ function truncateAnsi(text: string, max: number): string {
 			if (src[i] === "\x1b" && src[i + 1] === "[") {
 				const end = src.indexOf("m", i + 2);
 				if (end !== -1) {
-					raw += src.slice(i, end + 1);
-					i = end + 1;
-					continue;
+					const seq = src.slice(i, end + 1);
+					if (/^\x1b\[[0-9;]*m$/.test(seq)) {
+						raw += seq;
+						i = end + 1;
+						continue;
+					}
 				}
 			}
 			raw += src[i];
@@ -139,9 +143,12 @@ function truncateAnsi(text: string, max: number): string {
 			if (src[i] === "m") {
 				const escStart = src.lastIndexOf("\x1b[", i);
 				if (escStart !== -1 && escStart < i) {
-					// This is an ANSI sequence — don't count, skip past it
-					i = escStart - 1;
-					continue;
+					const seq = src.slice(escStart, i + 1);
+					if (/^\x1b\[[0-9;]*m$/.test(seq)) {
+						// This is an ANSI sequence — don't count, skip past it
+						i = escStart - 1;
+						continue;
+					}
 				}
 			}
 			visible++;
@@ -173,8 +180,11 @@ export function tailText(text: string, max: number): string {
 		if (flat[i] === "m") {
 			const escStart = flat.lastIndexOf("\x1b[", i);
 			if (escStart !== -1 && escStart < i) {
-				i = escStart - 1;
-				continue;
+				const seq = flat.slice(escStart, i + 1);
+				if (/^\x1b\[[0-9;]*m$/.test(seq)) {
+					i = escStart - 1;
+					continue;
+				}
 			}
 		}
 		visible++;

package/render.ts

@@ -22,7 +22,7 @@ import {
 	isFlowError,
 	isFlowSuccess,
 } from "./types.js";
-import { formatTokens, formatFixedTokens, formatFlowUsage, formatCompactStats, formatExpandedStats, formatFlowTypeName, truncateChars, tailText } from "./render-utils.js";
+import { formatFixedTokens, formatCompactStats, formatFlowTypeName, truncateChars, tailText, getTruncationBudget } 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 Text(truncateChars(header, maxWidth), 0, 0));
 
 	// dir: line (intent/objective)
 	if (r.intent) {
-		text += `\n${theme.fg("dim", "├─ dir:")} ${theme.fg("dim", truncateChars(r.intent, 50))}`;
+		container.addChild(new Text(`${theme.fg("dim", "├─ dir:")} ${theme.fg("dim", truncateChars(r.intent, getTruncationBudget(8)))}`, 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, 50)}`;
+		container.addChild(new Text(`${theme.fg("dim", "├─ exe:")} ${truncateChars(exeStr, getTruncationBudget(8))}`, 0, 0));
 	}
 
 	// log: line (last assistant text or streaming)
 	if (flowOutput) {
-		text += `\n${theme.fg("dim", "└─ log:")} ${theme.fg("dim", truncateChars(flowOutput, 50))}`;
+		container.addChild(new Text(`${theme.fg("dim", "└─ log:")} ${theme.fg("dim", truncateChars(flowOutput, getTruncationBudget(8)))}`, 0, 0));
 	} else if (streamingText) {
-		text += `\n${theme.fg("dim", "└─ log:")} ${theme.fg("dim", tailText(streamingText, 50))}`;
+		container.addChild(new Text(`${theme.fg("dim", "└─ log:")} ${theme.fg("dim", tailText(streamingText, getTruncationBudget(8)))}`, 0, 0));
 	} else if (error && r.errorMessage) {
-		text += `\n${theme.fg("dim", "└─ log:")} ${theme.fg("error", truncateChars(r.errorMessage, 50))}`;
+		container.addChild(new Text(`${theme.fg("dim", "└─ log:")} ${theme.fg("error", truncateChars(r.errorMessage, getTruncationBudget(8)))}`, 0, 0));
 	} else {
-		text += `\n${theme.fg("dim", "└─ log:")} ${theme.fg("dim", "[n/a]")}`;
+		container.addChild(new Text(`${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,29 +364,29 @@ function renderActivityPanel(
 		if (error && r.stopReason) {
 			headerLine += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
 		}
-		container.addChild(new Text(headerLine, 0, 0));
+		container.addChild(new Text(truncateChars(headerLine, maxWidth), 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, 50))}`, 0, 0));
+			container.addChild(new Text(`${theme.fg("dim", indent + "├─ dir:")} ${theme.fg("dim", truncateChars(r.intent, getTruncationBudget(11)))}`, 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, 50)}`, 0, 0));
+			container.addChild(new Text(`${theme.fg("dim", indent + "├─ exe:")} ${truncateChars(exeStr, getTruncationBudget(11))}`, 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));
+			container.addChild(new Text(`${theme.fg("dim", indent + "└─ log:")} ${theme.fg("dim", truncateChars(lastText, getTruncationBudget(11)))}`, 0, 0));
 		} else if (error && r.errorMessage) {
-			container.addChild(new Text(`${theme.fg("dim", indent + "└─ log:")} ${theme.fg("error", truncateChars(r.errorMessage, 50))}`, 0, 0));
+			container.addChild(new Text(`${theme.fg("dim", indent + "└─ log:")} ${theme.fg("error", truncateChars(r.errorMessage, getTruncationBudget(11)))}`, 0, 0));
 		} else {
 			container.addChild(new Text(`${theme.fg("dim", indent + "└─ log:")} ${theme.fg("dim", "[n/a]")}`, 0, 0));
 		}

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) {