agentic-qe 2.5.10 → 2.6.0

package/dist/code-intelligence/search/RRFFusion.js

@@ -0,0 +1,223 @@
+"use strict";
+/**
+ * Reciprocal Rank Fusion (RRF)
+ *
+ * Combines rankings from multiple search methods
+ * using the RRF formula:
+ *   score(d) = Σ 1 / (k + rank_i(d))
+ *
+ * Where k is a constant (typically 60) that determines
+ * how much weight is given to lower-ranked results.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RRFFusion = void 0;
+const types_js_1 = require("./types.js");
+class RRFFusion {
+    constructor(config = {}) {
+        this.config = { ...types_js_1.DEFAULT_RRF_CONFIG, ...config };
+    }
+    /**
+     * Fuse multiple ranked lists using RRF.
+     *
+     * @param rankedLists - Array of search result lists, each ordered by relevance
+     * @param topK - Number of results to return
+     * @returns Fused results ordered by RRF score
+     */
+    fuse(rankedLists, topK = 10) {
+        // Build RRF rankings
+        const rankings = new Map();
+        for (let listIndex = 0; listIndex < rankedLists.length; listIndex++) {
+            const list = rankedLists[listIndex];
+            for (let rank = 0; rank < list.length; rank++) {
+                const result = list[rank];
+                const existing = rankings.get(result.id);
+                if (existing) {
+                    // Update existing ranking
+                    existing.ranks[listIndex] = rank + 1; // 1-indexed rank
+                    existing.originalScores[listIndex] = result.score;
+                }
+                else {
+                    // Create new ranking
+                    const ranks = new Array(rankedLists.length).fill(Infinity);
+                    const originalScores = new Array(rankedLists.length).fill(0);
+                    ranks[listIndex] = rank + 1;
+                    originalScores[listIndex] = result.score;
+                    rankings.set(result.id, {
+                        id: result.id,
+                        ranks,
+                        rrfScore: 0,
+                        originalScores,
+                    });
+                }
+            }
+        }
+        // Calculate RRF scores
+        for (const ranking of rankings.values()) {
+            ranking.rrfScore = this.calculateRRFScore(ranking.ranks);
+        }
+        // Filter by minimum source agreement if configured
+        const filteredRankings = Array.from(rankings.values()).filter(ranking => {
+            const sourcesWithRank = ranking.ranks.filter(r => r !== Infinity).length;
+            return sourcesWithRank >= this.config.minSourceAgreement;
+        });
+        // Sort by RRF score
+        filteredRankings.sort((a, b) => b.rrfScore - a.rrfScore);
+        // Build result objects
+        // We need the original results to get full data
+        const resultMap = new Map();
+        for (const list of rankedLists) {
+            for (const result of list) {
+                if (!resultMap.has(result.id)) {
+                    resultMap.set(result.id, result);
+                }
+            }
+        }
+        return filteredRankings.slice(0, topK).map(ranking => {
+            const original = resultMap.get(ranking.id);
+            return {
+                ...original,
+                score: ranking.rrfScore,
+                bm25Score: ranking.originalScores[0] || undefined,
+                vectorScore: ranking.originalScores[1] || undefined,
+            };
+        });
+    }
+    /**
+     * Fuse with weighted RRF.
+     * Allows different weights for each source.
+     */
+    fuseWeighted(rankedLists, weights, topK = 10) {
+        if (weights.length !== rankedLists.length) {
+            throw new Error('Weights array must match number of ranked lists');
+        }
+        // Normalize weights
+        const totalWeight = weights.reduce((a, b) => a + b, 0);
+        const normalizedWeights = weights.map(w => w / totalWeight);
+        // Build weighted RRF rankings
+        const rankings = new Map();
+        for (let listIndex = 0; listIndex < rankedLists.length; listIndex++) {
+            const list = rankedLists[listIndex];
+            const weight = normalizedWeights[listIndex];
+            for (let rank = 0; rank < list.length; rank++) {
+                const result = list[rank];
+                const rrfContribution = weight / (this.config.k + rank + 1);
+                const existing = rankings.get(result.id);
+                if (existing) {
+                    existing.rrfScore += rrfContribution;
+                    existing.ranks[listIndex] = rank + 1;
+                    existing.originalScores[listIndex] = result.score;
+                }
+                else {
+                    const ranks = new Array(rankedLists.length).fill(Infinity);
+                    const originalScores = new Array(rankedLists.length).fill(0);
+                    ranks[listIndex] = rank + 1;
+                    originalScores[listIndex] = result.score;
+                    rankings.set(result.id, {
+                        id: result.id,
+                        ranks,
+                        rrfScore: rrfContribution,
+                        originalScores,
+                    });
+                }
+            }
+        }
+        // Filter and sort
+        const filteredRankings = Array.from(rankings.values()).filter(ranking => {
+            const sourcesWithRank = ranking.ranks.filter(r => r !== Infinity).length;
+            return sourcesWithRank >= this.config.minSourceAgreement;
+        });
+        filteredRankings.sort((a, b) => b.rrfScore - a.rrfScore);
+        // Build results
+        const resultMap = new Map();
+        for (const list of rankedLists) {
+            for (const result of list) {
+                if (!resultMap.has(result.id)) {
+                    resultMap.set(result.id, result);
+                }
+            }
+        }
+        return filteredRankings.slice(0, topK).map(ranking => {
+            const original = resultMap.get(ranking.id);
+            return {
+                ...original,
+                score: ranking.rrfScore,
+                bm25Score: ranking.originalScores[0] || undefined,
+                vectorScore: ranking.originalScores[1] || undefined,
+            };
+        });
+    }
+    /**
+     * Simple weighted score fusion (alternative to RRF).
+     * Combines normalized scores directly.
+     */
+    fuseScores(rankedLists, weights, topK = 10) {
+        if (weights.length !== rankedLists.length) {
+            throw new Error('Weights array must match number of ranked lists');
+        }
+        // Normalize weights
+        const totalWeight = weights.reduce((a, b) => a + b, 0);
+        const normalizedWeights = weights.map(w => w / totalWeight);
+        // Find max scores for normalization
+        const maxScores = rankedLists.map(list => list.length > 0 ? Math.max(...list.map(r => r.score)) : 1);
+        // Build combined scores
+        const scores = new Map();
+        for (let listIndex = 0; listIndex < rankedLists.length; listIndex++) {
+            const list = rankedLists[listIndex];
+            const weight = normalizedWeights[listIndex];
+            const maxScore = maxScores[listIndex];
+            for (const result of list) {
+                const normalizedScore = result.score / maxScore;
+                const weightedScore = normalizedScore * weight;
+                const existing = scores.get(result.id);
+                if (existing) {
+                    existing.combinedScore += weightedScore;
+                    existing.originalScores[listIndex] = result.score;
+                }
+                else {
+                    const originalScores = new Array(rankedLists.length).fill(0);
+                    originalScores[listIndex] = result.score;
+                    scores.set(result.id, {
+                        combinedScore: weightedScore,
+                        originalScores,
+                        result,
+                    });
+                }
+            }
+        }
+        // Sort and return
+        const sortedScores = Array.from(scores.values())
+            .sort((a, b) => b.combinedScore - a.combinedScore);
+        return sortedScores.slice(0, topK).map(({ combinedScore, originalScores, result }) => ({
+            ...result,
+            score: combinedScore,
+            bm25Score: originalScores[0] || undefined,
+            vectorScore: originalScores[1] || undefined,
+        }));
+    }
+    /**
+     * Calculate RRF score from ranks.
+     */
+    calculateRRFScore(ranks) {
+        let score = 0;
+        for (const rank of ranks) {
+            if (rank !== Infinity) {
+                score += 1 / (this.config.k + rank);
+            }
+        }
+        return score;
+    }
+    /**
+     * Get configuration.
+     */
+    getConfig() {
+        return { ...this.config };
+    }
+    /**
+     * Update configuration.
+     */
+    updateConfig(config) {
+        this.config = { ...this.config, ...config };
+    }
+}
+exports.RRFFusion = RRFFusion;
+//# sourceMappingURL=RRFFusion.js.map

package/dist/code-intelligence/search/RRFFusion.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"RRFFusion.js","sourceRoot":"","sources":["../../../src/code-intelligence/search/RRFFusion.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAEH,yCAKoB;AAEpB,MAAa,SAAS;IAGpB,YAAY,SAA6B,EAAE;QACzC,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,6BAAkB,EAAE,GAAG,MAAM,EAAE,CAAC;IACrD,CAAC;IAED;;;;;;OAMG;IACH,IAAI,CAAC,WAA6B,EAAE,OAAe,EAAE;QACnD,qBAAqB;QACrB,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAsB,CAAC;QAE/C,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,WAAW,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;YACpE,MAAM,IAAI,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC;YAEpC,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC;gBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC1B,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAEzC,IAAI,QAAQ,EAAE,CAAC;oBACb,0BAA0B;oBAC1B,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,iBAAiB;oBACvD,QAAQ,CAAC,cAAc,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;gBACpD,CAAC;qBAAM,CAAC;oBACN,qBAAqB;oBACrB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBAC3D,MAAM,cAAc,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;oBAC7D,KAAK,CAAC,SAAS,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC;oBAC5B,cAAc,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;oBAEzC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE;wBACtB,EAAE,EAAE,MAAM,CAAC,EAAE;wBACb,KAAK;wBACL,QAAQ,EAAE,CAAC;wBACX,cAAc;qBACf,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,uBAAuB;QACvB,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YACxC,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC3D,CAAC;QAED,mDAAmD;QACnD,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE;YACtE,MAAM,eAAe,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,MAAM,CAAC;YACzE,OAAO,eAAe,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,oBAAoB;QACpB,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;QAEzD,uBAAuB;QACvB,gDAAgD;QAChD,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;QAClD,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;YAC/B,KAAK,MAAM,MAAM,IAAI,IAAI,EAAE,CAAC;gBAC1B,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;oBAC9B,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;YACnD,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAE,CAAC;YAC5C,OAAO;gBACL,GAAG,QAAQ;gBACX,KAAK,EAAE,OAAO,CAAC,QAAQ;gBACvB,SAAS,EAAE,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,SAAS;gBACjD,WAAW,EAAE,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,SAAS;aACpD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACH,YAAY,CACV,WAA6B,EAC7B,OAAiB,EACjB,OAAe,EAAE;QAEjB,IAAI,OAAO,CAAC,MAAM,KAAK,WAAW,CAAC,MAAM,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QAED,oBAAoB;QACpB,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QACvD,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,WAAW,CAAC,CAAC;QAE5D,8BAA8B;QAC9B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAsB,CAAC;QAE/C,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,WAAW,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;YACpE,MAAM,IAAI,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC;YACpC,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAE5C,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC;gBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC1B,MAAM,eAAe,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC;gBAE5D,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACzC,IAAI,QAAQ,EAAE,CAAC;oBACb,QAAQ,CAAC,QAAQ,IAAI,eAAe,CAAC;oBACrC,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC;oBACrC,QAAQ,CAAC,cAAc,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;gBACpD,CAAC;qBAAM,CAAC;oBACN,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBAC3D,MAAM,cAAc,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;oBAC7D,KAAK,CAAC,SAAS,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC;oBAC5B,cAAc,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;oBAEzC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE;wBACtB,EAAE,EAAE,MAAM,CAAC,EAAE;wBACb,KAAK;wBACL,QAAQ,EAAE,eAAe;wBACzB,cAAc;qBACf,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,kBAAkB;QAClB,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE;YACtE,MAAM,eAAe,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,MAAM,CAAC;YACzE,OAAO,eAAe,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;QAEzD,gBAAgB;QAChB,MAAM,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;QAClD,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;YAC/B,KAAK,MAAM,MAAM,IAAI,IAAI,EAAE,CAAC;gBAC1B,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;oBAC9B,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;YACnD,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAE,CAAC;YAC5C,OAAO;gBACL,GAAG,QAAQ;gBACX,KAAK,EAAE,OAAO,CAAC,QAAQ;gBACvB,SAAS,EAAE,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,SAAS;gBACjD,WAAW,EAAE,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,SAAS;aACpD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACH,UAAU,CACR,WAA6B,EAC7B,OAAiB,EACjB,OAAe,EAAE;QAEjB,IAAI,OAAO,CAAC,MAAM,KAAK,WAAW,CAAC,MAAM,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QAED,oBAAoB;QACpB,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QACvD,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,WAAW,CAAC,CAAC;QAE5D,oCAAoC;QACpC,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CACvC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAC1D,CAAC;QAEF,wBAAwB;QACxB,MAAM,MAAM,GAAG,IAAI,GAAG,EAIlB,CAAC;QAEL,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,WAAW,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;YACpE,MAAM,IAAI,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC;YACpC,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAC5C,MAAM,QAAQ,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC;YAEtC,KAAK,MAAM,MAAM,IAAI,IAAI,EAAE,CAAC;gBAC1B,MAAM,eAAe,GAAG,MAAM,CAAC,KAAK,GAAG,QAAQ,CAAC;gBAChD,MAAM,aAAa,GAAG,eAAe,GAAG,MAAM,CAAC;gBAE/C,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACvC,IAAI,QAAQ,EAAE,CAAC;oBACb,QAAQ,CAAC,aAAa,IAAI,aAAa,CAAC;oBACxC,QAAQ,CAAC,cAAc,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;gBACpD,CAAC;qBAAM,CAAC;oBACN,MAAM,cAAc,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;oBAC7D,cAAc,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;oBAEzC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE;wBACpB,aAAa,EAAE,aAAa;wBAC5B,cAAc;wBACd,MAAM;qBACP,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,kBAAkB;QAClB,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;aAC7C,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,GAAG,CAAC,CAAC,aAAa,CAAC,CAAC;QAErD,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;YACrF,GAAG,MAAM;YACT,KAAK,EAAE,aAAa;YACpB,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,IAAI,SAAS;YACzC,WAAW,EAAE,cAAc,CAAC,CAAC,CAAC,IAAI,SAAS;SAC5C,CAAC,CAAC,CAAC;IACN,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,KAAe;QACvC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtB,KAAK,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,MAA0B;QACrC,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;IAC9C,CAAC;CACF;AA7PD,8BA6PC"}

package/dist/code-intelligence/search/VectorSearch.d.ts

@@ -0,0 +1,157 @@
+/**
+ * Vector Search
+ *
+ * Provides semantic vector search using embeddings.
+ * Delegates to CodeChunkStore for RuVector-backed persistence,
+ * with in-memory fallback for testing.
+ *
+ * @module code-intelligence/search/VectorSearch
+ */
+import type { SearchResult } from './types.js';
+import { CodeChunkStore, CodeChunkStoreConfig } from '../storage/CodeChunkStore.js';
+export interface VectorSearchConfig {
+    /**
+     * Number of dimensions in embedding vectors.
+     */
+    dimensions: number;
+    /**
+     * Distance metric for similarity.
+     */
+    metric: 'cosine' | 'euclidean' | 'dot';
+    /**
+     * PostgreSQL connection configuration for RuVector.
+     * If not provided, uses in-memory storage.
+     */
+    database?: CodeChunkStoreConfig;
+    /**
+     * Index type for pgvector (HNSW or IVFFlat).
+     */
+    indexType: 'hnsw' | 'ivfflat' | 'none';
+    /**
+     * HNSW M parameter (connections per layer).
+     */
+    hnswM: number;
+    /**
+     * HNSW ef_construction parameter.
+     */
+    hnswEfConstruction: number;
+    /**
+     * HNSW ef_search parameter (higher = more accurate but slower).
+     */
+    hnswEfSearch: number;
+}
+export interface VectorDocument {
+    /** Document ID */
+    id: string;
+    /** Embedding vector */
+    embedding: number[];
+    /** Document metadata */
+    metadata: {
+        filePath: string;
+        content: string;
+        startLine: number;
+        endLine: number;
+        entityType?: string;
+        entityName?: string;
+    };
+}
+export declare const DEFAULT_VECTOR_SEARCH_CONFIG: VectorSearchConfig;
+/**
+ * VectorSearch - Unified interface for vector search
+ *
+ * Uses CodeChunkStore (RuVector PostgreSQL) when database is configured,
+ * otherwise falls back to in-memory storage for testing.
+ */
+export declare class VectorSearch {
+    private config;
+    private documents;
+    private chunkStore;
+    private useDatabase;
+    constructor(config?: Partial<VectorSearchConfig>);
+    /**
+     * Initialize the vector search (connect to database if configured).
+     */
+    initialize(): Promise<void>;
+    /**
+     * Add a document with its embedding.
+     */
+    addDocument(doc: VectorDocument): Promise<void>;
+    /**
+     * Add multiple documents.
+     */
+    addDocuments(docs: VectorDocument[]): Promise<void>;
+    /**
+     * Remove a document.
+     */
+    removeDocument(docId: string): Promise<boolean>;
+    /**
+     * Search for similar documents.
+     */
+    search(queryEmbedding: number[], topK?: number, filter?: {
+        language?: string;
+        entityType?: string;
+        filePattern?: string;
+    }): Promise<SearchResult[]>;
+    /**
+     * Get a document by ID.
+     */
+    getDocument(docId: string): Promise<VectorDocument | null>;
+    /**
+     * Get all document IDs.
+     */
+    getDocumentIds(): string[];
+    /**
+     * Get document count.
+     */
+    getDocumentCount(): number;
+    /**
+     * Clear all documents.
+     */
+    clear(): Promise<void>;
+    /**
+     * Get configuration.
+     */
+    getConfig(): VectorSearchConfig;
+    /**
+     * Close database connection.
+     */
+    close(): Promise<void>;
+    /**
+     * Check if using database storage.
+     */
+    isUsingDatabase(): boolean;
+    /**
+     * Get the underlying CodeChunkStore (for advanced operations).
+     */
+    getChunkStore(): CodeChunkStore | null;
+    /**
+     * Health check
+     */
+    healthCheck(): Promise<{
+        healthy: boolean;
+        mode: 'database' | 'memory';
+        documentCount: number;
+        error?: string;
+    }>;
+    /**
+     * Search in memory using brute force.
+     */
+    private searchInMemory;
+    /**
+     * Calculate similarity between two vectors.
+     */
+    private calculateSimilarity;
+    /**
+     * Cosine similarity.
+     */
+    private cosineSimilarity;
+    /**
+     * Euclidean distance.
+     */
+    private euclideanDistance;
+    /**
+     * Dot product.
+     */
+    private dotProduct;
+}
+//# sourceMappingURL=VectorSearch.d.ts.map

package/dist/code-intelligence/search/VectorSearch.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"VectorSearch.d.ts","sourceRoot":"","sources":["../../../src/code-intelligence/search/VectorSearch.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AAEpF,MAAM,WAAW,kBAAkB;IACjC;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IAEnB;;OAEG;IACH,MAAM,EAAE,QAAQ,GAAG,WAAW,GAAG,KAAK,CAAC;IAEvC;;;OAGG;IACH,QAAQ,CAAC,EAAE,oBAAoB,CAAC;IAEhC;;OAEG;IACH,SAAS,EAAE,MAAM,GAAG,SAAS,GAAG,MAAM,CAAC;IAEvC;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,kBAAkB,EAAE,MAAM,CAAC;IAE3B;;OAEG;IACH,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,cAAc;IAC7B,kBAAkB;IAClB,EAAE,EAAE,MAAM,CAAC;IAEX,uBAAuB;IACvB,SAAS,EAAE,MAAM,EAAE,CAAC;IAEpB,wBAAwB;IACxB,QAAQ,EAAE;QACR,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;QAChB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,CAAC;CACH;AAED,eAAO,MAAM,4BAA4B,EAAE,kBAO1C,CAAC;AAEF;;;;;GAKG;AACH,qBAAa,YAAY;IACvB,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,SAAS,CAA0C;IAC3D,OAAO,CAAC,UAAU,CAA+B;IACjD,OAAO,CAAC,WAAW,CAAU;gBAEjB,MAAM,GAAE,OAAO,CAAC,kBAAkB,CAAM;IAYpD;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAMjC;;OAEG;IACG,WAAW,CAAC,GAAG,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAuBrD;;OAEG;IACG,YAAY,CAAC,IAAI,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBzD;;OAEG;IACG,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAUrD;;OAEG;IACG,MAAM,CACV,cAAc,EAAE,MAAM,EAAE,EACxB,IAAI,GAAE,MAAW,EACjB,MAAM,CAAC,EAAE;QACP,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GACA,OAAO,CAAC,YAAY,EAAE,CAAC;IA+B1B;;OAEG;IACG,WAAW,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC;IAShE;;OAEG;IACH,cAAc,IAAI,MAAM,EAAE;IAI1B;;OAEG;IACH,gBAAgB,IAAI,MAAM;IAI1B;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ5B;;OAEG;IACH,SAAS,IAAI,kBAAkB;IAI/B;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAO5B;;OAEG;IACH,eAAe,IAAI,OAAO;IAI1B;;OAEG;IACH,aAAa,IAAI,cAAc,GAAG,IAAI;IAItC;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC;QAC3B,OAAO,EAAE,OAAO,CAAC;QACjB,IAAI,EAAE,UAAU,GAAG,QAAQ,CAAC;QAC5B,aAAa,EAAE,MAAM,CAAC;QACtB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;IAoBF;;OAEG;IACH,OAAO,CAAC,cAAc;IA2CtB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAa3B;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAexB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IASzB;;OAEG;IACH,OAAO,CAAC,UAAU;CAOnB"}