@code-rag/core 0.1.0

package/dist/retrieval/query-analyzer.test.js

@@ -0,0 +1,236 @@
+import { describe, it, expect } from 'vitest';
+import { QueryAnalyzer } from './query-analyzer.js';
+describe('QueryAnalyzer', () => {
+    const analyzer = new QueryAnalyzer();
+    describe('analyze returns correct structure', () => {
+        it('should return all required fields', () => {
+            const result = analyzer.analyze('hello world');
+            expect(result).toHaveProperty('originalQuery', 'hello world');
+            expect(result).toHaveProperty('intent');
+            expect(result).toHaveProperty('entities');
+            expect(result).toHaveProperty('suggestedFilters');
+            expect(result).toHaveProperty('expandedTerms');
+            expect(Array.isArray(result.entities)).toBe(true);
+            expect(Array.isArray(result.expandedTerms)).toBe(true);
+        });
+        it('should preserve the original query string', () => {
+            const query = 'where is MyClass defined';
+            const result = analyzer.analyze(query);
+            expect(result.originalQuery).toBe(query);
+        });
+    });
+    describe('intent detection', () => {
+        const intentCases = [
+            // find_definition
+            { query: 'where is MyClass defined', expected: 'find_definition' },
+            { query: 'definition of parseConfig', expected: 'find_definition' },
+            { query: 'find definition of HybridSearch', expected: 'find_definition' },
+            { query: 'show me the definition', expected: 'find_definition' },
+            { query: 'where is MyClass declared', expected: 'find_definition' },
+            { query: 'declaration of SearchResult', expected: 'find_definition' },
+            // find_usage
+            { query: 'who calls parseConfig', expected: 'find_usage' },
+            { query: 'usage of DependencyGraph', expected: 'find_usage' },
+            { query: 'where is MyClass used', expected: 'find_usage' },
+            { query: 'references to SearchResult', expected: 'find_usage' },
+            { query: 'callers of initialize', expected: 'find_usage' },
+            { query: 'consumers of the API', expected: 'find_usage' },
+            { query: 'find usage of handleRequest', expected: 'find_usage' },
+            { query: 'used by other modules', expected: 'find_usage' },
+            // understand_module
+            { query: 'how does HybridSearch work', expected: 'understand_module' },
+            { query: 'explain the indexer module', expected: 'understand_module' },
+            { query: 'what does parseConfig do', expected: 'understand_module' },
+            { query: 'understand the graph module', expected: 'understand_module' },
+            { query: 'overview of the embedding system', expected: 'understand_module' },
+            { query: 'describe the chunker', expected: 'understand_module' },
+            { query: 'how is BM25Index implemented', expected: 'understand_module' },
+            // find_similar
+            { query: 'similar to HybridSearch', expected: 'find_similar' },
+            { query: 'something like parseConfig', expected: 'find_similar' },
+            { query: 'alternatives to LanceDB', expected: 'find_similar' },
+            { query: 'related to the search module', expected: 'find_similar' },
+            // general
+            { query: 'search function', expected: 'general' },
+            { query: 'what are the main modules', expected: 'general' },
+            { query: 'list all functions', expected: 'general' },
+        ];
+        for (const { query, expected } of intentCases) {
+            it(`should detect "${expected}" for: "${query}"`, () => {
+                const result = analyzer.analyze(query);
+                expect(result.intent).toBe(expected);
+            });
+        }
+    });
+    describe('entity extraction', () => {
+        it('should extract PascalCase identifiers as class entities', () => {
+            const result = analyzer.analyze('find DependencyGraph');
+            const classEntities = result.entities.filter((e) => e.type === 'class');
+            expect(classEntities).toContainEqual({
+                type: 'class',
+                value: 'DependencyGraph',
+            });
+        });
+        it('should extract camelCase identifiers as function entities', () => {
+            const result = analyzer.analyze('where is parseConfig defined');
+            const funcEntities = result.entities.filter((e) => e.type === 'function');
+            expect(funcEntities).toContainEqual({
+                type: 'function',
+                value: 'parseConfig',
+            });
+        });
+        it('should extract file paths as file entities', () => {
+            const result = analyzer.analyze('look at src/graph/dependency-graph.ts');
+            const fileEntities = result.entities.filter((e) => e.type === 'file');
+            expect(fileEntities).toContainEqual({
+                type: 'file',
+                value: 'src/graph/dependency-graph.ts',
+            });
+        });
+        it('should not extract common English words as entities', () => {
+            const result = analyzer.analyze('find the class');
+            const entities = result.entities.map((e) => e.value);
+            expect(entities).not.toContain('the');
+            expect(entities).not.toContain('find');
+        });
+        it('should deduplicate entities', () => {
+            const result = analyzer.analyze('MyClass and MyClass');
+            const myClassEntities = result.entities.filter((e) => e.value === 'MyClass');
+            expect(myClassEntities).toHaveLength(1);
+        });
+        it('should extract multiple identifiers', () => {
+            const result = analyzer.analyze('MyClass uses parseConfig from HybridSearch');
+            expect(result.entities.length).toBeGreaterThanOrEqual(3);
+        });
+        it('should return empty entities for plain text query', () => {
+            const result = analyzer.analyze('search for something');
+            // "something" doesn't match PascalCase or camelCase multi-word
+            expect(result.entities).toHaveLength(0);
+        });
+    });
+    describe('filter suggestion', () => {
+        it('should suggest typescript language filter', () => {
+            const result = analyzer.analyze('find typescript functions');
+            expect(result.suggestedFilters.languages).toContain('typescript');
+        });
+        it('should suggest language filter for abbreviated names', () => {
+            const result = analyzer.analyze('show me ts classes');
+            expect(result.suggestedFilters.languages).toContain('typescript');
+        });
+        it('should suggest python language filter', () => {
+            const result = analyzer.analyze('find python modules');
+            expect(result.suggestedFilters.languages).toContain('python');
+        });
+        it('should suggest chunk type filter for "function"', () => {
+            const result = analyzer.analyze('list all functions');
+            expect(result.suggestedFilters.chunkTypes).toContain('function');
+        });
+        it('should suggest chunk type filter for "class"', () => {
+            const result = analyzer.analyze('find the class definition');
+            expect(result.suggestedFilters.chunkTypes).toContain('class');
+        });
+        it('should suggest chunk type filter for "interface"', () => {
+            const result = analyzer.analyze('show all interfaces');
+            expect(result.suggestedFilters.chunkTypes).toContain('interface');
+        });
+        it('should suggest file path filter', () => {
+            const result = analyzer.analyze('look at src/graph/dependency-graph.ts');
+            expect(result.suggestedFilters.filePaths).toContain('src/graph/dependency-graph.ts');
+        });
+        it('should return empty filters when no language or type mentioned', () => {
+            const result = analyzer.analyze('hello world');
+            expect(result.suggestedFilters.languages).toBeUndefined();
+            expect(result.suggestedFilters.chunkTypes).toBeUndefined();
+            expect(result.suggestedFilters.filePaths).toBeUndefined();
+        });
+        it('should detect multiple languages', () => {
+            const result = analyzer.analyze('compare typescript and python');
+            expect(result.suggestedFilters.languages).toContain('typescript');
+            expect(result.suggestedFilters.languages).toContain('python');
+        });
+    });
+    describe('edge cases', () => {
+        it('should return general intent for empty string', () => {
+            const result = analyzer.analyze('');
+            expect(result.intent).toBe('general');
+            expect(result.entities).toHaveLength(0);
+            expect(result.expandedTerms).toHaveLength(0);
+            expect(result.originalQuery).toBe('');
+        });
+        it('should return general intent for whitespace-only string', () => {
+            const result = analyzer.analyze('   \t\n  ');
+            expect(result.intent).toBe('general');
+            expect(result.entities).toHaveLength(0);
+            expect(result.expandedTerms).toHaveLength(0);
+        });
+        it('should return general intent for oversized string', () => {
+            const longQuery = 'a'.repeat(3000);
+            const result = analyzer.analyze(longQuery);
+            expect(result.intent).toBe('general');
+            expect(result.entities).toHaveLength(0);
+            expect(result.expandedTerms).toHaveLength(0);
+            expect(result.originalQuery).toBe(longQuery);
+        });
+        it('should handle strings with special regex characters', () => {
+            const result = analyzer.analyze('find [MyClass] (deprecated)');
+            // Should not throw, should extract MyClass as class entity
+            expect(result).toBeDefined();
+            const classEntities = result.entities.filter((e) => e.type === 'class');
+            expect(classEntities).toContainEqual({
+                type: 'class',
+                value: 'MyClass',
+            });
+        });
+    });
+    describe('term expansion', () => {
+        it('should include original query terms', () => {
+            const result = analyzer.analyze('search function');
+            expect(result.expandedTerms).toContain('search');
+            expect(result.expandedTerms).toContain('function');
+        });
+        it('should expand "test" with related terms', () => {
+            const result = analyzer.analyze('test');
+            expect(result.expandedTerms).toContain('test');
+            expect(result.expandedTerms).toContain('spec');
+            expect(result.expandedTerms).toContain('describe');
+            expect(result.expandedTerms).toContain('it');
+            expect(result.expandedTerms).toContain('expect');
+        });
+        it('should expand "error" with related terms', () => {
+            const result = analyzer.analyze('error handling');
+            expect(result.expandedTerms).toContain('error');
+            expect(result.expandedTerms).toContain('exception');
+            expect(result.expandedTerms).toContain('throw');
+            expect(result.expandedTerms).toContain('catch');
+        });
+        it('should expand "config" with related terms', () => {
+            const result = analyzer.analyze('config setup');
+            expect(result.expandedTerms).toContain('config');
+            expect(result.expandedTerms).toContain('configuration');
+            expect(result.expandedTerms).toContain('settings');
+        });
+        it('should expand "auth" with related terms', () => {
+            const result = analyzer.analyze('auth flow');
+            expect(result.expandedTerms).toContain('auth');
+            expect(result.expandedTerms).toContain('authentication');
+            expect(result.expandedTerms).toContain('login');
+        });
+        it('should expand "database" with related terms', () => {
+            const result = analyzer.analyze('database operations');
+            expect(result.expandedTerms).toContain('database');
+            expect(result.expandedTerms).toContain('db');
+            expect(result.expandedTerms).toContain('query');
+        });
+        it('should not duplicate terms during expansion', () => {
+            const result = analyzer.analyze('test spec');
+            const testCount = result.expandedTerms.filter((t) => t === 'test').length;
+            expect(testCount).toBe(1);
+        });
+        it('should preserve non-expandable terms', () => {
+            const result = analyzer.analyze('foobar baz');
+            expect(result.expandedTerms).toContain('foobar');
+            expect(result.expandedTerms).toContain('baz');
+        });
+    });
+});
+//# sourceMappingURL=query-analyzer.test.js.map

package/dist/retrieval/query-analyzer.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-analyzer.test.js","sourceRoot":"","sources":["../../src/retrieval/query-analyzer.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAGpD,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,MAAM,QAAQ,GAAG,IAAI,aAAa,EAAE,CAAC;IAErC,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;QACjD,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAE/C,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,eAAe,EAAE,aAAa,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YAC/C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,KAAK,GAAG,0BAA0B,CAAC;YACzC,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,MAAM,WAAW,GAAoD;YACnE,kBAAkB;YAClB,EAAE,KAAK,EAAE,0BAA0B,EAAE,QAAQ,EAAE,iBAAiB,EAAE;YAClE,EAAE,KAAK,EAAE,2BAA2B,EAAE,QAAQ,EAAE,iBAAiB,EAAE;YACnE,EAAE,KAAK,EAAE,iCAAiC,EAAE,QAAQ,EAAE,iBAAiB,EAAE;YACzE,EAAE,KAAK,EAAE,wBAAwB,EAAE,QAAQ,EAAE,iBAAiB,EAAE;YAChE,EAAE,KAAK,EAAE,2BAA2B,EAAE,QAAQ,EAAE,iBAAiB,EAAE;YACnE,EAAE,KAAK,EAAE,6BAA6B,EAAE,QAAQ,EAAE,iBAAiB,EAAE;YAErE,aAAa;YACb,EAAE,KAAK,EAAE,uBAAuB,EAAE,QAAQ,EAAE,YAAY,EAAE;YAC1D,EAAE,KAAK,EAAE,0BAA0B,EAAE,QAAQ,EAAE,YAAY,EAAE;YAC7D,EAAE,KAAK,EAAE,uBAAuB,EAAE,QAAQ,EAAE,YAAY,EAAE;YAC1D,EAAE,KAAK,EAAE,4BAA4B,EAAE,QAAQ,EAAE,YAAY,EAAE;YAC/D,EAAE,KAAK,EAAE,uBAAuB,EAAE,QAAQ,EAAE,YAAY,EAAE;YAC1D,EAAE,KAAK,EAAE,sBAAsB,EAAE,QAAQ,EAAE,YAAY,EAAE;YACzD,EAAE,KAAK,EAAE,6BAA6B,EAAE,QAAQ,EAAE,YAAY,EAAE;YAChE,EAAE,KAAK,EAAE,uBAAuB,EAAE,QAAQ,EAAE,YAAY,EAAE;YAE1D,oBAAoB;YACpB,EAAE,KAAK,EAAE,4BAA4B,EAAE,QAAQ,EAAE,mBAAmB,EAAE;YACtE,EAAE,KAAK,EAAE,4BAA4B,EAAE,QAAQ,EAAE,mBAAmB,EAAE;YACtE,EAAE,KAAK,EAAE,0BAA0B,EAAE,QAAQ,EAAE,mBAAmB,EAAE;YACpE,EAAE,KAAK,EAAE,6BAA6B,EAAE,QAAQ,EAAE,mBAAmB,EAAE;YACvE,EAAE,KAAK,EAAE,kCAAkC,EAAE,QAAQ,EAAE,mBAAmB,EAAE;YAC5E,EAAE,KAAK,EAAE,sBAAsB,EAAE,QAAQ,EAAE,mBAAmB,EAAE;YAChE,EAAE,KAAK,EAAE,8BAA8B,EAAE,QAAQ,EAAE,mBAAmB,EAAE;YAExE,eAAe;YACf,EAAE,KAAK,EAAE,yBAAyB,EAAE,QAAQ,EAAE,cAAc,EAAE;YAC9D,EAAE,KAAK,EAAE,4BAA4B,EAAE,QAAQ,EAAE,cAAc,EAAE;YACjE,EAAE,KAAK,EAAE,yBAAyB,EAAE,QAAQ,EAAE,cAAc,EAAE;YAC9D,EAAE,KAAK,EAAE,8BAA8B,EAAE,QAAQ,EAAE,cAAc,EAAE;YAEnE,UAAU;YACV,EAAE,KAAK,EAAE,iBAAiB,EAAE,QAAQ,EAAE,SAAS,EAAE;YACjD,EAAE,KAAK,EAAE,2BAA2B,EAAE,QAAQ,EAAE,SAAS,EAAE;YAC3D,EAAE,KAAK,EAAE,oBAAoB,EAAE,QAAQ,EAAE,SAAS,EAAE;SACrD,CAAC;QAEF,KAAK,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,WAAW,EAAE,CAAC;YAC9C,EAAE,CAAC,kBAAkB,QAAQ,WAAW,KAAK,GAAG,EAAE,GAAG,EAAE;gBACrD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACvC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;YACxD,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;YACxE,MAAM,CAAC,aAAa,CAAC,CAAC,cAAc,CAAC;gBACnC,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,iBAAiB;aACzB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,8BAA8B,CAAC,CAAC;YAChE,MAAM,YAAY,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;YAC1E,MAAM,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC;gBAClC,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,aAAa;aACrB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,uCAAuC,CAAC,CAAC;YACzE,MAAM,YAAY,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;YACtE,MAAM,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC;gBAClC,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,+BAA+B;aACvC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACrD,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACtC,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;YACvD,MAAM,eAAe,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAC5C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,SAAS,CAC7B,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,4CAA4C,CAAC,CAAC;YAC9E,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;YACxD,+DAA+D;YAC/D,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,uCAAuC,CAAC,CAAC;YACzE,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,SAAS,CACjD,+BAA+B,CAChC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;YACxE,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,aAAa,EAAE,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,aAAa,EAAE,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,aAAa,EAAE,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,+BAA+B,CAAC,CAAC;YACjE,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,SAAS,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,6BAA6B,CAAC,CAAC;YAC/D,2DAA2D;YAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;YACxE,MAAM,CAAC,aAAa,CAAC,CAAC,cAAc,CAAC;gBACnC,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,SAAS;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,SAAS,GAAG,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,MAAM,CAAC,CAAC,MAAM,CAAC;YAC1E,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/retrieval/token-budget.d.ts

@@ -0,0 +1,51 @@
+import type { SearchResult } from '../types/index.js';
+import type { ExpandedContext, RelatedChunk } from './context-expander.js';
+export interface TokenBudgetConfig {
+    /** Maximum total tokens for assembled context. Default 8000. */
+    maxTokens: number;
+    /** Tokens reserved for the LLM response. Default 2000. */
+    reserveForAnswer: number;
+    /** Fraction of available budget for primary results. Default 0.6. */
+    primaryWeight: number;
+    /** Fraction of available budget for related context. Default 0.3. */
+    relatedWeight: number;
+    /** Fraction of available budget for graph excerpt. Default 0.1. */
+    graphWeight: number;
+}
+export interface AssembledContext {
+    /** Final assembled context string with formatted sections. */
+    content: string;
+    /** Primary results that fit within the budget. */
+    primaryChunks: SearchResult[];
+    /** Related chunks that fit within the budget. */
+    relatedChunks: RelatedChunk[];
+    /** Estimated total token count of the assembled content. */
+    tokenCount: number;
+    /** Whether results were truncated to fit the budget. */
+    truncated: boolean;
+}
+export declare class TokenBudgetOptimizer {
+    private readonly config;
+    constructor(config?: Partial<TokenBudgetConfig>);
+    /**
+     * Assemble an ExpandedContext into a token-budgeted context string.
+     *
+     * Pipeline:
+     * 1. Calculate available budget: maxTokens - reserveForAnswer
+     * 2. Allocate budget: primary * primaryWeight, related * relatedWeight, graph * graphWeight
+     * 3. Fill primary results first (highest score first)
+     * 4. Fill related chunks (closest relationship first)
+     * 5. Add graph excerpt if budget remains
+     * 6. Build formatted context string
+     */
+    assemble(expanded: ExpandedContext): AssembledContext;
+    /** Estimate token count using the text.length / 4 approximation. */
+    estimateTokens(text: string): number;
+    /**
+     * Fill a budget with items, returning what fits.
+     * Uses the formatted version of each item to estimate tokens.
+     */
+    private fillBudget;
+    /** Format the graph excerpt as a simplified text representation. */
+    private formatGraphExcerpt;
+}

package/dist/retrieval/token-budget.js

@@ -0,0 +1,141 @@
+const DEFAULT_CONFIG = {
+    maxTokens: 8000,
+    reserveForAnswer: 2000,
+    primaryWeight: 0.6,
+    relatedWeight: 0.3,
+    graphWeight: 0.1,
+};
+export class TokenBudgetOptimizer {
+    config;
+    constructor(config) {
+        this.config = { ...DEFAULT_CONFIG, ...config };
+    }
+    /**
+     * Assemble an ExpandedContext into a token-budgeted context string.
+     *
+     * Pipeline:
+     * 1. Calculate available budget: maxTokens - reserveForAnswer
+     * 2. Allocate budget: primary * primaryWeight, related * relatedWeight, graph * graphWeight
+     * 3. Fill primary results first (highest score first)
+     * 4. Fill related chunks (closest relationship first)
+     * 5. Add graph excerpt if budget remains
+     * 6. Build formatted context string
+     */
+    assemble(expanded) {
+        const availableBudget = Math.max(0, this.config.maxTokens - this.config.reserveForAnswer);
+        const primaryBudget = Math.floor(availableBudget * this.config.primaryWeight);
+        const relatedBudget = Math.floor(availableBudget * this.config.relatedWeight);
+        const graphBudget = Math.floor(availableBudget * this.config.graphWeight);
+        // Step 1: Fill primary results (sorted by score descending)
+        const sortedPrimary = [...expanded.primaryResults].sort((a, b) => b.score - a.score);
+        const { items: includedPrimary, tokensUsed: primaryTokensUsed } = this.fillBudget(sortedPrimary, primaryBudget, formatPrimaryChunk);
+        // Step 2: Fill related chunks (sorted by distance ascending)
+        const sortedRelated = [...expanded.relatedChunks].sort((a, b) => a.distance - b.distance);
+        const { items: includedRelated, tokensUsed: relatedTokensUsed } = this.fillBudget(sortedRelated, relatedBudget, formatRelatedChunk);
+        // Step 3: Build graph excerpt string
+        const graphString = this.formatGraphExcerpt(expanded.graphExcerpt);
+        const graphTokens = this.estimateTokens(graphString);
+        const includedGraph = graphTokens <= graphBudget ? graphString : '';
+        const graphTokensUsed = includedGraph ? graphTokens : 0;
+        // Step 4: Build formatted context string
+        const sections = [];
+        if (includedPrimary.length > 0) {
+            sections.push('## Primary Results\n');
+            for (const chunk of includedPrimary) {
+                sections.push(formatPrimaryChunk(chunk));
+            }
+        }
+        if (includedRelated.length > 0) {
+            sections.push('## Related Context\n');
+            for (const related of includedRelated) {
+                sections.push(formatRelatedChunk(related));
+            }
+        }
+        if (includedGraph) {
+            sections.push('## Dependency Graph\n');
+            sections.push(includedGraph);
+        }
+        const content = sections.join('\n');
+        const totalTokens = primaryTokensUsed + relatedTokensUsed + graphTokensUsed;
+        const truncated = includedPrimary.length < expanded.primaryResults.length ||
+            includedRelated.length < expanded.relatedChunks.length ||
+            (!includedGraph &&
+                expanded.graphExcerpt.nodes.length > 0 &&
+                graphTokens > graphBudget);
+        return {
+            content,
+            primaryChunks: includedPrimary,
+            relatedChunks: includedRelated,
+            tokenCount: totalTokens,
+            truncated,
+        };
+    }
+    /** Estimate token count using the text.length / 4 approximation. */
+    estimateTokens(text) {
+        return Math.ceil(text.length / 4);
+    }
+    /**
+     * Fill a budget with items, returning what fits.
+     * Uses the formatted version of each item to estimate tokens.
+     */
+    fillBudget(items, budget, formatter) {
+        const included = [];
+        let tokensUsed = 0;
+        for (const item of items) {
+            const formatted = formatter(item);
+            const tokens = this.estimateTokens(formatted);
+            if (tokensUsed + tokens > budget) {
+                break;
+            }
+            included.push(item);
+            tokensUsed += tokens;
+        }
+        return { items: included, tokensUsed };
+    }
+    /** Format the graph excerpt as a simplified text representation. */
+    formatGraphExcerpt(excerpt) {
+        if (excerpt.nodes.length === 0)
+            return '';
+        const lines = [];
+        lines.push(`Nodes: ${excerpt.nodes.join(', ')}`);
+        for (const edge of excerpt.edges) {
+            lines.push(`${edge.from} --[${edge.type}]--> ${edge.to}`);
+        }
+        return lines.join('\n');
+    }
+}
+/** Format a primary search result chunk for context output. */
+function formatPrimaryChunk(result) {
+    const header = result.metadata?.name
+        ? `### ${result.metadata.name} (${result.metadata.chunkType})`
+        : '### (unnamed chunk)';
+    const filePath = result.chunk?.filePath ?? '';
+    const lines = result.chunk?.startLine !== undefined && result.chunk?.endLine !== undefined
+        ? ` [L${result.chunk.startLine}-${result.chunk.endLine}]`
+        : '';
+    const locationLine = filePath ? `File: ${filePath}${lines}` : '';
+    const parts = [header];
+    if (locationLine)
+        parts.push(locationLine);
+    if (result.nlSummary)
+        parts.push(result.nlSummary);
+    parts.push('```');
+    parts.push(result.content);
+    parts.push('```');
+    return parts.join('\n');
+}
+/** Format a related chunk for context output. */
+function formatRelatedChunk(related) {
+    const label = `[${related.relationship}, distance=${related.distance}]`;
+    const name = related.chunk.metadata?.name ?? 'unknown';
+    const filePath = related.chunk.chunk?.filePath ?? '';
+    const parts = [`### ${name} ${label}`];
+    if (filePath)
+        parts.push(`File: ${filePath}`);
+    if (related.chunk.nlSummary)
+        parts.push(related.chunk.nlSummary);
+    parts.push('```');
+    parts.push(related.chunk.content);
+    parts.push('```');
+    return parts.join('\n');
+}

package/dist/retrieval/token-budget.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"token-budget.js","sourceRoot":"","sources":["../../src/retrieval/token-budget.ts"],"names":[],"mappings":"AA6BA,MAAM,cAAc,GAAsB;IACxC,SAAS,EAAE,IAAI;IACf,gBAAgB,EAAE,IAAI;IACtB,aAAa,EAAE,GAAG;IAClB,aAAa,EAAE,GAAG;IAClB,WAAW,EAAE,GAAG;CACjB,CAAC;AAEF,MAAM,OAAO,oBAAoB;IACd,MAAM,CAAoB;IAE3C,YAAY,MAAmC;QAC7C,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;IACjD,CAAC;IAED;;;;;;;;;;OAUG;IACH,QAAQ,CAAC,QAAyB;QAChC,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAC1F,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAC9E,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAC9E,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAE1E,4DAA4D;QAC5D,MAAM,aAAa,GAAG,CAAC,GAAG,QAAQ,CAAC,cAAc,CAAC,CAAC,IAAI,CACrD,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAC5B,CAAC;QACF,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,iBAAiB,EAAE,GAC7D,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,aAAa,EAAE,kBAAkB,CAAC,CAAC;QAEpE,6DAA6D;QAC7D,MAAM,aAAa,GAAG,CAAC,GAAG,QAAQ,CAAC,aAAa,CAAC,CAAC,IAAI,CACpD,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAClC,CAAC;QACF,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,iBAAiB,EAAE,GAC7D,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,aAAa,EAAE,kBAAkB,CAAC,CAAC;QAEpE,qCAAqC;QACrC,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QACnE,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QACrD,MAAM,aAAa,GAAG,WAAW,IAAI,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;QACpE,MAAM,eAAe,GAAG,aAAa,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;QAExD,yCAAyC;QACzC,MAAM,QAAQ,GAAa,EAAE,CAAC;QAE9B,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,QAAQ,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YACtC,KAAK,MAAM,KAAK,IAAI,eAAe,EAAE,CAAC;gBACpC,QAAQ,CAAC,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;QAED,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,QAAQ,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YACtC,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;gBACtC,QAAQ,CAAC,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QAED,IAAI,aAAa,EAAE,CAAC;YAClB,QAAQ,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;YACvC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC/B,CAAC;QAED,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,WAAW,GAAG,iBAAiB,GAAG,iBAAiB,GAAG,eAAe,CAAC;QAE5E,MAAM,SAAS,GACb,eAAe,CAAC,MAAM,GAAG,QAAQ,CAAC,cAAc,CAAC,MAAM;YACvD,eAAe,CAAC,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,MAAM;YACtD,CAAC,CAAC,aAAa;gBACb,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC;gBACtC,WAAW,GAAG,WAAW,CAAC,CAAC;QAE/B,OAAO;YACL,OAAO;YACP,aAAa,EAAE,eAAe;YAC9B,aAAa,EAAE,eAAe;YAC9B,UAAU,EAAE,WAAW;YACvB,SAAS;SACV,CAAC;IACJ,CAAC;IAED,oEAAoE;IACpE,cAAc,CAAC,IAAY;QACzB,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACpC,CAAC;IAED;;;OAGG;IACK,UAAU,CAChB,KAAU,EACV,MAAc,EACd,SAA8B;QAE9B,MAAM,QAAQ,GAAQ,EAAE,CAAC;QACzB,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;YAE9C,IAAI,UAAU,GAAG,MAAM,GAAG,MAAM,EAAE,CAAC;gBACjC,MAAM;YACR,CAAC;YAED,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpB,UAAU,IAAI,MAAM,CAAC;QACvB,CAAC;QAED,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC;IACzC,CAAC;IAED,oEAAoE;IAC5D,kBAAkB,CAAC,OAG1B;QACC,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAE1C,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,CAAC,IAAI,CAAC,UAAU,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEjD,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YACjC,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,OAAO,IAAI,CAAC,IAAI,QAAQ,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5D,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;CACF;AAED,+DAA+D;AAC/D,SAAS,kBAAkB,CAAC,MAAoB;IAC9C,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,EAAE,IAAI;QAClC,CAAC,CAAC,OAAO,MAAM,CAAC,QAAQ,CAAC,IAAI,KAAK,MAAM,CAAC,QAAQ,CAAC,SAAS,GAAG;QAC9D,CAAC,CAAC,qBAAqB,CAAC;IAE1B,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,EAAE,QAAQ,IAAI,EAAE,CAAC;IAC9C,MAAM,KAAK,GACT,MAAM,CAAC,KAAK,EAAE,SAAS,KAAK,SAAS,IAAI,MAAM,CAAC,KAAK,EAAE,OAAO,KAAK,SAAS;QAC1E,CAAC,CAAC,MAAM,MAAM,CAAC,KAAK,CAAC,SAAS,IAAI,MAAM,CAAC,KAAK,CAAC,OAAO,GAAG;QACzD,CAAC,CAAC,EAAE,CAAC;IAET,MAAM,YAAY,GAAG,QAAQ,CAAC,CAAC,CAAC,SAAS,QAAQ,GAAG,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAEjE,MAAM,KAAK,GAAG,CAAC,MAAM,CAAC,CAAC;IACvB,IAAI,YAAY;QAAE,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC3C,IAAI,MAAM,CAAC,SAAS;QAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACnD,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClB,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC3B,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAElB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,iDAAiD;AACjD,SAAS,kBAAkB,CAAC,OAAqB;IAC/C,MAAM,KAAK,GAAG,IAAI,OAAO,CAAC,YAAY,cAAc,OAAO,CAAC,QAAQ,GAAG,CAAC;IACxE,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,IAAI,SAAS,CAAC;IACvD,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,IAAI,EAAE,CAAC;IAErD,MAAM,KAAK,GAAG,CAAC,OAAO,IAAI,IAAI,KAAK,EAAE,CAAC,CAAC;IACvC,IAAI,QAAQ;QAAE,KAAK,CAAC,IAAI,CAAC,SAAS,QAAQ,EAAE,CAAC,CAAC;IAC9C,IAAI,OAAO,CAAC,KAAK,CAAC,SAAS;QAAE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IACjE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClB,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAClC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAElB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}

package/dist/retrieval/token-budget.test.d.ts

@@ -0,0 +1 @@
+export {};