@robhowley/pi-openrouter 0.11.1 → 0.12.0

package/extensions/openrouter/hooks.ts

@@ -9,7 +9,12 @@ import { createSessionState, type SessionState } from './session-state.js';
 import { writeLocalUsage, type LocalUsageEvent } from './local-usage.js';
 import { loadCache, getCacheAgeMs, formatDuration } from './models/cache.js';
 import { mapOpenRouterModels } from './models/mapper.js';
-import { includeBuiltinRouterModels, isSyncEnabled } from './models/sync.js';
+import {
+  filterModelsForCatalogMode,
+  includeBuiltinRouterModels,
+  isSyncEnabled,
+  setActiveCatalogState,
+} from './models/sync.js';
 import { loadOpenRouterStatusBar } from './status-bar.js';
 
 let sessionState: SessionState | null = null;
@@ -90,8 +95,13 @@ export async function loadStartupCacheState(
   }
 
   try {
-    const { configs } = await mapOpenRouterModels(cache.models);
-    const configsWithRouters = includeBuiltinRouterModels(configs);
+    const filteredCacheModels = filterModelsForCatalogMode(cache.models, cache.catalogMode);
+    const { configs, skipped, skippedDetails } = await mapOpenRouterModels(filteredCacheModels);
+    const configsWithRouters = includeBuiltinRouterModels(configs, cache.catalogMode);
+    const effectiveSkippedDetails = cache.skippedDetails ?? skippedDetails;
+    const skippedCount =
+      effectiveSkippedDetails.length > 0 ? effectiveSkippedDetails.length : skipped;
+    const cacheAgeMs = getCacheAgeMs(cache);
 
     pi.registerProvider('openrouter', {
       baseUrl: 'https://openrouter.ai/api/v1',
@@ -101,9 +111,19 @@ export async function loadStartupCacheState(
       authHeader: true,
     });
 
+    setActiveCatalogState({
+      mode: cache.catalogMode,
+      registeredModelIds: configsWithRouters.map((config) => config.id),
+      registeredCount: configsWithRouters.length,
+      skippedCount,
+      skippedDetails: effectiveSkippedDetails,
+      source: 'cache',
+      cacheAgeMs,
+    });
+
     startupState.info = {
       count: configsWithRouters.length,
-      age: formatDuration(getCacheAgeMs(cache)),
+      age: formatDuration(cacheAgeMs),
     };
   } catch (error) {
     startupState.warning = `OpenRouter: cached models found but failed to register: ${error instanceof Error ? error.message : String(error)}`;

package/extensions/openrouter/models/__tests__/cache.test.ts

@@ -45,11 +45,12 @@ describe('loadCache', () => {
   });
 
   it('should return parsed cache when file exists and is valid', async () => {
-    const mockCache = createMockCache({ timestamp: 1234567890 });
+    const mockCache = createMockCache({ catalogMode: 'free-only', timestamp: 1234567890 });
     await saveCache(mockCache);
 
     const result = await loadCache();
     expect(result).not.toBeNull();
+    expect(result!.catalogMode).toBe('free-only');
     expect(result!.timestamp).toBe(1234567890);
     expect(result!.models).toHaveLength(1);
     expect(result!.models[0]!.id).toBe('test/model');
@@ -73,6 +74,23 @@ describe('loadCache', () => {
     expect(result).toBeNull();
   });
 
+  it('should normalize old cache files without catalogMode to full', async () => {
+    await mkdir(testCacheDir, { recursive: true });
+    const cacheFile = join(testCacheDir, 'models-cache.json');
+    const oldCache = createMockCache({ timestamp: 1234567890 });
+    const legacyCache = {
+      models: oldCache.models,
+      skippedDetails: oldCache.skippedDetails,
+      timestamp: oldCache.timestamp,
+    };
+
+    await writeFile(cacheFile, JSON.stringify(legacyCache));
+
+    const result = await loadCache();
+    expect(result).not.toBeNull();
+    expect(result!.catalogMode).toBe('full');
+  });
+
   it('should return null when cache timestamp is too far in the future', async () => {
     await mkdir(testCacheDir, { recursive: true });
     const cacheFile = join(testCacheDir, 'models-cache.json');
@@ -98,6 +116,7 @@ describe('saveCache', () => {
     // Verify it can be loaded back
     const loaded = await loadCache();
     expect(loaded).not.toBeNull();
+    expect(loaded!.catalogMode).toBe('full');
     expect(loaded!.timestamp).toBe(mockCache.timestamp);
     expect(loaded!.models).toEqual(mockCache.models);
   });