pi-prompt-template-model 0.8.2 → 0.9.1

package/CHANGELOG.md

@@ -2,6 +2,22 @@
 
 ## [Unreleased]
 
+## [0.9.1] - 2026-04-26
+
+### Fixed
+- `boomerang: true` prompt collapses now signal rewind to keep current files automatically, avoiding the restore-options prompt during collapse.
+
+## [0.9.0] - 2026-04-25
+
+### Added
+- Added `boomerang: true` prompt frontmatter so non-chain templates, including looped templates, can run through prompt-template-model and then collapse their execution context back to the pre-run branch.
+
+### Fixed
+- Migrated extension tool schemas from `@sinclair/typebox` to `typebox` 1.x so packaged installs follow Pi's current extension runtime contract.
+
+### Changed
+- Added `typebox` as a runtime dependency for packaged installs.
+
 ## [0.8.2] - 2026-04-21
 
 ### Added

package/README.md

@@ -73,6 +73,7 @@ All fields are optional. Templates that don't use any extension features (no `mo
 | `rotate` | `false` | When `true` and looping, cycle through models in the `model` list instead of using fallback semantics. Thinking levels can also be comma-separated to pair with each model. |
 | `fresh` | `false` | When looping, collapse the conversation between iterations to a brief summary instead of carrying the full context forward. Saves tokens on long loops. |
 | `converge` | `true` | When looping, stop early if an iteration makes no file changes. Set `false` to always run every iteration. |
+| `boomerang` | `false` | After a non-chain prompt finishes, collapse its execution context back to the branch point with a brief summary. Works with loops, including `fresh` loop summaries. Useful for review prompts like `/double-check`. |
 | `worktree` | `false` | When `true`, parallel delegated work runs in separate git worktrees. Valid on chain templates with `parallel()` steps, on delegated prompts with `parallel: N`, and on compare templates via `bestOfN.worktree`. |
 
 ### Delegation
@@ -291,7 +292,7 @@ This repo ships one example compare prompt under `examples/`:
 - `examples/best-of-n.md` installs as `/best-of-n`, runs in the current repo, and shows mixed workers, mixed reviewers, and an optional final apply phase.
 - Smoke test: `/best-of-n smoke test`.
 
-Install them manually from this repo checkout (or from the installed package directory):
+Install it manually from this repo checkout (or from the installed package directory):
 
 ```bash
 PTM_DIR=/path/to/pi-prompt-template-model

package/index.ts

@@ -16,7 +16,7 @@ import {
 	type SubagentOverride,
 } from "./args.js";
 import { parseChainSteps, parseChainDeclaration, type ChainStep, type ChainStepOrParallel, type ParallelChainStep } from "./chain-parser.js";
-import { generateChainStepSummary, generateIterationSummary, didIterationMakeChanges, getIterationEntries, wasIterationAborted } from "./loop-utils.js";
+import { generateBoomerangSummary, generateChainStepSummary, generateIterationSummary, didIterationMakeChanges, getIterationEntries, wasIterationAborted } from "./loop-utils.js";
 import { selectModelCandidate } from "./model-selection.js";
 import { notify, summarizePromptDiagnostics, diagnosticsFingerprint } from "./notifications.js";
 import { preparePromptExecution, renderPromptForResolvedModel } from "./prompt-execution.js";
@@ -56,6 +56,12 @@ interface FreshCollapse {
 	totalIterations: number | null;
 }
 
+interface BoomerangCollapse {
+	targetId: string;
+	task: string;
+	previousSummaries: string[];
+}
+
 interface PendingSkillMessage {
 	customType: "skill-loaded";
 	content: string;
@@ -108,6 +114,7 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 	let chainActive = false;
 	let loopState: LoopState | null = null;
 	let freshCollapse: FreshCollapse | null = null;
+	let boomerangCollapse: BoomerangCollapse | null = null;
 	let accumulatedSummaries: string[] = [];
 	let lastDiagnostics = "";
 	let storedCommandCtx: ExtensionCommandContext | null = null;
@@ -918,6 +925,28 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 		}
 	}
 
+	async function collapseBoomerangPrompt(
+		ctx: ExtensionContext,
+		name: string,
+		targetId: string | null,
+		previousSummaries: string[] = [],
+	) {
+		if (!targetId) {
+			notify(ctx, `Cannot boomerang prompt \`${name}\`: no session entry to return to.`, "warning");
+			return;
+		}
+
+		boomerangCollapse = { targetId, task: name, previousSummaries };
+		try {
+			(globalThis as typeof globalThis & { __boomerangCollapseInProgress?: boolean }).__boomerangCollapseInProgress = true;
+			const result = await ctx.navigateTree(targetId, { summarize: true });
+			if (result.cancelled) notify(ctx, `Boomerang cancelled for prompt \`${name}\``, "warning");
+		} finally {
+			(globalThis as typeof globalThis & { __boomerangCollapseInProgress?: boolean }).__boomerangCollapseInProgress = false;
+			boomerangCollapse = null;
+		}
+	}
+
 	async function runPromptLoop(
 		name: string,
 		cleanedArgs: string,
@@ -942,10 +971,11 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 		let currentThinking = savedThinking;
 		const shouldRestore = initialPrompt.restore;
 		const useFresh = freshFlag || initialPrompt.fresh === true;
+		const shouldBoomerang = initialPrompt.boomerang === true;
 		const effectiveMax = totalIterations ?? UNLIMITED_LOOP_CAP;
 		const isUnlimited = totalIterations === null;
 		const useConverge = converge && initialPrompt.converge !== false;
-		const anchorId = useFresh ? ctx.sessionManager.getLeafId() : null;
+		const anchorId = useFresh || shouldBoomerang ? ctx.sessionManager.getLeafId() : null;
 
 		loopState = { currentIteration: 1, totalIterations };
 		accumulatedSummaries = [];
@@ -955,6 +985,7 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 		let loopErrorState: ExecutionErrorState = { hasError: false, error: undefined };
 		let lastDelegatedText: string | undefined;
 		let loopAborted = false;
+		let boomerangPreviousSummaries: string[] = [];
 
 		try {
 			for (let i = 0; i < effectiveMax; i++) {
@@ -1024,7 +1055,7 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 					break;
 				}
 
-				if (anchorId && i < effectiveMax - 1) {
+				if (useFresh && anchorId && i < effectiveMax - 1) {
 					freshCollapse = { targetId: anchorId, task: name, iteration: i + 1, totalIterations };
 					const result = await ctx.navigateTree(anchorId, { summarize: true });
 					freshCollapse = null;
@@ -1049,9 +1080,11 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 				"loop",
 			);
 
+			boomerangPreviousSummaries = accumulatedSummaries;
 			loopState = null;
 			pendingSkillMessage = undefined;
 			freshCollapse = null;
+			boomerangCollapse = null;
 			accumulatedSummaries = [];
 			updateLoopStatus(ctx);
 
@@ -1069,6 +1102,10 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 			await ctx.waitForIdle();
 		}
 
+		if (!loopErrorState.hasError && !loopAborted && shouldBoomerang) {
+			await collapseBoomerangPrompt(ctx, name, anchorId, boomerangPreviousSummaries);
+		}
+
 		if (loopErrorState.hasError) {
 			throw loopErrorState.error;
 		}
@@ -1402,6 +1439,7 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 			chainActive = false;
 			loopState = null;
 			freshCollapse = null;
+			boomerangCollapse = null;
 			accumulatedSummaries = [];
 			updateLoopStatus(ctx);
 			if (ctx.hasUI) {
@@ -1549,6 +1587,7 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 		};
 		const savedModel = getCurrentModel(ctx);
 		const savedThinking = pi.getThinkingLevel();
+		const boomerangTargetId = effectivePrompt.boomerang ? ctx.sessionManager.getLeafId() : null;
 		const stepResult = await executePromptStep(
 			effectivePrompt,
 			parseCommandArgs(argsWithoutSubagent),
@@ -1573,6 +1612,10 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 				previousThinking = savedThinking;
 			}
 		}
+
+		if (effectivePrompt.boomerang) {
+			await collapseBoomerangPrompt(ctx, name, boomerangTargetId);
+		}
 	}
 
 	function resetSessionScopedState(ctx: ExtensionContext) {
@@ -1581,6 +1624,7 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 		previousModel = undefined;
 		previousThinking = undefined;
 		runtimeModel = ctx.model;
+		boomerangCollapse = null;
 		toolManager.clearQueue();
 		refreshPrompts(ctx.cwd, ctx);
 	}
@@ -1644,6 +1688,15 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 	});
 
 	pi.on("session_before_tree", async (event) => {
+		if (boomerangCollapse && event.preparation.targetId === boomerangCollapse.targetId) {
+			const summary = generateBoomerangSummary(event.preparation.entriesToSummarize, boomerangCollapse.task);
+			return {
+				summary: {
+					summary: [...boomerangCollapse.previousSummaries, summary].join("\n\n---\n\n"),
+				},
+			};
+		}
+
 		if (!freshCollapse) return;
 		if (event.preparation.targetId !== freshCollapse.targetId) return;
 

package/loop-utils.ts

@@ -88,7 +88,7 @@ function collectSummaryData(entries: SessionEntry[]): CollectedSummaryData {
 	};
 }
 
-function formatSummary(header: string, entries: SessionEntry[]): string {
+function formatSummary(header: string, entries: SessionEntry[], preserveOutcome = false): string {
 	const { filesRead, filesWritten, commandCount, lastAssistantText } = collectSummaryData(entries);
 
 	let summary = header;
@@ -102,9 +102,11 @@ function formatSummary(header: string, entries: SessionEntry[]): string {
 	}
 
 	if (lastAssistantText) {
-		const cleaned = lastAssistantText.replace(/\n+/g, " ").trim();
-		const truncated = cleaned.slice(0, 500);
-		summary += `\nOutcome: ${truncated}${cleaned.length > 500 ? "..." : ""}`;
+		const cleaned = preserveOutcome
+			? lastAssistantText.replace(/\r\n?/g, "\n").trim()
+			: lastAssistantText.replace(/\n+/g, " ").trim();
+		const outcome = preserveOutcome || cleaned.length <= 500 ? cleaned : `${cleaned.slice(0, 500)}...`;
+		summary += `\nOutcome: ${outcome}`;
 	}
 
 	return summary;
@@ -117,6 +119,10 @@ export function generateIterationSummary(entries: SessionEntry[], task: string,
 	return formatSummary(header, entries);
 }
 
+export function generateBoomerangSummary(entries: SessionEntry[], task: string): string {
+	return formatSummary(`[Boomerang]\nTask: "${task}"`, entries, true);
+}
+
 export function generateChainStepSummary(entries: SessionEntry[], stepLabel: string, stepNumber: number): string {
 	return formatSummary(`Step ${stepNumber} — ${stepLabel}:`, entries);
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-prompt-template-model",
-  "version": "0.8.2",
+  "version": "0.9.1",
   "type": "module",
   "description": "Prompt template model selector extension for pi coding agent",
   "author": "Nico Bailon",
@@ -49,6 +49,9 @@
   "scripts": {
     "test": "tsx --test test/**/*.test.ts"
   },
+  "dependencies": {
+    "typebox": "^1.1.24"
+  },
   "devDependencies": {
     "@mariozechner/pi-agent-core": "^0.65.0",
     "@mariozechner/pi-ai": "^0.65.0",

package/prompt-loader.ts

@@ -74,6 +74,7 @@ export interface PromptWithModel {
 	fresh?: boolean;
 	loop?: number | null;
 	converge?: boolean;
+	boomerang?: boolean;
 	parallel?: number;
 	worktree?: boolean;
 	deterministic?: DeterministicStep;
@@ -303,6 +304,31 @@ function normalizeRotate(
 	return false;
 }
 
+function normalizeBoomerang(
+	value: unknown,
+	filePath: string,
+	source: PromptSource,
+	diagnostics: PromptLoaderDiagnostic[],
+): boolean {
+	if (value === undefined) return false;
+	if (typeof value === "boolean") return value;
+	if (typeof value === "string") {
+		const normalized = value.trim().toLowerCase();
+		if (normalized === "true") return true;
+		if (normalized === "false") return false;
+	}
+
+	diagnostics.push(
+		createDiagnostic(
+			"invalid-boomerang",
+			filePath,
+			source,
+			`Using default boomerang=false for ${filePath}: frontmatter field "boomerang" must be true or false.`,
+		),
+	);
+	return false;
+}
+
 function normalizeLoop(
 	value: unknown,
 	filePath: string,
@@ -1643,6 +1669,18 @@ function loadPromptsWithModelFromDir(
 				const fresh = normalizeFresh(frontmatter.fresh, fullPath, source, diagnostics);
 				const loop = normalizeLoop(frontmatter.loop, fullPath, source, diagnostics);
 				const converge = normalizeConverge(frontmatter.converge, fullPath, source, diagnostics);
+				let boomerang = normalizeBoomerang(frontmatter.boomerang, fullPath, source, diagnostics);
+				if (chain && boomerang) {
+					diagnostics.push(
+						createDiagnostic(
+							"invalid-boomerang-chain",
+							fullPath,
+							source,
+							`Ignoring boomerang in ${fullPath}: frontmatter fields "chain" and "boomerang" cannot be combined.`,
+						),
+					);
+					boomerang = false;
+				}
 				if (loop !== undefined && deterministic !== undefined) {
 					diagnostics.push(
 						createDiagnostic(
@@ -1695,6 +1733,7 @@ function loadPromptsWithModelFromDir(
 					fresh === true ||
 					loop !== undefined ||
 					converge === false ||
+					boomerang === true ||
 					safeParallel !== undefined ||
 					deterministic !== undefined ||
 					hasLineup ||
@@ -1721,6 +1760,7 @@ function loadPromptsWithModelFromDir(
 					fresh: fresh || undefined,
 					loop: loop !== undefined ? loop : undefined,
 					converge: converge === false ? false : undefined,
+					boomerang: boomerang || undefined,
 					parallel: safeParallel,
 					worktree: safeWorktree,
 					deterministic,
@@ -1821,6 +1861,7 @@ export function buildPromptCommandDescription(prompt: PromptWithModel): string {
 	const thinkingValue = prompt.thinkingLevels ? prompt.thinkingLevels.join(",") : prompt.thinking;
 	const thinkingLabel = thinkingValue ? ` ${thinkingValue}` : "";
 	const loopLabel = prompt.loop !== undefined ? ` loop:${prompt.loop === null ? "unlimited" : prompt.loop}` : "";
+	const boomerangLabel = prompt.boomerang ? " boomerang" : "";
 	const subagentLabel = prompt.subagent ? ` subagent:${prompt.subagent === true ? "delegate" : prompt.subagent}` : "";
 	const parallelLabel = prompt.parallel !== undefined ? ` parallel:${prompt.parallel}` : "";
 	const deterministicLabel = prompt.deterministic ? ` deterministic-step:${prompt.deterministic.handoff}` : "";
@@ -1831,7 +1872,7 @@ export function buildPromptCommandDescription(prompt: PromptWithModel): string {
 	const inheritContextLabel = prompt.inheritContext ? " fork" : "";
 	const worktreeLabel = prompt.worktree ? " worktree" : "";
 	const details =
-		`[${modelLabel}${rotateLabel}${thinkingLabel}${skillLabel}${loopLabel}${subagentLabel}${parallelLabel}${deterministicLabel}${workersLabel}${reviewersLabel}${finalApplierLabel}${cwdLabel}${inheritContextLabel}${worktreeLabel}] ${sourceLabel}`;
+		`[${modelLabel}${rotateLabel}${thinkingLabel}${skillLabel}${loopLabel}${boomerangLabel}${subagentLabel}${parallelLabel}${deterministicLabel}${workersLabel}${reviewersLabel}${finalApplierLabel}${cwdLabel}${inheritContextLabel}${worktreeLabel}] ${sourceLabel}`;
 	return prompt.description ? `${prompt.description} ${details}` : details;
 }
 

package/tool-manager.ts

@@ -2,7 +2,7 @@ import { mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { homedir } from "node:os";
 import { join } from "node:path";
 import type { ExtensionAPI, ExtensionCommandContext, ExtensionContext } from "@mariozechner/pi-coding-agent";
-import { Type } from "@sinclair/typebox";
+import { Type } from "typebox";
 import { notify } from "./notifications.js";
 
 export interface ToolManagerDeps {