@diegopetrucci/pi-extensions 0.1.21 → 0.1.22

package/README.md

@@ -5,7 +5,7 @@ A collection of [pi](https://github.com/earendil-works/pi-mono) agent extensions
 - [`confirm-destructive`](./extensions/confirm-destructive): Confirms before destructive session actions like clear, switch, and fork.
 - [`context-cap`](./extensions/context-cap): Caps effective model context windows at 200k tokens by default so pi avoids the `dumb zone`; toggle temporarily with `/context-cap`.
 - [`context-inspector`](./extensions/context-inspector): Adds `/context`, a local self-contained HTML dashboard that breaks down where the current session context is going, with category overview, top offenders, and drilldown search.
-- [`librarian`](./extensions/librarian): Adds a GitHub research scout that asks whether to use an opt-in local repo checkout cache under the OS user cache directory, with cached repos expiring after 30 days of non-use.
+- [`librarian`](./extensions/librarian): Adds a GitHub research scout with a local repo checkout cache enabled by default under the OS user cache directory, toggleable with `/librarian-cache`, with cached repos expiring after 30 days of non-use.
 - [`minimal-footer`](./extensions/minimal-footer): Replaces pi's built-in footer with a minimal configurable two-line layout: branch/repo on the first line, context/model on the second, optional `DUMB ZONE`, plus OpenAI Codex 5-hour and 7-day usage when available.
 - [`notify`](./extensions/notify): Sends configurable terminal, desktop, bell, and sound notifications when pi finishes and is ready for input.
 - [`openai-fast`](./extensions/openai-fast): Adds `/fast` to enable OpenAI Codex Fast mode for ChatGPT-auth GPT-5.4 and GPT-5.5 by injecting the priority service tier.
@@ -26,7 +26,7 @@ pi install npm:@diegopetrucci/pi-extensions
 Or pin the GitHub package to this release:
 
 ```bash
-pi install git:github.com/diegopetrucci/pi-extensions@v0.1.21
+pi install git:github.com/diegopetrucci/pi-extensions@v0.1.22
 ```
 
 Or a specific extension:

package/extensions/librarian/README.md

@@ -1,8 +1,8 @@
 # librarian
 
-A pi GitHub research scout inspired by `pi-librarian`, with an opt-in local checkout cache.
+A pi GitHub research scout inspired by `pi-librarian`, with a local checkout cache enabled by default.
 
-When the `librarian` tool runs, it asks whether to cache/reuse repository checkouts locally. If you say no, cancel, time out, or run without UI, it uses GitHub API/search and temporary fetched files only.
+When the `librarian` tool runs, it can cache/reuse repository checkouts locally. Use `/librarian-cache off` to force GitHub API/search and temporary fetched files only, or `/librarian-cache on` to re-enable cached local checkouts.
 
 ## Install
 
@@ -35,10 +35,21 @@ Then reload pi:
 - Tool name: `librarian`
 - Uses a restricted subagent with `bash` and `read`
 - Uses `gh` for GitHub search/API access
-- Asks on each call whether to use cached local checkouts
-- Defaults to no checkout/cache
+- Uses cached local checkouts by default
+- Toggle cache behavior for future calls with `/librarian-cache on | off | toggle | status`
 - Cached repos are removed lazily after 30 days without use
 
+## Commands
+
+```text
+/librarian-cache status
+/librarian-cache off
+/librarian-cache on
+/librarian-cache toggle
+```
+
+The command works in interactive mode, RPC mode, and print/JSON mode. It writes a global preference to `~/.pi/agent/extensions/librarian.json`, so separate non-UI invocations use the same setting. In non-UI modes, command feedback is written to stderr so stdout remains usable for normal output or JSON events.
+
 ## Cache location
 
 macOS:

package/extensions/librarian/index.ts

@@ -25,11 +25,14 @@ const CACHE_TTL_DAYS = 30;
 const CACHE_TTL_MS = CACHE_TTL_DAYS * 24 * 60 * 60 * 1000;
 const CACHE_METADATA_FILE = ".pi-librarian-cache.json";
 const CACHE_MARKER_FILE = ".pi-librarian-cache-used";
+const CACHE_CONFIG_FILE = "librarian.json";
 
 type LibrarianStatus = "running" | "done" | "error" | "aborted";
 
 type CacheMode = "disabled" | "enabled";
 
+const DEFAULT_CACHE_MODE: CacheMode = "enabled";
+
 type ToolCall = {
 	id: string;
 	name: string;
@@ -252,30 +255,67 @@ async function cleanupExpiredCache(cacheRoot: string): Promise<{ deleted: number
 	}
 }
 
-async function askForCache(ctx: ExtensionContext, cacheRoot: string): Promise<{ enabled: boolean; reason: string }> {
-	if (!ctx.hasUI) return { enabled: false, reason: "no UI available; using GitHub API/temp files only" };
+function getCacheConfigPath(): string {
+	return path.join(getAgentDir(), "extensions", CACHE_CONFIG_FILE);
+}
+
+function parseCacheMode(value: unknown): CacheMode | undefined {
+	if (value === "enabled" || value === "on" || value === true) return "enabled";
+	if (value === "disabled" || value === "off" || value === false) return "disabled";
+	return undefined;
+}
 
+async function readCachePreference(): Promise<CacheMode> {
 	try {
-		const enabled = await ctx.ui.confirm(
-			"Librarian repo cache",
-			[
-				"Cache/reuse local GitHub repo checkouts for this Librarian call?",
-				"",
-				`Cache directory: ${cacheRoot}`,
-				`Repos unused for ${CACHE_TTL_DAYS} days are removed lazily on future Librarian calls.`,
-				"",
-				"Choose No to use GitHub API and temporary fetched files only.",
-			].join("\n"),
-			{ timeout: 30_000 },
+		const raw = await fs.readFile(getCacheConfigPath(), "utf8");
+		const parsed = JSON.parse(raw) as {
+			cacheMode?: unknown;
+			cacheEnabled?: unknown;
+			cache?: { mode?: unknown; enabled?: unknown };
+		};
+		return (
+			parseCacheMode(parsed.cacheMode) ??
+			parseCacheMode(parsed.cache?.mode) ??
+			parseCacheMode(parsed.cacheEnabled) ??
+			parseCacheMode(parsed.cache?.enabled) ??
+			DEFAULT_CACHE_MODE
 		);
+	} catch {
+		return DEFAULT_CACHE_MODE;
+	}
+}
 
-		return enabled
-			? { enabled: true, reason: "user opted into cached local checkouts" }
-			: { enabled: false, reason: "user declined or prompt timed out" };
-	} catch (error) {
-		const message = error instanceof Error ? error.message : String(error);
-		return { enabled: false, reason: `cache prompt failed (${message}); using GitHub API/temp files only` };
+async function writeCachePreference(preference: CacheMode): Promise<void> {
+	const configPath = getCacheConfigPath();
+	const config = {
+		cacheMode: preference,
+		cacheEnabled: preference === "enabled",
+		updatedAt: new Date().toISOString(),
+	};
+
+	await fs.mkdir(path.dirname(configPath), { recursive: true });
+	await fs.writeFile(configPath, `${JSON.stringify(config, null, 2)}\n`, "utf8");
+}
+
+function resolveCacheDecision(preference: CacheMode): { enabled: boolean; reason: string } {
+	if (preference === "enabled") {
+		return { enabled: true, reason: "cache preference enabled; using cached local checkouts" };
+	}
+
+	return { enabled: false, reason: "cache preference disabled; using GitHub API/temp files only" };
+}
+
+function formatCachePreference(preference: CacheMode): string {
+	return preference === "enabled" ? "on" : "off";
+}
+
+function notifyCommand(ctx: ExtensionContext, message: string, type: "info" | "warning" | "error" = "info"): void {
+	if (ctx.hasUI) {
+		ctx.ui.notify(message, type);
+		return;
 	}
+
+	console.error(message);
 }
 
 function resolveToolPath(cwd: string, rawPath: string): string {
@@ -438,13 +478,81 @@ function isAbortLikeError(error: unknown): boolean {
 }
 
 export default function librarianExtension(pi: ExtensionAPI) {
+	let cachePreference: CacheMode = DEFAULT_CACHE_MODE;
+
+	pi.on("session_start", async () => {
+		cachePreference = await readCachePreference();
+	});
+
+	pi.registerCommand("librarian-cache", {
+		description: "Toggle Librarian local checkout cache for future librarian calls",
+		getArgumentCompletions: (prefix) => {
+			const commands = ["on", "off", "toggle", "status"];
+			const query = prefix.trim().toLowerCase();
+			const matches = commands.filter((command) => command.startsWith(query));
+			return matches.length > 0 ? matches.map((value) => ({ value, label: value })) : null;
+		},
+		handler: async (args, ctx) => {
+			const action = args.trim().toLowerCase() || "status";
+			const cacheRoot = getUserCacheReposRoot();
+			const configPath = getCacheConfigPath();
+
+			const setPreference = async (mode: CacheMode): Promise<string | undefined> => {
+				cachePreference = mode;
+				try {
+					await writeCachePreference(mode);
+					return undefined;
+				} catch (error) {
+					const message = error instanceof Error ? error.message : String(error);
+					return `Preference changed for this process, but could not save ${configPath}: ${message}`;
+				}
+			};
+
+			const formatSetMessage = (mode: CacheMode, warning?: string): string => {
+				const main = mode === "enabled"
+					? `Librarian cache enabled. Future librarian calls will reuse local checkouts under ${cacheRoot}.`
+					: "Librarian cache disabled. Future librarian calls will use GitHub API/search and temporary fetched files only.";
+				return warning ? `${main} ${warning}` : main;
+			};
+
+			if (action === "on" || action === "enable") {
+				const warning = await setPreference("enabled");
+				notifyCommand(ctx, formatSetMessage("enabled", warning), warning ? "warning" : "info");
+				return;
+			}
+
+			if (action === "off" || action === "disable") {
+				const warning = await setPreference("disabled");
+				notifyCommand(ctx, formatSetMessage("disabled", warning), warning ? "warning" : "info");
+				return;
+			}
+
+			if (action === "toggle") {
+				const next = cachePreference === "enabled" ? "disabled" : "enabled";
+				const warning = await setPreference(next);
+				notifyCommand(ctx, formatSetMessage(next, warning), warning ? "warning" : "info");
+				return;
+			}
+
+			if (action === "status") {
+				notifyCommand(
+					ctx,
+					`Librarian cache is ${formatCachePreference(cachePreference)}. Cache directory: ${cacheRoot}. Config: ${configPath}. Repos unused for ${CACHE_TTL_DAYS} days are removed lazily.`,
+				);
+				return;
+			}
+
+			notifyCommand(ctx, "Usage: /librarian-cache on | off | toggle | status", "warning");
+		},
+	});
+
 	pi.registerTool({
 		name: "librarian",
 		label: "Librarian",
 		description:
-			"GitHub research scout for coding and personal-assistant tasks. Use when the answer likely lives in GitHub repos, exact repo/path locations are unknown, or you'd otherwise do exploratory gh search/tree probes plus local rg/read inspection. Librarian asks whether to use an optional 30-day local checkout cache, otherwise it behaves like API-only pi-librarian.",
+			"GitHub research scout for coding and personal-assistant tasks. Use when the answer likely lives in GitHub repos, exact repo/path locations are unknown, or you'd otherwise do exploratory gh search/tree probes plus local rg/read inspection. Librarian uses an optional 30-day local checkout cache by default; toggle it with /librarian-cache.",
 		promptSnippet:
-			"Research GitHub repositories with evidence-first path and line citations; optionally ask the user to cache local repo checkouts.",
+			"Research GitHub repositories with evidence-first path and line citations; local checkout cache is enabled by default and user-toggleable with /librarian-cache.",
 		promptGuidelines: [
 			"Use librarian when the answer likely requires exploratory GitHub repository search or line-cited evidence from external repos.",
 			"Do not use librarian for files already present in the current workspace unless the user asks for external GitHub research.",
@@ -472,7 +580,7 @@ export default function librarianExtension(pi: ExtensionAPI) {
 			await fs.mkdir(path.join(workspace, "repos"), { recursive: true });
 
 			const cacheRoot = getUserCacheReposRoot();
-			const cacheDecision = await askForCache(ctx, cacheRoot);
+			const cacheDecision = resolveCacheDecision(cachePreference);
 			if (cacheDecision.enabled) await fs.mkdir(cacheRoot, { recursive: true });
 			const cleanup = cacheDecision.enabled ? await cleanupExpiredCache(cacheRoot) : { deleted: 0, errors: [] };
 

package/extensions/librarian/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@diegopetrucci/pi-librarian",
   "version": "0.1.0",
-  "description": "A pi GitHub research scout that can optionally cache local repo checkouts under the user's OS cache directory.",
+  "description": "A pi GitHub research scout with a toggleable local repo checkout cache under the user's OS cache directory.",
   "keywords": ["pi-package", "pi", "github", "research", "subagent", "cache"],
   "license": "MIT",
   "repository": {

package/extensions/openai-fast/README.md

@@ -25,6 +25,7 @@ Fast mode is only injected when all of these are true:
 ## Commands
 
 ```text
+/fast
 /fast status
 /fast on
 /fast off
@@ -32,7 +33,7 @@ Fast mode is only injected when all of these are true:
 /fast toggle
 ```
 
-`/fast on` and `/fast off` are temporary session/runtime overrides. Use `/fast auto` to reload and follow config defaults again.
+Run `/fast` without arguments to pick an action from a menu. `/fast on` and `/fast off` are temporary session/runtime overrides. Use `/fast auto` to reload and follow config defaults again.
 
 The extension defaults to off so installing the full collection does not accidentally spend Fast-mode credits.
 

package/extensions/openai-fast/index.ts

@@ -249,7 +249,17 @@ export default function openAIFastExtension(pi: ExtensionAPI) {
 		},
 		handler: async (args, ctx) => {
 			const state = getState(ctx);
-			const action = args.trim().toLowerCase() || "status";
+			let action = args.trim().toLowerCase();
+
+			if (!action) {
+				if (!ctx.hasUI) {
+					action = "status";
+				} else {
+					const selection = await ctx.ui.select("OpenAI Fast mode", COMMANDS);
+					if (!selection) return;
+					action = selection;
+				}
+			}
 
 			if (action === "on" || action === "enable") {
 				state.override = "on";

package/extensions/openai-fast/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-openai-fast",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "A pi extension that enables OpenAI Codex Fast mode for ChatGPT-auth GPT-5.4 and GPT-5.5 by injecting the priority service tier.",
   "keywords": ["pi-package", "pi", "openai", "codex", "fast"],
   "license": "MIT",

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@diegopetrucci/pi-extensions",
-  "version": "0.1.21",
-  "description": "A collection of pi extensions, including a GitHub librarian with opt-in local repo checkout caching, a minimal custom footer, an Amp-style oracle, a 200k context cap for auto-compaction, a local HTML context inspector, OpenAI Codex Fast mode controls, quiet one-line collapsed invocation previews, a permission gate for dangerous bash commands, confirm-before-destructive session actions, and terminal notifications when pi is ready for input.",
+  "version": "0.1.22",
+  "description": "A collection of pi extensions, including a GitHub librarian with toggleable local repo checkout caching, a minimal custom footer, an Amp-style oracle, a 200k context cap for auto-compaction, a local HTML context inspector, OpenAI Codex Fast mode controls, quiet one-line collapsed invocation previews, a permission gate for dangerous bash commands, confirm-before-destructive session actions, and terminal notifications when pi is ready for input.",
   "keywords": ["pi-package", "pi", "terminal", "agent"],
   "license": "MIT",
   "repository": {