pi-model-sort 0.1.2 → 0.1.4

package/model-sort.ts

@@ -149,10 +149,29 @@ function patchFilterModels(getLastUsed: () => Record<string, number>): void {
   origFilterModels = proto.filterModels as (query: string) => void;
 
   proto.filterModels = function (this: Record<string, unknown>, query: string) {
-    origFilterModels!.call(this, query);
+    // Suppress the original's updateList() call — we'll call it once after
+    // re-sorting to avoid a double-render.
+    const origUpdateList = this.updateList as () => void;
+    this.updateList = (() => {}) as unknown as () => void;
+
+    try {
+      origFilterModels!.call(this, query);
+    } finally {
+      this.updateList = origUpdateList as unknown as () => void;
+    }
+
+    // Empty query: nothing to re-sort (activeModels is already sorted by our
+    // sortModels/loadModels patches), just render the original result.
+    if (!query) {
+      origUpdateList.call(this);
+      return;
+    }
 
     const filtered = this.filteredModels as Array<{ provider: string; id: string; model: unknown }> | undefined;
-    if (!filtered || filtered.length <= 1 || !query) return;
+    if (!filtered || filtered.length <= 1) {
+      origUpdateList.call(this);
+      return;
+    }
 
     const lastUsed = getLastUsed();
     this.filteredModels = sortByLastUsed(filtered, lastUsed, buildCurrentModelKey(this));
@@ -168,6 +187,9 @@ function patchFilterModels(getLastUsed: () => Record<string, number>): void {
         this.selectedIndex = newIndex;
       }
     }
+
+    // Render once with the final sorted list.
+    origUpdateList.call(this);
   };
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-model-sort",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "Sort models in pi's /model selector by last usage — most recently used models appear first",
   "type": "module",
   "author": "Tom X Nguyen",