@sprinterai/runtime 0.4.1 → 0.5.0

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":""}

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

@@ -0,0 +1,169 @@
+import { describe, it, expect } from 'vitest';
+import { createInMemorySearch } from './in-memory-search.js';
+// ─── Test Data ───────────────────────────────────────────────
+const chunks = [
+    {
+        id: 'c-1',
+        documentId: 'doc-1',
+        content: 'The quick brown fox jumps over the lazy dog',
+        chunkIndex: 0,
+        pageNumber: 1,
+        metadata: { source: 'test' },
+        embedding: [1, 0, 0],
+    },
+    {
+        id: 'c-2',
+        documentId: 'doc-1',
+        content: 'A lazy cat sleeps on the mat',
+        chunkIndex: 1,
+        pageNumber: 1,
+        metadata: null,
+        embedding: [0, 1, 0],
+    },
+    {
+        id: 'c-3',
+        documentId: 'doc-2',
+        content: 'Machine learning algorithms process data efficiently',
+        chunkIndex: 0,
+        pageNumber: null,
+        metadata: null,
+        embedding: [0, 0, 1],
+    },
+    {
+        id: 'c-4',
+        documentId: 'doc-2',
+        content: 'No embedding available for this chunk',
+        chunkIndex: 1,
+        pageNumber: null,
+        metadata: null,
+        // no embedding
+    },
+];
+describe('createInMemorySearch', () => {
+    describe('textSearch', () => {
+        it('finds chunks matching query terms', async () => {
+            const { textSearch } = createInMemorySearch(chunks);
+            const results = await textSearch('lazy dog', { tenantId: 't-1' });
+            expect(results.length).toBeGreaterThan(0);
+            // c-1 has both "lazy" and "dog" -> rank 1.0
+            const c1 = results.find((r) => r.id === 'c-1');
+            expect(c1).toBeDefined();
+            expect(c1.rank).toBe(1.0); // both terms match
+            // c-2 has "lazy" but not "dog" -> rank 0.5
+            const c2 = results.find((r) => r.id === 'c-2');
+            expect(c2).toBeDefined();
+            expect(c2.rank).toBe(0.5);
+        });
+        it('returns empty for non-matching queries', async () => {
+            const { textSearch } = createInMemorySearch(chunks);
+            const results = await textSearch('quantum physics', { tenantId: 't-1' });
+            expect(results).toHaveLength(0);
+        });
+        it('is case-insensitive', async () => {
+            const { textSearch } = createInMemorySearch(chunks);
+            const results = await textSearch('QUICK BROWN', { tenantId: 't-1' });
+            expect(results.length).toBeGreaterThan(0);
+            expect(results[0].id).toBe('c-1');
+        });
+        it('filters by documentIds when provided', async () => {
+            const { textSearch } = createInMemorySearch(chunks);
+            const results = await textSearch('the', {
+                tenantId: 't-1',
+                documentIds: ['doc-2'],
+            });
+            // Only c-4 from doc-2 contains "the" (in "this")
+            // Actually "the" doesn't appear in doc-2 chunks... let me check
+            // c-3: "Machine learning algorithms process data efficiently" - no "the"
+            // c-4: "No embedding available for this chunk" - contains "this" but not "the"
+            // So no results from doc-2 for "the"
+            for (const r of results) {
+                expect(r.documentId).toBe('doc-2');
+            }
+        });
+        it('returns results sorted by rank descending', async () => {
+            const { textSearch } = createInMemorySearch(chunks);
+            const results = await textSearch('lazy', { tenantId: 't-1' });
+            for (let i = 1; i < results.length; i++) {
+                expect(results[i - 1].rank).toBeGreaterThanOrEqual(results[i].rank);
+            }
+        });
+        it('handles empty query', async () => {
+            const { textSearch } = createInMemorySearch(chunks);
+            const results = await textSearch('', { tenantId: 't-1' });
+            expect(results).toHaveLength(0);
+        });
+        it('handles whitespace-only query', async () => {
+            const { textSearch } = createInMemorySearch(chunks);
+            const results = await textSearch('   ', { tenantId: 't-1' });
+            expect(results).toHaveLength(0);
+        });
+    });
+    describe('vectorSearch', () => {
+        it('finds chunks by cosine similarity', async () => {
+            const { vectorSearch } = createInMemorySearch(chunks);
+            // Query embedding close to c-1's embedding [1, 0, 0]
+            const results = await vectorSearch([0.9, 0.1, 0], { tenantId: 't-1' });
+            expect(results.length).toBeGreaterThan(0);
+            expect(results[0].id).toBe('c-1');
+            expect(results[0].similarity).toBeGreaterThan(0.9);
+        });
+        it('only includes chunks with embeddings', async () => {
+            const { vectorSearch } = createInMemorySearch(chunks);
+            const results = await vectorSearch([1, 1, 1], { tenantId: 't-1' });
+            // c-4 has no embedding, so it should not appear
+            const ids = results.map((r) => r.id);
+            expect(ids).not.toContain('c-4');
+            expect(results).toHaveLength(3); // c-1, c-2, c-3
+        });
+        it('returns results sorted by similarity descending', async () => {
+            const { vectorSearch } = createInMemorySearch(chunks);
+            const results = await vectorSearch([1, 0.5, 0.2], { tenantId: 't-1' });
+            for (let i = 1; i < results.length; i++) {
+                expect(results[i - 1].similarity).toBeGreaterThanOrEqual(results[i].similarity);
+            }
+        });
+        it('filters by documentIds when provided', async () => {
+            const { vectorSearch } = createInMemorySearch(chunks);
+            const results = await vectorSearch([1, 1, 1], {
+                tenantId: 't-1',
+                documentIds: ['doc-1'],
+            });
+            for (const r of results) {
+                expect(r.documentId).toBe('doc-1');
+            }
+            expect(results).toHaveLength(2); // c-1 and c-2
+        });
+        it('returns empty when no chunks have embeddings', async () => {
+            const noEmbeddingChunks = [
+                {
+                    id: 'c-x',
+                    documentId: 'doc-1',
+                    content: 'no embedding',
+                    chunkIndex: 0,
+                    pageNumber: null,
+                    metadata: null,
+                },
+            ];
+            const { vectorSearch } = createInMemorySearch(noEmbeddingChunks);
+            const results = await vectorSearch([1, 0], { tenantId: 't-1' });
+            expect(results).toHaveLength(0);
+        });
+    });
+    describe('combined usage', () => {
+        it('both search functions work together', async () => {
+            const { textSearch, vectorSearch } = createInMemorySearch(chunks);
+            const textResults = await textSearch('fox', { tenantId: 't-1' });
+            const vecResults = await vectorSearch([1, 0, 0], { tenantId: 't-1' });
+            expect(textResults.length).toBeGreaterThan(0);
+            expect(vecResults.length).toBeGreaterThan(0);
+        });
+        it('works with empty chunks array', async () => {
+            const { textSearch, vectorSearch } = createInMemorySearch([]);
+            const textResults = await textSearch('anything', { tenantId: 't-1' });
+            const vecResults = await vectorSearch([1, 0, 0], { tenantId: 't-1' });
+            expect(textResults).toHaveLength(0);
+            expect(vecResults).toHaveLength(0);
+        });
+    });
+});
+//# sourceMappingURL=in-memory-search.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"in-memory-search.test.js","sourceRoot":"","sources":["../../src/search/in-memory-search.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAE9C,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAG1D,gEAAgE;AAEhE,MAAM,MAAM,GAAoB;IAC9B;QACE,EAAE,EAAE,KAAK;QACT,UAAU,EAAE,OAAO;QACnB,OAAO,EAAE,6CAA6C;QACtD,UAAU,EAAE,CAAC;QACb,UAAU,EAAE,CAAC;QACb,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;QAC5B,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;KACrB;IACD;QACE,EAAE,EAAE,KAAK;QACT,UAAU,EAAE,OAAO;QACnB,OAAO,EAAE,8BAA8B;QACvC,UAAU,EAAE,CAAC;QACb,UAAU,EAAE,CAAC;QACb,QAAQ,EAAE,IAAI;QACd,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;KACrB;IACD;QACE,EAAE,EAAE,KAAK;QACT,UAAU,EAAE,OAAO;QACnB,OAAO,EAAE,sDAAsD;QAC/D,UAAU,EAAE,CAAC;QACb,UAAU,EAAE,IAAI;QAChB,QAAQ,EAAE,IAAI;QACd,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;KACrB;IACD;QACE,EAAE,EAAE,KAAK;QACT,UAAU,EAAE,OAAO;QACnB,OAAO,EAAE,uCAAuC;QAChD,UAAU,EAAE,CAAC;QACb,UAAU,EAAE,IAAI;QAChB,QAAQ,EAAE,IAAI;QACd,eAAe;KAChB;CACF,CAAC;AAEF,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,EAAE,UAAU,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEpD,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAElE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC1C,4CAA4C;YAC5C,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,CAAC;YAC/C,MAAM,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;YACzB,MAAM,CAAC,EAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,mBAAmB;YAE/C,2CAA2C;YAC3C,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,CAAC;YAC/C,MAAM,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;YACzB,MAAM,CAAC,EAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,EAAE,UAAU,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEpD,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,iBAAiB,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAEzE,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;YACnC,MAAM,EAAE,UAAU,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEpD,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,aAAa,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAErE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,EAAE,UAAU,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEpD,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE;gBACtC,QAAQ,EAAE,KAAK;gBACf,WAAW,EAAE,CAAC,OAAO,CAAC;aACvB,CAAC,CAAC;YAEH,iDAAiD;YACjD,gEAAgE;YAChE,yEAAyE;YACzE,+EAA+E;YAC/E,qCAAqC;YACrC,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBACxB,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,EAAE,UAAU,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEpD,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAE9D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACxC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACtE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;YACnC,MAAM,EAAE,UAAU,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEpD,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAE1D,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,EAAE,UAAU,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEpD,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAE7D,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,KAAK,IAAI,EAAE;YACjD,MAAM,EAAE,YAAY,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEtD,qDAAqD;YACrD,MAAM,OAAO,GAAG,MAAM,YAAa,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAExE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAClC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,EAAE,YAAY,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEtD,MAAM,OAAO,GAAG,MAAM,YAAa,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAEpE,gDAAgD;YAChD,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACrC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACjC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,gBAAgB;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,EAAE,YAAY,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEtD,MAAM,OAAO,GAAG,MAAM,YAAa,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAExE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACxC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;YAClF,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,EAAE,YAAY,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEtD,MAAM,OAAO,GAAG,MAAM,YAAa,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE;gBAC7C,QAAQ,EAAE,KAAK;gBACf,WAAW,EAAE,CAAC,OAAO,CAAC;aACvB,CAAC,CAAC;YAEH,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBACxB,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,iBAAiB,GAAoB;gBACzC;oBACE,EAAE,EAAE,KAAK;oBACT,UAAU,EAAE,OAAO;oBACnB,OAAO,EAAE,cAAc;oBACvB,UAAU,EAAE,CAAC;oBACb,UAAU,EAAE,IAAI;oBAChB,QAAQ,EAAE,IAAI;iBACf;aACF,CAAC;YAEF,MAAM,EAAE,YAAY,EAAE,GAAG,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;YAEjE,MAAM,OAAO,GAAG,MAAM,YAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAEjE,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAElE,MAAM,WAAW,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YACjE,MAAM,UAAU,GAAG,MAAM,YAAa,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAEvE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,GAAG,oBAAoB,CAAC,EAAE,CAAC,CAAC;YAE9D,MAAM,WAAW,GAAG,MAAM,UAAU,CAAC,UAAU,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YACtE,MAAM,UAAU,GAAG,MAAM,YAAa,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAEvE,MAAM,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACpC,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/search/index.d.ts

@@ -0,0 +1,6 @@
+export { cosineSimilarity } from './cosine-similarity.js';
+export { reciprocalRankFusion, createHybridSearch } from './hybrid-search.js';
+export type { TextSearchHit, VectorSearchHit, HybridSearchConfig } from './hybrid-search.js';
+export { createInMemorySearch } from './in-memory-search.js';
+export type { InMemoryChunk } from './in-memory-search.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/search/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/search/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAC3E,YAAY,EAAE,aAAa,EAAE,eAAe,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAC1F,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,YAAY,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC"}

package/dist/search/index.js

@@ -0,0 +1,4 @@
+export { cosineSimilarity } from './cosine-similarity.js';
+export { reciprocalRankFusion, createHybridSearch } from './hybrid-search.js';
+export { createInMemorySearch } from './in-memory-search.js';
+//# sourceMappingURL=index.js.map

package/dist/search/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/search/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAE3E,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC"}

package/dist/testing/in-memory-agent-store.d.ts

@@ -1,7 +1,8 @@
-import type { AgentStore, AgentRecord } from "@sprinterai/core";
+import type { AgentStore, AgentRecord, CreateAgentInput, UpdateAgentInput } from '@sprinterai/core';
 /** In-memory AgentStore for testing. */
 export declare class InMemoryAgentStore implements AgentStore {
     private agents;
+    private nextId;
     /** Seed an agent for test setup. */
     seedAgent(agent: AgentRecord): void;
     getAgentById(id: string): Promise<AgentRecord | null>;
@@ -10,6 +11,9 @@ export declare class InMemoryAgentStore implements AgentStore {
     listAgents(options?: {
         enabledOnly?: boolean;
     }): Promise<AgentRecord[]>;
+    createAgent(input: CreateAgentInput): Promise<AgentRecord>;
+    updateAgent(id: string, input: UpdateAgentInput): Promise<AgentRecord | null>;
+    deleteAgent(id: string): Promise<void>;
     /** Test helper: clear all data. */
     clear(): void;
 }

package/dist/testing/in-memory-agent-store.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"in-memory-agent-store.d.ts","sourceRoot":"","sources":["../../src/testing/in-memory-agent-store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAEhE,wCAAwC;AACxC,qBAAa,kBAAmB,YAAW,UAAU;IACnD,OAAO,CAAC,MAAM,CAAkC;IAEhD,oCAAoC;IACpC,SAAS,CAAC,KAAK,EAAE,WAAW,GAAG,IAAI;IAI7B,YAAY,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAIrD,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAOzD,oBAAoB,IAAI,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAOnD,UAAU,CAAC,OAAO,CAAC,EAAE;QACzB,WAAW,CAAC,EAAE,OAAO,CAAC;KACvB,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAQ1B,mCAAmC;IACnC,KAAK,IAAI,IAAI;CAGd"}
+{"version":3,"file":"in-memory-agent-store.d.ts","sourceRoot":"","sources":["../../src/testing/in-memory-agent-store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAEpG,wCAAwC;AACxC,qBAAa,kBAAmB,YAAW,UAAU;IACnD,OAAO,CAAC,MAAM,CAAkC;IAChD,OAAO,CAAC,MAAM,CAAK;IAEnB,oCAAoC;IACpC,SAAS,CAAC,KAAK,EAAE,WAAW,GAAG,IAAI;IAI7B,YAAY,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAIrD,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAOzD,oBAAoB,IAAI,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAOnD,UAAU,CAAC,OAAO,CAAC,EAAE;QAAE,WAAW,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAQvE,WAAW,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;IAQ1D,WAAW,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAQ7E,WAAW,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI5C,mCAAmC;IACnC,KAAK,IAAI,IAAI;CAGd"}

package/dist/testing/in-memory-agent-store.js

@@ -1,6 +1,7 @@
 /** In-memory AgentStore for testing. */
 export class InMemoryAgentStore {
     agents = new Map();
+    nextId = 1;
     /** Seed an agent for test setup. */
     seedAgent(agent) {
         this.agents.set(agent.id, agent);
@@ -29,6 +30,24 @@ export class InMemoryAgentStore {
         }
         return all;
     }
+    async createAgent(input) {
+        const id = `agent-${String(this.nextId++)}`;
+        const now = new Date().toISOString();
+        const record = { ...input, id, created_at: now, updated_at: now };
+        this.agents.set(id, record);
+        return record;
+    }
+    async updateAgent(id, input) {
+        const existing = this.agents.get(id);
+        if (!existing)
+            return null;
+        const updated = { ...existing, ...input, updated_at: new Date().toISOString() };
+        this.agents.set(id, updated);
+        return updated;
+    }
+    async deleteAgent(id) {
+        this.agents.delete(id);
+    }
     /** Test helper: clear all data. */
     clear() {
         this.agents.clear();

package/dist/testing/in-memory-agent-store.js.map

@@ -1 +1 @@
-{"version":3,"file":"in-memory-agent-store.js","sourceRoot":"","sources":["../../src/testing/in-memory-agent-store.ts"],"names":[],"mappings":"AAEA,wCAAwC;AACxC,MAAM,OAAO,kBAAkB;IACrB,MAAM,GAAG,IAAI,GAAG,EAAuB,CAAC;IAEhD,oCAAoC;IACpC,SAAS,CAAC,KAAkB;QAC1B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,EAAU;QAC3B,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,IAAY;QAC/B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,IAAI,KAAK,CAAC,IAAI,KAAK,IAAI;gBAAE,OAAO,KAAK,CAAC;QACxC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,oBAAoB;QACxB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,IAAI,KAAK,CAAC,UAAU;gBAAE,OAAO,KAAK,CAAC;QACrC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAEhB;QACC,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAC7C,IAAI,OAAO,EAAE,WAAW,EAAE,CAAC;YACzB,OAAO,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACzC,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,mCAAmC;IACnC,KAAK;QACH,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;CACF"}
+{"version":3,"file":"in-memory-agent-store.js","sourceRoot":"","sources":["../../src/testing/in-memory-agent-store.ts"],"names":[],"mappings":"AAEA,wCAAwC;AACxC,MAAM,OAAO,kBAAkB;IACrB,MAAM,GAAG,IAAI,GAAG,EAAuB,CAAC;IACxC,MAAM,GAAG,CAAC,CAAC;IAEnB,oCAAoC;IACpC,SAAS,CAAC,KAAkB;QAC1B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,EAAU;QAC3B,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,IAAY;QAC/B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,IAAI,KAAK,CAAC,IAAI,KAAK,IAAI;gBAAE,OAAO,KAAK,CAAC;QACxC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,oBAAoB;QACxB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,IAAI,KAAK,CAAC,UAAU;gBAAE,OAAO,KAAK,CAAC;QACrC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAAmC;QAClD,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAC7C,IAAI,OAAO,EAAE,WAAW,EAAE,CAAC;YACzB,OAAO,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACzC,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAuB;QACvC,MAAM,EAAE,GAAG,SAAS,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;QAC5C,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,MAAM,GAAgB,EAAE,GAAG,KAAK,EAAE,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC;QAC/E,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QAC5B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,EAAU,EAAE,KAAuB;QACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACrC,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC;QAC3B,MAAM,OAAO,GAAgB,EAAE,GAAG,QAAQ,EAAE,GAAG,KAAK,EAAE,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC;QAC7F,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC7B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,EAAU;QAC1B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACzB,CAAC;IAED,mCAAmC;IACnC,KAAK;QACH,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;CACF"}