pi-agent-flow 1.8.0 → 1.8.1

package/package.json

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

package/src/index.ts

@@ -634,7 +634,7 @@ function sanitizeForkSnapshot(snapshot: string | null, cache: Map<string, Compre
 
 function computeActiveTools(optimize: boolean): string[] {
 	return optimize
-		? ["batch_read", "batch", "batch_bash_poll", "flow"]
+		? ["batch_read", "batch_bash_poll", "flow"]
 		: ["read", "write", "edit", "batch", "bash", "flow", "web"];
 }
 

package/src/render-utils.ts

@@ -51,13 +51,16 @@ export function formatCompactStats(
 	usage: Partial<UsageStats>,
 	model?: string,
 	maxWidth?: number,
-	options: { skipTokens?: boolean } = {},
+	options: { skipTokens?: boolean; skipContext?: boolean; hideModel?: boolean } = {},
 ): string {
 	const tokenParts = [`↑ ${formatFixedTokens(usage.input || 0)}`, `↓ ${formatFixedTokens(usage.output || 0)}`];
-	const runtimeParts = [`tps: ${formatTps(usage.smoothedTps)}`, `ctx: ${formatFixedTokens(usage.contextTokens || 0)}`];
+	let runtimeParts = [`tps: ${formatTps(usage.smoothedTps)}`];
+	if (!options.skipContext) {
+		runtimeParts.push(`ctx: ${formatFixedTokens(usage.contextTokens || 0)}`);
+	}
 	const parts = options.skipTokens ? runtimeParts : [...tokenParts, ...runtimeParts];
 
-	const displayModel = model ? model.replace(/^[^/]+\//, "") : undefined;
+	const displayModel = options.hideModel ? undefined : (model ? model.replace(/^[^/]+\//, "") : undefined);
 	let result = parts.join(" · ") + (displayModel ? ` · ${displayModel}` : "");
 	if (maxWidth && visibleLength(result) > maxWidth) {
 		// Drop model first.

package/src/render.ts

@@ -325,9 +325,10 @@ function renderFlowCollapsed(
 ): Container {
 	const container = new Container();
 	const maxWidth = process.stdout.columns ?? 80;
-	const stats = formatCompactStats(r.usage, r.model, maxWidth, { skipTokens: true });
+	const stats = formatCompactStats(r.usage, r.model, maxWidth, { skipTokens: true, skipContext: true, hideModel: true });
 	const typeName = formatCollapsedFlowHeaderTypeName(r.type);
-	let header = `${theme.fg("accent", theme.bold(typeName))}${theme.fg("dim", `· ${stats}`)}`;
+	const modelLabel = r.model ? r.model.replace(/^[^/]+\//, "") : "";
+	let header = `${theme.fg("accent", theme.bold(typeName))}${theme.fg("dim", modelLabel ? ` ${modelLabel} - ` : " ")}${theme.fg("dim", stats)}`;
 	if (error && r.stopReason) header += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
 	container.addChild(new TruncatedText(header, 0, 0));
 
@@ -457,13 +458,14 @@ function renderActivityPanel(
 	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, maxWidth, { skipTokens: true });
+		const stats = formatCompactStats(r.usage, r.model, maxWidth, { skipTokens: true, skipContext: true, hideModel: true });
 		const error = isFlowError(r);
 		const typeName = formatCollapsedFlowHeaderTypeName(r.type);
 
 		// Header line
 		const headerPrefix = isLast ? "└─" : "├─";
-		let headerLine = `${theme.fg("dim", headerPrefix)} ${theme.fg("accent", theme.bold(typeName))}${theme.fg("dim", `· ${stats}`)}`;
+		const modelLabel = r.model ? r.model.replace(/^[^/]+\//, "") : "";
+		let headerLine = `${theme.fg("dim", headerPrefix)} ${theme.fg("accent", theme.bold(typeName))}${theme.fg("dim", modelLabel ? ` ${modelLabel} - ` : " ")}${theme.fg("dim", stats)}`;
 		if (error && r.stopReason) {
 			headerLine += ` ${theme.fg("error", `[${r.stopReason}]`)}`;
 		}