@shanepadgett/tau-agent 0.28.1 → 0.30.0

package/extensions/context-pruning/prune.ts

@@ -1,199 +0,0 @@
-import { resolve } from "node:path";
-import { type ExtensionContext, type SessionEntry } from "@earendil-works/pi-coding-agent";
-import { type Static, Type } from "typebox";
-import type { ContextPruneDeferredFileV2, ContextPruneDetailsV2 } from "../../shared/context-pruning-state.ts";
-import { prepareAutoreadMessage, type PreparedAutoreadMessage } from "../../shared/autoread.ts";
-import { MAX_COMPLETE_FILE_SNAPSHOT_BYTES } from "../../shared/full-file-knowledge.ts";
-
-export const contextPruneParameters = Type.Object(
-	{
-		keepFiles: Type.Array(
-			Type.Object({ path: Type.String(), relevance: Type.String() }, { additionalProperties: false }),
-		),
-		keepToolCalls: Type.Array(
-			Type.Object({ toolCallId: Type.String(), relevance: Type.String() }, { additionalProperties: false }),
-		),
-		deferFiles: Type.Array(
-			Type.Object(
-				{ path: Type.String(), reason: Type.String(), relevantWhen: Type.String() },
-				{ additionalProperties: false },
-			),
-		),
-	},
-	{ additionalProperties: false },
-);
-
-export type ContextPruneInput = Static<typeof contextPruneParameters>;
-
-interface ContextPruneExecutionOptions {
-	toolCallId: string;
-	params: ContextPruneInput;
-	signal: AbortSignal | undefined;
-	ctx: ExtensionContext;
-	generation: number;
-	currentGeneration: () => number;
-}
-
-interface ContextPruneToolResult {
-	content: Array<{ type: "text"; text: string }>;
-	details: ContextPruneDetailsV2;
-}
-
-interface ContextPruneExecutionResult {
-	result: ContextPruneToolResult;
-	autoreads: PreparedAutoreadMessage[];
-}
-
-export async function executeContextPrune(options: ContextPruneExecutionOptions): Promise<ContextPruneExecutionResult> {
-	assertCurrent(options);
-	const retainedToolCallIds = [...new Set(options.params.keepToolCalls.map((selection) => selection.toolCallId))];
-	const retainedTools = new Set(retainedToolCallIds);
-	const { prunedToolCallIds, prunedAutoreadRowIds } = collectPrunedRows(
-		options.ctx.sessionManager.getBranch(),
-		options.toolCallId,
-		retainedTools,
-	);
-	const warnings: string[] = [];
-	const preparedSnapshots: PreparedAutoreadMessage[] = [];
-	const keptPaths = new Set<string>();
-	for (const selection of options.params.keepFiles) {
-		assertCurrent(options);
-		const path = normalizePath(selection.path);
-		const pathKey = resolve(options.ctx.cwd, path);
-		if (keptPaths.has(pathKey)) continue;
-		keptPaths.add(pathKey);
-		try {
-			preparedSnapshots.push(
-				await prepareAutoreadMessage({
-					rowId: `${options.toolCallId}:${preparedSnapshots.length}`,
-					path,
-					cwd: options.ctx.cwd,
-					source: "context-pruning",
-					batchId: options.toolCallId,
-					signal: options.signal,
-					isLifecycleCurrent: () => options.generation === options.currentGeneration(),
-					maximumBytes: MAX_COMPLETE_FILE_SNAPSHOT_BYTES,
-				}),
-			);
-		} catch (error) {
-			if (options.signal?.aborted || options.generation !== options.currentGeneration()) throw error;
-			warnings.push(`${path}: ${errorMessage(error)}`);
-		}
-	}
-
-	const deferredFiles: ContextPruneDeferredFileV2[] = [];
-	const deferredPaths = new Set<string>();
-	for (const selection of options.params.deferFiles) {
-		const path = normalizePath(selection.path);
-		const pathKey = resolve(options.ctx.cwd, path);
-		if (keptPaths.has(pathKey) || deferredPaths.has(pathKey)) continue;
-		deferredPaths.add(pathKey);
-		deferredFiles.push({ path, reason: selection.reason, relevantWhen: selection.relevantWhen });
-	}
-	assertCurrent(options);
-
-	const retainedAutoreadRowIds = preparedSnapshots.map((snapshot) => snapshot.details.rowId);
-	const refreshedFiles = preparedSnapshots.map((snapshot) => ({
-		path: snapshot.details.path,
-		rowId: snapshot.details.rowId,
-	}));
-	const details: ContextPruneDetailsV2 = {
-		v: 2,
-		anchorToolCallId: options.toolCallId,
-		prunedToolCallIds,
-		prunedAutoreadRowIds,
-		retainedToolCallIds,
-		retainedAutoreadRowIds,
-		refreshedFiles,
-		deferredFiles,
-		warnings,
-	};
-	const status =
-		warnings.length === 0
-			? "Context checkpoint applied. Continue with the next action stated before this call."
-			: `Context checkpoint applied with ${warnings.length} warning${warnings.length === 1 ? "" : "s"}:\n${warnings.map((warning) => `- ${warning}`).join("\n")}\nContinue with the next action stated before this call.`;
-	return {
-		result: {
-			content: [
-				{ type: "text", text: status },
-				...(deferredFiles.length === 0 ? [] : [{ type: "text" as const, text: deferredFileText(deferredFiles) }]),
-			],
-			details,
-		},
-		autoreads: preparedSnapshots,
-	};
-}
-
-function collectPrunedRows(
-	branch: readonly SessionEntry[],
-	anchorToolCallId: string,
-	retainedToolCallIds: ReadonlySet<string>,
-): { prunedToolCallIds: string[]; prunedAutoreadRowIds: string[] } {
-	let anchorIndex = -1;
-	for (let index = branch.length - 1; index >= 0; index -= 1) {
-		const entry = branch[index];
-		if (
-			entry?.type === "message" &&
-			entry.message.role === "assistant" &&
-			entry.message.content.some(
-				(block) => block.type === "toolCall" && block.id === anchorToolCallId && block.name === "context_prune",
-			)
-		) {
-			anchorIndex = index;
-			break;
-		}
-	}
-	if (anchorIndex < 0) return { prunedToolCallIds: [], prunedAutoreadRowIds: [] };
-	const prunedToolCallIds = new Set<string>();
-	const prunedAutoreadRowIds = new Set<string>();
-	for (let index = 0; index < anchorIndex; index += 1) {
-		const entry = branch[index];
-		if (entry?.type === "message" && entry.message.role === "assistant") {
-			for (const block of entry.message.content) {
-				if (block.type === "toolCall" && !retainedToolCallIds.has(block.id)) prunedToolCallIds.add(block.id);
-			}
-			continue;
-		}
-		if (entry?.type !== "custom_message" || entry.customType !== "tau.autoread") continue;
-		if (isRecord(entry.details) && typeof entry.details.rowId === "string") {
-			prunedAutoreadRowIds.add(entry.details.rowId);
-		}
-		continue;
-	}
-	for (let index = 0; index < anchorIndex; index += 1) {
-		const entry = branch[index];
-		if (entry?.type !== "message" || entry.message.role !== "toolResult") continue;
-		if (entry.message.toolName !== "context_prune" || !isRecord(entry.message.details)) continue;
-		const files = entry.message.details.refreshedFiles;
-		if (!Array.isArray(files)) continue;
-		for (const file of files)
-			if (isRecord(file) && typeof file.rowId === "string") prunedAutoreadRowIds.add(file.rowId);
-	}
-	return { prunedToolCallIds: [...prunedToolCallIds], prunedAutoreadRowIds: [...prunedAutoreadRowIds] };
-}
-
-function deferredFileText(files: readonly ContextPruneDeferredFileV2[]): string {
-	return [
-		"Deferred files are advisory. Reconsider them only when their condition applies:",
-		...files.map((file) => `- ${file.path}: ${file.reason} Relevant when: ${file.relevantWhen}`),
-	].join("\n");
-}
-
-function normalizePath(path: string): string {
-	return path.replace(/^@/, "");
-}
-
-function assertCurrent(options: ContextPruneExecutionOptions): void {
-	options.signal?.throwIfAborted();
-	if (options.generation !== options.currentGeneration()) {
-		throw new Error("Context checkpoint crossed a session lifecycle boundary");
-	}
-}
-
-function errorMessage(error: unknown): string {
-	return error instanceof Error ? error.message : String(error);
-}
-
-function isRecord(value: unknown): value is Record<string, unknown> {
-	return typeof value === "object" && value !== null && !Array.isArray(value);
-}

package/extensions/context-pruning/render.ts

@@ -1,251 +0,0 @@
-import type { AgentToolResult, Theme } from "@earendil-works/pi-coding-agent";
-import { Text, truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
-import { Marker } from "@shanepadgett/tau-tui";
-import { parseContextPruneDetailsV2 } from "../../shared/context-pruning-state.ts";
-import { formatToolRowTitle, type ToolRowStateStore } from "../../shared/tool-row-state.ts";
-import type { ContextPruneInput } from "./prune.ts";
-
-const MAX_EXPANDED_ITEMS = 20;
-const MAX_EXPANDED_LINE_CHARACTERS = 240;
-const MAX_EXPANDED_TEXT_CHARACTERS = 3_000;
-const MAX_WARNING_CHARACTERS = 1_000;
-
-export type ContextPruningNudgeDetailsV3 =
-	| {
-			v: 3;
-			kind: "automatic";
-			tokens: number;
-			boundaryTokens: number;
-			reminder: number;
-			tier: number;
-			tierCount: number;
-			tierFloor: number;
-			anchorToolCallId: string | null;
-			suppressedThroughTokens: number;
-	  }
-	| {
-			v: 3;
-			kind: "manual";
-			tokens: null;
-			boundaryTokens: null;
-			reminder: null;
-			tier: null;
-			tierCount: null;
-			tierFloor: null;
-			anchorToolCallId: string | null;
-			suppressedThroughTokens: null;
-	  };
-
-export function parseContextPruningNudgeDetailsV3(value: unknown): ContextPruningNudgeDetailsV3 | undefined {
-	if (!isRecord(value)) return undefined;
-	const keys = [
-		"v",
-		"kind",
-		"tokens",
-		"boundaryTokens",
-		"reminder",
-		"tier",
-		"tierCount",
-		"tierFloor",
-		"anchorToolCallId",
-		"suppressedThroughTokens",
-	];
-	if (Object.keys(value).length !== keys.length || !keys.every((key) => Object.hasOwn(value, key))) return undefined;
-	if (value.v !== 3 || (value.kind !== "automatic" && value.kind !== "manual")) return undefined;
-	if (
-		value.anchorToolCallId !== null &&
-		(typeof value.anchorToolCallId !== "string" || value.anchorToolCallId.length === 0)
-	) {
-		return undefined;
-	}
-	if (value.kind === "manual") {
-		if (
-			value.tokens !== null ||
-			value.boundaryTokens !== null ||
-			value.reminder !== null ||
-			value.tier !== null ||
-			value.tierCount !== null ||
-			value.tierFloor !== null ||
-			value.suppressedThroughTokens !== null
-		)
-			return undefined;
-		return {
-			v: 3,
-			kind: "manual",
-			tokens: null,
-			boundaryTokens: null,
-			reminder: null,
-			tier: null,
-			tierCount: null,
-			tierFloor: null,
-			anchorToolCallId: value.anchorToolCallId,
-			suppressedThroughTokens: null,
-		};
-	}
-	if (
-		!isTokenCount(value.tokens) ||
-		!isBoundary(value.boundaryTokens) ||
-		!isReminder(value.reminder) ||
-		!isTier(value.tier) ||
-		!isTier(value.tierCount) ||
-		!isTierFloor(value.tierFloor) ||
-		!isTokenCount(value.suppressedThroughTokens) ||
-		value.boundaryTokens > value.tokens ||
-		value.boundaryTokens <= value.suppressedThroughTokens ||
-		value.tier > value.tierCount ||
-		value.tierFloor > value.tierCount ||
-		value.tier !== Math.max(Math.min(value.reminder, value.tierCount), value.tierFloor) ||
-		(value.anchorToolCallId === null && value.suppressedThroughTokens !== 0)
-	) {
-		return undefined;
-	}
-	const interval = value.boundaryTokens / value.reminder;
-	if (
-		!Number.isSafeInteger(interval) ||
-		interval < 1 ||
-		value.tokens - value.boundaryTokens >= interval ||
-		value.suppressedThroughTokens % interval !== 0
-	)
-		return undefined;
-	return {
-		v: 3,
-		kind: "automatic",
-		tokens: value.tokens,
-		boundaryTokens: value.boundaryTokens,
-		reminder: value.reminder,
-		tier: value.tier,
-		tierCount: value.tierCount,
-		tierFloor: value.tierFloor,
-		anchorToolCallId: value.anchorToolCallId,
-		suppressedThroughTokens: value.suppressedThroughTokens,
-	};
-}
-
-export function renderContextPruningNudge(details: unknown, theme: Theme): Marker | undefined {
-	const parsed = parseContextPruningNudgeDetailsV3(details);
-	if (!parsed) return undefined;
-	return new Marker({
-		theme,
-		state: "muted",
-		label: "Context:",
-		parts:
-			parsed.kind === "manual"
-				? ["Prune requested."]
-				: [
-						formatTokens(parsed.tokens),
-						...(parsed.tier === parsed.tierCount ? ["Prune now."] : parsed.tier > 1 ? ["Prune soon."] : []),
-					],
-	});
-}
-
-export function renderContextPruneCall(
-	args: ContextPruneInput,
-	theme: Theme,
-	context: {
-		rowState: ToolRowStateStore;
-		rowId: string;
-		invalidate: () => void;
-		lastComponent: unknown;
-	},
-): Text {
-	context.rowState.watch(context.rowId, context.invalidate);
-	const component = context.lastComponent instanceof Text ? context.lastComponent : new Text("", 0, 0);
-	const count =
-		(Array.isArray(args.keepFiles) ? args.keepFiles.length : 0) +
-		(Array.isArray(args.keepToolCalls) ? args.keepToolCalls.length : 0) +
-		(Array.isArray(args.deferFiles) ? args.deferFiles.length : 0);
-	component.setText(
-		`${formatToolRowTitle(context.rowState, context.rowId, "context_prune", theme)} ${theme.fg("muted", `${count} selection${count === 1 ? "" : "s"}`)}`,
-	);
-	return component;
-}
-
-export function renderContextPruneResult(
-	result: AgentToolResult<unknown>,
-	expanded: boolean,
-	theme: Theme,
-	lastComponent: unknown,
-): Text {
-	const component = lastComponent instanceof Text ? lastComponent : new Text("", 0, 0);
-	const details = parseContextPruneDetailsV2(result.details);
-	if (!details) {
-		component.setText(
-			theme.fg(
-				"warning",
-				boundedText(firstResultText(result) || "context_prune returned invalid details", MAX_WARNING_CHARACTERS),
-			),
-		);
-		return component;
-	}
-	const pruned = details.prunedToolCallIds.length + details.prunedAutoreadRowIds.length;
-	const retained = details.retainedToolCallIds.length + details.retainedAutoreadRowIds.length;
-	let text = theme.fg(
-		details.warnings.length === 0 ? "success" : "warning",
-		`Checkpoint · pruned ${pruned} · retained ${retained} · refreshed ${details.refreshedFiles.length} · deferred ${details.deferredFiles.length}${details.warnings.length === 0 ? "" : ` · warnings ${details.warnings.length}`}`,
-	);
-	if (expanded) {
-		const lines = [
-			...details.prunedToolCallIds.map((id) => `pruned tool: ${id}`),
-			...details.prunedAutoreadRowIds.map((id) => `pruned autoread: ${id}`),
-			...details.retainedToolCallIds.map((id) => `retained tool: ${id}`),
-			...details.retainedAutoreadRowIds.map((id) => `retained autoread: ${id}`),
-			...details.refreshedFiles.map((file) => `refreshed: ${file.path} (${file.rowId})`),
-			...details.deferredFiles.map((file) => `deferred: ${file.path} — ${file.reason}; when ${file.relevantWhen}`),
-			...details.warnings.map((warning) => `warning: ${warning}`),
-		];
-		const shown = lines.slice(0, MAX_EXPANDED_ITEMS).map((line) => boundedText(line, MAX_EXPANDED_LINE_CHARACTERS));
-		if (lines.length > shown.length) shown.push(`… ${lines.length - shown.length} more`);
-		let remainingWidth = Math.max(0, MAX_EXPANDED_TEXT_CHARACTERS - visibleWidth(text));
-		for (const line of shown) {
-			if (remainingWidth <= 1) break;
-			const styled = theme.fg("dim", line);
-			const availableWidth = remainingWidth - 1;
-			text += `\n${truncateToWidth(styled, availableWidth, "…")}`;
-			remainingWidth -= 1 + Math.min(visibleWidth(styled), availableWidth);
-			if (visibleWidth(styled) > availableWidth) break;
-		}
-	}
-	component.setText(text);
-	return component;
-}
-
-function boundedText(text: string, maximumCharacters: number): string {
-	if (text.length <= maximumCharacters) return text;
-	return `${text.slice(0, Math.max(0, maximumCharacters - 1))}…`;
-}
-
-function firstResultText(result: AgentToolResult<unknown>): string {
-	for (const part of result.content) if (part.type === "text") return part.text;
-	return "";
-}
-
-function formatTokens(count: number): string {
-	if (count < 1_000) return `${count}`;
-	if (count < 10_000) return `${(count / 1_000).toFixed(1)}k`;
-	if (count < 1_000_000) return `${Math.round(count / 1_000)}k`;
-	return `${(count / 1_000_000).toFixed(1)}M`;
-}
-
-function isTokenCount(value: unknown): value is number {
-	return typeof value === "number" && Number.isSafeInteger(value) && value >= 0;
-}
-
-function isBoundary(value: unknown): value is number {
-	return isTokenCount(value) && value > 0;
-}
-
-function isReminder(value: unknown): value is number {
-	return typeof value === "number" && Number.isSafeInteger(value) && value >= 1;
-}
-
-function isTier(value: unknown): value is number {
-	return typeof value === "number" && Number.isInteger(value) && value >= 1 && value <= 5;
-}
-
-function isTierFloor(value: unknown): value is number {
-	return typeof value === "number" && Number.isInteger(value) && value >= 0 && value <= 5;
-}
-
-function isRecord(value: unknown): value is Record<string, unknown> {
-	return typeof value === "object" && value !== null && !Array.isArray(value);
-}

package/extensions/context-pruning/settings.ts

@@ -1,39 +0,0 @@
-import { Type } from "typebox";
-import { defineTauExtensionSettings } from "../../shared/settings/define.ts";
-
-const DEFAULT_NUDGE_INSTRUCTIONS: [string, ...string[]] = [
-	"No prune is required yet unless broad exploration has converged or substantial evidence is already irrelevant. Continue coherent work.",
-	"Move toward a pruning point now. Finish the current coherent step, then prune before starting another broad exploration. Managed context is materially increasing model cost.",
-	"Prune now before further tool work. Continuing with stale managed context is wasting money.",
-];
-
-export default defineTauExtensionSettings({
-	key: "contextPruning",
-	defaults: {
-		enabled: true as boolean,
-		nudgeEveryTokens: 30_000 as number,
-		nudgeInstructions: DEFAULT_NUDGE_INSTRUCTIONS,
-	},
-	schema: Type.Object(
-		{
-			enabled: Type.Optional(Type.Boolean({ default: true, description: "Enable context pruning." })),
-			nudgeEveryTokens: Type.Optional(
-				Type.Integer({
-					default: 30_000,
-					minimum: 1,
-					description: "Active-context token interval between automatic pruning hints.",
-				}),
-			),
-			nudgeInstructions: Type.Optional(
-				Type.Array(Type.String({ minLength: 1, maxLength: 2_000 }), {
-					default: DEFAULT_NUDGE_INSTRUCTIONS,
-					minItems: 1,
-					maxItems: 5,
-					description:
-						"Ordered automatic pruning instructions. Later reminders repeat the final instruction, and the final tier requires an anchor before further tool work.",
-				}),
-			),
-		},
-		{ additionalProperties: false },
-	),
-});

package/extensions/subagent/agents/review.md

@@ -1,68 +0,0 @@
----
-name: review
-description: Perform an adversarial, read-only review for correctness, runtime risks, duplication, and over- or under-engineering
-tools:
-  - read
-  - bash
-  - outline
-  - show
-  - discover
-  - ast_search
-  - deps
-  - reverse_deps
-  - callers
-  - callees
-  - references
-  - implementations
-  - impact
-  - context
-names:
-  - Auditor
-  - Inspector
-  - Skeptic
-  - Examiner
-  - Sentinel
-model: openai-codex/gpt-5.6-sol
-thinking: high
----
-
-Stay inside delegated change. Answer two questions:
-
-1. Is runtime behavior correct?
-2. Is this the simplest implementation of requested behavior?
-
-Find concrete failures or needless complexity. Report. Stop. No mutations, unrelated review, or broad concern inventory.
-
-## Evidence ladder
-
-Use cheapest evidence that settles each question. Escalate only when answer could change verdict.
-
-1. **Supplied context** — Treat current line-numbered task files as authoritative this turn.
-2. **Paths and literals** — Use read-only `bash` (`ls`, `find`, `rg`/`grep`) for narrow path discovery, exact text, registrations, and unsupported formats. Use ranged `read` when formatting or source context matters.
-3. **Structure and reuse** — Default to `outline`. Use `discover` only when changed code may duplicate an existing repository API but name or path is unknown. Use `ast_search` for a concrete risky or duplicated source shape.
-4. **Exact declarations** — Use `show` with path + name (+ line when needed). Retrieve only contract, body, imports, or nearby lines needed for verdict.
-5. **Runtime relationships** — Use `callers`, `callees`, `references`, or `implementations` after selecting a declaration. Use `deps`/`reverse_deps` for file imports. Use `impact` for full blast radius and `context` for one bounded declaration pack.
-
-Keep roots and result limits narrow. Structural evidence proves bounded syntax, not dynamic dispatch. Preserve inferred and ambiguous labels.
-
-## Review procedure
-
-1. Extract requested behavior, changed scope, and repository constraints from task and supplied files.
-2. Inspect changed declarations. Form only actionable runtime and simplicity questions.
-3. Runtime: follow shortest relevant path through callers, state transitions, boundaries, and error handling. Check realistic sequences, not theoretical branch inventory.
-4. Simplicity: identify wrappers, helpers, option bags, duplicated logic, or staged abstractions that can disappear without changing requested behavior. Report only material reductions in concepts or ownership.
-5. Inspect surrounding code only to confirm suspected failure, contract mismatch, missed caller, or existing simpler API.
-6. Stop when both questions have evidence-backed answers.
-
-Treat implementation as untrusted. Reject speculation, personal style preferences, and complexity complaints without concrete cost. Do not modify files.
-
-## Output
-
-List findings first, ordered by severity. Each finding needs:
-
-- severity and direct title;
-- exact file, line range, and declaration when one exists;
-- runtime failure mechanism or concrete complexity cost;
-- smallest credible fix direction.
-
-Then list only unresolved questions that materially affect runtime correctness. No findings: say runtime appears correct and implementation is already simplest credible version. Briefly name inspected scope. No preamble, search log, broad summary, or repeated evidence.

package/extensions/turn-budget/README.md

@@ -1,14 +0,0 @@
-# Turn Budget
-
-Adds soft turn budget hints for Tau sessions.
-
-Turn Budget helps agents spend fewer provider cycles by nudging them to batch related tool work. It counts
-tool-using turns per user prompt, sends visible steering messages at configured intervals, and extends the soft cap
-instead of blocking work.
-
-## Settings
-
-- `enabled`: defaults to `true`
-- `turnLimit`: defaults to `30`
-- `nudgeEveryTurns`: defaults to `5`
-- `softCapIncrement`: defaults to `10`

package/extensions/turn-budget/index.ts

@@ -1,116 +0,0 @@
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-import { loadTauExtensionSettings } from "../../shared/settings/load.ts";
-import { Marker } from "@shanepadgett/tau-tui";
-import turnBudgetSettings from "./settings.ts";
-
-const MARKER_TYPE = "tau.turn-budget.marker";
-
-interface Settings {
-	enabled: boolean;
-	turnLimit: number;
-	nudgeEveryTurns: number;
-	softCapIncrement: number;
-}
-
-type Hint =
-	| { kind: "normal"; used: number; cap: number }
-	| { kind: "extended"; used: number; previousCap: number; newCap: number };
-
-interface MarkerDetails {
-	used: number;
-	cap: number;
-	extended: boolean;
-}
-
-export default function turnBudgetExtension(pi: ExtensionAPI): void {
-	let settings = normalizeSettings(turnBudgetSettings.defaults);
-	let turnCount = 0;
-	let softCap = settings.turnLimit;
-
-	pi.registerMessageRenderer<MarkerDetails>(MARKER_TYPE, (message, _options, theme) => {
-		const details = readMarkerDetails(message.details);
-		if (!details) return undefined;
-		return new Marker({
-			theme,
-			state: "muted",
-			label: "Turn Budget:",
-			parts: [`${details.used}/${details.cap}`, ...(details.extended ? ["Soft cap extended."] : [])],
-		});
-	});
-
-	pi.on("session_start", async (_event, ctx) => {
-		settings = normalizeSettings(await loadTauExtensionSettings(ctx, turnBudgetSettings));
-	});
-
-	pi.on("agent_start", () => {
-		turnCount = 0;
-		softCap = settings.turnLimit;
-	});
-
-	pi.on("turn_end", (event) => {
-		if (!settings.enabled) return undefined;
-		if (event.toolResults.length === 0) return undefined;
-		turnCount += 1;
-		if (turnCount >= softCap) {
-			const hint: Hint = {
-				kind: "extended",
-				used: turnCount,
-				previousCap: softCap,
-				newCap: turnCount + settings.softCapIncrement,
-			};
-			sendTurnBudgetMessage(pi, hint);
-			softCap = hint.newCap;
-			return undefined;
-		}
-		if (turnCount % settings.nudgeEveryTurns === 0) {
-			const hint: Hint = { kind: "normal", used: turnCount, cap: softCap };
-			sendTurnBudgetMessage(pi, hint);
-		}
-		return undefined;
-	});
-}
-
-function sendTurnBudgetMessage(pi: ExtensionAPI, hint: Hint): void {
-	pi.sendMessage<MarkerDetails>({
-		customType: MARKER_TYPE,
-		content: formatSteeringMessage(hint),
-		display: true,
-		details: markerDetails(hint),
-	});
-}
-
-function formatSteeringMessage(hint: Hint): string {
-	const instruction =
-		"Internal steering instruction. Work within it silently. Do not mention or acknowledge turn counts, budget messages, or budget summaries.";
-	if (hint.kind === "extended") {
-		return `${instruction} Turn budget: ${hint.used}/${hint.previousCap} turns used for this user prompt. Soft cap extended to ${hint.newCap}. Batch tools when more tool work remains.`;
-	}
-	return `${instruction} Turn budget: ${hint.used}/${hint.cap} turns used for this user prompt. Batch tools when more tool work remains.`;
-}
-
-function normalizeSettings(value: typeof turnBudgetSettings.defaults): Settings {
-	return {
-		enabled: value.enabled ?? true,
-		turnLimit: Number.isInteger(value.turnLimit) && value.turnLimit > 0 ? value.turnLimit : 30,
-		nudgeEveryTurns: Number.isInteger(value.nudgeEveryTurns) && value.nudgeEveryTurns > 0 ? value.nudgeEveryTurns : 5,
-		softCapIncrement:
-			Number.isInteger(value.softCapIncrement) && value.softCapIncrement > 0 ? value.softCapIncrement : 10,
-	};
-}
-
-function markerDetails(hint: Hint): MarkerDetails {
-	return hint.kind === "extended"
-		? { used: hint.used, cap: hint.newCap, extended: true }
-		: { used: hint.used, cap: hint.cap, extended: false };
-}
-
-function readMarkerDetails(value: unknown): MarkerDetails | undefined {
-	if (!value || typeof value !== "object") return undefined;
-	const record = value as Record<string, unknown>;
-	const used = record.used;
-	const cap = record.cap;
-	if (typeof used !== "number" || !Number.isInteger(used) || used < 0) return undefined;
-	if (typeof cap !== "number" || !Number.isInteger(cap) || cap < 1) return undefined;
-	if (typeof record.extended !== "boolean") return undefined;
-	return { used, cap, extended: record.extended };
-}