@shanepadgett/tau-agent 0.27.0 → 0.28.1

package/{extensions/explore → shared}/autoread.ts

@@ -2,11 +2,10 @@ import { createHash } from "node:crypto";
 import { readFile } from "node:fs/promises";
 import { resolve } from "node:path";
 import type { ExtensionAPI, Theme } from "@earendil-works/pi-coding-agent";
-import { onTauEvent, type TauAgentEvents } from "../../shared/events.js";
-import type { ToolRowStateStore } from "../../shared/tool-row-state.js";
 import { Marker, type MarkerState } from "@shanepadgett/tau-tui";
-import { createCompleteFileMeta } from "./full-file-knowledge.ts";
-import type { ReadCacheMetaV1 } from "./read-cache.ts";
+import { onTauEvent, type TauAgentEvents } from "./events.js";
+import { createCompleteFileMeta, type ReadCacheMetaV1 } from "./full-file-knowledge.ts";
+import type { ToolRowStateStore } from "./tool-row-state.js";
 
 const AUTOREAD_MESSAGE_TYPE = "tau.autoread";
 
@@ -30,6 +29,17 @@ export interface PreparedAutoreadMessage {
 	details: AutoreadDetails & { status: "read"; readCache: ReadCacheMetaV1 };
 }
 
+export type PrepareAutoreadMessage = (options: {
+	rowId: string;
+	path: string;
+	cwd: string;
+	source: string;
+	batchId: string;
+	signal: AbortSignal | undefined;
+	isLifecycleCurrent: () => boolean;
+	maximumBytes?: number;
+}) => Promise<PreparedAutoreadMessage>;
+
 export async function prepareAutoreadMessage(options: {
 	rowId: string;
 	path: string;
@@ -76,7 +86,11 @@ export async function prepareAutoreadMessage(options: {
 	};
 }
 
-export function registerAutoread(pi: ExtensionAPI, rowState: ToolRowStateStore): void {
+export function registerAutoread(
+	pi: ExtensionAPI,
+	rowState: ToolRowStateStore,
+	prepare: PrepareAutoreadMessage = prepareAutoreadMessage,
+): void {
 	let lifecycleGeneration = 0;
 	pi.on("session_start", () => {
 		lifecycleGeneration += 1;
@@ -90,7 +104,7 @@ export function registerAutoread(pi: ExtensionAPI, rowState: ToolRowStateStore):
 	pi.on("session_shutdown", () => {
 		lifecycleGeneration += 1;
 	});
-	onTauEvent(pi, "explore.autoread", "tau:autoread.requested", async (data) => {
+	onTauEvent(pi, "shared.autoread", "tau:autoread.requested", async (data) => {
 		const event = readAutoreadRequestedEvent(data);
 		if (!event) return;
 		const generation = lifecycleGeneration;
@@ -105,7 +119,7 @@ export function registerAutoread(pi: ExtensionAPI, rowState: ToolRowStateStore):
 					batchId: event.batchId,
 				} satisfies Omit<AutoreadDetails, "status" | "error">;
 				try {
-					const message = await prepareAutoreadMessage({
+					const message = await prepare({
 						...details,
 						signal: undefined,
 						isLifecycleCurrent: () => generation === lifecycleGeneration,

package/shared/events.ts

@@ -15,7 +15,7 @@ export type TauAgentEvents = {
 		disposition: "notify" | "discard";
 	};
 	"tau:file-mutation.applied": {
-		source: "patch" | "locatorEdit";
+		source: "patch";
 		toolCallId: string;
 		cwd: string;
 		status: "completed" | "partial" | "failed";

package/shared/full-file-knowledge.ts

@@ -0,0 +1,49 @@
+type ReadCacheMode = "baseline" | "recovery" | "unchanged" | "diff";
+type ReadCachePresentation = "plain" | "line-numbered";
+
+export interface ReadCacheMetaV1 {
+	v: 1;
+	pathKey: string;
+	scopeKey: string;
+	presentation: ReadCachePresentation;
+	servedHash: string;
+	baseHash?: string;
+	mode: ReadCacheMode;
+	baselineTokens: number;
+	returnedTokens: number;
+	totalLines: number;
+	summary: string;
+}
+
+const COMPLETE_FILE_SCOPE = "full";
+export const MAX_COMPLETE_FILE_SNAPSHOT_BYTES = 1024 * 1024;
+
+export function createCompleteFileMeta(options: {
+	pathKey: string;
+	presentation: ReadCachePresentation;
+	servedHash: string;
+	mode: ReadCacheMode;
+	sourceText: string;
+	returnedText: string;
+	totalLines: number;
+	summary: string;
+	baseHash?: string;
+}): ReadCacheMetaV1 {
+	return {
+		v: 1,
+		pathKey: options.pathKey,
+		scopeKey: COMPLETE_FILE_SCOPE,
+		presentation: options.presentation,
+		servedHash: options.servedHash,
+		baseHash: options.baseHash,
+		mode: options.mode,
+		baselineTokens: estimateTokens(options.sourceText),
+		returnedTokens: estimateTokens(options.returnedText),
+		totalLines: options.totalLines,
+		summary: options.summary,
+	};
+}
+
+function estimateTokens(text: string): number {
+	return Math.ceil(text.length / 4);
+}

package/extensions/explore/ast-guidance.ts

@@ -1,67 +0,0 @@
-import {
-	AST_LANGUAGE_REGISTRY,
-	astLanguageForPath,
-	formatAstLanguageLabels,
-	type AstLanguage,
-} from "./ast-languages.ts";
-import { collectPaths } from "./traverse.ts";
-
-export const AST_DISCOVERY_BUDGET = 4096;
-
-export interface AstGuidanceRequest {
-	cwd: string;
-	workerLanguages(): Promise<readonly AstLanguage[]>;
-	discoveryBudget: number;
-}
-
-export async function effectiveAstGuidance(request: AstGuidanceRequest): Promise<string | undefined> {
-	let detected: Set<AstLanguage>;
-	try {
-		const entries = await collectPaths({
-			cwd: request.cwd,
-			root: request.cwd,
-			maxEntries: request.discoveryBudget,
-			includeRoot: false,
-			includeHidden: true,
-			includeIgnored: false,
-			includeNoise: false,
-		});
-		detected = new Set(
-			entries.flatMap((entry) => {
-				if (entry.type !== "file") return [];
-				const language = astLanguageForPath(entry.absolutePath);
-				return language ? [language] : [];
-			}),
-		);
-	} catch {
-		return undefined;
-	}
-
-	if (detected.size === 0) return undefined;
-	let workerLanguages: readonly AstLanguage[];
-	try {
-		workerLanguages = await request.workerLanguages();
-	} catch {
-		return undefined;
-	}
-	const available = new Set(workerLanguages);
-	const languages = AST_LANGUAGE_REGISTRY.filter(
-		(language) => detected.has(language.workerLanguage) && available.has(language.workerLanguage),
-	);
-	if (languages.length === 0) return undefined;
-	const labels = formatAstLanguageLabels(languages, "and");
-	return `## Explore source policy
-
-AST-backed exploration is available here for ${labels}.
-
-- Use the smallest structural query that can answer the current question. Stop when the result is sufficient; expand only for a specific unresolved question.
-- Identify the current job before exploring: locate, reuse, edit, explain, or debug. Do not preload implementation, callers, tests, or documentation needed only by a later stage.
-- For an unfamiliar repository or subtree, use recursive \`outline\` when available. Otherwise find a likely package, then outline that directory. For a known package or file, outline it directly.
-- Outline large Markdown files first, then retrieve only the relevant heading section with \`symbol\`.
-- For reuse work with an unknown location, use \`api_discover\` and prefer \`packageSurface\`, \`sourceExport\`, or \`public\` results. Inspect private declarations only when targeted implementation work requires them.
-- Narrow paths, exact \`names\`, query work, and result limits as soon as the likely target is known.
-- Treat a clear outline or API candidate signature as enough. Use \`symbol(signature)\` only when that selected contract needs closer inspection, \`signatureWithDocs\` only when attached documentation matters, and declaration views only for exact implementation source.
-- Discover references, callers, implementations, and affected tests after selecting a change target and when that impact information can change the plan. Do not inspect tests up front unless the task starts from test behavior or a failure.
-- Use \`ast_search\` for code shapes and \`grep\` for literal text. Use targeted \`read\` for exact formatting, comments, unsupported files, or source outside declaration boundaries.
-- Prefer locator edits for a known declaration or body when they express the complete change cleanly. Use textual patching when the change crosses structural boundaries or depends on exact surrounding text.`;
-}

package/extensions/explore/ast-languages.ts

@@ -1,61 +0,0 @@
-import { basename, extname } from "node:path";
-
-export const AST_LANGUAGE_REGISTRY = [
-	{
-		id: "typescript",
-		label: "TypeScript",
-		extensions: [".ts"],
-		workerLanguage: "typeScript",
-	},
-	{ id: "tsx", label: "TSX", extensions: [".tsx"], workerLanguage: "tsx" },
-	{ id: "odin", label: "Odin", extensions: [".odin"], workerLanguage: "odin" },
-	{ id: "go", label: "Go", extensions: [".go"], workerLanguage: "go" },
-	{ id: "rust", label: "Rust", extensions: [".rs"], workerLanguage: "rust" },
-	{ id: "csharp", label: "C#", extensions: [".cs"], workerLanguage: "cSharp" },
-	{ id: "java", label: "Java", extensions: [".java"], workerLanguage: "java" },
-	{
-		id: "kotlin",
-		label: "Kotlin",
-		extensions: [".kt", ".ktm", ".kts"],
-		workerLanguage: "kotlin",
-	},
-	{ id: "swift", label: "Swift", extensions: [".swift"], workerLanguage: "swift" },
-	{
-		id: "markdown",
-		label: "Markdown",
-		extensions: [".md", ".markdown", ".mdown"],
-		workerLanguage: "markdown",
-	},
-] as const;
-
-export type AstLanguageDefinition = (typeof AST_LANGUAGE_REGISTRY)[number];
-export type AstLanguage = AstLanguageDefinition["workerLanguage"];
-
-export function isAstLanguage(value: unknown): value is AstLanguage {
-	return AST_LANGUAGE_REGISTRY.some((language) => language.workerLanguage === value);
-}
-
-export function astLanguageForPath(path: string): AstLanguage | undefined {
-	const extension = extname(basename(path)).toLowerCase();
-	for (const language of AST_LANGUAGE_REGISTRY) {
-		const extensions: readonly string[] = language.extensions;
-		if (extensions.includes(extension)) return language.workerLanguage;
-	}
-	return undefined;
-}
-
-export function requireAstLanguageForPath(path: string): AstLanguage {
-	const language = astLanguageForPath(path);
-	if (language) return language;
-	throw new Error(`Unsupported outline file type: ${extname(path) || "no extension"}`);
-}
-
-export function formatAstLanguageLabels(
-	languages: readonly Pick<AstLanguageDefinition, "label">[],
-	conjunction: "and" | "or",
-): string {
-	const labels = languages.map((language) => language.label);
-	if (labels.length < 2) return labels[0] ?? "";
-	if (labels.length === 2) return `${labels[0]} ${conjunction} ${labels[1]}`;
-	return `${labels.slice(0, -1).join(", ")}, ${conjunction} ${labels.at(-1)}`;
-}