@henryqw/pi-herdr-rename 0.3.1 → 1.0.1

package/README.md

@@ -1,52 +1,53 @@
 # `@henryqw/pi-herdr-rename`
 
-Pi extension that gives conversations short, model-generated chat titles. It stores the title as the Pi session name and renames the current Herdr pane; the enclosing Herdr tab is renamed only when that pane is the tab's sole pane.
+Give each conversation a short semantic title: Pi session name, current Herdr pane, and a Git branch when the checkout is still generated.
 
 ## Install
 
 ```bash
+pi install npm:@henryqw/pi-task-models
 pi install npm:@henryqw/pi-herdr-rename
 ```
 
-Remove with:
+`pi-task-models` provides `/task-models` for the rename routes. Requires Pi Coding Agent 0.84.2+.
 
-```bash
-pi remove npm:@henryqw/pi-herdr-rename
-```
-
-## Behavior
-
-- On the first real, non-empty text prompt in a new session, title generation starts in the background and does not delay the main Pi response. Extension-injected prompts, empty prompts, and image-only input are ignored.
-- Successful titles are lowercase and, by default, at most four words and 40 characters. The first 1,000 characters of user text are sent to the rename model; prompt content is never logged.
-- A successful title updates the Pi session name and current Herdr pane. The enclosing Herdr tab is updated only when the current tab has one pane. Outside Herdr, only the Pi session name changes.
-- Resuming a named session reapplies its saved title without another rename-model request. Automatic rename-model errors show a warning; other automatic failures stay quiet. Failures do not change labels, fall back locally, or retry.
+## Use
 
-## Manual rename
+| Surface | Purpose |
+| --- | --- |
+| first real user prompt | Generate a title in the background. Does not delay the main reply. |
+| `/rename` | Generate a title from up to three recent user/assistant rounds. |
+| `/task-models` | Configure the `pi-herdr-rename/rename` profile (default `fast`). |
 
-Run `/rename` to generate a title from up to the three most recent user/assistant rounds. It uses text only, caps each message at 1,000 characters and the complete context at 4,000 characters, and applies the same Pi and Herdr rules. An animated widget shows `renaming...`, briefly changes to `renamed to <title>`, then disappears. The command warns without changing anything when no user text exists or generation fails. If requests overlap, the latest rename request wins.
+Successful titles use `type: subject`, lowercase words, and by default at most four words and 40 characters. Example: `fix: extension name` → Git branch `fix/extension-name`.
 
-## Rename model
+In a linked worktree, a detached checkout or Herdr `worktree/...` branch is renamed; an existing non-generated branch stays. A generated workspace label such as `worktree-brave-meadow-4aa8` becomes the semantic branch; custom workspace names stay. The enclosing Herdr tab updates only when this pane is the tab's only pane. Outside Herdr, only the Pi session name changes.
 
-Run `/rename-model` to choose the model from Pi's available authenticated text models. Until a valid selection is saved, every session start prompts you to run the command and title generation remains disabled.
+Tries the assigned profile primary, then fallback. Never substitutes the current session model. No viable route leaves titles unchanged. Resuming a named session reapplies the saved title without another model request.
 
-The selection is saved in:
+## Config
 
-`getAgentDir()/config/pi-herdr-rename.json` (normally `~/.pi/agent/config/pi-herdr-rename.json`)
+`~/.pi/agent/config/pi-herdr-rename.json`
 
 ```json
 {
-  "model": "provider/model",
   "maxWords": 4,
   "maxChars": 40
 }
 ```
 
-`maxWords` and `maxChars` accept positive integers and default to 4 and 40. Invalid limits use their defaults. Missing, malformed, or unavailable model selection never falls back to another model.
+`maxWords` must be at least 2 and `maxChars` at least 6; defaults are 4 and 40. Invalid limits use defaults. Model routes live in `~/.pi/agent/config/pi-task-models.json`. Malformed shared task-model config is reported, left unchanged, and never changes a title.
+
+## Remove
+
+```bash
+pi remove npm:@henryqw/pi-herdr-rename
+```
 
 ## Development
 
 ```bash
-npm test
-npm run typecheck
-npm run pack:check
+npm test --workspace @henryqw/pi-herdr-rename
+npm run typecheck --workspace @henryqw/pi-herdr-rename
+npm run pack:check --workspace @henryqw/pi-herdr-rename
 ```

package/extensions/rename.ts

@@ -1,5 +1,5 @@
-import { mkdir, readFile, writeFile } from "node:fs/promises";
-import { dirname, join } from "node:path";
+import { readFile } from "node:fs/promises";
+import { join } from "node:path";
 import {
 	BorderedLoader,
 	getAgentDir,
@@ -7,44 +7,90 @@ import {
 	type ExtensionContext,
 } from "@earendil-works/pi-coding-agent";
 import { createHerdrClient } from "@henryqw/pi-herdr";
+import {
+	orderedProfileRoutes,
+	readTaskModelsConfig,
+	resolveTaskModelRoute,
+	type ResolvedTaskRoute,
+} from "@henryqw/pi-task-models";
 
 const WIDGET_KEY = "pi-herdr-rename";
 const WIDGET_RESULT_MS = 2_000;
 const MAX_MESSAGE_CHARS = 1_000;
-const MAX_CONTEXT_CHARS = 4_000;
+const MAX_CONTEXT_CHARS = 2_000;
 const DEFAULT_MAX_WORDS = 4;
 const DEFAULT_MAX_CHARS = 40;
+const RENAME_TASK = "pi-herdr-rename/rename";
+const DEFAULT_RENAME_PROFILE = "fast" as const;
+const HERDR_DEFAULT_WORKTREE_NAME = /^(?:worktree[-/])?(?:brave|calm|clear|green|lucky|quiet|rapid|silver)-(?:river|cloud|field|forest|harbor|meadow|stone|valley)-[0-9a-f]{4}$/;
 const configPath = () => join(getAgentDir(), "config", "pi-herdr-rename.json");
 
-type RenameConfig = { model?: string; maxWords: number; maxChars: number };
+type RenameConfig = { maxWords: number; maxChars: number };
 
 class RenameModelError extends Error {}
 
-const positiveInteger = (value: unknown, fallback: number) =>
-	typeof value === "number" && Number.isInteger(value) && value > 0 ? value : fallback;
+const positiveInteger = (value: unknown, fallback: number, minimum = 1) =>
+	typeof value === "number" && Number.isInteger(value) && value >= minimum ? value : fallback;
+
+function configuredRenameRoutes(ctx: ExtensionContext): ResolvedTaskRoute[] {
+	let config;
+	try {
+		config = readTaskModelsConfig();
+	} catch {
+		throw new RenameModelError("Couldn't read task model config. Run /task-models.");
+	}
+
+	const profileName = config.tasks[RENAME_TASK] ?? DEFAULT_RENAME_PROFILE;
+	const profile = config.profiles[profileName];
+	if (!profile) {
+		throw new RenameModelError(`Rename task profile ${profileName} is not configured. Run /task-models.`);
+	}
+
+	const routes = orderedProfileRoutes(profile)
+		.map((route) => resolveTaskModelRoute(ctx, route))
+		.filter((route): route is ResolvedTaskRoute => route !== undefined);
+	if (!routes.length) {
+		throw new RenameModelError(`Rename task profile ${profileName} has no available route. Run /task-models.`);
+	}
+	return routes;
+}
+
+function branchFromTitle(title: string): string | undefined {
+	const match = /^([a-z][a-z0-9-]*): ([a-z0-9]+(?: [a-z0-9]+)*)$/.exec(title);
+	return match ? `${match[1]}/${match[2].replaceAll(" ", "-")}` : undefined;
+}
+
+function branchAvailable(candidate: string, branches: string[]): boolean {
+	return branches.every((branch) => branch !== candidate && !branch.startsWith(`${candidate}/`) && !candidate.startsWith(`${branch}/`));
+}
+
+function availableBranch(branch: string, branches: string[]): string {
+	const [type, subject] = branch.split("/");
+	for (let suffix = 1; suffix <= branches.length + 1; suffix++) {
+		const candidate = suffix === 1 ? branch : `${type}/${subject}-${suffix}`;
+		if (branchAvailable(candidate, branches)) return candidate;
+	}
+	for (let suffix = 2; suffix <= branches.length + 2; suffix++) {
+		const candidate = `${type}-${suffix}/${subject}`;
+		if (branchAvailable(candidate, branches)) return candidate;
+	}
+	throw new Error("Could not choose an available semantic branch.");
+}
 
 async function configured(): Promise<RenameConfig> {
 	try {
 		const config: unknown = JSON.parse(await readFile(configPath(), "utf8"));
 		if (config && typeof config === "object" && !Array.isArray(config)) {
-			const values = config as { model?: unknown; maxWords?: unknown; maxChars?: unknown };
+			const values = config as { maxWords?: unknown; maxChars?: unknown };
 			return {
-				model: typeof values.model === "string" && /^[^\s/]+\/\S+$/.test(values.model) ? values.model : undefined,
-				maxWords: positiveInteger(values.maxWords, DEFAULT_MAX_WORDS),
-				maxChars: positiveInteger(values.maxChars, DEFAULT_MAX_CHARS),
+				maxWords: positiveInteger(values.maxWords, DEFAULT_MAX_WORDS, 2),
+				maxChars: positiveInteger(values.maxChars, DEFAULT_MAX_CHARS, 6),
 			};
 		}
 	} catch {}
 	return { maxWords: DEFAULT_MAX_WORDS, maxChars: DEFAULT_MAX_CHARS };
 }
 
-async function saveModel(model: string): Promise<void> {
-	const path = configPath();
-	const { maxWords, maxChars } = await configured();
-	await mkdir(dirname(path), { recursive: true });
-	await writeFile(path, `${JSON.stringify({ model, maxWords, maxChars }, null, 2)}\n`, "utf8");
-}
-
 function messageText(content: unknown): string {
 	if (typeof content === "string") return content;
 	if (!Array.isArray(content)) return "";
@@ -99,40 +145,67 @@ function recentConversation(ctx: ExtensionContext, fallback?: string): string |
 }
 
 async function generateTitle(text: string, ctx: ExtensionContext, signal: AbortSignal): Promise<string> {
-	const { model: key, maxWords, maxChars } = await configured();
-	if (!key) throw new Error("Rename model is not configured. Run /rename-model.");
-	const separator = key.indexOf("/");
-	const provider = key.slice(0, separator);
-	const id = key.slice(separator + 1);
-	const model = ctx.modelRegistry
-		.getAvailable()
-		.find((candidate) => candidate.provider === provider && candidate.id === id && candidate.input.includes("text"));
-	if (!model) throw new Error(`Rename model unavailable: ${key}. Run /rename-model.`);
-
-	const response = await ctx.modelRegistry.complete(
-		model,
-		{
-			systemPrompt: `Return only a short chat title for the current conversation topic, prioritizing the most recent user intent: lowercase, at most ${maxWords} words, and at most ${maxChars} characters.`,
-			messages: [{ role: "user", content: text.slice(0, MAX_CONTEXT_CHARS), timestamp: Date.now() }],
-		},
-		{ signal, maxRetries: 0, maxTokens: 64 },
-	);
-	if (response.stopReason === "error") {
-		throw new RenameModelError(response.errorMessage || "Rename model failed.");
-	}
-	if (response.stopReason !== "stop") throw new Error("Rename model did not return a complete title.");
-
-	const title = response.content
-		.filter((part) => part.type === "text")
-		.map((part) => part.text)
-		.join(" ")
-		.trim()
-		.toLowerCase()
-		.replace(/\s+/g, " ");
-	if (!title || title.length > maxChars || title.split(" ").length > maxWords) {
-		throw new Error("Rename model returned an invalid title.");
+	const { maxWords, maxChars } = await configured();
+	const completionContext = {
+		systemPrompt: `Return only a semantic title for latest user intent. Format: type: subject. Use lowercase type and lowercase alphanumeric subject words separated by spaces. No other punctuation; at most ${maxWords} words and at most ${maxChars} characters.`,
+		messages: [{ role: "user" as const, content: text.slice(0, MAX_CONTEXT_CHARS), timestamp: Date.now() }],
+	};
+	const complete = async (route: ResolvedTaskRoute) => {
+		let auth;
+		try {
+			auth = await ctx.modelRegistry.getApiKeyAndHeaders(route.model);
+		} catch (error) {
+			if (signal.aborted) throw error;
+			throw new RenameModelError("Couldn't authenticate rename task model.");
+		}
+		if (!auth.ok) throw new RenameModelError("Couldn't authenticate rename task model.");
+
+		const provider = ctx.modelRegistry.getProvider(route.model.provider);
+		if (!provider) throw new RenameModelError("Rename task model provider is unavailable.");
+		const model = auth.baseUrl ? { ...route.model, baseUrl: auth.baseUrl } : route.model;
+
+		let response;
+		try {
+			// streamSimple maps the shared thinking level through this registered model's metadata.
+			response = await provider.streamSimple(model, completionContext, {
+				apiKey: auth.apiKey,
+				headers: auth.headers,
+				env: auth.env,
+				signal,
+				maxRetries: 0,
+				maxTokens: 64,
+				...(route.thinkingLevel === "off" ? {} : { reasoning: route.thinkingLevel }),
+			}).result();
+		} catch (error) {
+			if (signal.aborted) throw error;
+			throw new RenameModelError(error instanceof Error ? error.message : "Rename task model failed.");
+		}
+		if (response.stopReason === "error") throw new RenameModelError(response.errorMessage || "Rename task model failed.");
+		if (response.stopReason !== "stop") throw new RenameModelError("Rename task model did not return a complete title.");
+		return response;
+	};
+
+	let failure: RenameModelError | undefined;
+	for (const route of configuredRenameRoutes(ctx)) {
+		try {
+			const response = await complete(route);
+			const title = response.content
+				.filter((part) => part.type === "text")
+				.map((part) => part.text)
+				.join(" ")
+				.trim()
+				.toLowerCase()
+				.replace(/\s+/g, " ");
+			if (!title || title.length > maxChars || title.split(" ").length > maxWords || !branchFromTitle(title)) {
+				throw new RenameModelError("Rename task model returned an invalid title.");
+			}
+			return title;
+		} catch (error) {
+			if (signal.aborted || !(error instanceof RenameModelError)) throw error;
+			failure = error;
+		}
 	}
-	return title;
+	throw failure ?? new RenameModelError("Rename task model routes failed.");
 }
 
 export default function herdrRenameExtension(pi: ExtensionAPI): void {
@@ -157,7 +230,8 @@ export default function herdrRenameExtension(pi: ExtensionAPI): void {
 		if (!isCurrent(request, controller)) return;
 
 		const paneResponse: unknown = await herdr.json(["pane", "get", paneId], { signal: controller.signal });
-		const tabId = (paneResponse as { result?: { pane?: { tab_id?: unknown } } }).result?.pane?.tab_id;
+		const pane = (paneResponse as { result?: { pane?: { tab_id?: unknown; workspace_id?: unknown } } }).result?.pane;
+		const tabId = pane?.tab_id;
 		if (typeof tabId !== "string" || !tabId) throw new Error("Herdr pane response omitted tab_id.");
 		if (!isCurrent(request, controller)) return;
 
@@ -167,6 +241,40 @@ export default function herdrRenameExtension(pi: ExtensionAPI): void {
 		if (paneCount === 1 && isCurrent(request, controller)) {
 			await herdr.run(["tab", "rename", tabId, title], { signal: controller.signal });
 		}
+		if (!isCurrent(request, controller)) return;
+
+		const workspaceId = pane?.workspace_id;
+		if (typeof workspaceId !== "string" || !workspaceId) throw new Error("Herdr pane response omitted workspace_id.");
+		const workspaceResponse: unknown = await herdr.json(["workspace", "get", workspaceId], { signal: controller.signal });
+		const workspace = (workspaceResponse as { result?: { workspace?: { label?: unknown; worktree?: { checkout_path?: unknown; is_linked_worktree?: unknown } } } }).result?.workspace;
+		const workspaceName = workspace?.label;
+		if (typeof workspaceName !== "string") throw new Error("Herdr workspace response omitted label.");
+		const worktree = workspace?.worktree;
+		const checkoutPath = worktree?.checkout_path;
+		if (worktree?.is_linked_worktree !== true || typeof checkoutPath !== "string" || !checkoutPath) return;
+
+		const runGit = async (args: string[]) => {
+			const result = await pi.exec("git", args, { cwd: checkoutPath, signal: controller.signal });
+			if (result.code !== 0 || result.killed) {
+				throw new Error(`git ${args.join(" ")} failed: ${result.stderr.trim() || result.stdout.trim() || (result.killed ? "killed" : `exit ${result.code}`)}`);
+			}
+			return result.stdout.trim();
+		};
+
+		let branch = await runGit(["branch", "--show-current"]);
+		if (!branch || branch.startsWith("worktree/")) {
+			const generatedBranch = branchFromTitle(title);
+			if (!generatedBranch || !isCurrent(request, controller)) return;
+			const branches = (await runGit(["for-each-ref", "--format=%(refname:short)", "refs/heads"]))
+				.split("\n")
+				.filter(Boolean);
+			const semanticBranch = availableBranch(generatedBranch, branches);
+			await runGit(branch ? ["branch", "-m", semanticBranch] : ["switch", "-c", semanticBranch]);
+			branch = semanticBranch;
+		}
+		if (HERDR_DEFAULT_WORKTREE_NAME.test(workspaceName) && isCurrent(request, controller)) {
+			await herdr.run(["workspace", "rename", workspaceId, branch], { signal: controller.signal });
+		}
 	};
 
 	const begin = () => {
@@ -205,14 +313,20 @@ export default function herdrRenameExtension(pi: ExtensionAPI): void {
 		ctx.ui.setWidget(WIDGET_KEY, undefined);
 	};
 
-	pi.on("session_start", async (_event, ctx) => {
+	pi.on("session_start", (_event, ctx) => {
 		clearWidget(ctx);
 		active?.abort();
 		active = undefined;
 		sequence++;
 		latestUserText = latestSessionUserText(ctx);
-		if (!(await configured()).model) {
-			ctx.ui.notify("Run /rename-model to configure chat title generation.", "warning");
+		try {
+			const taskModels = readTaskModelsConfig();
+			const profileName = taskModels.tasks[RENAME_TASK] ?? DEFAULT_RENAME_PROFILE;
+			if (!taskModels.profiles[profileName]) {
+				ctx.ui.notify(`Configure rename task profile ${profileName} with /task-models.`, "warning");
+			}
+		} catch {
+			ctx.ui.notify("Couldn't read task model config. Run /task-models.", "warning");
 		}
 		const title = pi.getSessionName();
 		automaticStarted = Boolean(title || latestUserText);
@@ -271,27 +385,4 @@ export default function herdrRenameExtension(pi: ExtensionAPI): void {
 			widgetTimer.unref?.();
 		},
 	});
-
-	pi.registerCommand("rename-model", {
-		description: "Choose the model used to generate chat titles",
-		handler: async (_args, ctx) => {
-			const models = ctx.modelRegistry
-				.getAvailable()
-				.filter((model) => model.input.includes("text"))
-				.map((model) => `${model.provider}/${model.id}`)
-				.sort();
-			if (!models.length) {
-				ctx.ui.notify("No authenticated text models are available.", "warning");
-				return;
-			}
-			const selected = await ctx.ui.select("Rename model", models);
-			if (!selected) return;
-			try {
-				await saveModel(selected);
-				ctx.ui.notify(`Rename model saved: ${selected}`, "info");
-			} catch {
-				ctx.ui.notify("Couldn't save rename model config.", "warning");
-			}
-		},
-	});
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@henryqw/pi-herdr-rename",
-  "version": "0.3.1",
+  "version": "1.0.1",
   "description": "Generate short Pi chat titles and rename the current Herdr location.",
   "keywords": [
     "pi-package",
@@ -25,7 +25,8 @@
     "pack:check": "npm pack --dry-run"
   },
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": "*"
+    "@earendil-works/pi-ai": "^0.84.2",
+    "@earendil-works/pi-coding-agent": "^0.84.2"
   },
   "repository": {
     "type": "git",
@@ -44,6 +45,7 @@
     ]
   },
   "dependencies": {
-    "@henryqw/pi-herdr": "^0.1.0"
+    "@henryqw/pi-herdr": "^0.1.0",
+    "@henryqw/pi-task-models": "^0.1.0"
   }
 }