pi-agent-flow 0.1.0-alpha.1 → 0.1.0-alpha.3

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 } from "./runner-events.js";
+import { processFlowJsonLine, drainStreamingText } from "./runner-events.js";
 import {
 	type SingleResult,
 	type FlowDetails,
@@ -224,11 +224,12 @@ export async function runFlow(opts: RunFlowOptions): Promise<SingleResult> {
 	};
 
 	const emitUpdate = () => {
+		const streaming = drainStreamingText(result);
 		onUpdate?.({
 			content: [
 				{
 					type: "text",
-					text: getFlowOutput(result.messages) || "(running...)",
+					text: streaming || getFlowOutput(result.messages) || "(running...)",
 				},
 			],
 			details: makeDetails([result]),

package/index.ts

@@ -352,16 +352,6 @@ export default function (pi: ExtensionAPI) {
 
 		const discovery = discoverFlows(ctx.cwd, "all");
 		discoveredFlows = discovery.flows;
-
-		if (discoveredFlows.length > 0 && ctx.hasUI) {
-			const list = discoveredFlows
-				.map((f) => `  - ${f.name} (${f.source})`)
-				.join("\n");
-			ctx.ui.notify(
-				`Found ${discoveredFlows.length} flow(s):\n${list}`,
-				"info",
-			);
-		}
 	});
 
 	// Inject available flows into the system prompt
@@ -389,9 +379,8 @@ Multiple independent flows? Batch them into one call:
 
 Each call renders as:
 
-routing to:
-  • flow [explore] — Map the full directory structure...
-  • flow [review] — Audit security and quality...
+• flow [explore] — Map the full directory structure...
+• flow [review] — Audit security and quality...
 
 Each flow returns:
 
@@ -499,10 +488,12 @@ flow [type] accomplished
 					};
 				}
 
-				const emitProgress = () => {
+				let lastStreamingText = "";
+				const emitProgress = (streamingText?: string) => {
 					if (!onUpdate) return;
+					if (streamingText) lastStreamingText = streamingText;
 					onUpdate({
-						content: [{ type: "text", text: `Flow: ${allResults.filter((r) => r.exitCode !== -1).length}/${allResults.length} done` }],
+						content: [{ type: "text", text: lastStreamingText || "" }],
 						details: makeDetails([...allResults]),
 					});
 				};
@@ -537,7 +528,7 @@ flow [type] accomplished
 							onUpdate: (partial) => {
 								if (partial.details?.results[0]) {
 									allResults[index] = partial.details.results[0];
-									emitProgress();
+									emitProgress(partial.content?.[0]?.text);
 								}
 							},
 							makeDetails,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-agent-flow",
-  "version": "0.1.0-alpha.1",
+  "version": "0.1.0-alpha.3",
   "description": "Flow-state delegation extension for Pi coding agent.",
   "type": "module",
   "main": "index.ts",
@@ -36,10 +36,11 @@
   },
   "scripts": {
     "lint": "tsc --noEmit",
-    "test": "echo \"no tests yet\" && exit 0"
+    "test": "vitest run"
   },
   "devDependencies": {
-    "typescript": "^5.0.0"
+    "typescript": "^5.0.0",
+    "vitest": "^3.2.4"
   },
   "peerDependencies": {
     "@mariozechner/pi-agent-core": ">=0.37.0",

package/render.ts

@@ -20,31 +20,7 @@ import {
 	isFlowError,
 	isFlowSuccess,
 } from "./types.js";
-
-// ---------------------------------------------------------------------------
-// Formatting helpers
-// ---------------------------------------------------------------------------
-
-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 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(" ");
-}
+import { formatTokens, formatFlowUsage, formatCompactStats, truncateChars, tailText } from "./render-utils.js";
 
 function shortenPath(p: string): string {
 	const home = os.homedir();
@@ -140,25 +116,8 @@ function truncateText(text: string): string {
 export function renderFlowCall(args: Record<string, any>, theme: { fg: ThemeFg; bold: (s: string) => string }): Text {
 	const flows = args.flow as Array<{ type: string; intent: string }> | undefined;
 
-	if (flows && flows.length > 0) {
-		if (flows.length === 1) {
-			const f = flows[0];
-			const text =
-				theme.fg("toolTitle", "routing to ") +
-				theme.fg("accent", `flow [${f.type}]`) +
-				theme.fg("dim", ` — ${truncateText(f.intent)}`);
-			return new Text(text, 0, 0);
-		}
-
-		let text = theme.fg("toolTitle", "routing to:");
-		for (const f of flows) {
-			text +=
-				`\n  ${theme.fg("muted", "•")} ` +
-				theme.fg("accent", `flow [${f.type}]`) +
-				theme.fg("dim", ` — ${truncateText(f.intent)}`);
-		}
-		return new Text(text, 0, 0);
-	}
+	// Minimal — renderFlowResult owns the full display
+	return new Text("", 0, 0);
 
 	return new Text(theme.fg("muted", "(empty flow call)"), 0, 0);
 }
@@ -173,13 +132,14 @@ export function renderFlowResult(
 	theme: { fg: ThemeFg; bold: (s: string) => string },
 ): Container | Text {
 	const details = result.details as FlowDetails | undefined;
+	const streamingText = result.content?.[0]?.type === "text" ? result.content[0].text : undefined;
+
 	if (!details || details.results.length === 0) {
-		const first = result.content[0];
-		return new Text(first?.type === "text" && first.text ? first.text : "(no output)", 0, 0);
+		return new Text(streamingText || "", 0, 0);
 	}
 
 	if (details.results.length === 1) {
-		return renderSingleFlowResult(details.results[0], expanded, theme);
+		return renderSingleFlowResult(details.results[0], expanded, theme, streamingText);
 	}
 
 	return renderMultiFlowResult(details, expanded, theme);
@@ -193,6 +153,7 @@ function renderSingleFlowResult(
 	r: SingleResult,
 	expanded: boolean,
 	theme: { fg: ThemeFg; bold: (s: string) => string },
+	streamingText?: string,
 ): Container | Text {
 	const error = isFlowError(r);
 	const icon = flowStatusIcon(r, theme);
@@ -202,7 +163,7 @@ function renderSingleFlowResult(
 	if (expanded) {
 		return renderFlowExpanded(r, icon, error, displayItems, flowOutput, theme);
 	}
-	return renderFlowCollapsed(r, icon, error, flowOutput, theme);
+	return renderFlowCollapsed(r, icon, error, flowOutput, theme, streamingText);
 }
 
 function renderFlowExpanded(
@@ -257,24 +218,34 @@ function renderFlowExpanded(
 	return container;
 }
 
+
+
 function renderFlowCollapsed(
 	r: SingleResult,
 	icon: string,
 	error: boolean,
 	flowOutput: string,
 	theme: { fg: ThemeFg; bold: (s: string) => string },
+	streamingText?: string,
 ): Text {
-	const usageStr = formatFlowUsage(r.usage, r.model);
-	let text = `${icon} ${theme.fg("toolTitle", theme.bold(`[${r.type}]`))}${theme.fg("muted", ` (${r.agentSource})`)}`;
-	if (usageStr) text += `   ${theme.fg("dim", usageStr)}`;
+	const stats = formatCompactStats(r.usage, r.model);
+	let text = `${theme.fg("accent", r.type)} ─ ${theme.fg("dim", stats)}`;
 	if (error && r.stopReason) text += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
 
-	if (error && r.errorMessage) {
-		text += `\n${theme.fg("error", `Error: ${r.errorMessage}`)}`;
-	} else if (flowOutput) {
-		text += `\n${renderFlowReport(truncateText(flowOutput), theme)}`;
-	} else {
-		text += `\n${theme.fg(error ? "error" : "muted", getFlowSummaryText(r))}`;
+	// Intent line
+	const hasOutput = !!(flowOutput || streamingText || (error && r.errorMessage));
+	if (r.intent) {
+		const prefix = hasOutput ? "├" : "└";
+		text += `\n${theme.fg("muted", `${prefix}─ int:`)} ${theme.fg("dim", truncateChars(r.intent, 40))}`;
+	}
+
+	// Output line
+	if (flowOutput) {
+		text += `\n${theme.fg("muted", "└─ msg:")} ${renderFlowReport(truncateChars(flowOutput, 25), theme)}`;
+	} else if (streamingText) {
+		text += `\n${theme.fg("muted", "└─ msg:")} ${theme.fg("dim", tailText(streamingText, 40))}`;
+	} else if (error && r.errorMessage) {
+		text += `\n${theme.fg("muted", "└─ msg:")} ${theme.fg("error", truncateChars(r.errorMessage, 25))}`;
 	}
 
 	return new Text(text, 0, 0);
@@ -358,15 +329,12 @@ function renderMultiFlowCollapsed(
 	let text = `${icon} ${theme.fg("toolTitle", theme.bold("flow "))}${theme.fg("accent", `${successCount}/${results.length} flows`)}`;
 
 	for (const r of results) {
-		const rIcon = flowStatusIcon(r, theme);
 		const flowOutput = getFlowOutput(r.messages);
 		const usageStr = formatFlowUsage(r.usage, r.model);
-		text += `\n\n${theme.fg("muted", "─── ")}${theme.fg("accent", `[${r.type}]`)} ${rIcon}`;
-		if (usageStr) text += `   ${theme.fg("dim", usageStr)}`;
+		text += `\n${theme.fg("accent", `flow [${r.type}]`)}`;
+		if (usageStr) text += ` ${theme.fg("dim", usageStr)}`;
 		if (flowOutput) {
 			text += `\n${renderFlowReport(truncateText(flowOutput), theme)}`;
-		} else {
-			text += `\n${theme.fg("muted", getFlowSummaryText(r))}`;
 		}
 	}
 

package/runner-events.js

@@ -14,6 +14,51 @@ function getSeenFlowMessageSignatures(result) {
   return result.__seenMessageSignatures;
 }
 
+function getStreamingTextBuffer(result) {
+  if (!Object.prototype.hasOwnProperty.call(result, "__streamingTextBuffer")) {
+    Object.defineProperty(result, "__streamingTextBuffer", {
+      value: "",
+      enumerable: false,
+      configurable: false,
+      writable: true,
+    });
+    Object.defineProperty(result, "__lastEmittedWordCount", {
+      value: 0,
+      enumerable: false,
+      configurable: false,
+      writable: true,
+    });
+  }
+  return result.__streamingTextBuffer;
+}
+
+/**
+ * Drain the accumulated streaming text buffer and return it.
+ * Updates the last-emitted word count for threshold tracking.
+ */
+export function drainStreamingText(result) {
+  const buf = getStreamingTextBuffer(result);
+  if (!buf) return "";
+  result.__streamingTextBuffer = "";
+  result.__lastEmittedWordCount = 0;
+  return buf;
+}
+
+/**
+ * Accumulate a text or thinking delta into the streaming buffer.
+ * Returns true if the caller should emit an update.
+ */
+function accumulateStreamingDelta(result, delta) {
+  if (!delta) return false;
+  const buf = getStreamingTextBuffer(result);
+  result.__streamingTextBuffer = buf + delta;
+  if (result.__streamingTextBuffer.length - result.__lastEmittedWordCount >= 40) {
+    result.__lastEmittedWordCount = result.__streamingTextBuffer.length;
+    return true;
+  }
+  return false;
+}
+
 function stableStringify(value) {
   if (value === null || typeof value !== "object") {
     return JSON.stringify(value);
@@ -98,6 +143,18 @@ export function processFlowEvent(event, result) {
       result.sawAgentEnd = true;
       return addFlowAssistantMessages(result, event.messages);
 
+    case "message_update": {
+      const evt = event.assistantMessageEvent;
+      if (!evt || typeof evt !== "object") return false;
+      if (evt.type === "text_delta") {
+        return accumulateStreamingDelta(result, evt.delta);
+      }
+      if (evt.type === "thinking_delta") {
+        return accumulateStreamingDelta(result, evt.delta);
+      }
+      return false;
+    }
+
     default:
       return false;
   }