@stfade/pi-read-delegator 1.0.11 → 1.0.12

package/config.d.ts

@@ -11,6 +11,7 @@ export interface ReadDelegatorConfig {
     blocked_tools: string[];
     allowed_bash_write_commands: string[];
     orchestrator_prompt: string;
+    reader_model: string;
     language: string;
 }
 /**

package/config.js

@@ -45,6 +45,7 @@ exports.saveConfig = saveConfig;
 const fs = __importStar(require("fs"));
 const os = __importStar(require("os"));
 const path = __importStar(require("path"));
+const ui_1 = require("./ui");
 // ---------------------------------------------------------------------------
 // Defaults
 // ---------------------------------------------------------------------------
@@ -62,6 +63,7 @@ const DEFAULT_CONFIG = {
         "cp",
     ],
     orchestrator_prompt: "You are an orchestrator. For any file reading, searching, or listing operation, you MUST use the subagent tool with subagent='reader'. Do not use read/grep/find/ls yourself. If you need to run a shell command that only reads (like cat, grep, find, ls), also delegate it to the reader subagent.",
+    reader_model: "lmstudio/nvidia/nemotron-3-nano-4b",
     language: "auto",
 };
 // ---------------------------------------------------------------------------
@@ -104,7 +106,7 @@ function loadConfig() {
     }
     catch (err) {
         // File is missing, unreadable, or invalid JSON → overwrite with defaults
-        console.warn(`[pi-read-delegator] Corrupted config file at ${filePath}. Overwriting with defaults. Error: ${err}`);
+        (0, ui_1.rawWarn)(`Corrupted config file at ${filePath}. Overwriting with defaults. Error: ${err}`);
         try {
             ensureDir(path.dirname(filePath));
             fs.writeFileSync(filePath, JSON.stringify(DEFAULT_CONFIG, null, 2), "utf-8");
@@ -126,11 +128,11 @@ function saveConfig(config, options) {
     try {
         fs.writeFileSync(filePath, JSON.stringify(config, null, 2), "utf-8");
         if (!options?.silent) {
-            console.log(`[pi-read-delegator] Config saved to ${filePath}`);
+            (0, ui_1.rawLog)(`Config saved to ${filePath}`);
         }
     }
     catch (err) {
-        console.error(`[pi-read-delegator] Failed to save config: ${err}`);
+        (0, ui_1.rawError)(`Failed to save config: ${err}`);
         throw err;
     }
 }
@@ -157,6 +159,9 @@ function mergeDefaults(partial, defaults) {
         orchestrator_prompt: typeof p.orchestrator_prompt === "string"
             ? p.orchestrator_prompt
             : defaults.orchestrator_prompt,
+        reader_model: typeof p.reader_model === "string"
+            ? p.reader_model
+            : defaults.reader_model,
         language: typeof p.language === "string" ? p.language : defaults.language,
     };
 }

package/index.js

@@ -85,32 +85,64 @@ const READ_BASH_COMMANDS = new Set([
 function readerPath() {
     return path.join(os.homedir(), ".pi", "agent", "agents", "reader.md");
 }
-async function ensureReaderTemplate() {
+function syncReaderTemplate(model) {
     const rp = readerPath();
-    if (fs.existsSync(rp))
-        return;
-    const content = [
-        "---",
-        "name: reader",
-        "description: Token-efficient code reader that returns minimal results.",
-        "tools: read, grep, find, ls",
-        "model: lmstudio/nvidia/nemotron-3-nano-4b",
-        "---",
-        "",
-        "You are a token-efficient analyst. Execute read/search/list tasks and return",
-        "ONLY the essential result. Maximum 10 lines. Use bullet summaries.",
-        "Never dump entire files. Focus only on what was asked.",
-    ].join("\n");
     try {
         const dir = path.dirname(rp);
         fs.mkdirSync(dir, { recursive: true });
-        await fs.promises.writeFile(rp, content, "utf8");
+        const content = [
+            "---",
+            "name: reader",
+            "description: Token-efficient code reader that returns minimal results.",
+            "tools: read, grep, find, ls",
+            `model: ${model}`,
+            "---",
+            "",
+            "You are a token-efficient analyst. Execute read/search/list tasks and return",
+            "ONLY the essential result. Maximum 10 lines. Use bullet summaries.",
+            "Never dump entire files. Focus only on what was asked.",
+        ].join("\n");
+        fs.writeFileSync(rp, content, "utf8");
     }
     catch {
-        // read-only home directory — template creation is best-effort
+        // read-only home directory — template sync is best-effort
     }
 }
 // ---------------------------------------------------------------------------
+// Model picker
+// ---------------------------------------------------------------------------
+/**
+ * Interactive model picker using Pi's model registry.
+ * Shows a select UI with all configured models and updates config + reader.md
+ * when the user picks a new model.
+ *
+ * Returns the selected model string ("provider/model") or undefined if the
+ * picker is unavailable or the user cancels.
+ */
+async function pickReaderModel(config, ctx) {
+    try {
+        const models = ctx.modelRegistry?.getAvailable?.() ?? [];
+        if (models.length === 0)
+            return undefined;
+        const options = models.map((m) => `${m.provider ?? "?"}/${m.id}`);
+        if (!options.includes(config.reader_model)) {
+            options.unshift(config.reader_model);
+        }
+        const selected = await ctx.ui.select("Choose reader model (ESC to keep current)", options);
+        if (selected && selected !== config.reader_model) {
+            config.reader_model = selected;
+            (0, config_1.saveConfig)(config, { silent: true });
+            syncReaderTemplate(selected);
+            ctx.ui.notify("✅ Reader model set to: " + selected, "info");
+            return selected;
+        }
+    }
+    catch {
+        // modelRegistry or ui.select not available — fallback
+    }
+    return undefined;
+}
+// ---------------------------------------------------------------------------
 // Tool helpers
 // ---------------------------------------------------------------------------
 /** Determine which tools stay active after blocking read tools.
@@ -182,6 +214,8 @@ async function default_1(pi) {
             depsChecked = true;
             if (!depsReady)
                 return;
+            // First install: pick reader model from available models
+            await pickReaderModel(config, ctx);
         }
         // --- Tool blocking ---
         if (config.enabled) {
@@ -229,9 +263,13 @@ async function default_1(pi) {
     // 4. /read-delegator command
     // -----------------------------------------------------------------------
     pi.registerCommand("read-delegator", {
-        description: "Manage read delegation (on|off|status)",
-        handler: (args, ctx) => {
-            const sub = args?.trim().toLowerCase() ?? "status";
+        description: "Manage read delegation (on|off|status|model)",
+        handler: async (args, ctx) => {
+            const raw = args?.trim() ?? "";
+            const spaceIdx = raw.indexOf(" ");
+            const sub = spaceIdx >= 0
+                ? raw.slice(0, spaceIdx).toLowerCase()
+                : raw.toLowerCase() || "status";
             switch (sub) {
                 case "on":
                 case "enable": {
@@ -255,12 +293,31 @@ async function default_1(pi) {
                     ctx.ui.setStatus("read-delegator", (0, ui_1.msg)("status_off"));
                     return;
                 }
+                case "model": {
+                    const modelArg = spaceIdx >= 0 ? raw.slice(spaceIdx + 1).trim() : "";
+                    if (!modelArg) {
+                        // Interactive picker via available models
+                        const picked = await pickReaderModel(config, ctx);
+                        if (picked === undefined) {
+                            ctx.ui.notify("Current reader model: " +
+                                config.reader_model +
+                                "\n\nChange: /read-delegator model <provider/model>", "info");
+                        }
+                        return;
+                    }
+                    config.reader_model = modelArg;
+                    (0, config_1.saveConfig)(config, { silent: true });
+                    syncReaderTemplate(modelArg);
+                    ctx.ui.notify("✅ Reader model set to: " + modelArg, "info");
+                    return;
+                }
                 case "status":
                 default: {
                     const lines = [
                         "Read delegation: " +
                             (config.enabled ? "🟢 enabled" : "🔴 disabled"),
                         "Reader subagent: " + config.reader_subagent_name,
+                        "Reader model: " + config.reader_model,
                         "Dependencies: " + (depsReady ? "✅ ready" : "❌ missing"),
                         "Blocked tools: " + config.blocked_tools.join(", "),
                     ];
@@ -270,9 +327,7 @@ async function default_1(pi) {
             }
         },
     });
-    // -----------------------------------------------------------------------
-    // 5. Background: ensure reader.md template
-    // -----------------------------------------------------------------------
-    await ensureReaderTemplate();
+    // 5. Background: sync reader.md template from config
+    syncReaderTemplate(config.reader_model);
 }
 //# sourceMappingURL=index.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stfade/pi-read-delegator",
-  "version": "1.0.11",
+  "version": "1.0.12",
   "description": "Pi extension that delegates all read operations to a Reader subagent, blocking read tools from the main model",
   "main": "index.js",
   "types": "index.d.ts",

package/reader-manager.js

@@ -91,9 +91,43 @@ function piSubagentsDir() {
 function piSubagentsPackageJson() {
     return path.join(piSubagentsDir(), "package.json");
 }
+function piSettingsPath() {
+    return path.join(os.homedir(), ".pi", "agent", "settings.json");
+}
+/**
+ * Register pi-subagents in Pi's package list (settings.json).
+ *
+ * npm install puts the package on disk, but Pi's `pi list` command reads
+ * from settings.json's `packages` array. Without this step, the user
+ * won't see pi-subagents in their package list.
+ */
+function registerPiSubagentsPackage() {
+    try {
+        const settingsPath = piSettingsPath();
+        let settings = {};
+        if (fs.existsSync(settingsPath)) {
+            const raw = fs.readFileSync(settingsPath, "utf-8");
+            settings = JSON.parse(raw);
+        }
+        const packages = Array.isArray(settings.packages)
+            ? [...settings.packages]
+            : [];
+        const pkgName = "npm:pi-subagents";
+        if (!packages.includes(pkgName)) {
+            packages.push(pkgName);
+            settings.packages = packages;
+            fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2) + "\n", "utf-8");
+            (0, ui_1.rawLog)("✅ Registered pi-subagents in settings.json package list.");
+        }
+    }
+    catch (err) {
+        (0, ui_1.rawWarn)("⚠️ Failed to register pi-subagents in settings.json: " + String(err));
+    }
+}
 async function checkDependencies(prompt, onProgress) {
     // Already installed — nothing to do
     if (fs.existsSync(piSubagentsPackageJson())) {
+        registerPiSubagentsPackage();
         return true;
     }
     (0, ui_1.rawWarn)("⚠️ pi-subagents is not installed.");
@@ -114,6 +148,7 @@ async function checkDependencies(prompt, onProgress) {
             encoding: "utf-8",
         });
         (0, ui_1.rawLog)("✅ pi-subagents installed via npm.");
+        registerPiSubagentsPackage();
         onProgress?.("done");
     }
     catch (firstErr) {

package/tool-blocker.js

@@ -16,6 +16,7 @@ exports.tempAllowOnce = tempAllowOnce;
 exports.isBlocked = isBlocked;
 exports.getBlockedTools = getBlockedTools;
 exports.reset = reset;
+const ui_1 = require("./ui");
 // ---------------------------------------------------------------------------
 // Tool blocker state
 // ---------------------------------------------------------------------------
@@ -40,10 +41,10 @@ function blockTools(agent, tools) {
             removedTools.set(toolName, definition);
             try {
                 agent.removeTool(toolName);
-                console.log(`[pi-read-delegator] Blocked tool: ${toolName}`);
+                (0, ui_1.rawLog)(`Blocked tool: ${toolName}`);
             }
             catch (err) {
-                console.error(`[pi-read-delegator] Failed to block tool "${toolName}": ${err}`);
+                (0, ui_1.rawError)(`Failed to block tool "${toolName}": ${err}`);
             }
         }
     }
@@ -56,10 +57,10 @@ function restoreTools(agent) {
     for (const [toolName, definition] of removedTools.entries()) {
         try {
             agent.addTool(definition);
-            console.log(`[pi-read-delegator] Restored tool: ${toolName}`);
+            (0, ui_1.rawLog)(`Restored tool: ${toolName}`);
         }
         catch (err) {
-            console.error(`[pi-read-delegator] Failed to restore tool "${toolName}": ${err}`);
+            (0, ui_1.rawError)(`Failed to restore tool "${toolName}": ${err}`);
         }
     }
     removedTools.clear();
@@ -92,7 +93,7 @@ async function tempAllowOnce(agent, blockedTools, callback) {
                 restoredNow.push(toolName);
             }
             catch (err) {
-                console.error(`[pi-read-delegator] Failed to temporarily restore "${toolName}": ${err}`);
+                (0, ui_1.rawError)(`Failed to temporarily restore "${toolName}": ${err}`);
             }
         }
     }
@@ -113,7 +114,7 @@ async function tempAllowOnce(agent, blockedTools, callback) {
                     agent.removeTool(toolName);
                 }
                 catch (err) {
-                    console.error(`[pi-read-delegator] Failed to re-block "${toolName}": ${err}`);
+                    (0, ui_1.rawError)(`Failed to re-block "${toolName}": ${err}`);
                 }
             }
         }