pi-hide-providers 0.1.3 → 0.1.5

package/README.md

@@ -30,7 +30,8 @@ Pi has `enabledModels` in `settings.json` as an allowlist, but maintaining it ma
 - The `/model` selector, `Ctrl+P` cycling, `--list-models`, and session restoration all see only visible models
 - `/hide-models reset` unpatches the registry — all models return immediately
 - Changes via `/hide-models add` and `/hide-models remove` take effect immediately (no reload needed)
-- Interactive `/hide-models` command — no-args opens the TUI selector; subcommands for adding, removing, and inspecting rules
+- Interactive `/hide-models` TUI lists visible models first in user-toggle order, then hidden models in their original registry order, and keeps the cursor at the same index after each toggle — matching pi core's `/scoped-models` UX
+- `/hide-models` subcommands for adding, removing, and inspecting rules
 
 No `settings.json` is modified. No 250+ entry explosion. No allowlist semantics.
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pi-hide-providers",
-  "version": "0.1.3",
-  "description": "Hide providers and models from pi's model selector \u2014 filter the /model list and Ctrl+P cycling via a configurable blocklist",
+  "version": "0.1.5",
+  "description": "Hide providers and models from pi's model selector — filter the /model list and Ctrl+P cycling via a configurable blocklist",
   "type": "module",
   "author": "Tom X Nguyen",
   "license": "MIT",

package/src/provider-selector.ts

@@ -61,6 +61,10 @@ export class HideProviderSelectorComponent implements Component {
   // model data
   private allItems: DisplayItem[] = [];
 
+  // visible model fullIds in the order they were unhidden (mirrors pi core's
+  // enabledIds cycle order in the scoped models selector)
+  private visibleIds: string[] = [];
+
   // current hide rules
   private hiddenRules: HideRule[] = [];
 
@@ -93,7 +97,7 @@ export class HideProviderSelectorComponent implements Component {
     this.done = done;
     this.hiddenRules = deduplicateRules(currentRules);
 
-    // Build display items
+    // Build display items in the original registry order.
     for (const m of allModels) {
       this.allItems.push({
         fullId: `${m.provider}/${m.id}`,
@@ -103,7 +107,12 @@ export class HideProviderSelectorComponent implements Component {
         hidden: isHidden(currentRules, m.provider, m.id),
       });
     }
-    this.filteredItems = [...this.allItems];
+    // Track visible models in original order initially; toggle order changes
+    // during the session, matching pi core's /scoped-models behavior.
+    this.visibleIds = this.allItems
+      .filter((i) => !i.hidden)
+      .map((i) => i.fullId);
+    this.filteredItems = this.buildItems();
 
     this.searchInput = new Input();
     this.listContainer = new Container();
@@ -268,26 +277,59 @@ export class HideProviderSelectorComponent implements Component {
 
   private refresh(): void {
     const query = this.searchInput.getValue();
+
+    // Recompute hidden status, then rebuild the list in pi core's order:
+    // visible models in user-toggle order, followed by hidden models in
+    // original registry order.
+    for (const item of this.allItems) {
+      item.hidden = isHidden(this.hiddenRules, item.provider, item.modelId);
+    }
+
+    const items = this.buildItems();
     this.filteredItems = query
       ? fuzzyFilter(
-          this.allItems,
+          items,
           query,
           (i) => `${i.provider} ${i.modelId} ${i.provider}/${i.modelId} ${i.modelName}`,
         )
-      : [...this.allItems];
-
-    // Update hidden status on all items (rules may have changed)
-    for (const item of this.filteredItems) {
-      item.hidden = isHidden(this.hiddenRules, item.provider, item.modelId);
-    }
+      : items;
 
+    // Keep the cursor at the same numeric index after re-sorting. The selected
+    // model may change, matching pi core's /scoped-models behavior.
     this.selectedIndex = Math.min(
       this.selectedIndex,
       Math.max(0, this.filteredItems.length - 1),
     );
+
     this.updateList();
   }
 
+  /** Build the display order used by pi core's scoped models selector:
+   *  visible models first in the order they were unhidden, then hidden models
+   *  in their original registry order.
+   */
+  private buildItems(): DisplayItem[] {
+    const visibleSet = new Set(this.visibleIds);
+    const result: DisplayItem[] = [];
+
+    // Visible models in user-toggle order.
+    for (const fullId of this.visibleIds) {
+      const item = this.allItems.find((i) => i.fullId === fullId);
+      if (item && !item.hidden) {
+        result.push(item);
+      }
+    }
+
+    // Hidden models in original registry order.
+    for (const item of this.allItems) {
+      if (item.hidden && !visibleSet.has(item.fullId)) {
+        result.push(item);
+      }
+    }
+
+    return result;
+  }
+
   private updateList(): void {
     this.listContainer.clear();
 
@@ -398,12 +440,20 @@ export class HideProviderSelectorComponent implements Component {
             !(r.provider === item.provider && r.model === item.modelId),
         );
       }
+
+      // Newly visible model goes to the end of the visible order (pi core).
+      if (!this.visibleIds.includes(item.fullId)) {
+        this.visibleIds.push(item.fullId);
+      }
     } else {
       // Add a rule for this specific model
       this.hiddenRules = deduplicateRules([
         ...this.hiddenRules,
         { provider: item.provider, model: item.modelId },
       ]);
+
+      // Remove from the visible order.
+      this.visibleIds = this.visibleIds.filter((id) => id !== item.fullId);
     }
     this.hasChanges = true;
     this.refresh();
@@ -421,12 +471,25 @@ export class HideProviderSelectorComponent implements Component {
       this.hiddenRules = this.hiddenRules.filter(
         (r) => r.provider !== provider,
       );
+
+      // Append newly visible provider models to the end of the visible order
+      // in original registry order (pi core).
+      const providerIds = providerItems.map((i) => i.fullId);
+      this.visibleIds = [
+        ...this.visibleIds,
+        ...providerIds.filter((id) => !this.visibleIds.includes(id)),
+      ];
     } else {
       // Hide all — add a single provider-level rule
       this.hiddenRules = deduplicateRules([
         ...this.hiddenRules,
         { provider },
       ]);
+
+      // Remove provider models from the visible order.
+      this.visibleIds = this.visibleIds.filter(
+        (id) => !providerItems.some((i) => i.fullId === id),
+      );
     }
     this.hasChanges = true;
     this.refresh();
@@ -461,6 +524,10 @@ export class HideProviderSelectorComponent implements Component {
         }
       }
     }
+
+    // Remove hidden models from the visible order.
+    const hiddenFullIds = new Set(items.map((i) => i.fullId));
+    this.visibleIds = this.visibleIds.filter((id) => !hiddenFullIds.has(id));
   }
 
   /** Remove hide rules for the given items. */
@@ -513,6 +580,13 @@ export class HideProviderSelectorComponent implements Component {
         }
       }
     }
+
+    // Append newly visible models to the end of the visible order (pi core).
+    for (const item of items) {
+      if (!this.visibleIds.includes(item.fullId)) {
+        this.visibleIds.push(item.fullId);
+      }
+    }
   }
 
   /** Close and pass results to `done`. */