@code-rag/core 0.1.0

package/dist/retrieval/context-expander.test.js

@@ -0,0 +1,339 @@
+import { describe, it, expect } from 'vitest';
+import { ContextExpander } from './context-expander.js';
+import { DependencyGraph } from '../graph/dependency-graph.js';
+function makeNode(id, overrides = {}) {
+    return {
+        id,
+        filePath: `src/${id}.ts`,
+        symbols: [],
+        type: 'module',
+        ...overrides,
+    };
+}
+function makeEdge(source, target, type = 'imports') {
+    return { source, target, type };
+}
+function makeSearchResult(chunkId, overrides = {}) {
+    return {
+        chunkId,
+        content: `content of ${chunkId}`,
+        nlSummary: `summary of ${chunkId}`,
+        score: 0.9,
+        method: 'hybrid',
+        metadata: {
+            chunkType: 'function',
+            name: chunkId,
+            declarations: [],
+            imports: [],
+            exports: [],
+        },
+        ...overrides,
+    };
+}
+function buildTestGraph() {
+    const graph = new DependencyGraph();
+    // A imports B, B imports C, D imports A
+    graph.addNode(makeNode('A'));
+    graph.addNode(makeNode('B'));
+    graph.addNode(makeNode('C'));
+    graph.addNode(makeNode('D'));
+    graph.addEdge(makeEdge('A', 'B'));
+    graph.addEdge(makeEdge('B', 'C'));
+    graph.addEdge(makeEdge('D', 'A'));
+    return graph;
+}
+describe('ContextExpander', () => {
+    describe('basic expansion', () => {
+        it('should return primary results unchanged', () => {
+            const graph = buildTestGraph();
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            lookupMap.set('C', makeSearchResult('C'));
+            lookupMap.set('D', makeSearchResult('D'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const primary = [makeSearchResult('A')];
+            const result = expander.expand(primary);
+            expect(result.primaryResults).toEqual(primary);
+            expect(result.primaryResults).toHaveLength(1);
+        });
+        it('should find related chunks via BFS traversal', () => {
+            const graph = buildTestGraph();
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            lookupMap.set('C', makeSearchResult('C'));
+            lookupMap.set('D', makeSearchResult('D'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const primary = [makeSearchResult('A')];
+            const result = expander.expand(primary);
+            // A -> B (depth 1), B -> C (depth 2), D -> A (depth 1)
+            const relatedIds = result.relatedChunks.map((rc) => rc.chunk.chunkId);
+            expect(relatedIds).toContain('B');
+            expect(relatedIds).toContain('C');
+            expect(relatedIds).toContain('D');
+        });
+        it('should not include primary results in related chunks', () => {
+            const graph = buildTestGraph();
+            const lookupMap = new Map();
+            lookupMap.set('A', makeSearchResult('A'));
+            lookupMap.set('B', makeSearchResult('B'));
+            lookupMap.set('C', makeSearchResult('C'));
+            lookupMap.set('D', makeSearchResult('D'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            // A and B are both primary results
+            const primary = [makeSearchResult('A'), makeSearchResult('B')];
+            const result = expander.expand(primary);
+            const relatedIds = result.relatedChunks.map((rc) => rc.chunk.chunkId);
+            expect(relatedIds).not.toContain('A');
+            expect(relatedIds).not.toContain('B');
+        });
+        it('should deduplicate related chunks', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A'));
+            graph.addNode(makeNode('B'));
+            graph.addNode(makeNode('C'));
+            graph.addEdge(makeEdge('A', 'C'));
+            graph.addEdge(makeEdge('B', 'C'));
+            const lookupMap = new Map();
+            lookupMap.set('C', makeSearchResult('C'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            // Both A and B connect to C; C should appear only once
+            const primary = [makeSearchResult('A'), makeSearchResult('B')];
+            const result = expander.expand(primary);
+            const cChunks = result.relatedChunks.filter((rc) => rc.chunk.chunkId === 'C');
+            expect(cChunks).toHaveLength(1);
+        });
+        it('should skip related nodes that have no chunk lookup', () => {
+            const graph = buildTestGraph();
+            // Only provide lookup for B, not C or D
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const primary = [makeSearchResult('A')];
+            const result = expander.expand(primary);
+            const relatedIds = result.relatedChunks.map((rc) => rc.chunk.chunkId);
+            expect(relatedIds).toContain('B');
+            expect(relatedIds).not.toContain('C');
+            expect(relatedIds).not.toContain('D');
+        });
+    });
+    describe('maxRelated limiting', () => {
+        it('should respect maxRelated parameter', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A'));
+            const lookupMap = new Map();
+            for (let i = 0; i < 20; i++) {
+                const id = `N${i}`;
+                graph.addNode(makeNode(id));
+                graph.addEdge(makeEdge('A', id));
+                lookupMap.set(id, makeSearchResult(id));
+            }
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const primary = [makeSearchResult('A')];
+            const result = expander.expand(primary, 5);
+            expect(result.relatedChunks.length).toBeLessThanOrEqual(5);
+        });
+        it('should default maxRelated to 10', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A'));
+            const lookupMap = new Map();
+            for (let i = 0; i < 20; i++) {
+                const id = `N${i}`;
+                graph.addNode(makeNode(id));
+                graph.addEdge(makeEdge('A', id));
+                lookupMap.set(id, makeSearchResult(id));
+            }
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const primary = [makeSearchResult('A')];
+            const result = expander.expand(primary);
+            expect(result.relatedChunks.length).toBeLessThanOrEqual(10);
+        });
+    });
+    describe('relationship classification', () => {
+        it('should classify outgoing edge as "imports"', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A'));
+            graph.addNode(makeNode('B'));
+            graph.addEdge(makeEdge('A', 'B', 'imports'));
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const result = expander.expand([makeSearchResult('A')]);
+            const bRelated = result.relatedChunks.find((rc) => rc.chunk.chunkId === 'B');
+            expect(bRelated?.relationship).toBe('imports');
+        });
+        it('should classify incoming edge as "imported_by"', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A'));
+            graph.addNode(makeNode('B'));
+            graph.addEdge(makeEdge('B', 'A', 'imports'));
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const result = expander.expand([makeSearchResult('A')]);
+            const bRelated = result.relatedChunks.find((rc) => rc.chunk.chunkId === 'B');
+            expect(bRelated?.relationship).toBe('imported_by');
+        });
+        it('should classify test files as "test_for"', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A', { filePath: 'src/parser.ts' }));
+            graph.addNode(makeNode('B', { filePath: 'src/parser.test.ts' }));
+            graph.addEdge(makeEdge('B', 'A'));
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const result = expander.expand([makeSearchResult('A')]);
+            const bRelated = result.relatedChunks.find((rc) => rc.chunk.chunkId === 'B');
+            expect(bRelated?.relationship).toBe('test_for');
+        });
+        it('should classify spec files as "test_for"', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A', { filePath: 'src/parser.ts' }));
+            graph.addNode(makeNode('B', { filePath: 'src/parser.spec.ts' }));
+            graph.addEdge(makeEdge('B', 'A'));
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const result = expander.expand([makeSearchResult('A')]);
+            const bRelated = result.relatedChunks.find((rc) => rc.chunk.chunkId === 'B');
+            expect(bRelated?.relationship).toBe('test_for');
+        });
+        it('should classify implements edge as "interface_of"', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A'));
+            graph.addNode(makeNode('B'));
+            graph.addEdge(makeEdge('A', 'B', 'implements'));
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const result = expander.expand([makeSearchResult('A')]);
+            const bRelated = result.relatedChunks.find((rc) => rc.chunk.chunkId === 'B');
+            expect(bRelated?.relationship).toBe('interface_of');
+        });
+        it('should classify same-directory nodes as "sibling"', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A', { filePath: 'src/graph/foo.ts' }));
+            graph.addNode(makeNode('B', { filePath: 'src/graph/bar.ts' }));
+            // Connected but no direct edge between them — through a shared node
+            graph.addNode(makeNode('C', { filePath: 'src/other/baz.ts' }));
+            graph.addEdge(makeEdge('A', 'C'));
+            graph.addEdge(makeEdge('B', 'C'));
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const result = expander.expand([makeSearchResult('A')]);
+            const bRelated = result.relatedChunks.find((rc) => rc.chunk.chunkId === 'B');
+            // B is reachable via A -> C <- B (depth 2), and they share src/graph/
+            expect(bRelated?.relationship).toBe('sibling');
+        });
+    });
+    describe('distance computation', () => {
+        it('should assign distance 1 to direct neighbors', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A'));
+            graph.addNode(makeNode('B'));
+            graph.addEdge(makeEdge('A', 'B'));
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const result = expander.expand([makeSearchResult('A')]);
+            const bRelated = result.relatedChunks.find((rc) => rc.chunk.chunkId === 'B');
+            expect(bRelated?.distance).toBe(1);
+        });
+        it('should assign distance 2 to second-hop neighbors', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A'));
+            graph.addNode(makeNode('B'));
+            graph.addNode(makeNode('C'));
+            graph.addEdge(makeEdge('A', 'B'));
+            graph.addEdge(makeEdge('B', 'C'));
+            const lookupMap = new Map();
+            lookupMap.set('C', makeSearchResult('C'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const result = expander.expand([makeSearchResult('A')]);
+            const cRelated = result.relatedChunks.find((rc) => rc.chunk.chunkId === 'C');
+            expect(cRelated?.distance).toBe(2);
+        });
+        it('should sort related chunks by distance ascending', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A'));
+            graph.addNode(makeNode('B'));
+            graph.addNode(makeNode('C'));
+            graph.addEdge(makeEdge('A', 'B'));
+            graph.addEdge(makeEdge('B', 'C'));
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            lookupMap.set('C', makeSearchResult('C'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const result = expander.expand([makeSearchResult('A')]);
+            expect(result.relatedChunks.length).toBeGreaterThanOrEqual(2);
+            for (let i = 1; i < result.relatedChunks.length; i++) {
+                const prev = result.relatedChunks[i - 1];
+                const curr = result.relatedChunks[i];
+                expect(prev.distance).toBeLessThanOrEqual(curr.distance);
+            }
+        });
+    });
+    describe('graph excerpt', () => {
+        it('should include nodes from primary and related results', () => {
+            const graph = buildTestGraph();
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            lookupMap.set('C', makeSearchResult('C'));
+            lookupMap.set('D', makeSearchResult('D'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const result = expander.expand([makeSearchResult('A')]);
+            expect(result.graphExcerpt.nodes).toContain('A');
+            expect(result.graphExcerpt.nodes).toContain('B');
+        });
+        it('should include edges between relevant nodes', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A'));
+            graph.addNode(makeNode('B'));
+            graph.addEdge(makeEdge('A', 'B'));
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            const result = expander.expand([makeSearchResult('A')]);
+            expect(result.graphExcerpt.edges).toContainEqual({
+                from: 'A',
+                to: 'B',
+                type: 'imports',
+            });
+        });
+        it('should deduplicate graph edges', () => {
+            const graph = new DependencyGraph();
+            graph.addNode(makeNode('A'));
+            graph.addNode(makeNode('B'));
+            graph.addEdge(makeEdge('A', 'B'));
+            const lookupMap = new Map();
+            lookupMap.set('B', makeSearchResult('B'));
+            const expander = new ContextExpander(graph, (id) => lookupMap.get(id));
+            // Pass A twice as primary to trigger duplicate edge collection
+            const result = expander.expand([
+                makeSearchResult('A'),
+                makeSearchResult('A', { chunkId: 'A' }),
+            ]);
+            const abEdges = result.graphExcerpt.edges.filter((e) => e.from === 'A' && e.to === 'B');
+            expect(abEdges).toHaveLength(1);
+        });
+    });
+    describe('empty inputs', () => {
+        it('should handle empty results array', () => {
+            const graph = new DependencyGraph();
+            const expander = new ContextExpander(graph, () => undefined);
+            const result = expander.expand([]);
+            expect(result.primaryResults).toHaveLength(0);
+            expect(result.relatedChunks).toHaveLength(0);
+            expect(result.graphExcerpt.nodes).toHaveLength(0);
+            expect(result.graphExcerpt.edges).toHaveLength(0);
+        });
+        it('should handle results with no graph nodes', () => {
+            const graph = new DependencyGraph();
+            const expander = new ContextExpander(graph, () => undefined);
+            const result = expander.expand([makeSearchResult('X')]);
+            expect(result.primaryResults).toHaveLength(1);
+            expect(result.relatedChunks).toHaveLength(0);
+        });
+    });
+});
+//# sourceMappingURL=context-expander.test.js.map

package/dist/retrieval/context-expander.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-expander.test.js","sourceRoot":"","sources":["../../src/retrieval/context-expander.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAI/D,SAAS,QAAQ,CAAC,EAAU,EAAE,YAAgC,EAAE;IAC9D,OAAO;QACL,EAAE;QACF,QAAQ,EAAE,OAAO,EAAE,KAAK;QACxB,OAAO,EAAE,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CACf,MAAc,EACd,MAAc,EACd,OAA0B,SAAS;IAEnC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;AAClC,CAAC;AAED,SAAS,gBAAgB,CACvB,OAAe,EACf,YAAmC,EAAE;IAErC,OAAO;QACL,OAAO;QACP,OAAO,EAAE,cAAc,OAAO,EAAE;QAChC,SAAS,EAAE,cAAc,OAAO,EAAE;QAClC,KAAK,EAAE,GAAG;QACV,MAAM,EAAE,QAAQ;QAChB,QAAQ,EAAE;YACR,SAAS,EAAE,UAAU;YACrB,IAAI,EAAE,OAAO;YACb,YAAY,EAAE,EAAE;YAChB,OAAO,EAAE,EAAE;YACX,OAAO,EAAE,EAAE;SACZ;QACD,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,cAAc;IACrB,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;IACpC,wCAAwC;IACxC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;IAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;IAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;IAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;IAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IAClC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IAClC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IAClC,OAAO,KAAK,CAAC;AACf,CAAC;AAED,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,KAAK,GAAG,cAAc,EAAE,CAAC;YAC/B,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1C,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1C,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,OAAO,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YACxC,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAExC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,KAAK,GAAG,cAAc,EAAE,CAAC;YAC/B,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1C,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1C,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,OAAO,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YACxC,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAExC,uDAAuD;YACvD,MAAM,UAAU,GAAG,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACtE,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YAClC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YAClC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,KAAK,GAAG,cAAc,EAAE,CAAC;YAC/B,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1C,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1C,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1C,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,mCAAmC;YACnC,MAAM,OAAO,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC/D,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAExC,MAAM,UAAU,GAAG,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACtE,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAClC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAElC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,uDAAuD;YACvD,MAAM,OAAO,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC/D,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAExC,MAAM,OAAO,GAAG,MAAM,CAAC,aAAa,CAAC,MAAM,CACzC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,GAAG,CACjC,CAAC;YACF,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,KAAK,GAAG,cAAc,EAAE,CAAC;YAC/B,wCAAwC;YACxC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,OAAO,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YACxC,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAExC,MAAM,UAAU,GAAG,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACtE,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YAClC,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAElD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC;gBACnB,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;gBAC5B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;gBACjC,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,gBAAgB,CAAC,EAAE,CAAC,CAAC,CAAC;YAC1C,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,OAAO,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YACxC,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAE3C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAElD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC;gBACnB,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;gBAC5B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;gBACjC,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,gBAAgB,CAAC,EAAE,CAAC,CAAC,CAAC;YAC1C,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,OAAO,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YACxC,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAExC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;QAC3C,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,SAAS,CAAC,CAAC,CAAC;YAE7C,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAExD,MAAM,QAAQ,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,CACxC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,GAAG,CACjC,CAAC;YACF,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,SAAS,CAAC,CAAC,CAAC;YAE7C,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAExD,MAAM,QAAQ,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,CACxC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,GAAG,CACjC,CAAC;YACF,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC;YAC5D,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,oBAAoB,EAAE,CAAC,CAAC,CAAC;YACjE,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAElC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAExD,MAAM,QAAQ,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,CACxC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,GAAG,CACjC,CAAC;YACF,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC;YAC5D,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,oBAAoB,EAAE,CAAC,CAAC,CAAC;YACjE,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAElC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAExD,MAAM,QAAQ,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,CACxC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,GAAG,CACjC,CAAC;YACF,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,CAAC;YAEhD,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAExD,MAAM,QAAQ,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,CACxC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,GAAG,CACjC,CAAC;YACF,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;YAC/D,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;YAC/D,oEAAoE;YACpE,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;YAC/D,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAClC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAElC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAExD,MAAM,QAAQ,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,CACxC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,GAAG,CACjC,CAAC;YACF,sEAAsE;YACtE,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAElC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAExD,MAAM,QAAQ,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,CACxC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,GAAG,CACjC,CAAC;YACF,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAClC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAElC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAExD,MAAM,QAAQ,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,CACxC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,KAAK,GAAG,CACjC,CAAC;YACF,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAClC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAElC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1C,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAExD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;YAC9D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACrD,MAAM,IAAI,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAE,CAAC;gBAC1C,MAAM,IAAI,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC,CAAE,CAAC;gBACtC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,KAAK,GAAG,cAAc,EAAE,CAAC;YAC/B,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1C,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1C,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAExD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAElC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAExD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC;gBAC/C,IAAI,EAAE,GAAG;gBACT,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,SAAS;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAElC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE1C,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,+DAA+D;YAC/D,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;gBAC7B,gBAAgB,CAAC,GAAG,CAAC;gBACrB,gBAAgB,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;aACxC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,MAAM,CAC9C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,GAAG,IAAI,CAAC,CAAC,EAAE,KAAK,GAAG,CACtC,CAAC;YACF,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAEnC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;YACpC,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAExD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/retrieval/cross-encoder-reranker.d.ts

@@ -0,0 +1,16 @@
+import { type Result } from 'neverthrow';
+import type { SearchResult } from '../types/search.js';
+import { ReRankerError, type ReRanker } from '../types/provider.js';
+export type { ReRanker };
+export { ReRankerError };
+export interface CrossEncoderConfig {
+    model: string;
+    baseUrl?: string;
+    topN: number;
+    timeout?: number;
+}
+export declare class CrossEncoderReRanker implements ReRanker {
+    private readonly config;
+    constructor(config: CrossEncoderConfig);
+    rerank(query: string, results: SearchResult[]): Promise<Result<SearchResult[], ReRankerError>>;
+}

package/dist/retrieval/cross-encoder-reranker.js

@@ -0,0 +1,90 @@
+import { ok, err } from 'neverthrow';
+import { ReRankerError } from '../types/provider.js';
+export { ReRankerError };
+const DEFAULT_BASE_URL = 'http://localhost:11434';
+const DEFAULT_TIMEOUT = 30_000;
+/** Neutral midpoint score assigned when scoring fails for a single result. */
+const DEFAULT_SCORE = 50;
+const MAX_QUERY_LENGTH = 500;
+const MAX_CONTENT_LENGTH = 2000;
+function buildScoringPrompt(query, result) {
+    const chunkType = result.metadata?.chunkType ?? 'unknown';
+    const name = result.metadata?.name ?? 'unnamed';
+    const truncatedQuery = query.length > MAX_QUERY_LENGTH
+        ? query.slice(0, MAX_QUERY_LENGTH) + '...'
+        : query;
+    const truncatedContent = result.content.length > MAX_CONTENT_LENGTH
+        ? result.content.slice(0, MAX_CONTENT_LENGTH) + '...'
+        : result.content;
+    return [
+        'Rate relevance 0-100 of this code to the query. Reply with ONLY the number.',
+        `<query>${truncatedQuery}</query>`,
+        `<code type="${chunkType}" name="${name}">`,
+        truncatedContent,
+        '</code>',
+        'Score:',
+    ].join('\n');
+}
+function parseScore(response) {
+    const match = response.match(/-?\d+/);
+    if (!match) {
+        return DEFAULT_SCORE;
+    }
+    const score = parseInt(match[0], 10);
+    return Math.max(0, Math.min(100, score));
+}
+export class CrossEncoderReRanker {
+    config;
+    constructor(config) {
+        this.config = {
+            model: config.model,
+            baseUrl: config.baseUrl ?? DEFAULT_BASE_URL,
+            topN: config.topN,
+            timeout: config.timeout ?? DEFAULT_TIMEOUT,
+        };
+    }
+    async rerank(query, results) {
+        if (results.length === 0) {
+            return ok([]);
+        }
+        const topN = Math.min(this.config.topN, results.length);
+        const toRerank = results.slice(0, topN);
+        const remaining = results.slice(topN);
+        const scored = [];
+        for (const result of toRerank) {
+            const prompt = buildScoringPrompt(query, result);
+            try {
+                const response = await globalThis.fetch(`${this.config.baseUrl}/api/generate`, {
+                    method: 'POST',
+                    headers: { 'Content-Type': 'application/json' },
+                    body: JSON.stringify({
+                        model: this.config.model,
+                        prompt,
+                        stream: false,
+                    }),
+                    signal: AbortSignal.timeout(this.config.timeout),
+                });
+                if (!response.ok) {
+                    scored.push({ result, score: DEFAULT_SCORE });
+                    continue;
+                }
+                const data = (await response.json());
+                const score = parseScore(data.response);
+                scored.push({ result, score });
+            }
+            catch (error) {
+                // If no results have been scored yet, Ollama is likely unreachable
+                if (scored.length === 0) {
+                    const message = error instanceof Error ? error.message : 'Unknown error';
+                    return err(new ReRankerError(`Ollama request failed: ${message}`));
+                }
+                // Transient error on a later call — assign default score and continue
+                scored.push({ result, score: DEFAULT_SCORE });
+            }
+        }
+        // Sort by score descending
+        scored.sort((a, b) => b.score - a.score);
+        const reranked = scored.map((s) => s.result);
+        return ok([...reranked, ...remaining]);
+    }
+}

package/dist/retrieval/cross-encoder-reranker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cross-encoder-reranker.js","sourceRoot":"","sources":["../../src/retrieval/cross-encoder-reranker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,GAAG,EAAe,MAAM,YAAY,CAAC;AAElD,OAAO,EAAE,aAAa,EAAiB,MAAM,sBAAsB,CAAC;AAGpE,OAAO,EAAE,aAAa,EAAE,CAAC;AAazB,MAAM,gBAAgB,GAAG,wBAAwB,CAAC;AAClD,MAAM,eAAe,GAAG,MAAM,CAAC;AAC/B,8EAA8E;AAC9E,MAAM,aAAa,GAAG,EAAE,CAAC;AACzB,MAAM,gBAAgB,GAAG,GAAG,CAAC;AAC7B,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAEhC,SAAS,kBAAkB,CAAC,KAAa,EAAE,MAAoB;IAC7D,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,EAAE,SAAS,IAAI,SAAS,CAAC;IAC1D,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,EAAE,IAAI,IAAI,SAAS,CAAC;IAChD,MAAM,cAAc,GAAG,KAAK,CAAC,MAAM,GAAG,gBAAgB;QACpD,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC,GAAG,KAAK;QAC1C,CAAC,CAAC,KAAK,CAAC;IACV,MAAM,gBAAgB,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,kBAAkB;QACjE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,kBAAkB,CAAC,GAAG,KAAK;QACrD,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;IACnB,OAAO;QACL,6EAA6E;QAC7E,UAAU,cAAc,UAAU;QAClC,eAAe,SAAS,WAAW,IAAI,IAAI;QAC3C,gBAAgB;QAChB,SAAS;QACT,QAAQ;KACT,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,SAAS,UAAU,CAAC,QAAgB;IAClC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACtC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,aAAa,CAAC;IACvB,CAAC;IACD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,EAAE,CAAC,CAAC;IACtC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,OAAO,oBAAoB;IACd,MAAM,CAA+B;IAEtD,YAAY,MAA0B;QACpC,IAAI,CAAC,MAAM,GAAG;YACZ,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,gBAAgB;YAC3C,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,eAAe;SAC3C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CACV,KAAa,EACb,OAAuB;QAEvB,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;QAChB,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEtC,MAAM,MAAM,GAAmD,EAAE,CAAC;QAElE,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC9B,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAEjD,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAC,KAAK,CACrC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,eAAe,EACrC;oBACE,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;oBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wBACnB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;wBACxB,MAAM;wBACN,MAAM,EAAE,KAAK;qBACd,CAAC;oBACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;iBACjD,CACF,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC;oBAC9C,SAAS;gBACX,CAAC;gBAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA2B,CAAC;gBAC/D,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACxC,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YACjC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,mEAAmE;gBACnE,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACxB,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;oBAC3D,OAAO,GAAG,CACR,IAAI,aAAa,CAAC,0BAA0B,OAAO,EAAE,CAAC,CACvD,CAAC;gBACJ,CAAC;gBACD,sEAAsE;gBACtE,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;QAED,2BAA2B;QAC3B,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAEzC,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;QAC7C,OAAO,EAAE,CAAC,CAAC,GAAG,QAAQ,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC;IACzC,CAAC;CACF"}

package/dist/retrieval/cross-encoder-reranker.test.d.ts

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