@shanepadgett/tau-agent 0.28.0 → 0.28.1

package/extensions/explore/ast/ir.ts

@@ -83,6 +83,12 @@ export type Decl = {
 	endOffset: number;
 	/** Decl start → body open; signature = source.slice(startOffset, signatureEndOffset). */
 	signatureEndOffset: number;
+	/** Initializer/body spans replaced with compact markers when a signature cannot be one contiguous source slice. */
+	signatureOmissions?: Array<{
+		startOffset: number;
+		endOffset: number;
+		replacement: string;
+	}>;
 	/** Absent when no body. */
 	bodyStartOffset?: number;
 	bodyEndOffset?: number;

package/extensions/explore/ast/languages/typescript.ts

@@ -300,6 +300,32 @@ function valueIsFunction(value: Node | null): boolean {
 	);
 }
 
+function compactInitializer(value: Node): Decl["signatureOmissions"] {
+	const omissions: NonNullable<Decl["signatureOmissions"]> = [];
+	for (const callable of value.descendantsOfType([
+		ARROW_FUNCTION,
+		FUNCTION_EXPRESSION,
+		FUNCTION_DECLARATION,
+		GENERATOR_FUNCTION_DECLARATION,
+	])) {
+		const body = field(callable, "body");
+		if (body === null) continue;
+		if (
+			omissions.some((omission) => body.startIndex >= omission.startOffset && body.endIndex <= omission.endOffset)
+		) {
+			continue;
+		}
+		omissions.push({
+			startOffset: body.startIndex,
+			endOffset: body.endIndex,
+			replacement: body.type === "statement_block" ? "{ … }" : "…",
+		});
+	}
+	if (omissions.length > 0) return omissions;
+	if (value.startPosition.row === value.endPosition.row) return undefined;
+	return [{ startOffset: value.startIndex, endOffset: value.endIndex, replacement: "…" }];
+}
+
 function walkStatementList(nodes: readonly Node[], owner: string, source: string, exportedDefault: boolean): Decl[] {
 	const decls: Decl[] = [];
 	for (const node of nodes) {
@@ -482,7 +508,10 @@ function declsFromExport(node: Node, owner: string, source: string): Decl[] {
 			},
 			body,
 		);
-		if (body === null) decl.signatureEndOffset = node.endIndex;
+		if (body === null) {
+			decl.signatureEndOffset = node.endIndex;
+			decl.signatureOmissions = compactInitializer(value);
+		}
 		return [applyDoc(decl, doc)];
 	}
 
@@ -523,7 +552,10 @@ function variableDecls(node: Node, owner: string, exported: boolean, doc: DocSpa
 			},
 			body,
 		);
-		if (body === null) decl.signatureEndOffset = spanNode.endIndex;
+		if (body === null) {
+			decl.signatureEndOffset = spanNode.endIndex;
+			if (value !== null) decl.signatureOmissions = compactInitializer(value);
+		}
 		out.push(i === 0 ? applyDoc(decl, doc) : decl);
 	}
 	return out;
@@ -583,6 +615,9 @@ function classMembers(body: Node, owner: string, source: string): Decl[] {
 				doc,
 				visibilityOf(node),
 			);
+			if (decl !== undefined && !isFn && value !== null) {
+				decl.signatureOmissions = compactInitializer(value);
+			}
 			if (decl !== undefined) out.push(decl);
 			continue;
 		}

package/extensions/explore/ast/slice.ts

@@ -4,7 +4,20 @@ import type { Decl } from "./ir.ts";
 
 /** Decl signature: [startOffset, signatureEndOffset) — no docs, no body. */
 export function signatureText(decl: Decl, source: string): string {
-	return source.slice(decl.startOffset, decl.signatureEndOffset);
+	return signatureSlice(decl, source, decl.startOffset);
+}
+
+function signatureSlice(decl: Decl, source: string, startOffset: number): string {
+	if (decl.signatureOmissions === undefined) return source.slice(startOffset, decl.signatureEndOffset);
+
+	let text = "";
+	let cursor = startOffset;
+	for (const omission of decl.signatureOmissions) {
+		if (omission.startOffset < cursor || omission.endOffset > decl.signatureEndOffset) continue;
+		text += source.slice(cursor, omission.startOffset) + omission.replacement;
+		cursor = omission.endOffset;
+	}
+	return text + source.slice(cursor, decl.signatureEndOffset);
 }
 
 /** Attached doc comment text, or undefined when none. */
@@ -27,5 +40,5 @@ export function signatureWithDocsText(decl: Decl, source: string): string {
 	if (decl.docStartOffset === undefined || decl.docEndOffset === undefined) {
 		return signatureText(decl, source);
 	}
-	return source.slice(decl.docStartOffset, decl.signatureEndOffset);
+	return signatureSlice(decl, source, decl.docStartOffset);
 }

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@shanepadgett/tau-agent",
-	"version": "0.28.0",
+	"version": "0.28.1",
 	"description": "Tau is a custom agentic harness built with pi extensions",
 	"type": "module",
 	"main": "./src/index.ts",
@@ -35,7 +35,7 @@
 	],
 	"dependencies": {
 		"@ast-grep/wasm": "0.45.0",
-		"@shanepadgett/tau-tui": "0.28.0",
+		"@shanepadgett/tau-tui": "0.28.1",
 		"@vscode/tree-sitter-wasm": "0.3.1",
 		"image-size": "2.0.2",
 		"smol-toml": "1.7.0",