@henryqw/pi-herdr-rename 0.4.0 → 1.0.2

package/README.md

@@ -1,52 +1,59 @@
 # `@henryqw/pi-herdr-rename`
 
-Pi extension that gives conversations semantic model-generated titles. It stores each title as Pi session name and renames current Herdr pane. In a linked worktree, it creates a semantic Git branch for detached or Herdr-generated branches, then uses that branch for a generated workspace label; enclosing Herdr tab is renamed only when pane is 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:
+Requires Pi Coding Agent 0.84.2+.
 
-```bash
-pi remove npm:@henryqw/pi-herdr-rename
-```
+## With
 
-## Behavior
+| Package | Why |
+| --- | --- |
+| `@henryqw/pi-task-models` | Required. Shared model profiles for title generation. |
 
-- 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 use `type: subject`, lowercase words, and by default at most four words and 40 characters. For example, `fix: extension name` maps to Git branch `fix/extension-name`. First 1,000 characters of user text go to rename model; prompt content is never logged.
-- A successful title updates Pi session name and current Herdr pane. In a linked worktree, a detached checkout gets `git switch -c <generated-branch>`; a Herdr-generated `worktree/...` branch gets renamed. Conflicting local refs add numeric suffix such as `-2`. Existing non-generated branch remains unchanged. If workspace label still matches generated default pattern such as `worktree-brave-meadow-4aa8`, it is renamed to semantic branch; custom workspace names stay unchanged. Enclosing Herdr tab updates only when current tab has one pane. Outside Herdr, only Pi session name changes.
-- Resuming a named session reapplies saved title without another rename-model request. A configured rename-model transport/fetch failure tries current text-capable main model once. A final automatic rename-model error shows a warning; other automatic failures stay quiet. Failures do not change labels or retry.
+## Use
 
-## Manual rename
+| Surface | Type | Purpose |
+| --- | --- | --- |
+| `/rename` | command | Generate a title from up to three recent user/assistant rounds. |
 
-Run `/rename` to generate a semantic title from up to three most recent user/assistant rounds. It uses text only, caps each message at 1,000 characters and complete context at 2,000 characters, and applies same Pi, Git, and Herdr rules. Animated widget shows `renaming...`, briefly changes to `renamed to <title>`, then disappears. Command warns without changing anything when no user text exists or generation fails. If title-generation requests overlap, latest title request wins.
+First real user prompt also generates a title in the background and does not delay the main reply.
 
-## Rename model
+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`.
 
-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.
+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.
 
-The selection is saved in:
+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.
 
-`getAgentDir()/config/pi-herdr-rename.json` (normally `~/.pi/agent/config/pi-herdr-rename.json`)
+## Config
+
+`~/.pi/agent/config/pi-herdr-rename.json`
 
 ```json
 {
-  "model": "provider/model",
   "maxWords": 4,
   "maxChars": 40
 }
 ```
 
-`maxWords` must be at least 2 and `maxChars` at least 6; defaults are 4 and 40. Invalid limits use defaults. Missing, malformed, or unavailable model selection does not fall back; a configured-model transport/fetch failure uses current text-capable main model once.
+`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,6 +7,12 @@ 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;
@@ -14,18 +20,41 @@ const MAX_MESSAGE_CHARS = 1_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 };
-
-const isTransportFailure = (message: string) => /\b(?:fetch failed|network[- ]error|connection[- ]error|timed? out|timeout)\b/i.test(message);
+type RenameConfig = { maxWords: number; maxChars: number };
 
 class RenameModelError extends Error {}
 
 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;
@@ -52,9 +81,8 @@ 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, 2),
 				maxChars: positiveInteger(values.maxChars, DEFAULT_MAX_CHARS, 6),
 			};
@@ -63,13 +91,6 @@ async function configured(): Promise<RenameConfig> {
 	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 "";
@@ -124,60 +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 { 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 (target: NonNullable<ExtensionContext["model"]>) => {
+	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 {
-			response = await ctx.modelRegistry.complete(target, completionContext, { signal, maxRetries: 0, maxTokens: 64 });
+			// 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 model failed.");
+			throw new RenameModelError(error instanceof Error ? error.message : "Rename task model failed.");
 		}
-		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.");
+		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 response: Awaited<ReturnType<typeof complete>>;
-	try {
-		response = await complete(model);
-	} catch (error) {
-		const fallback = ctx.model;
-		if (
-			!(error instanceof RenameModelError && isTransportFailure(error.message)) ||
-			signal.aborted ||
-			!fallback?.input.includes("text") ||
-			(fallback.provider === model.provider && fallback.id === model.id)
-		) {
-			throw error;
+	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;
 		}
-		response = await complete(fallback);
-	}
-
-	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 Error("Rename model returned an invalid title.");
 	}
-	return title;
+	throw failure ?? new RenameModelError("Rename task model routes failed.");
 }
 
 export default function herdrRenameExtension(pi: ExtensionAPI): void {
@@ -285,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);
@@ -351,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.4.0",
+  "version": "1.0.2",
   "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"
   }
 }