glancey 2.0.0

package/dist/__tests__/dashboard/state.test.js

@@ -0,0 +1,124 @@
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { DashboardStateManager } from '../../dashboard/state.js';
+describe('DashboardStateManager', () => {
+    let stateManager;
+    beforeEach(() => {
+        stateManager = new DashboardStateManager();
+    });
+    describe('setIndexer', () => {
+        it('should set and use the indexer', async () => {
+            const mockIndexer = {
+                getStatus: vi.fn().mockResolvedValue({
+                    indexed: true,
+                    fileCount: 10,
+                    chunkCount: 50,
+                    lastUpdated: '2024-01-01T00:00:00.000Z',
+                    indexPath: '/test/.glancey',
+                }),
+            };
+            stateManager.setIndexer(mockIndexer);
+            const status = await stateManager.getStatus();
+            expect(status).toEqual({
+                indexed: true,
+                fileCount: 10,
+                chunkCount: 50,
+                lastUpdated: '2024-01-01T00:00:00.000Z',
+                indexPath: '/test/.glancey',
+            });
+            expect(mockIndexer.getStatus).toHaveBeenCalled();
+        });
+        it('should return null if no indexer is set', async () => {
+            const status = await stateManager.getStatus();
+            expect(status).toBeNull();
+        });
+    });
+    describe('setConfig', () => {
+        it('should set and retrieve config', () => {
+            const config = {
+                patterns: ['**/*.ts'],
+                excludePatterns: ['**/node_modules/**'],
+            };
+            stateManager.setConfig(config);
+            expect(stateManager.getConfig()).toEqual(config);
+        });
+    });
+    describe('setProjectPath', () => {
+        it('should set and retrieve project path', () => {
+            stateManager.setProjectPath('/test/project');
+            expect(stateManager.getProjectPath()).toBe('/test/project');
+        });
+    });
+    describe('indexing state', () => {
+        it('should track indexing state', () => {
+            expect(stateManager.isIndexingInProgress()).toBe(false);
+            stateManager.onIndexingStart();
+            expect(stateManager.isIndexingInProgress()).toBe(true);
+            stateManager.onIndexingComplete({ filesIndexed: 5, chunksCreated: 20 });
+            expect(stateManager.isIndexingInProgress()).toBe(false);
+        });
+        it('should track last progress', () => {
+            expect(stateManager.getLastProgress()).toBeNull();
+            const progress = {
+                phase: 'embedding',
+                current: 5,
+                total: 10,
+                message: 'Embedding chunks...',
+            };
+            stateManager.onProgress(progress);
+            expect(stateManager.getLastProgress()).toEqual(progress);
+        });
+        it('should clear last progress on indexing complete', () => {
+            stateManager.onIndexingStart();
+            stateManager.onProgress({
+                phase: 'embedding',
+                current: 5,
+                total: 10,
+                message: 'Embedding chunks...',
+            });
+            expect(stateManager.getLastProgress()).not.toBeNull();
+            stateManager.onIndexingComplete({ filesIndexed: 5, chunksCreated: 20 });
+            expect(stateManager.getLastProgress()).toBeNull();
+        });
+    });
+    describe('events', () => {
+        it('should emit indexing:start event', () => {
+            const listener = vi.fn();
+            stateManager.on('indexing:start', listener);
+            stateManager.onIndexingStart();
+            expect(listener).toHaveBeenCalled();
+        });
+        it('should emit progress event', () => {
+            const listener = vi.fn();
+            stateManager.on('progress', listener);
+            const progress = {
+                phase: 'chunking',
+                current: 3,
+                total: 10,
+                message: 'Chunking files...',
+            };
+            stateManager.onProgress(progress);
+            expect(listener).toHaveBeenCalledWith(progress);
+        });
+        it('should emit indexing:complete event', () => {
+            const listener = vi.fn();
+            stateManager.on('indexing:complete', listener);
+            const result = { filesIndexed: 5, chunksCreated: 20 };
+            stateManager.onIndexingComplete(result);
+            expect(listener).toHaveBeenCalledWith(result);
+        });
+        it('should emit status:change event', () => {
+            const listener = vi.fn();
+            stateManager.on('status:change', listener);
+            const status = {
+                indexed: true,
+                fileCount: 10,
+                chunkCount: 50,
+                lastUpdated: '2024-01-01T00:00:00.000Z',
+                indexPath: '/test/.glancey',
+            };
+            stateManager.onStatusChange(status);
+            expect(listener).toHaveBeenCalledWith(status);
+        });
+    });
+});
+//# sourceMappingURL=state.test.js.map

package/dist/__tests__/dashboard/state.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"state.test.js","sourceRoot":"","sources":["../../../src/__tests__/dashboard/state.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AAEjE,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,IAAI,YAAmC,CAAC;IAExC,UAAU,CAAC,GAAG,EAAE;QACd,YAAY,GAAG,IAAI,qBAAqB,EAAE,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,WAAW,GAAG;gBAClB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBACnC,OAAO,EAAE,IAAI;oBACb,SAAS,EAAE,EAAE;oBACb,UAAU,EAAE,EAAE;oBACd,WAAW,EAAE,0BAA0B;oBACvC,SAAS,EAAE,gBAAgB;iBAC5B,CAAC;aACH,CAAC;YAEF,YAAY,CAAC,UAAU,CAAC,WAAoB,CAAC,CAAC;YAC9C,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,SAAS,EAAE,CAAC;YAE9C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,EAAE;gBACb,UAAU,EAAE,EAAE;gBACd,WAAW,EAAE,0BAA0B;gBACvC,SAAS,EAAE,gBAAgB;aAC5B,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,SAAS,EAAE,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,MAAM,GAAG;gBACb,QAAQ,EAAE,CAAC,SAAS,CAAC;gBACrB,eAAe,EAAE,CAAC,oBAAoB,CAAC;aACxC,CAAC;YAEF,YAAY,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAC/B,MAAM,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,YAAY,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YAC7C,MAAM,CAAC,YAAY,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,CAAC,YAAY,CAAC,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAExD,YAAY,CAAC,eAAe,EAAE,CAAC;YAC/B,MAAM,CAAC,YAAY,CAAC,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEvD,YAAY,CAAC,kBAAkB,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;YACxE,MAAM,CAAC,YAAY,CAAC,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,CAAC,YAAY,CAAC,eAAe,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC;YAElD,MAAM,QAAQ,GAAG;gBACf,KAAK,EAAE,WAAoB;gBAC3B,OAAO,EAAE,CAAC;gBACV,KAAK,EAAE,EAAE;gBACT,OAAO,EAAE,qBAAqB;aAC/B,CAAC;YAEF,YAAY,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAClC,MAAM,CAAC,YAAY,CAAC,eAAe,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,YAAY,CAAC,eAAe,EAAE,CAAC;YAC/B,YAAY,CAAC,UAAU,CAAC;gBACtB,KAAK,EAAE,WAAW;gBAClB,OAAO,EAAE,CAAC;gBACV,KAAK,EAAE,EAAE;gBACT,OAAO,EAAE,qBAAqB;aAC/B,CAAC,CAAC;YAEH,MAAM,CAAC,YAAY,CAAC,eAAe,EAAE,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAEtD,YAAY,CAAC,kBAAkB,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;YACxE,MAAM,CAAC,YAAY,CAAC,eAAe,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,YAAY,CAAC,EAAE,CAAC,gBAAgB,EAAE,QAAQ,CAAC,CAAC;YAE5C,YAAY,CAAC,eAAe,EAAE,CAAC;YAE/B,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,YAAY,CAAC,EAAE,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YAEtC,MAAM,QAAQ,GAAG;gBACf,KAAK,EAAE,UAAmB;gBAC1B,OAAO,EAAE,CAAC;gBACV,KAAK,EAAE,EAAE;gBACT,OAAO,EAAE,mBAAmB;aAC7B,CAAC;YACF,YAAY,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAElC,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,YAAY,CAAC,EAAE,CAAC,mBAAmB,EAAE,QAAQ,CAAC,CAAC;YAE/C,MAAM,MAAM,GAAG,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC;YACtD,YAAY,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAExC,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,YAAY,CAAC,EAAE,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;YAE3C,MAAM,MAAM,GAAG;gBACb,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,EAAE;gBACb,UAAU,EAAE,EAAE;gBACd,WAAW,EAAE,0BAA0B;gBACvC,SAAS,EAAE,gBAAgB;aAC5B,CAAC;YACF,YAAY,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAEpC,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/dashboard/token-tracking.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=token-tracking.test.d.ts.map

package/dist/__tests__/dashboard/token-tracking.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"token-tracking.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/dashboard/token-tracking.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/dashboard/token-tracking.test.js

@@ -0,0 +1,315 @@
+import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
+import * as fs from 'fs';
+import { TokenSavingsTracker } from '../../dashboard/token-tracking.js';
+// Mock fs module
+vi.mock('fs');
+describe('TokenSavingsTracker', () => {
+    let tracker;
+    const mockProjectPath = '/test/project';
+    beforeEach(() => {
+        vi.clearAllMocks();
+        tracker = new TokenSavingsTracker();
+        // Default: file doesn't exist
+        vi.mocked(fs.existsSync).mockReturnValue(false);
+    });
+    afterEach(() => {
+        vi.useRealTimers();
+    });
+    describe('setProjectPath', () => {
+        it('should load persisted data when file exists', () => {
+            const persistedData = {
+                events: [
+                    {
+                        type: 'search_code',
+                        timestamp: 1234567890,
+                        charsReturned: 100,
+                        charsAvoided: 500,
+                        filesAvoided: 2,
+                    },
+                ],
+                sessionStart: 1234567800,
+            };
+            vi.mocked(fs.existsSync).mockReturnValue(true);
+            vi.mocked(fs.readFileSync).mockReturnValue(JSON.stringify(persistedData));
+            tracker.setProjectPath(mockProjectPath);
+            const stats = tracker.getStats();
+            expect(stats.operationCount).toBe(1);
+            expect(stats.sessionStart).toBe(1234567800);
+        });
+        it('should validate events on load and filter invalid ones', () => {
+            const persistedData = {
+                events: [
+                    // Valid event
+                    {
+                        type: 'search_code',
+                        timestamp: 1234567890,
+                        charsReturned: 100,
+                        charsAvoided: 500,
+                        filesAvoided: 2,
+                    },
+                    // Invalid: missing type
+                    {
+                        timestamp: 1234567891,
+                        charsReturned: 50,
+                        charsAvoided: 250,
+                        filesAvoided: 1,
+                    },
+                    // Invalid: wrong type
+                    {
+                        type: 'invalid_type',
+                        timestamp: 1234567892,
+                        charsReturned: 50,
+                        charsAvoided: 250,
+                        filesAvoided: 1,
+                    },
+                    // Invalid: missing required field
+                    {
+                        type: 'search_code',
+                        timestamp: 1234567893,
+                        charsReturned: 50,
+                        // missing charsAvoided and filesAvoided
+                    },
+                ],
+                sessionStart: 1234567800,
+            };
+            vi.mocked(fs.existsSync).mockReturnValue(true);
+            vi.mocked(fs.readFileSync).mockReturnValue(JSON.stringify(persistedData));
+            tracker.setProjectPath(mockProjectPath);
+            const stats = tracker.getStats();
+            // Only the first valid event should be loaded
+            expect(stats.operationCount).toBe(1);
+        });
+        it('should handle missing file gracefully', () => {
+            vi.mocked(fs.existsSync).mockReturnValue(false);
+            tracker.setProjectPath(mockProjectPath);
+            const stats = tracker.getStats();
+            expect(stats.operationCount).toBe(0);
+        });
+        it('should handle corrupted JSON gracefully', () => {
+            vi.mocked(fs.existsSync).mockReturnValue(true);
+            vi.mocked(fs.readFileSync).mockReturnValue('not valid json');
+            tracker.setProjectPath(mockProjectPath);
+            const stats = tracker.getStats();
+            expect(stats.operationCount).toBe(0);
+        });
+    });
+    describe('event recording', () => {
+        beforeEach(() => {
+            vi.useFakeTimers();
+            tracker.setProjectPath(mockProjectPath);
+        });
+        it('should record search_code events with correct estimates', () => {
+            tracker.recordSearchCode(1000, 3, 50);
+            const stats = tracker.getStats();
+            expect(stats.operationCount).toBe(1);
+            expect(stats.byType.search_code.count).toBe(1);
+            expect(stats.byType.search_code.charsReturned).toBe(1000);
+            // charsAvoided = max(0, 100 * 60 - 1000) = 5000
+            expect(stats.byType.search_code.charsAvoided).toBe(5000);
+        });
+        it('should record search_similar events', () => {
+            tracker.recordSearchSimilar(500, 5);
+            const stats = tracker.getStats();
+            expect(stats.byType.search_similar.count).toBe(1);
+            expect(stats.byType.search_similar.charsReturned).toBe(500);
+        });
+        it('should record get_symbols_overview events', () => {
+            tracker.recordSymbolsOverview(200, 100);
+            const stats = tracker.getStats();
+            expect(stats.byType.get_symbols_overview.count).toBe(1);
+            // charsAvoided = max(0, 100 * 60 - 200) = 5800
+            expect(stats.byType.get_symbols_overview.charsAvoided).toBe(5800);
+        });
+        it('should record find_symbol events', () => {
+            tracker.recordFindSymbol(300, 20);
+            const stats = tracker.getStats();
+            expect(stats.byType.find_symbol.count).toBe(1);
+        });
+        it('should record summarize_codebase events', () => {
+            tracker.recordSummarizeCodebase(1500, 100);
+            const stats = tracker.getStats();
+            expect(stats.byType.summarize_codebase.count).toBe(1);
+        });
+        it('should record list_concepts events', () => {
+            tracker.recordListConcepts(400, 5);
+            const stats = tracker.getStats();
+            expect(stats.byType.list_concepts.count).toBe(1);
+        });
+        it('should record search_by_concept events', () => {
+            tracker.recordSearchByConcept(600, 10);
+            const stats = tracker.getStats();
+            expect(stats.byType.search_by_concept.count).toBe(1);
+        });
+        it('should not record negative charsAvoided', () => {
+            // Return more chars than estimated, should clamp to 0
+            tracker.recordSearchCode(10000, 1, 10);
+            const stats = tracker.getStats();
+            expect(stats.byType.search_code.charsAvoided).toBe(0);
+        });
+    });
+    describe('getStats', () => {
+        beforeEach(() => {
+            vi.useFakeTimers();
+            tracker.setProjectPath(mockProjectPath);
+        });
+        it('should calculate correct totals', () => {
+            tracker.recordSearchCode(100, 2, 50);
+            tracker.recordFindSymbol(200, 30);
+            const stats = tracker.getStats();
+            expect(stats.operationCount).toBe(2);
+            expect(stats.charsReturned).toBe(300);
+        });
+        it('should calculate efficiency percentage correctly', () => {
+            // Record event where we return 1000 chars and avoid 5000
+            tracker.recordSearchCode(1000, 2, 50);
+            const stats = tracker.getStats();
+            // Total chars = 1000 + 5000 = 6000
+            // Efficiency = 5000 / 6000 = 83.33% -> rounds to 83%
+            expect(stats.efficiencyPercent).toBe(83);
+        });
+        it('should return 0% efficiency when no events', () => {
+            const stats = tracker.getStats();
+            expect(stats.efficiencyPercent).toBe(0);
+        });
+        it('should calculate tokens saved correctly', () => {
+            // 4 chars per token
+            tracker.recordSearchCode(1000, 2, 50);
+            const stats = tracker.getStats();
+            // charsAvoided = 5000, tokens = 5000 / 4 = 1250
+            expect(stats.tokensSaved).toBe(1250);
+        });
+    });
+    describe('event rotation', () => {
+        beforeEach(() => {
+            vi.useFakeTimers();
+            tracker.setProjectPath(mockProjectPath);
+        });
+        it('should rotate events when exceeding MAX_EVENTS', () => {
+            // Record more than 1000 events
+            for (let i = 0; i < 1005; i++) {
+                tracker.recordSearchCode(100, 1, 10);
+            }
+            const stats = tracker.getStats();
+            // Should be capped at 1000
+            expect(stats.operationCount).toBe(1000);
+        });
+        it('should keep most recent events when rotating', () => {
+            // Load persisted data with 1005 events
+            const events = [];
+            for (let i = 0; i < 1005; i++) {
+                events.push({
+                    type: 'search_code',
+                    timestamp: 1000 + i,
+                    charsReturned: 100,
+                    charsAvoided: 500,
+                    filesAvoided: 1,
+                });
+            }
+            vi.mocked(fs.existsSync).mockReturnValue(true);
+            vi.mocked(fs.readFileSync).mockReturnValue(JSON.stringify({ events, sessionStart: 1000 }));
+            tracker.setProjectPath(mockProjectPath);
+            const recentEvents = tracker.getRecentEvents(5);
+            // Should have the most recent events (highest timestamps)
+            expect(recentEvents[0].timestamp).toBeGreaterThan(1000);
+        });
+    });
+    describe('debounced saving', () => {
+        beforeEach(() => {
+            vi.useFakeTimers();
+            vi.mocked(fs.existsSync).mockReturnValue(true);
+            vi.mocked(fs.mkdirSync).mockReturnValue(undefined);
+            vi.mocked(fs.writeFileSync).mockReturnValue(undefined);
+            tracker.setProjectPath(mockProjectPath);
+        });
+        it('should debounce multiple rapid saves', () => {
+            tracker.recordSearchCode(100, 1, 10);
+            tracker.recordSearchCode(200, 2, 20);
+            tracker.recordSearchCode(300, 3, 30);
+            // Should not have saved yet
+            expect(fs.writeFileSync).not.toHaveBeenCalled();
+            // Advance past debounce delay
+            vi.advanceTimersByTime(1500);
+            // Should have saved once with all events
+            expect(fs.writeFileSync).toHaveBeenCalledTimes(1);
+            const savedData = JSON.parse(fs.writeFileSync.mock.calls[0][1]);
+            expect(savedData.events.length).toBe(3);
+        });
+    });
+    describe('onUpdate callback', () => {
+        beforeEach(() => {
+            vi.useFakeTimers();
+            tracker.setProjectPath(mockProjectPath);
+        });
+        it('should call onUpdate callback when events are recorded', () => {
+            const callback = vi.fn();
+            tracker.setOnUpdate(callback);
+            tracker.recordSearchCode(100, 1, 10);
+            expect(callback).toHaveBeenCalledTimes(1);
+        });
+        it('should call onUpdate for each event recorded', () => {
+            const callback = vi.fn();
+            tracker.setOnUpdate(callback);
+            tracker.recordSearchCode(100, 1, 10);
+            tracker.recordFindSymbol(200, 20);
+            expect(callback).toHaveBeenCalledTimes(2);
+        });
+    });
+    describe('getRecentEvents', () => {
+        beforeEach(() => {
+            vi.useFakeTimers();
+            tracker.setProjectPath(mockProjectPath);
+        });
+        it('should return last N events', () => {
+            for (let i = 0; i < 20; i++) {
+                tracker.recordSearchCode(100 + i, 1, 10);
+            }
+            const recent = tracker.getRecentEvents(5);
+            expect(recent.length).toBe(5);
+            // Should be the last 5 (highest charsReturned values)
+            expect(recent[0].charsReturned).toBe(115);
+            expect(recent[4].charsReturned).toBe(119);
+        });
+        it('should return all events if fewer than limit', () => {
+            tracker.recordSearchCode(100, 1, 10);
+            tracker.recordSearchCode(200, 2, 20);
+            const recent = tracker.getRecentEvents(10);
+            expect(recent.length).toBe(2);
+        });
+        it('should default to 10 events', () => {
+            for (let i = 0; i < 20; i++) {
+                tracker.recordSearchCode(100, 1, 10);
+            }
+            const recent = tracker.getRecentEvents();
+            expect(recent.length).toBe(10);
+        });
+    });
+    describe('filesAvoided estimates', () => {
+        beforeEach(() => {
+            vi.useFakeTimers();
+            tracker.setProjectPath(mockProjectPath);
+        });
+        it('should estimate filesAvoided for search_code based on files searched', () => {
+            // 50 files searched, 3 matched -> avoided ~7 files (min(10, 50) - 3)
+            tracker.recordSearchCode(100, 3, 50);
+            const events = tracker.getRecentEvents(1);
+            expect(events[0].filesAvoided).toBe(7);
+        });
+        it('should estimate filesAvoided for search_similar as at least 3', () => {
+            tracker.recordSearchSimilar(100, 2);
+            const events = tracker.getRecentEvents(1);
+            expect(events[0].filesAvoided).toBe(3);
+        });
+        it('should estimate filesAvoided for summarize_codebase capped at 20', () => {
+            tracker.recordSummarizeCodebase(100, 100);
+            const events = tracker.getRecentEvents(1);
+            expect(events[0].filesAvoided).toBe(20);
+        });
+        it('should estimate filesAvoided for list_concepts as 3x clusters capped at 15', () => {
+            tracker.recordListConcepts(100, 10);
+            const events = tracker.getRecentEvents(1);
+            // 10 clusters * 3 = 30, capped at 15
+            expect(events[0].filesAvoided).toBe(15);
+        });
+    });
+});
+//# sourceMappingURL=token-tracking.test.js.map

package/dist/__tests__/dashboard/token-tracking.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"token-tracking.test.js","sourceRoot":"","sources":["../../../src/__tests__/dashboard/token-tracking.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,EAAE,mBAAmB,EAA0B,MAAM,mCAAmC,CAAC;AAEhG,iBAAiB;AACjB,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEd,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,OAA4B,CAAC;IACjC,MAAM,eAAe,GAAG,eAAe,CAAC;IAExC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,GAAG,IAAI,mBAAmB,EAAE,CAAC;QACpC,8BAA8B;QAC9B,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,aAAa,GAAG;gBACpB,MAAM,EAAE;oBACN;wBACE,IAAI,EAAE,aAAa;wBACnB,SAAS,EAAE,UAAU;wBACrB,aAAa,EAAE,GAAG;wBAClB,YAAY,EAAE,GAAG;wBACjB,YAAY,EAAE,CAAC;qBAChB;iBACF;gBACD,YAAY,EAAE,UAAU;aACzB,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC;YAE1E,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YACxC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YAEjC,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,aAAa,GAAG;gBACpB,MAAM,EAAE;oBACN,cAAc;oBACd;wBACE,IAAI,EAAE,aAAa;wBACnB,SAAS,EAAE,UAAU;wBACrB,aAAa,EAAE,GAAG;wBAClB,YAAY,EAAE,GAAG;wBACjB,YAAY,EAAE,CAAC;qBAChB;oBACD,wBAAwB;oBACxB;wBACE,SAAS,EAAE,UAAU;wBACrB,aAAa,EAAE,EAAE;wBACjB,YAAY,EAAE,GAAG;wBACjB,YAAY,EAAE,CAAC;qBAChB;oBACD,sBAAsB;oBACtB;wBACE,IAAI,EAAE,cAAc;wBACpB,SAAS,EAAE,UAAU;wBACrB,aAAa,EAAE,EAAE;wBACjB,YAAY,EAAE,GAAG;wBACjB,YAAY,EAAE,CAAC;qBAChB;oBACD,kCAAkC;oBAClC;wBACE,IAAI,EAAE,aAAa;wBACnB,SAAS,EAAE,UAAU;wBACrB,aAAa,EAAE,EAAE;wBACjB,wCAAwC;qBACzC;iBACF;gBACD,YAAY,EAAE,UAAU;aACzB,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC;YAE1E,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YACxC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YAEjC,8CAA8C;YAC9C,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAEhD,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YACxC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YAEjC,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;YAE7D,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YACxC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YAEjC,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,OAAO,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAEtC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1D,gDAAgD;YAChD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,OAAO,CAAC,mBAAmB,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YAEpC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,OAAO,CAAC,qBAAqB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAExC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACxD,+CAA+C;YAC/C,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAElC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,OAAO,CAAC,uBAAuB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAE3C,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,OAAO,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YAEnC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,OAAO,CAAC,qBAAqB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAEvC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,sDAAsD;YACtD,OAAO,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAEvC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACrC,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAElC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YAEjC,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,yDAAyD;YACzD,OAAO,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAEtC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,mCAAmC;YACnC,qDAAqD;YACrD,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,oBAAoB;YACpB,OAAO,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAEtC,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,gDAAgD;YAChD,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,+BAA+B;YAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9B,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACvC,CAAC;YAED,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,2BAA2B;YAC3B,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,uCAAuC;YACvC,MAAM,MAAM,GAAwB,EAAE,CAAC;YACvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9B,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,aAAa;oBACnB,SAAS,EAAE,IAAI,GAAG,CAAC;oBACnB,aAAa,EAAE,GAAG;oBAClB,YAAY,EAAE,GAAG;oBACjB,YAAY,EAAE,CAAC;iBAChB,CAAC,CAAC;YACL,CAAC;YAED,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAE3F,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YACxC,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEhD,0DAA0D;YAC1D,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,aAAa,EAAE,CAAC;YACnB,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YACnD,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YACvD,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACrC,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACrC,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAErC,4BAA4B;YAC5B,MAAM,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAEhD,8BAA8B;YAC9B,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAE7B,yCAAyC;YACzC,MAAM,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CACzB,EAAE,CAAC,aAA0C,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAW,CAC1E,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAE9B,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAErC,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAE9B,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACrC,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAElC,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,OAAO,CAAC,gBAAgB,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAC3C,CAAC;YAED,MAAM,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAE1C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9B,sDAAsD;YACtD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACrC,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAErC,MAAM,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;YAE3C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACvC,CAAC;YAED,MAAM,MAAM,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;YAEzC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;YAC9E,qEAAqE;YACrE,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAErC,MAAM,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,OAAO,CAAC,mBAAmB,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YAEpC,MAAM,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,OAAO,CAAC,uBAAuB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAE1C,MAAM,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,OAAO,CAAC,kBAAkB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAEpC,MAAM,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC1C,qCAAqC;YACrC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/embeddings/factory.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=factory.test.d.ts.map

package/dist/__tests__/embeddings/factory.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/embeddings/factory.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/embeddings/factory.test.js

@@ -0,0 +1,157 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { createSuccessFetch } from '../mocks/fetch.mock.js';
+import { DEFAULT_OLLAMA_MODEL } from '../../embeddings/ollama.js';
+/** Helper to create a mock /api/tags response with the default model available */
+function createTagsResponseWithDefaultModel() {
+    return { models: [{ name: DEFAULT_OLLAMA_MODEL }] };
+}
+/** Helper to create a mock Gemini embedding response */
+function createGeminiEmbeddingResponse() {
+    return { embedding: { values: [0.1, 0.2] } };
+}
+// We need to dynamically import createEmbeddingBackend after setting env vars
+async function getCreateEmbeddingBackend() {
+    // Reset module cache to pick up new env vars
+    vi.resetModules();
+    const module = await import('../../embeddings/index.js');
+    return module.createEmbeddingBackend;
+}
+describe('createEmbeddingBackend', () => {
+    const originalEnv = { ...process.env };
+    beforeEach(() => {
+        // Clear relevant env vars
+        delete process.env.GEMINI_API_KEY;
+        delete process.env.OLLAMA_URL;
+    });
+    afterEach(() => {
+        // Restore original env
+        process.env = { ...originalEnv };
+        vi.unstubAllGlobals();
+        vi.resetModules();
+    });
+    describe('priority order', () => {
+        it('should prefer Gemini when GEMINI_API_KEY is set', async () => {
+            process.env.GEMINI_API_KEY = 'test-gemini-key';
+            const mockFetch = vi.fn().mockResolvedValue({
+                ok: true,
+                status: 200,
+                json: async () => createGeminiEmbeddingResponse(),
+            });
+            vi.stubGlobal('fetch', mockFetch);
+            const createEmbeddingBackend = await getCreateEmbeddingBackend();
+            const { backend } = await createEmbeddingBackend();
+            expect(backend.name).toBe('gemini');
+        });
+        it('should fallback to Ollama when Gemini fails', async () => {
+            process.env.GEMINI_API_KEY = 'invalid-gemini-key';
+            const mockFetch = vi
+                .fn()
+                // Gemini fails
+                .mockResolvedValueOnce({ ok: false, status: 401, text: async () => 'Unauthorized' })
+                // Ollama succeeds (with default model available)
+                .mockResolvedValue({
+                ok: true,
+                status: 200,
+                json: async () => createTagsResponseWithDefaultModel(),
+            });
+            vi.stubGlobal('fetch', mockFetch);
+            const createEmbeddingBackend = await getCreateEmbeddingBackend();
+            const { backend } = await createEmbeddingBackend();
+            expect(backend.name).toBe('ollama');
+        });
+        it('should use Ollama when no API keys are set', async () => {
+            vi.stubGlobal('fetch', createSuccessFetch(createTagsResponseWithDefaultModel()));
+            const createEmbeddingBackend = await getCreateEmbeddingBackend();
+            const { backend } = await createEmbeddingBackend();
+            expect(backend.name).toBe('ollama');
+        });
+    });
+    describe('error when all backends fail', () => {
+        it('should throw when no backend is available', async () => {
+            const mockFetch = vi.fn().mockRejectedValue(new Error('Connection refused'));
+            vi.stubGlobal('fetch', mockFetch);
+            const createEmbeddingBackend = await getCreateEmbeddingBackend();
+            await expect(createEmbeddingBackend()).rejects.toThrow('No embedding backend available');
+        });
+    });
+    describe('config options', () => {
+        it('should pass apiKey to Gemini backend from config', async () => {
+            const mockFetch = vi.fn().mockResolvedValue({
+                ok: true,
+                status: 200,
+                json: async () => createGeminiEmbeddingResponse(),
+            });
+            vi.stubGlobal('fetch', mockFetch);
+            const createEmbeddingBackend = await getCreateEmbeddingBackend();
+            const { backend } = await createEmbeddingBackend({ apiKey: 'config-gemini-key' });
+            expect(backend.name).toBe('gemini');
+        });
+        it('should use OLLAMA_URL from environment', async () => {
+            process.env.OLLAMA_URL = 'http://remote-ollama:11434';
+            const mockFetch = vi.fn().mockImplementation(async (url) => {
+                expect(url).toContain('remote-ollama');
+                return { ok: true, status: 200, json: async () => createTagsResponseWithDefaultModel() };
+            });
+            vi.stubGlobal('fetch', mockFetch);
+            const createEmbeddingBackend = await getCreateEmbeddingBackend();
+            const { backend } = await createEmbeddingBackend();
+            expect(backend.name).toBe('ollama');
+        });
+        it('should prefer baseUrl from config over environment', async () => {
+            process.env.OLLAMA_URL = 'http://env-ollama:11434';
+            const mockFetch = vi.fn().mockImplementation(async (url) => {
+                expect(url).toContain('config-ollama');
+                return { ok: true, status: 200, json: async () => createTagsResponseWithDefaultModel() };
+            });
+            vi.stubGlobal('fetch', mockFetch);
+            const createEmbeddingBackend = await getCreateEmbeddingBackend();
+            const { backend } = await createEmbeddingBackend({ baseUrl: 'http://config-ollama:11434' });
+            expect(backend.name).toBe('ollama');
+        });
+    });
+    describe('explicit backend with fallback', () => {
+        it('should fallback to Ollama when explicitly configured Gemini fails', async () => {
+            process.env.GEMINI_API_KEY = 'test-key';
+            const mockFetch = vi
+                .fn()
+                // Gemini initialization fails with an error
+                .mockResolvedValueOnce({ ok: false, status: 401, text: async () => 'Invalid API key' })
+                // Ollama succeeds
+                .mockResolvedValue({
+                ok: true,
+                status: 200,
+                json: async () => createTagsResponseWithDefaultModel(),
+            });
+            vi.stubGlobal('fetch', mockFetch);
+            const createEmbeddingBackend = await getCreateEmbeddingBackend();
+            const { backend, fallback } = await createEmbeddingBackend({ backend: 'gemini' });
+            expect(backend.name).toBe('ollama');
+            expect(fallback).toBeDefined();
+            expect(fallback?.occurred).toBe(true);
+            expect(fallback?.originalBackend).toBe('gemini');
+            expect(fallback?.fallbackBackend).toBe('ollama');
+            expect(fallback?.reason).toContain('401');
+        });
+        it('should throw when explicitly configured Gemini fails and Ollama also fails', async () => {
+            process.env.GEMINI_API_KEY = 'test-key';
+            const mockFetch = vi.fn().mockRejectedValue(new Error('Connection refused'));
+            vi.stubGlobal('fetch', mockFetch);
+            const createEmbeddingBackend = await getCreateEmbeddingBackend();
+            await expect(createEmbeddingBackend({ backend: 'gemini' })).rejects.toThrow('Configured gemini backend failed');
+        });
+        it('should not have fallback info when explicitly configured backend succeeds', async () => {
+            process.env.GEMINI_API_KEY = 'test-key';
+            const mockFetch = vi.fn().mockResolvedValue({
+                ok: true,
+                status: 200,
+                json: async () => createGeminiEmbeddingResponse(),
+            });
+            vi.stubGlobal('fetch', mockFetch);
+            const createEmbeddingBackend = await getCreateEmbeddingBackend();
+            const { backend, fallback } = await createEmbeddingBackend({ backend: 'gemini' });
+            expect(backend.name).toBe('gemini');
+            expect(fallback).toBeUndefined();
+        });
+    });
+});
+//# sourceMappingURL=factory.test.js.map