@pi-kaush/pi-agent-mode 0.2.0 → 0.2.1

package/CHANGELOG.md

@@ -2,6 +2,10 @@
 
 ## Unreleased
 
+- Refresh Pi's local model catalog before `/agent` model selection and baseline
+  restoration, so a long-lived session sees newly configured profile candidates
+  without using the parent's model-cycling scope. Report refresh failures without
+  changing the active agent.
 - Add `profile` frontmatter: activate a named model profile from the shared machine-local `~/.pi/agent/profiles.yaml` (`@pi-kaush/pi-model-profiles`), walking candidates in order and applying the first available model and thinking level.
 - Reject agents declaring both `profile` and `model`; precedence is explicit: profile first, then model, then the session default.
 - Load `@pi-kaush/pi-model-profiles` lazily so a missing library degrades only profile support; `/agent` remains fully functional without it.

package/README.md

@@ -48,10 +48,12 @@ Agent frontmatter may declare a `model` (exact model, optional thinking
 suffix) or a `profile` — a named entry in the shared machine-local
 `~/.pi/agent/profiles.yaml` maintained by
 [@pi-kaush/pi-model-profiles](../pi-model-profiles). An agent must not declare
-both. Profile candidates are walked in order at activation and the first
-model that exists and is authenticated wins; once the session is running,
-models never switch mid-request. Agents without either keep the current
-session model.
+both. Before selecting a model, `/agent` refreshes Pi's local model catalog so
+models added during a long-lived session can be used; the parent session's
+`enabledModels` / `--models` cycling scope does not limit profile candidates.
+Profile candidates are walked in order at activation and the first model that
+exists and is authenticated wins; once the session is running, models never
+switch mid-request. Agents without either keep the current session model.
 
 `pi-agent-mode` loads `@pi-kaush/pi-model-profiles` lazily: if the library is
 missing, only agents that declare `profile:` fail (with an install hint) and

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-kaush/pi-agent-mode",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Activate a configured Pi agent as a persistent mode in the current session.",
   "license": "MIT",
   "author": "Kaushik Gopal",

package/src/index.ts

@@ -148,6 +148,26 @@ export function registerAgentMode(pi: ExtensionAPI) {
     pi.appendEntry(PI_AGENT_MODE_STATE_TYPE, state);
   }
 
+  async function refreshModelCatalog(ctx: ExtensionContext): Promise<boolean> {
+    try {
+      // Activation may happen after models.json changes in a long-lived session.
+      // Pi 0.80 accepts no options; newer Pi can suppress remote catalog fetches.
+      // The cycling scope is independent of the models an agent may select.
+      await (
+        ctx.modelRegistry.refresh as (options: {
+          allowNetwork: boolean;
+        }) => void | Promise<unknown>
+      )({ allowNetwork: false });
+      return true;
+    } catch (error) {
+      ctx.ui.notify(
+        `Could not refresh model catalog: ${error instanceof Error ? error.message : error}`,
+        "error",
+      );
+      return false;
+    }
+  }
+
   function findModel(modelSpec: string, ctx: ExtensionContext) {
     const spec = modelSpec.trim();
     const separator = spec.indexOf("/");
@@ -175,6 +195,7 @@ export function registerAgentMode(pi: ExtensionAPI) {
     if (ctx.model?.provider === model.provider && ctx.model.id === model.id)
       return true;
 
+    if (!(await refreshModelCatalog(ctx))) return false;
     const resolved = ctx.modelRegistry.find(model.provider, model.id);
     if (!resolved) {
       ctx.ui.notify(
@@ -290,6 +311,8 @@ export function registerAgentMode(pi: ExtensionAPI) {
       );
       return false;
     }
+    if ((agent.profile || agent.model) && !(await refreshModelCatalog(ctx)))
+      return false;
     if (agent.profile) {
       const applied = await applyProfileModels(agent, ctx);
       if (!applied) return false;