@shanepadgett/tau-agent 0.20.2 → 0.22.0

package/extensions/image-gen/index.ts

@@ -1,20 +1,20 @@
-import { defineTool, withFileMutationQueue, type ExtensionAPI } from "@earendil-works/pi-coding-agent";
-import { randomUUID } from "node:crypto";
-import { link, mkdir, readFile, rm, stat, writeFile } from "node:fs/promises";
-import { homedir } from "node:os";
-import { basename, dirname, extname, isAbsolute, join, resolve } from "node:path";
+import { defineTool, type ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import {
+	generateImage,
+	type GeneratedImageResult,
+	type GenerateImageRequest,
+	type ImageGenerationContext,
+	type ImageProvider,
+} from "@shanepadgett/tau-agent";
 import { type Static, Type } from "typebox";
-import { detectImageMimeType, editImage, generateImage, type EditImage, type GeneratedImage } from "./client.ts";
-import { OPENAI_IMAGE_MODEL, OPENAI_PROVIDER, XAI_IMAGE_MODEL, XAI_PROVIDER, type ImageProvider } from "./constants.ts";
 
-const MAX_INPUT_BYTES = 50 * 1024 * 1024;
 const MAX_INLINE_BYTES = 12 * 1024 * 1024;
 
 const imageGenSchema = Type.Object(
 	{
 		prompt: Type.String({ minLength: 1 }),
 		provider: Type.Optional(
-			Type.Union([Type.Literal("openai"), Type.Literal("xai")], {
+			Type.Union([Type.Literal("openai" satisfies ImageProvider), Type.Literal("xai" satisfies ImageProvider)], {
 				description:
 					"Image provider override. Omit to follow the parent model, preferring OpenAI for GPT and xAI for Grok.",
 			}),
@@ -28,19 +28,7 @@ const imageGenSchema = Type.Object(
 );
 
 type ImageGenParams = Static<typeof imageGenSchema>;
-
-interface ImageGenDetails {
-	path: string;
-	provider: ImageProvider;
-	model: typeof OPENAI_IMAGE_MODEL | typeof XAI_IMAGE_MODEL;
-	operation: "generate" | "edit";
-}
-
-function outputExtension(image: GeneratedImage): string {
-	if (image.mimeType === "image/png") return ".png";
-	if (image.mimeType === "image/webp") return ".webp";
-	return ".jpg";
-}
+type ImageGenDetails = Omit<GeneratedImageResult, "bytes">;
 
 export default function imageGenExtension(pi: ExtensionAPI): void {
 	pi.registerTool(
@@ -51,127 +39,30 @@ export default function imageGenExtension(pi: ExtensionAPI): void {
 				"Generate a requested raster image or AI-edit existing images with OpenAI GPT Image or xAI Grok Imagine. Omit provider to follow the parent model; set it to openai or xai to override. Omit referenced_image_paths to generate; pass one to three local paths to edit or compose. Omit path to use Tau's external image store; pass path only when the user explicitly requests a repository file or other destination. Returns the image for inspection.",
 			parameters: imageGenSchema,
 			async execute(_toolCallId, params: ImageGenParams, signal, onUpdate, ctx) {
-				signal?.throwIfAborted();
-				const prompt = params.prompt.trim();
-				if (!prompt) throw new Error("Image prompt cannot be empty");
-				const requestedPath = params.path?.startsWith("@") ? params.path.slice(1) : params.path;
-				if (requestedPath !== undefined && !requestedPath.trim()) throw new Error("Image path cannot be empty");
-				const requestedAbsolutePath = requestedPath
-					? isAbsolute(requestedPath)
-						? requestedPath
-						: resolve(ctx.cwd, requestedPath)
-					: undefined;
-				if (
-					requestedAbsolutePath &&
-					![".jpg", ".jpeg", ".png", ".webp"].includes(extname(requestedAbsolutePath).toLowerCase())
-				) {
-					throw new Error("Image path must end in .jpg, .jpeg, .png, or .webp");
-				}
-
-				const parentUsesXai =
-					ctx.model?.provider.toLowerCase() === XAI_PROVIDER ||
-					ctx.model?.id.toLowerCase().includes("grok") === true;
-				const preferredProvider: ImageProvider = params.provider ?? (parentUsesXai ? "xai" : "openai");
-				const providers: readonly ImageProvider[] = params.provider
-					? [params.provider]
-					: preferredProvider === "xai"
-						? ["xai", "openai"]
-						: ["openai", "xai"];
-				let provider: ImageProvider | undefined;
-				let token: string | undefined;
-				for (const candidate of providers) {
-					const candidateToken = await ctx.modelRegistry.getApiKeyForProvider(
-						candidate === "openai" ? OPENAI_PROVIDER : XAI_PROVIDER,
-					);
-					if (candidateToken) {
-						provider = candidate;
-						token = candidateToken;
-						break;
-					}
-				}
-				if (!provider || !token) {
-					if (params.provider === "openai") {
-						throw new Error("OpenAI Codex authentication is unavailable. Run /login openai-codex.");
-					}
-					if (params.provider === "xai") {
-						throw new Error("xAI authentication is unavailable. Run /login xai and choose a login method.");
-					}
-					throw new Error("Image generation authentication is unavailable. Run /login for OpenAI Codex or xAI.");
-				}
-				const model = provider === "openai" ? OPENAI_IMAGE_MODEL : XAI_IMAGE_MODEL;
-				const images: EditImage[] = [];
-				for (const path of params.referenced_image_paths ?? []) {
-					const rawPath = path.startsWith("@") ? path.slice(1) : path;
-					if (!rawPath.trim()) throw new Error("Referenced image path cannot be empty");
-					const absolutePath = isAbsolute(rawPath) ? rawPath : resolve(ctx.cwd, rawPath);
-					const metadata = await stat(absolutePath);
-					if (!metadata.isFile()) throw new Error(`Referenced image is not a file: ${absolutePath}`);
-					if (metadata.size > MAX_INPUT_BYTES) throw new Error(`Referenced image exceeds 50 MiB: ${absolutePath}`);
-					const bytes = await readFile(absolutePath);
-					if (bytes.length > MAX_INPUT_BYTES) throw new Error(`Referenced image exceeds 50 MiB: ${absolutePath}`);
-					const mimeType = detectImageMimeType(bytes);
-					if (!mimeType)
-						throw new Error(`Referenced image is not a supported PNG, JPEG, or WebP file: ${absolutePath}`);
-					images.push({ mimeType, data: bytes.toString("base64") });
-				}
-
-				const operation = images.length === 0 ? "generate" : "edit";
 				await onUpdate?.({
 					content: [
 						{
 							type: "text",
-							text:
-								operation === "generate"
-									? `Generating image with ${model}...`
-									: `Editing image with ${model}...`,
+							text: "Processing image request...",
 						},
 					],
 					details: undefined,
 				});
-				const generated =
-					operation === "generate"
-						? await generateImage(provider, prompt, token, signal)
-						: await editImage(provider, prompt, images, token, signal);
-				signal?.throwIfAborted();
-				const generatedExtension = outputExtension(generated);
-				if (requestedAbsolutePath) {
-					const requestedExtension = extname(requestedAbsolutePath).toLowerCase();
-					const matches =
-						requestedExtension === generatedExtension ||
-						(generatedExtension === ".jpg" && requestedExtension === ".jpeg");
-					if (!matches)
-						throw new Error(
-							`${provider === "openai" ? "OpenAI Codex" : "xAI"} returned ${generated.mimeType}; destination must end in ${generatedExtension}`,
-						);
-				}
-				const absolutePath =
-					requestedAbsolutePath ??
-					join(homedir(), ".local", "share", "tau-agent", "images", `image-${randomUUID()}${generatedExtension}`);
-
-				const outputDirectory = dirname(absolutePath);
-				await withFileMutationQueue(absolutePath, async () => {
-					await mkdir(outputDirectory, { recursive: true });
-					const temporaryPath = join(
-						outputDirectory,
-						`.${basename(absolutePath)}.${randomUUID()}.tmp${generatedExtension}`,
-					);
-					try {
-						await writeFile(temporaryPath, generated.bytes, { flag: "wx" });
-						signal?.throwIfAborted();
-						await link(temporaryPath, absolutePath);
-					} finally {
-						await rm(temporaryPath, { force: true });
-					}
-				});
-
-				const verb = operation === "generate" ? "Generated" : "Edited";
-				const details: ImageGenDetails = { path: absolutePath, provider, model, operation };
-				if (generated.bytes.length > MAX_INLINE_BYTES) {
+				const request = {
+					prompt: params.prompt,
+					provider: params.provider,
+					path: params.path,
+					referencedImagePaths: params.referenced_image_paths,
+					signal,
+				} satisfies GenerateImageRequest;
+				const { bytes, ...details } = await generateImage(ctx satisfies ImageGenerationContext, request);
+				const verb = details.operation === "generate" ? "Generated" : "Edited";
+				if (bytes.length > MAX_INLINE_BYTES) {
 					return {
 						content: [
 							{
 								type: "text",
-								text: `${verb} image saved to ${absolutePath}. The image exceeds the 12 MiB attachment limit, so it was not added to model context.`,
+								text: `${verb} image saved to ${details.path}. The image exceeds the 12 MiB attachment limit, so it was not added to model context.`,
 							},
 						],
 						details,
@@ -179,8 +70,8 @@ export default function imageGenExtension(pi: ExtensionAPI): void {
 				}
 				return {
 					content: [
-						{ type: "text", text: `${verb} image saved to ${absolutePath}` },
-						{ type: "image", data: generated.base64, mimeType: generated.mimeType },
+						{ type: "text", text: `${verb} image saved to ${details.path}` },
+						{ type: "image", data: bytes.toString("base64"), mimeType: details.mimeType },
 					],
 					details,
 				};

package/extensions/subagent/index.ts

@@ -145,7 +145,9 @@ Start a fresh thread with \`agent\` and \`task\`. Continue an existing thread wi
 
 Pass \`files\` when exact relevant files are already known. Tau autoreads current line-numbered snapshots into that child turn before it starts.
 
-Delegate one focused task per call. Children do not inherit parent messages. Include exact absolute reference paths when a child must inspect a repository outside the current working directory.`;
+Delegate one focused task per call. Children do not inherit parent messages. Include exact absolute reference paths when a child must inspect a repository outside the current working directory.
+
+Review limit: for one user task or coherent implementation batch, call the \`review\` agent at most twice. The first call is the initial review. The only permitted second call is a follow-up in the same retained thread to check fixes from that initial review. Never call \`review\` a third time for the same task. Do not call it for documentation-only work, trivial edits, small localized changes, or micro-adjustments after review; inspect and test those directly.`;
 		return { systemPrompt: `${event.systemPrompt}\n\n${prompt}` };
 	});
 	pi.registerTool(

package/extensions/tau-help/help.md

@@ -36,7 +36,7 @@ Adds `/context` to select reusable repository work scopes from `.pi/contexts`, a
 
 ## context-pruning
 
-Gives the agent `context_prune` for removing substantial stale tool evidence after broad exploration converges. The agent records durable conclusions before creating an anchor, can retain exact exchanges or current file knowledge, and continues the task automatically. Context markers occasionally steer the agent after tool-heavy growth and become more direct under pressure. Run `/prune` with no arguments to request an anchor manually. Session history remains unchanged, and anchors, markers, and pruned-row state follow the active branch.
+Gives the agent `context_prune` for creating a hard context checkpoint after broad exploration converges. Everything before the checkpoint leaves future model input unless the agent retains an exact tool exchange or carries a file forward as a fresh snapshot. File snapshot failures are reported without blocking the checkpoint. Deferred files remain as a small conditional note. Context markers use an ordered instruction ladder that escalates from informational guidance to pruning before further tool work. Run `/prune` with no arguments to request a checkpoint manually. Session history remains unchanged, and checkpoints, markers, and pruned-row state follow the active branch.
 
 ## explore
 
@@ -46,6 +46,10 @@ Replaces Pi’s filesystem inspection tools with compact Tau versions: `ls`, `fi
 
 Adds `/footer` to toggle and refresh Tau’s status footer.
 
+## handoff
+
+Adds `/handoff <goal>` to start a fresh linked session from the current conversation. Tau generates an opening prompt, autoreads the relevant project files it already knows about, and leaves the prompt in the editor for review instead of submitting it.
+
 ## ideas
 
 Adds `/ideas` to log rough ideas or open the ideas browser.

package/package.json

@@ -1,8 +1,13 @@
 {
 	"name": "@shanepadgett/tau-agent",
-	"version": "0.20.2",
+	"version": "0.22.0",
 	"description": "Tau is a custom agentic harness built with pi extensions",
 	"type": "module",
+	"main": "./src/index.ts",
+	"types": "./src/index.ts",
+	"exports": {
+		".": "./src/index.ts"
+	},
 	"license": "MIT",
 	"keywords": [
 		"pi-package",
@@ -18,6 +23,7 @@
 		"access": "public"
 	},
 	"files": [
+		"src/**/*",
 		"extensions/**/*",
 		"shared/**/*",
 		"prompts/**/*",
@@ -28,8 +34,9 @@
 		"README.md"
 	],
 	"dependencies": {
-		"@shanepadgett/tau-tui": "0.20.2",
+		"@shanepadgett/tau-tui": "0.22.0",
 		"@toon-format/toon": "2.3.0",
+		"image-size": "2.0.2",
 		"smol-toml": "1.7.0"
 	},
 	"peerDependencies": {

package/schemas/tau.schema.json

@@ -68,18 +68,21 @@
 							"maximum": 100,
 							"description": "Context growth interval between automatic pruning hints."
 						},
-						"pressurePercent": {
-							"type": "integer",
-							"default": 50,
-							"minimum": 1,
-							"maximum": 99,
-							"description": "Context usage above which pruning hints become urgent."
-						},
-						"minimumReclaimTokens": {
-							"type": "integer",
-							"default": 8000,
-							"minimum": 1,
-							"description": "Minimum estimated tokens a prune must reclaim."
+						"nudgeInstructions": {
+							"type": "array",
+							"items": {
+								"type": "string",
+								"minLength": 1,
+								"maxLength": 2000
+							},
+							"default": [
+								"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."
+							],
+							"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