@sprinterai/runtime 0.4.1 → 0.6.0

package/dist/search/hybrid-search.js

@@ -0,0 +1,111 @@
+/** RRF constant (standard value from literature). */
+const RRF_K = 60;
+/** Default search configuration values. */
+const DEFAULTS = {
+    vectorWeight: 0.7,
+    textWeight: 0.3,
+    minSimilarity: 0.3,
+    limit: 10,
+};
+/**
+ * Reciprocal Rank Fusion: combines two ranked lists into one.
+ * score = vectorWeight / (RRF_K + vectorRank) + textWeight / (RRF_K + textRank)
+ *
+ * Hits appearing in only one list receive the RRF contribution from that list only.
+ */
+export function reciprocalRankFusion(vectorHits, textHits, options) {
+    const vectorWeight = options.vectorWeight ?? DEFAULTS.vectorWeight;
+    const textWeight = options.textWeight ?? DEFAULTS.textWeight;
+    const minSimilarity = options.minSimilarity ?? DEFAULTS.minSimilarity;
+    const limit = options.limit ?? DEFAULTS.limit;
+    // Filter vector hits by minimum similarity
+    const filteredVectorHits = vectorHits.filter((hit) => hit.similarity >= minSimilarity);
+    // Rank vector hits by similarity (highest first -- already assumed sorted, but be safe)
+    const rankedVector = [...filteredVectorHits].sort((a, b) => b.similarity - a.similarity);
+    // Rank text hits by rank score (highest first)
+    const rankedText = [...textHits].sort((a, b) => b.rank - a.rank);
+    // Build vector rank map: id -> { rank (1-based), hit }
+    const vectorRankMap = new Map();
+    for (let i = 0; i < rankedVector.length; i++) {
+        vectorRankMap.set(rankedVector[i].id, {
+            rank: i + 1,
+            hit: rankedVector[i],
+        });
+    }
+    // Build text rank map: id -> { rank (1-based), hit }
+    const textRankMap = new Map();
+    for (let i = 0; i < rankedText.length; i++) {
+        textRankMap.set(rankedText[i].id, { rank: i + 1, hit: rankedText[i] });
+    }
+    // Collect all unique IDs
+    const allIds = new Set([...vectorRankMap.keys(), ...textRankMap.keys()]);
+    // Compute RRF score for each item
+    const results = [];
+    for (const id of allIds) {
+        const vectorEntry = vectorRankMap.get(id);
+        const textEntry = textRankMap.get(id);
+        const vectorRRF = vectorEntry ? vectorWeight / (RRF_K + vectorEntry.rank) : 0;
+        const textRRF = textEntry ? textWeight / (RRF_K + textEntry.rank) : 0;
+        // Use whichever hit we have for the base data
+        const hit = vectorEntry?.hit ?? textEntry?.hit;
+        if (!hit)
+            continue;
+        results.push({
+            id: hit.id,
+            documentId: hit.documentId,
+            content: hit.content,
+            chunkIndex: hit.chunkIndex,
+            pageNumber: hit.pageNumber,
+            metadata: hit.metadata,
+            vectorScore: vectorEntry?.hit.similarity ?? null,
+            textScore: textEntry?.hit.rank ?? null,
+            score: vectorRRF + textRRF,
+        });
+    }
+    // Sort by combined score descending
+    results.sort((a, b) => b.score - a.score);
+    // Apply limit
+    return results.slice(0, limit);
+}
+/** Create a hybrid search provider. */
+export function createHybridSearch(config) {
+    return {
+        async search(query, options) {
+            const canVector = config.embeddingProvider !== undefined && config.vectorSearch !== undefined;
+            if (canVector) {
+                // Generate query embedding and run both searches in parallel
+                const embedding = await config.embeddingProvider.generateEmbeddings([query]);
+                if (embedding !== null && embedding.length > 0) {
+                    const [vectorHits, textHits] = await Promise.all([
+                        config.vectorSearch(embedding[0], options),
+                        config.textSearch(query, options),
+                    ]);
+                    return reciprocalRankFusion(vectorHits, textHits, {
+                        vectorWeight: options.vectorWeight,
+                        textWeight: options.textWeight,
+                        minSimilarity: options.minSimilarity,
+                        limit: options.limit,
+                    });
+                }
+            }
+            // Fallback: text search only
+            const textHits = await config.textSearch(query, options);
+            // Convert text hits to SearchResult with text-only scoring
+            return textHits
+                .sort((a, b) => b.rank - a.rank)
+                .slice(0, options.limit ?? DEFAULTS.limit)
+                .map((hit, index) => ({
+                id: hit.id,
+                documentId: hit.documentId,
+                content: hit.content,
+                chunkIndex: hit.chunkIndex,
+                pageNumber: hit.pageNumber,
+                metadata: hit.metadata,
+                vectorScore: null,
+                textScore: hit.rank,
+                score: (options.textWeight ?? DEFAULTS.textWeight) / (RRF_K + index + 1),
+            }));
+        },
+    };
+}
+//# sourceMappingURL=hybrid-search.js.map

package/dist/search/hybrid-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid-search.js","sourceRoot":"","sources":["../../src/search/hybrid-search.ts"],"names":[],"mappings":"AAyCA,qDAAqD;AACrD,MAAM,KAAK,GAAG,EAAE,CAAC;AAEjB,2CAA2C;AAC3C,MAAM,QAAQ,GAAG;IACf,YAAY,EAAE,GAAG;IACjB,UAAU,EAAE,GAAG;IACf,aAAa,EAAE,GAAG;IAClB,KAAK,EAAE,EAAE;CACD,CAAC;AAEX;;;;;GAKG;AACH,MAAM,UAAU,oBAAoB,CAClC,UAA6B,EAC7B,QAAyB,EACzB,OAKC;IAED,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,QAAQ,CAAC,YAAY,CAAC;IACnE,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,QAAQ,CAAC,UAAU,CAAC;IAC7D,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,QAAQ,CAAC,aAAa,CAAC;IACtE,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC;IAE9C,2CAA2C;IAC3C,MAAM,kBAAkB,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,IAAI,aAAa,CAAC,CAAC;IAEvF,wFAAwF;IACxF,MAAM,YAAY,GAAG,CAAC,GAAG,kBAAkB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC;IAEzF,+CAA+C;IAC/C,MAAM,UAAU,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;IAEjE,uDAAuD;IACvD,MAAM,aAAa,GAAG,IAAI,GAAG,EAAkD,CAAC;IAChF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC7C,aAAa,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE;YACpC,IAAI,EAAE,CAAC,GAAG,CAAC;YACX,GAAG,EAAE,YAAY,CAAC,CAAC,CAAC;SACrB,CAAC,CAAC;IACL,CAAC;IAED,qDAAqD;IACrD,MAAM,WAAW,GAAG,IAAI,GAAG,EAAgD,CAAC;IAC5E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3C,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACzE,CAAC;IAED,yBAAyB;IACzB,MAAM,MAAM,GAAG,IAAI,GAAG,CAAS,CAAC,GAAG,aAAa,CAAC,IAAI,EAAE,EAAE,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAEjF,kCAAkC;IAClC,MAAM,OAAO,GAAmB,EAAE,CAAC;IAEnC,KAAK,MAAM,EAAE,IAAI,MAAM,EAAE,CAAC;QACxB,MAAM,WAAW,GAAG,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC1C,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAEtC,MAAM,SAAS,GAAG,WAAW,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9E,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,KAAK,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAEtE,8CAA8C;QAC9C,MAAM,GAAG,GAAG,WAAW,EAAE,GAAG,IAAI,SAAS,EAAE,GAAG,CAAC;QAC/C,IAAI,CAAC,GAAG;YAAE,SAAS;QAEnB,OAAO,CAAC,IAAI,CAAC;YACX,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,UAAU,EAAE,GAAG,CAAC,UAAU;YAC1B,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,UAAU,EAAE,GAAG,CAAC,UAAU;YAC1B,UAAU,EAAE,GAAG,CAAC,UAAU;YAC1B,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,WAAW,EAAE,WAAW,EAAE,GAAG,CAAC,UAAU,IAAI,IAAI;YAChD,SAAS,EAAE,SAAS,EAAE,GAAG,CAAC,IAAI,IAAI,IAAI;YACtC,KAAK,EAAE,SAAS,GAAG,OAAO;SAC3B,CAAC,CAAC;IACL,CAAC;IAED,oCAAoC;IACpC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IAE1C,cAAc;IACd,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;AACjC,CAAC;AAED,uCAAuC;AACvC,MAAM,UAAU,kBAAkB,CAAC,MAA0B;IAC3D,OAAO;QACL,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,OAA4B;YACtD,MAAM,SAAS,GAAG,MAAM,CAAC,iBAAiB,KAAK,SAAS,IAAI,MAAM,CAAC,YAAY,KAAK,SAAS,CAAC;YAE9F,IAAI,SAAS,EAAE,CAAC;gBACd,6DAA6D;gBAC7D,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,iBAAkB,CAAC,kBAAkB,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;gBAE9E,IAAI,SAAS,KAAK,IAAI,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC/C,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;wBAC/C,MAAM,CAAC,YAAa,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC;wBAC3C,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC;qBAClC,CAAC,CAAC;oBAEH,OAAO,oBAAoB,CAAC,UAAU,EAAE,QAAQ,EAAE;wBAChD,YAAY,EAAE,OAAO,CAAC,YAAY;wBAClC,UAAU,EAAE,OAAO,CAAC,UAAU;wBAC9B,aAAa,EAAE,OAAO,CAAC,aAAa;wBACpC,KAAK,EAAE,OAAO,CAAC,KAAK;qBACrB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,6BAA6B;YAC7B,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAEzD,2DAA2D;YAC3D,OAAO,QAAQ;iBACZ,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC;iBAC/B,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC;iBACzC,GAAG,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBACpB,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,WAAW,EAAE,IAAI;gBACjB,SAAS,EAAE,GAAG,CAAC,IAAI;gBACnB,KAAK,EAAE,CAAC,OAAO,CAAC,UAAU,IAAI,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,GAAG,KAAK,GAAG,CAAC,CAAC;aACzE,CAAC,CAAC,CAAC;QACR,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/search/hybrid-search.test.d.ts

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

package/dist/search/hybrid-search.test.d.ts.map

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

package/dist/search/hybrid-search.test.js

@@ -0,0 +1,238 @@
+import { describe, it, expect, vi } from 'vitest';
+import { reciprocalRankFusion, createHybridSearch } from './hybrid-search.js';
+// ─── Helpers ─────────────────────────────────────────────────
+function makeVectorHit(id, similarity, overrides) {
+    return {
+        id,
+        documentId: overrides?.documentId ?? 'doc-1',
+        content: overrides?.content ?? `content-${id}`,
+        chunkIndex: overrides?.chunkIndex ?? 0,
+        pageNumber: overrides?.pageNumber ?? null,
+        metadata: overrides?.metadata ?? null,
+        similarity,
+    };
+}
+function makeTextHit(id, rank, overrides) {
+    return {
+        id,
+        documentId: overrides?.documentId ?? 'doc-1',
+        content: overrides?.content ?? `content-${id}`,
+        chunkIndex: overrides?.chunkIndex ?? 0,
+        pageNumber: overrides?.pageNumber ?? null,
+        metadata: overrides?.metadata ?? null,
+        rank,
+    };
+}
+const RRF_K = 60; // Must match the constant in the module
+// ─── reciprocalRankFusion ────────────────────────────────────
+describe('reciprocalRankFusion', () => {
+    it('combines vector and text results with default weights', () => {
+        const vectorHits = [makeVectorHit('a', 0.9), makeVectorHit('b', 0.7)];
+        const textHits = [makeTextHit('a', 0.8), makeTextHit('c', 0.6)];
+        const results = reciprocalRankFusion(vectorHits, textHits, {});
+        // 'a' appears in both lists -> highest score
+        const resultA = results.find((r) => r.id === 'a');
+        expect(resultA).toBeDefined();
+        // vector rank 1, text rank 1
+        const expectedA = 0.7 / (RRF_K + 1) + 0.3 / (RRF_K + 1);
+        expect(resultA.score).toBeCloseTo(expectedA, 10);
+        // 'b' only in vector (rank 2)
+        const resultB = results.find((r) => r.id === 'b');
+        expect(resultB).toBeDefined();
+        expect(resultB.score).toBeCloseTo(0.7 / (RRF_K + 2), 10);
+        // 'c' only in text (rank 2)
+        const resultC = results.find((r) => r.id === 'c');
+        expect(resultC).toBeDefined();
+        expect(resultC.score).toBeCloseTo(0.3 / (RRF_K + 2), 10);
+    });
+    it('returns results ordered by score descending', () => {
+        const vectorHits = [makeVectorHit('a', 0.95), makeVectorHit('b', 0.8), makeVectorHit('c', 0.5)];
+        const textHits = [makeTextHit('c', 0.9), makeTextHit('b', 0.7), makeTextHit('a', 0.3)];
+        const results = reciprocalRankFusion(vectorHits, textHits, {});
+        for (let i = 1; i < results.length; i++) {
+            expect(results[i - 1].score).toBeGreaterThanOrEqual(results[i].score);
+        }
+    });
+    it('filters vector hits below minSimilarity', () => {
+        const vectorHits = [makeVectorHit('a', 0.9), makeVectorHit('b', 0.2)];
+        const textHits = [];
+        const results = reciprocalRankFusion(vectorHits, textHits, { minSimilarity: 0.3 });
+        expect(results).toHaveLength(1);
+        expect(results[0].id).toBe('a');
+    });
+    it('applies limit to restrict results', () => {
+        const vectorHits = Array.from({ length: 20 }, (_, i) => makeVectorHit(`v-${String(i)}`, 0.9 - i * 0.03));
+        const textHits = [];
+        const results = reciprocalRankFusion(vectorHits, textHits, { limit: 5 });
+        expect(results).toHaveLength(5);
+    });
+    it('handles empty vector hits', () => {
+        const textHits = [makeTextHit('a', 0.8), makeTextHit('b', 0.6)];
+        const results = reciprocalRankFusion([], textHits, {});
+        expect(results).toHaveLength(2);
+        expect(results[0].vectorScore).toBeNull();
+        expect(results[0].textScore).toBe(0.8);
+    });
+    it('handles empty text hits', () => {
+        const vectorHits = [makeVectorHit('a', 0.9)];
+        const results = reciprocalRankFusion(vectorHits, [], {});
+        expect(results).toHaveLength(1);
+        expect(results[0].textScore).toBeNull();
+        expect(results[0].vectorScore).toBe(0.9);
+    });
+    it('handles both lists empty', () => {
+        const results = reciprocalRankFusion([], [], {});
+        expect(results).toHaveLength(0);
+    });
+    it('respects custom weights', () => {
+        // With textWeight=0.9, vectorWeight=0.1, text-only hit should rank higher
+        const vectorHits = [makeVectorHit('vec-only', 0.95)];
+        const textHits = [makeTextHit('text-only', 0.95)];
+        const results = reciprocalRankFusion(vectorHits, textHits, {
+            vectorWeight: 0.1,
+            textWeight: 0.9,
+        });
+        expect(results).toHaveLength(2);
+        // text-only should be first due to higher weight
+        const textResult = results.find((r) => r.id === 'text-only');
+        const vecResult = results.find((r) => r.id === 'vec-only');
+        expect(textResult.score).toBeGreaterThan(vecResult.score);
+    });
+    it('preserves metadata and page numbers in results', () => {
+        const vectorHits = [
+            makeVectorHit('a', 0.9, {
+                pageNumber: 5,
+                metadata: { source: 'pdf' },
+                content: 'from vector',
+            }),
+        ];
+        const textHits = [];
+        const results = reciprocalRankFusion(vectorHits, textHits, {});
+        expect(results[0].pageNumber).toBe(5);
+        expect(results[0].metadata).toEqual({ source: 'pdf' });
+        expect(results[0].content).toBe('from vector');
+    });
+    it('uses vector hit data when item appears in both lists', () => {
+        const vectorHits = [makeVectorHit('shared', 0.9, { content: 'vector-version' })];
+        const textHits = [makeTextHit('shared', 0.8, { content: 'text-version' })];
+        const results = reciprocalRankFusion(vectorHits, textHits, {});
+        expect(results).toHaveLength(1);
+        // vectorEntry is checked first so vector data is used
+        expect(results[0].content).toBe('vector-version');
+        expect(results[0].vectorScore).toBe(0.9);
+        expect(results[0].textScore).toBe(0.8);
+    });
+});
+// ─── createHybridSearch ──────────────────────────────────────
+describe('createHybridSearch', () => {
+    it('runs vector + text search in parallel when embedding provider available', async () => {
+        const embeddingProvider = {
+            generateEmbeddings: vi.fn().mockResolvedValue([[0.1, 0.2, 0.3]]),
+            dimensions: 3,
+        };
+        const textSearch = vi.fn().mockResolvedValue([makeTextHit('a', 0.8)]);
+        const vectorSearch = vi.fn().mockResolvedValue([makeVectorHit('a', 0.9)]);
+        const provider = createHybridSearch({
+            embeddingProvider,
+            textSearch,
+            vectorSearch,
+        });
+        const results = await provider.search('hello', { tenantId: 't-1' });
+        expect(embeddingProvider.generateEmbeddings).toHaveBeenCalledWith(['hello']);
+        expect(textSearch).toHaveBeenCalledWith('hello', { tenantId: 't-1' });
+        expect(vectorSearch).toHaveBeenCalledWith([0.1, 0.2, 0.3], { tenantId: 't-1' });
+        expect(results).toHaveLength(1);
+        expect(results[0].vectorScore).toBe(0.9);
+        expect(results[0].textScore).toBe(0.8);
+    });
+    it('falls back to text-only when no embedding provider', async () => {
+        const textSearch = vi.fn().mockResolvedValue([makeTextHit('a', 0.8), makeTextHit('b', 0.5)]);
+        const provider = createHybridSearch({ textSearch });
+        const results = await provider.search('query', { tenantId: 't-1' });
+        expect(textSearch).toHaveBeenCalledWith('query', { tenantId: 't-1' });
+        expect(results).toHaveLength(2);
+        expect(results[0].vectorScore).toBeNull();
+        expect(results[0].textScore).toBe(0.8);
+    });
+    it('falls back to text-only when embedding generation returns null', async () => {
+        const embeddingProvider = {
+            generateEmbeddings: vi.fn().mockResolvedValue(null),
+            dimensions: 3,
+        };
+        const textSearch = vi.fn().mockResolvedValue([makeTextHit('a', 0.6)]);
+        const vectorSearch = vi.fn();
+        const provider = createHybridSearch({
+            embeddingProvider,
+            textSearch,
+            vectorSearch,
+        });
+        const results = await provider.search('query', { tenantId: 't-1' });
+        expect(vectorSearch).not.toHaveBeenCalled();
+        expect(results).toHaveLength(1);
+        expect(results[0].vectorScore).toBeNull();
+    });
+    it('falls back to text-only when no vectorSearch function', async () => {
+        const embeddingProvider = {
+            generateEmbeddings: vi.fn().mockResolvedValue([[0.1, 0.2]]),
+            dimensions: 2,
+        };
+        const textSearch = vi.fn().mockResolvedValue([makeTextHit('a', 0.7)]);
+        const provider = createHybridSearch({
+            embeddingProvider,
+            textSearch,
+            // no vectorSearch
+        });
+        const results = await provider.search('query', { tenantId: 't-1' });
+        expect(embeddingProvider.generateEmbeddings).not.toHaveBeenCalled();
+        expect(results).toHaveLength(1);
+        expect(results[0].vectorScore).toBeNull();
+    });
+    it('handles empty results from both search backends', async () => {
+        const textSearch = vi.fn().mockResolvedValue([]);
+        const provider = createHybridSearch({ textSearch });
+        const results = await provider.search('nothing matches', { tenantId: 't-1' });
+        expect(results).toHaveLength(0);
+    });
+    it('passes search options through to backends', async () => {
+        const embeddingProvider = {
+            generateEmbeddings: vi.fn().mockResolvedValue([[0.1]]),
+            dimensions: 1,
+        };
+        const textSearch = vi.fn().mockResolvedValue([]);
+        const vectorSearch = vi.fn().mockResolvedValue([]);
+        const provider = createHybridSearch({
+            embeddingProvider,
+            textSearch,
+            vectorSearch,
+        });
+        const options = {
+            tenantId: 't-1',
+            documentIds: ['doc-1'],
+            limit: 5,
+            vectorWeight: 0.8,
+            textWeight: 0.2,
+            minSimilarity: 0.5,
+        };
+        await provider.search('query', options);
+        expect(textSearch).toHaveBeenCalledWith('query', options);
+        expect(vectorSearch).toHaveBeenCalledWith([0.1], options);
+    });
+    it('applies limit in text-only fallback mode', async () => {
+        const textSearch = vi
+            .fn()
+            .mockResolvedValue(Array.from({ length: 20 }, (_, i) => makeTextHit(`t-${String(i)}`, 0.9 - i * 0.03)));
+        const provider = createHybridSearch({ textSearch });
+        const results = await provider.search('query', { tenantId: 't-1', limit: 3 });
+        expect(results).toHaveLength(3);
+    });
+    it('scores text-only results using RRF formula', async () => {
+        const textSearch = vi.fn().mockResolvedValue([makeTextHit('a', 0.9), makeTextHit('b', 0.5)]);
+        const provider = createHybridSearch({ textSearch });
+        const results = await provider.search('query', { tenantId: 't-1' });
+        // First result: textWeight / (RRF_K + 1) = 0.3 / 61
+        expect(results[0].score).toBeCloseTo(0.3 / 61, 10);
+        // Second result: textWeight / (RRF_K + 2) = 0.3 / 62
+        expect(results[1].score).toBeCloseTo(0.3 / 62, 10);
+    });
+});
+//# sourceMappingURL=hybrid-search.test.js.map

package/dist/search/hybrid-search.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid-search.test.js","sourceRoot":"","sources":["../../src/search/hybrid-search.test.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAElD,OAAO,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAG3E,gEAAgE;AAEhE,SAAS,aAAa,CACpB,EAAU,EACV,UAAkB,EAClB,SAAoC;IAEpC,OAAO;QACL,EAAE;QACF,UAAU,EAAE,SAAS,EAAE,UAAU,IAAI,OAAO;QAC5C,OAAO,EAAE,SAAS,EAAE,OAAO,IAAI,WAAW,EAAE,EAAE;QAC9C,UAAU,EAAE,SAAS,EAAE,UAAU,IAAI,CAAC;QACtC,UAAU,EAAE,SAAS,EAAE,UAAU,IAAI,IAAI;QACzC,QAAQ,EAAE,SAAS,EAAE,QAAQ,IAAI,IAAI;QACrC,UAAU;KACX,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,EAAU,EAAE,IAAY,EAAE,SAAkC;IAC/E,OAAO;QACL,EAAE;QACF,UAAU,EAAE,SAAS,EAAE,UAAU,IAAI,OAAO;QAC5C,OAAO,EAAE,SAAS,EAAE,OAAO,IAAI,WAAW,EAAE,EAAE;QAC9C,UAAU,EAAE,SAAS,EAAE,UAAU,IAAI,CAAC;QACtC,UAAU,EAAE,SAAS,EAAE,UAAU,IAAI,IAAI;QACzC,QAAQ,EAAE,SAAS,EAAE,QAAQ,IAAI,IAAI;QACrC,IAAI;KACL,CAAC;AACJ,CAAC;AAED,MAAM,KAAK,GAAG,EAAE,CAAC,CAAC,wCAAwC;AAE1D,gEAAgE;AAEhE,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QACtE,MAAM,QAAQ,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAEhE,MAAM,OAAO,GAAG,oBAAoB,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;QAE/D,6CAA6C;QAC7C,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,6BAA6B;QAC7B,MAAM,SAAS,GAAG,GAAG,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QACxD,MAAM,CAAC,OAAQ,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAElD,8BAA8B;QAC9B,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,MAAM,CAAC,OAAQ,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAE1D,4BAA4B;QAC5B,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,MAAM,CAAC,OAAQ,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAChG,MAAM,QAAQ,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAEvF,MAAM,OAAO,GAAG,oBAAoB,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;QAE/D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QACxE,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QACtE,MAAM,QAAQ,GAAoB,EAAE,CAAC;QAErC,MAAM,OAAO,GAAG,oBAAoB,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;QAEnF,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACrD,aAAa,CAAC,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,CAChD,CAAC;QACF,MAAM,QAAQ,GAAoB,EAAE,CAAC;QAErC,MAAM,OAAO,GAAG,oBAAoB,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QAEzE,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,QAAQ,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAEhE,MAAM,OAAO,GAAG,oBAAoB,CAAC,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;QAEvD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAE7C,MAAM,OAAO,GAAG,oBAAoB,CAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,CAAC;QACxC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,OAAO,GAAG,oBAAoB,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,0EAA0E;QAC1E,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,CAAC,WAAW,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC;QAElD,MAAM,OAAO,GAAG,oBAAoB,CAAC,UAAU,EAAE,QAAQ,EAAE;YACzD,YAAY,EAAE,GAAG;YACjB,UAAU,EAAE,GAAG;SAChB,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,iDAAiD;QACjD,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,WAAW,CAAC,CAAC;QAC7D,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAAC,CAAC;QAC3D,MAAM,CAAC,UAAW,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,SAAU,CAAC,KAAK,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,UAAU,GAAG;YACjB,aAAa,CAAC,GAAG,EAAE,GAAG,EAAE;gBACtB,UAAU,EAAE,CAAC;gBACb,QAAQ,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE;gBAC3B,OAAO,EAAE,aAAa;aACvB,CAAC;SACH,CAAC;QACF,MAAM,QAAQ,GAAoB,EAAE,CAAC;QAErC,MAAM,OAAO,GAAG,oBAAoB,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;QAE/D,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QACvD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,QAAQ,EAAE,GAAG,EAAE,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC;QACjF,MAAM,QAAQ,GAAG,CAAC,WAAW,CAAC,QAAQ,EAAE,GAAG,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;QAE3E,MAAM,OAAO,GAAG,oBAAoB,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;QAE/D,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,sDAAsD;QACtD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,gEAAgE;AAEhE,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,MAAM,iBAAiB,GAAsB;YAC3C,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAChE,UAAU,EAAE,CAAC;SACd,CAAC;QAEF,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;QACtE,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;QAE1E,MAAM,QAAQ,GAAG,kBAAkB,CAAC;YAClC,iBAAiB;YACjB,UAAU;YACV,YAAY;SACb,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEpE,MAAM,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QAC7E,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QACtE,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAChF,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;QAE7F,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpD,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEpE,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QACtE,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,iBAAiB,GAAsB;YAC3C,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;YACnD,UAAU,EAAE,CAAC;SACd,CAAC;QAEF,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;QACtE,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAE7B,MAAM,QAAQ,GAAG,kBAAkB,CAAC;YAClC,iBAAiB;YACjB,UAAU;YACV,YAAY;SACb,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEpE,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,iBAAiB,GAAsB;YAC3C,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAC3D,UAAU,EAAE,CAAC;SACd,CAAC;QAEF,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;QAEtE,MAAM,QAAQ,GAAG,kBAAkB,CAAC;YAClC,iBAAiB;YACjB,UAAU;YACV,kBAAkB;SACnB,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEpE,MAAM,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACpE,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;QAEjD,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpD,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,iBAAiB,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAE9E,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,iBAAiB,GAAsB;YAC3C,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YACtD,UAAU,EAAE,CAAC;SACd,CAAC;QAEF,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;QACjD,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;QAEnD,MAAM,QAAQ,GAAG,kBAAkB,CAAC;YAClC,iBAAiB;YACjB,UAAU;YACV,YAAY;SACb,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG;YACd,QAAQ,EAAE,KAAK;YACf,WAAW,EAAE,CAAC,OAAO,CAAC;YACtB,KAAK,EAAE,CAAC;YACR,YAAY,EAAE,GAAG;YACjB,UAAU,EAAE,GAAG;YACf,aAAa,EAAE,GAAG;SACnB,CAAC;QAEF,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAExC,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC1D,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,UAAU,GAAG,EAAE;aAClB,EAAE,EAAE;aACJ,iBAAiB,CAChB,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC,CACpF,CAAC;QAEJ,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpD,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QAE9E,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;QAE7F,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC;QAEpD,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEpE,oDAAoD;QACpD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,GAAG,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;QACnD,qDAAqD;QACrD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,GAAG,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/search/in-memory-search.d.ts

@@ -0,0 +1,17 @@
+import type { HybridSearchConfig } from './hybrid-search.js';
+/** A chunk stored in memory for search. */
+export interface InMemoryChunk {
+    id: string;
+    documentId: string;
+    content: string;
+    chunkIndex: number;
+    pageNumber: number | null;
+    metadata: Record<string, unknown> | null;
+    embedding?: number[];
+}
+/** Create an in-memory search backend for testing and simple use cases. */
+export declare function createInMemorySearch(chunks: InMemoryChunk[]): {
+    textSearch: HybridSearchConfig['textSearch'];
+    vectorSearch: HybridSearchConfig['vectorSearch'];
+};
+//# sourceMappingURL=in-memory-search.d.ts.map

package/dist/search/in-memory-search.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"in-memory-search.d.ts","sourceRoot":"","sources":["../../src/search/in-memory-search.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAkC,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAE1F,2CAA2C;AAC3C,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IACzC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;CACtB;AAmBD,2EAA2E;AAC3E,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,aAAa,EAAE,GAAG;IAC7D,UAAU,EAAE,kBAAkB,CAAC,YAAY,CAAC,CAAC;IAC7C,YAAY,EAAE,kBAAkB,CAAC,cAAc,CAAC,CAAC;CAClD,CAsDA"}

package/dist/search/in-memory-search.js

@@ -0,0 +1,59 @@
+import { cosineSimilarity } from './cosine-similarity.js';
+/**
+ * Simple keyword-based text scoring.
+ * Score = number of matched query terms / total query terms.
+ */
+function scoreText(content, query) {
+    const lowerContent = content.toLowerCase();
+    const terms = query
+        .toLowerCase()
+        .split(/\s+/)
+        .filter((t) => t.length > 0);
+    if (terms.length === 0)
+        return 0;
+    const matched = terms.filter((term) => lowerContent.includes(term)).length;
+    return matched / terms.length;
+}
+/** Create an in-memory search backend for testing and simple use cases. */
+export function createInMemorySearch(chunks) {
+    const textSearch = async (query, options) => {
+        let filtered = chunks;
+        // Filter by tenant -- in-memory chunks don't have tenant_id,
+        // so filter by documentIds if provided
+        if (options.documentIds && options.documentIds.length > 0) {
+            const docIdSet = new Set(options.documentIds);
+            filtered = filtered.filter((c) => docIdSet.has(c.documentId));
+        }
+        const scored = filtered
+            .map((chunk) => ({
+            id: chunk.id,
+            documentId: chunk.documentId,
+            content: chunk.content,
+            chunkIndex: chunk.chunkIndex,
+            pageNumber: chunk.pageNumber,
+            metadata: chunk.metadata,
+            rank: scoreText(chunk.content, query),
+        }))
+            .filter((hit) => hit.rank > 0);
+        return scored.sort((a, b) => b.rank - a.rank);
+    };
+    const vectorSearch = async (embedding, options) => {
+        let filtered = chunks.filter((c) => c.embedding !== undefined);
+        if (options.documentIds && options.documentIds.length > 0) {
+            const docIdSet = new Set(options.documentIds);
+            filtered = filtered.filter((c) => docIdSet.has(c.documentId));
+        }
+        const scored = filtered.map((chunk) => ({
+            id: chunk.id,
+            documentId: chunk.documentId,
+            content: chunk.content,
+            chunkIndex: chunk.chunkIndex,
+            pageNumber: chunk.pageNumber,
+            metadata: chunk.metadata,
+            similarity: cosineSimilarity(embedding, chunk.embedding),
+        }));
+        return scored.sort((a, b) => b.similarity - a.similarity);
+    };
+    return { textSearch, vectorSearch };
+}
+//# sourceMappingURL=in-memory-search.js.map

package/dist/search/in-memory-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"in-memory-search.js","sourceRoot":"","sources":["../../src/search/in-memory-search.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAcvD;;;GAGG;AACH,SAAS,SAAS,CAAC,OAAe,EAAE,KAAa;IAC/C,MAAM,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAC3C,MAAM,KAAK,GAAG,KAAK;SAChB,WAAW,EAAE;SACb,KAAK,CAAC,KAAK,CAAC;SACZ,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAE/B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IAEjC,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC;IAC3E,OAAO,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC;AAChC,CAAC;AAED,2EAA2E;AAC3E,MAAM,UAAU,oBAAoB,CAAC,MAAuB;IAI1D,MAAM,UAAU,GAAG,KAAK,EACtB,KAAa,EACb,OAA4B,EACF,EAAE;QAC5B,IAAI,QAAQ,GAAG,MAAM,CAAC;QAEtB,6DAA6D;QAC7D,uCAAuC;QACvC,IAAI,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1D,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAC9C,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ;aACpB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YACf,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,IAAI,EAAE,SAAS,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC;SACtC,CAAC,CAAC;aACF,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;QAEjC,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;IAChD,CAAC,CAAC;IAEF,MAAM,YAAY,GAAG,KAAK,EACxB,SAAmB,EACnB,OAA4B,EACA,EAAE;QAC9B,IAAI,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC;QAE/D,IAAI,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1D,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAC9C,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YACtC,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,UAAU,EAAE,gBAAgB,CAAC,SAAS,EAAE,KAAK,CAAC,SAAU,CAAC;SAC1D,CAAC,CAAC,CAAC;QAEJ,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC;IAC5D,CAAC,CAAC;IAEF,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC;AACtC,CAAC"}

package/dist/search/in-memory-search.test.d.ts

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

package/dist/search/in-memory-search.test.d.ts.map

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