pi-agent-flow 1.0.4 → 1.0.6

package/agents/architect.md

@@ -2,7 +2,7 @@
 name: architect
 description: Plan structure, break down requirements, design solutions
 tools: read, bash, find, grep, ls
-maxDepth: 2
+maxDepth: 0
 ---
 
 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.

package/agents/code.md

@@ -2,7 +2,7 @@
 name: code
 description: Implement features, fix bugs, write tests, deploy, and ship
 tools: read, write, edit, bash, find, grep, ls
-maxDepth: 2
+maxDepth: 0
 ---
 
 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.

package/agents/review.md

@@ -2,7 +2,7 @@
 name: review
 description: Audit security, quality, correctness
 tools: read, bash, find, grep, ls
-maxDepth: 2
+maxDepth: 0
 ---
 
 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.

package/flow.ts

@@ -49,7 +49,7 @@ function mergeStreamingUsage(
 		...actual,
 		...(estimatedOutputTokens > 0 ? { output: Math.max(actual.output, estimatedOutputTokens) } : {}),
 		...(ctxEstimate > 0 ? { contextTokens: Math.max(actual.contextTokens, ctxEstimate) } : {}),
-		...(smoothedTps > 0 ? { smoothedTps } : {}),
+		...(smoothedTps > 0 ? { smoothedTps: Math.max(actual.smoothedTps ?? 0, smoothedTps) } : {}),
 	};
 }
 
@@ -104,6 +104,8 @@ function buildFlowArgs(
 	intent: string,
 	forkSessionPath: string | null,
 	tieredModels?: { lite?: string; flash?: string; full?: string },
+	parentDepth: number = 0,
+	maxDepth: number = 0,
 ): string[] {
 	const args: string[] = [
 		"--mode",
@@ -138,17 +140,46 @@ function buildFlowArgs(
 	// No --append-system-prompt: child inherits parent's system prompt for cache hits.
 	// Flow instructions go in the intent message instead.
 
-	const flowDirectives =
-		`<flow-directive>\n` +
-		`The conversation history above is background context — use it for reference, but your sole focus is the intent below, and try to execute EXACTLY as requested in the intent.\n` +
-		`</flow-directive>`;
-
-	const flowInstructions = flow.systemPrompt.trim()
-		? `\n\n<system-directive>\n${flow.systemPrompt.trim()}\n</system-directive>`
+	const currentDepth = Math.max(0, Math.floor(parentDepth)) + 1;
+	const effectiveMaxDepth = Math.max(0, Math.floor(maxDepth));
+	const canDelegate = currentDepth < effectiveMaxDepth;
+	const availableTools = flow.tools?.join(", ") ?? "all";
+
+	// Phase 1: Context seal — sharp boundary declaring history sealed
+	const contextSeal =
+		`<context-seal>\n` +
+		`The conversation above is sealed — it is your session history for situational awareness only.\n` +
+		`Your task begins NOW. Do not respond to or continue anything from the history.\n` +
+		`</context-seal>`;
+
+	// Phase 2: Activation — role, tools, depth, delegation rules (dynamically generated)
+	const delegationRule = canDelegate
+		? `You may delegate to sub-flows (depth ${currentDepth}/${effectiveMaxDepth}).`
+		: `You may NOT delegate to sub-flows (depth limit reached).`;
+
+	const activation =
+		`\n\n<activation flow="${flow.name}" depth="${currentDepth}" tools="${availableTools}">\n` +
+		`You are a [${flow.name}] agent operating at depth ${currentDepth}.\n` +
+		`Available tools: ${availableTools}.\n` +
+		`${delegationRule}\n` +
+		`Do not attempt to use any tool outside the available set — it will fail.\n` +
+		`</activation>`;
+
+	// Phase 3: Directive — the flow's system prompt (renamed from <system-directive>)
+	const directive = flow.systemPrompt.trim()
+		? `\n\n<directive>\n${flow.systemPrompt.trim()}\n</directive>`
 		: "";
 
+	// Phase 4: Mission — the intent wrapped with execution contract
+	const mission =
+		`\n\n<mission>\n` +
+		`${intent}\n` +
+		`\nExecute this mission. Use only your available tools. If blocked, report why — do not guess.\n` +
+		`Follow the output format specified in your directive.\n` +
+		`</mission>`;
+
 	// -p must immediately precede the prompt so the CLI parser binds it correctly
-	args.push("-p", `${flowDirectives}${flowInstructions}\n\nIntent: ${intent}`);
+	args.push("-p", `${contextSeal}${activation}${directive}${mission}`);
 	return args;
 }
 
@@ -269,6 +300,8 @@ export async function runFlow(opts: RunFlowOptions): Promise<SingleResult> {
 			intent,
 			forkSessionTmpPath,
 			opts.tieredModels,
+			parentDepth,
+			maxDepth,
 		);
 		let wasAborted = false;
 
@@ -403,6 +436,15 @@ export async function runFlow(opts: RunFlowOptions): Promise<SingleResult> {
 		});
 
 		result.exitCode = exitCode;
+
+		// Persist final smoothed TPS into the result's usage so it survives after streaming ends.
+		// During streaming, emitUpdate() only merges smoothedTps into a temporary display object;
+		// without this, result.usage.smoothedTps stays at 0 and the UI shows a dash.
+		const finalSmoothedTps = drainSmoothedTps(result);
+		if (finalSmoothedTps > 0) {
+			result.usage.smoothedTps = finalSmoothedTps;
+		}
+
 		return normalizeFlowResult(result, wasAborted);
 	} finally {
 		cleanupFlowTempDir(forkSessionTmpDir);

package/index.ts

@@ -25,7 +25,7 @@ import {
 // Limits
 // ---------------------------------------------------------------------------
 
-const DEFAULT_MAX_DELEGATION_DEPTH = 3;
+const DEFAULT_MAX_DELEGATION_DEPTH = 0;
 const DEFAULT_PREVENT_CYCLE_DELEGATION = true;
 const FLOW_DEPTH_ENV = "PI_FLOW_DEPTH";
 const FLOW_MAX_DEPTH_ENV = "PI_FLOW_MAX_DEPTH";

package/package.json

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

package/runner-events.js

@@ -52,7 +52,7 @@ export function drainStreamingText(result) {
 const CHARS_PER_TOKEN = 4;
 
 /** EMA smoothing factor for tokens-per-second (higher = more responsive). */
-const EMA_ALPHA = 0.3;
+const EMA_ALPHA = 0.15;
 
 function getStreamingEstimate(result) {
   if (!Object.prototype.hasOwnProperty.call(result, "__streamingEstimate")) {