@shanepadgett/tau-agent 0.22.0 → 0.23.0

package/extensions/context-pruning/README.md

@@ -11,7 +11,7 @@ Before calling the tool, the agent writes visible prose containing the durable c
 The tool accepts three required lists:
 
 - `keepToolCalls` retains exact earlier tool exchanges by tool-call ID. Parallel calls remain independently selectable: retaining one call does not retain its siblings.
-- `keepFiles` reads each selected file from disk and carries its complete current contents forward as a fresh autoread snapshot. It does not require an earlier complete read.
+- `keepFiles` reads each selected file from disk and carries its complete current contents forward as a fresh autoread snapshot. Each successful snapshot appears as its own autoread marker below the compact checkpoint result. It does not require an earlier complete read.
 - `deferFiles` carries forward a short advisory note explaining why a file is irrelevant now and when to reconsider it.
 
 Duplicate selections are collapsed. A file selected in both `keepFiles` and `deferFiles` is kept. Missing, unreadable, or otherwise unsnapshotable files produce warnings in the successful tool result; they do not block the checkpoint or other file snapshots.

package/extensions/context-pruning/index.ts

@@ -117,7 +117,7 @@ export default function contextPruningExtension(pi: ExtensionAPI): void {
 					parameters: contextPruneParameters,
 					executionMode: "sequential",
 					async execute(toolCallId, params, signal, _onUpdate, executionContext) {
-						return executeContextPrune({
+						const execution = await executeContextPrune({
 							toolCallId,
 							params,
 							signal,
@@ -125,6 +125,10 @@ export default function contextPruningExtension(pi: ExtensionAPI): void {
 							generation: lifecycleGeneration,
 							currentGeneration: () => lifecycleGeneration,
 						});
+						for (const autoread of execution.autoreads) {
+							pi.sendMessage(autoread, { deliverAs: "steer" });
+						}
+						return execution.result;
 					},
 					renderCall(args, theme, context) {
 						return renderContextPruneCall(args, theme, {

package/extensions/context-pruning/prune.ts

@@ -34,11 +34,16 @@ interface ContextPruneExecutionOptions {
 	currentGeneration: () => number;
 }
 
-interface ContextPruneExecutionResult {
+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))];
@@ -91,8 +96,6 @@ export async function executeContextPrune(options: ContextPruneExecutionOptions)
 	const refreshedFiles = preparedSnapshots.map((snapshot) => ({
 		path: snapshot.details.path,
 		rowId: snapshot.details.rowId,
-		servedHash: snapshot.details.readCache.servedHash,
-		autoreadDetails: { ...snapshot.details },
 	}));
 	const details: ContextPruneDetailsV2 = {
 		v: 2,
@@ -110,12 +113,14 @@ export async function executeContextPrune(options: ContextPruneExecutionOptions)
 			? "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 {
-		content: [
-			{ type: "text", text: status },
-			...preparedSnapshots.map((snapshot) => ({ type: "text" as const, text: snapshot.content })),
-			...(deferredFiles.length === 0 ? [] : [{ type: "text" as const, text: deferredFileText(deferredFiles) }]),
-		],
-		details,
+		result: {
+			content: [
+				{ type: "text", text: status },
+				...(deferredFiles.length === 0 ? [] : [{ type: "text" as const, text: deferredFileText(deferredFiles) }]),
+			],
+			details,
+		},
+		autoreads: preparedSnapshots,
 	};
 }
 

package/extensions/explore/read-cache.ts

@@ -128,30 +128,25 @@ export function replayReadCache(
 		if (entry.type === "message" && "message" in entry) message = entry.message;
 		else if (entry.type === "custom_message") message = entry;
 		else continue;
-		const candidates = carriedFileMessages(message);
-		let acceptedCarryForward = false;
-		for (const candidate of candidates ?? [message]) {
-			const parsedMeta = readMetaFromMessage(candidate);
-			if (!parsedMeta) continue;
-			acceptedCarryForward = true;
-			const candidateRowId = readRowId(candidate);
-			if (candidateRowId && ignoredRowIds.has(candidateRowId)) continue;
+		const parsedMeta = readMetaFromMessage(message);
+		if (parsedMeta) {
+			const rowId = readRowId(message);
+			if (!rowId || ignoredRowIds.has(rowId)) continue;
 			const pathKey = resolve(cwd, parsedMeta.pathKey);
 			const meta = pathKey === parsedMeta.pathKey ? parsedMeta : { ...parsedMeta, pathKey };
-			if (candidateRowId && applyReadMeta(trust, failedPatchRecoveryPaths, candidate, meta, candidateRowId)) {
+			if (applyReadMeta(trust, failedPatchRecoveryPaths, message, meta, rowId)) {
 				const acceptedTrust = trust.get(meta.pathKey)?.get(meta.scopeKey);
-				if (acceptedTrust) {
+				if (acceptedTrust)
 					acceptedRows.push({
-						rowId: candidateRowId,
+						rowId,
 						pathKey: meta.pathKey,
 						scopeKey: meta.scopeKey,
 						meta,
 						dependencyRowIds: [...acceptedTrust.rowIds],
 					});
-				}
 			}
+			continue;
 		}
-		if (acceptedCarryForward) continue;
 		for (const path of failedPatchPaths(message, cwd)) failedPatchRecoveryPaths.add(path);
 	}
 	const completeFileChains = new Map<string, CompleteFileDependencyChain>();
@@ -168,25 +163,6 @@ export function replayReadCache(
 	return { scopeTrust: trust, acceptedRows, completeFileChains, failedPatchRecoveryPaths };
 }
 
-function carriedFileMessages(message: unknown): unknown[] | undefined {
-	if (!isRecord(message) || message.role !== "toolResult" || message.toolName !== "context_prune") return undefined;
-	if (!isRecord(message.details) || message.details.v !== 2 || !Array.isArray(message.details.refreshedFiles)) {
-		return undefined;
-	}
-	if (!Array.isArray(message.content)) {
-		return undefined;
-	}
-	const messages: unknown[] = [];
-	for (let index = 0; index < message.details.refreshedFiles.length; index += 1) {
-		const file = message.details.refreshedFiles[index];
-		const part = message.content[index + 1];
-		if (!isRecord(file) || !isRecord(file.autoreadDetails)) continue;
-		if (!isRecord(part) || part.type !== "text" || typeof part.text !== "string") continue;
-		messages.push({ customType: "tau.autoread", content: part.text, details: file.autoreadDetails });
-	}
-	return messages;
-}
-
 function applyReadMeta(
 	trust: Map<string, Map<string, ReadCacheScopeTrust>>,
 	failedPatchRecoveryPaths: Set<string>,

package/extensions/subagent/cmux-dashboard.ts

@@ -71,54 +71,82 @@ function cap(text: string | undefined, limit: number): string {
 	return `${normalized.slice(0, limit - 1)}…`;
 }
 
+function tableCell(text: string, limit: number): string {
+	const compact = cap(text.replace(/\s+/g, " ").trim(), limit);
+	return compact.replaceAll("|", "&#124;").replaceAll("<", "&lt;").replaceAll(">", "&gt;");
+}
+
+function quote(text: string): string {
+	return cap(text, 1200)
+		.replaceAll("<", "&lt;")
+		.replaceAll(">", "&gt;")
+		.split("\n")
+		.map((line) => `> ${line}`)
+		.join("\n");
+}
+
+function elapsed(durationMs: number): string {
+	if (durationMs < 1000) return "<1s";
+	if (durationMs < 60_000) return `${(durationMs / 1000).toFixed(1)}s`;
+	const minutes = Math.floor(durationMs / 60_000);
+	const seconds = Math.floor((durationMs % 60_000) / 1000);
+	return `${minutes}m ${seconds}s`;
+}
+
+function dashboardState(status: SubagentInvocationSnapshot["status"]): string {
+	switch (status) {
+		case "waiting":
+			return "queued";
+		case "completed":
+			return "done";
+		case "aborted":
+			return "stopped";
+		default:
+			return status;
+	}
+}
+
 export function formatDashboardMarkdown(snapshots: readonly SubagentInvocationSnapshot[]): string {
 	const ordered = [...snapshots].sort(
 		(a, b) => a.startedAt - b.startedAt || a.invocationId.localeCompare(b.invocationId),
 	);
-	const activeCount = ordered.filter(
-		(snapshot) => snapshot.status !== "completed" && snapshot.status !== "failed" && snapshot.status !== "aborted",
+	const activeCount = ordered.filter((snapshot) =>
+		["waiting", "starting", "running"].includes(snapshot.status),
 	).length;
-	const lines = [
-		"# Subagent dashboard",
-		"",
-		`Active: ${activeCount} · Completed: ${ordered.length - activeCount}`,
-		"",
-	];
+	const doneCount = ordered.filter((snapshot) => snapshot.status === "completed").length;
+	const failedCount = ordered.filter((snapshot) => snapshot.status === "failed").length;
+	const stoppedCount = ordered.filter((snapshot) => snapshot.status === "aborted").length;
+	const counts = [`${activeCount} active`, `${doneCount} done`];
+	if (failedCount > 0) counts.push(`${failedCount} failed`);
+	if (stoppedCount > 0) counts.push(`${stoppedCount} stopped`);
+	const lines = ["# Subagents", "", counts.join(" · "), ""];
 	if (ordered.length === 0) {
-		lines.push("_No active invocations._", "");
+		lines.push("_No subagents._", "");
 		return lines.join("\n");
 	}
+	lines.push("| Agent | State | Request | Last tool | Calls | Time |", "| --- | --- | --- | --- | ---: | ---: |");
 	for (const details of ordered) {
-		const identity = `${details.displayName} (${details.agent})`;
+		const latest = details.actions.at(-1);
+		const currentTool = details.currentActivity?.match(/^\S+/)?.[0];
+		const lastTool = currentTool ?? latest?.tool ?? "";
 		lines.push(
-			`## ${identity}`,
-			"",
-			`\`${details.status}${details.phase ? ` · ${details.phase}` : ""}\` · inv ${details.invocationId} · tools ${details.toolCalls} · ${(details.durationMs / 1000).toFixed(1)}s`,
-			`model ${details.model} · thinking ${details.thinkingLevel}`,
+			`| ${tableCell(`${details.displayName} (${details.agent})`, 48)} | ${dashboardState(details.status)} | ${tableCell(details.task, 90)} | ${tableCell(lastTool, 32) || "—"} | ${details.toolCalls} | ${elapsed(details.durationMs)} |`,
+		);
+	}
+	lines.push("", "## Inputs", "");
+	for (const details of ordered) {
+		lines.push(
+			`### ${tableCell(details.displayName, 80)}`,
 			"",
-			"### Task",
+			`${tableCell(details.agent, 80)} · ${tableCell(details.invocationId, 80)}`,
 			"",
-			cap(details.task, 1200) || "_(empty)_",
+			quote(details.task) || "> _(empty)_",
 		);
-		if (details.currentActivity) {
-			lines.push("", "### Current", "", cap(details.currentActivity, 400));
-		}
-		if (details.actions.length > 0) {
-			lines.push("", "### Actions", "");
-			for (const action of details.actions.slice(-12)) {
-				const mark = action.error ? "!" : "·";
-				lines.push(`- ${mark} ${cap(action.summary, 200)}`);
-			}
-			if (details.omittedActions > 0) {
-				lines.push(`- _${details.omittedActions} earlier actions omitted_`);
-			}
-		}
-		if (details.response) {
-			lines.push("", "### Response", "", cap(details.response, 3000));
-		}
-		if (details.error) {
-			lines.push("", "### Error", "", cap(details.error, 800));
+		if (details.files.length > 0) {
+			lines.push("", "**Files**", "");
+			for (const file of details.files) lines.push(`- ${tableCell(file, 300)}`);
 		}
+		if (details.error) lines.push("", "**Error**", "", quote(details.error));
 		lines.push("");
 	}
 	return lines.join("\n");

package/extensions/subagent/run.ts

@@ -77,6 +77,7 @@ export interface SubagentDetails {
 export interface SubagentInvocationSnapshot extends SubagentDetails {
 	invocationId: string;
 	startedAt: number;
+	files: readonly string[];
 }
 
 export interface SubagentThread {

package/extensions/subagent/runtime.ts

@@ -243,6 +243,7 @@ export class SubagentRuntime {
 			}),
 			invocationId,
 			startedAt,
+			files: [...(options.files ?? [])],
 		};
 		const active: ActiveInvocation = {
 			snapshot: initialSnapshot,
@@ -330,6 +331,7 @@ export class SubagentRuntime {
 				...details,
 				invocationId,
 				startedAt,
+				files: current.snapshot.files,
 				threadId: details.threadId ?? threadId,
 				agent: details.agent || agent,
 				displayName: details.displayName || displayName,

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@shanepadgett/tau-agent",
-	"version": "0.22.0",
+	"version": "0.23.0",
 	"description": "Tau is a custom agentic harness built with pi extensions",
 	"type": "module",
 	"main": "./src/index.ts",
@@ -34,7 +34,7 @@
 		"README.md"
 	],
 	"dependencies": {
-		"@shanepadgett/tau-tui": "0.22.0",
+		"@shanepadgett/tau-tui": "0.23.0",
 		"@toon-format/toon": "2.3.0",
 		"image-size": "2.0.2",
 		"smol-toml": "1.7.0"

package/shared/context-pruning-state.ts

@@ -5,8 +5,6 @@ const CONTEXT_PRUNE_TOOL = "context_prune";
 export interface ContextPruneRefreshedFileV2 {
 	path: string;
 	rowId: string;
-	servedHash: string;
-	autoreadDetails: Record<string, unknown>;
 }
 
 export interface ContextPruneDeferredFileV2 {
@@ -116,13 +114,7 @@ function parseRefreshedFiles(value: unknown): ContextPruneRefreshedFileV2[] | un
 	const files: ContextPruneRefreshedFileV2[] = [];
 	for (const item of value) {
 		if (!isRecord(item) || !isNonEmptyString(item.path) || !isNonEmptyString(item.rowId)) return undefined;
-		if (!isNonEmptyString(item.servedHash) || !isRecord(item.autoreadDetails)) return undefined;
-		files.push({
-			path: item.path,
-			rowId: item.rowId,
-			servedHash: item.servedHash,
-			autoreadDetails: { ...item.autoreadDetails },
-		});
+		files.push({ path: item.path, rowId: item.rowId });
 	}
 	return files;
 }