@fgv/ts-agent-memory 5.1.0-39 → 5.1.0-41

package/dist/test/unit/vector/inMemoryCosineIndex.test.js

@@ -5,6 +5,10 @@
 import '@fgv/ts-utils-jest';
 import { fail, succeed } from '@fgv/ts-utils';
 import { InMemoryCosineIndex } from '../../../index';
+/** A scope-qualified target from a `(scope, id)` pair. */
+function target(scope, id) {
+    return { scope: scope, id: id };
+}
 /** A trivial record carrying just the id + a marker body, for rebuild tests. */
 function record(id, body = `body-${id}`) {
     return {
@@ -13,6 +17,10 @@ function record(id, body = `body-${id}`) {
         body
     };
 }
+/** A scope-qualified record entry for a rebuild source. */
+function scoped(scope, id, body) {
+    return { target: target(scope, id), record: record(id, body) };
+}
 /** A scripted record source for `rebuild`. */
 class FakeSource {
     constructor(result) {
@@ -24,30 +32,51 @@ class FakeSource {
 }
 describe('InMemoryCosineIndex', () => {
     describe('add', () => {
-        test('returns the id as the embedding ref and tracks size', async () => {
+        test('returns the scoped-target key as the embedding ref and tracks size', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
             expect(index.size).toBe(0);
-            expect(await index.add('a', Float32Array.from([1, 0]))).toSucceedWith('a');
+            const t = target('conv-a', 'turn-1');
+            expect(await index.add(t, Float32Array.from([1, 0]))).toSucceedWith(`conv-a\0turn-1`);
             expect(index.size).toBe(1);
         });
-        test('replaces the vector on a same-id add', async () => {
+        test('replaces the vector on a same-target add', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
-            expect(await index.add('a', Float32Array.from([1, 0]))).toSucceed();
-            expect(await index.add('a', Float32Array.from([0, 1]))).toSucceed();
+            const t = target('conv-a', 'turn-1');
+            expect(await index.add(t, Float32Array.from([1, 0]))).toSucceed();
+            expect(await index.add(t, Float32Array.from([0, 1]))).toSucceed();
             expect(index.size).toBe(1);
             // The query now matches the replacement vector ([0,1]) not the original.
             expect(await index.query(Float32Array.from([0, 1]), 1)).toSucceedAndSatisfy((hits) => {
                 expect(hits[0].score).toBeCloseTo(1);
             });
         });
+        test('same stem in different scopes are distinct entries — the second add does NOT clobber the first', async () => {
+            const index = InMemoryCosineIndex.create().orThrow();
+            const a = target('conv-a', 'turn-3');
+            const b = target('conv-b', 'turn-3');
+            // Two records with the identical stem, orthogonal embeddings.
+            expect(await index.add(a, Float32Array.from([1, 0]))).toSucceed();
+            expect(await index.add(b, Float32Array.from([0, 1]))).toSucceed();
+            // The bug: a bare-id index would have collapsed these to one entry.
+            expect(index.size).toBe(2);
+            // A query surfaces BOTH as distinct hits carrying distinct scoped targets.
+            expect(await index.query(Float32Array.from([1, 1]), 5)).toSucceedAndSatisfy((hits) => {
+                expect(hits).toHaveLength(2);
+                expect(hits.map((h) => h.target)).toEqual(expect.arrayContaining([a, b]));
+                // Each retains its OWN embedding: a matches [1,0], b matches [0,1].
+                const byScope = new Map(hits.map((h) => [h.target.scope, h.score]));
+                expect(byScope.get('conv-a')).toBeCloseTo(1 / Math.sqrt(2));
+                expect(byScope.get('conv-b')).toBeCloseTo(1 / Math.sqrt(2));
+            });
+        });
         test('fails loudly on an empty vector', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
-            expect(await index.add('a', new Float32Array(0))).toFailWith(/empty vector/i);
+            expect(await index.add(target('conv-a', 'turn-1'), new Float32Array(0))).toFailWith(/empty vector/i);
         });
         test('stores a defensive copy — mutating the caller buffer after add does not corrupt the index', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
             const buffer = Float32Array.from([1, 0]);
-            (await index.add('a', buffer)).orThrow();
+            (await index.add(target('conv-a', 'turn-1'), buffer)).orThrow();
             // Mutate the caller's buffer after the add; the stored vector must be unaffected.
             buffer[0] = 0;
             buffer[1] = 1;
@@ -58,22 +87,22 @@ describe('InMemoryCosineIndex', () => {
         });
         test('fails loudly on a dimension mismatch against the established dimension', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
-            expect(await index.add('a', Float32Array.from([1, 0]))).toSucceed();
-            expect(await index.add('b', Float32Array.from([1, 0, 0]))).toFailWith(/dimension 3 does not match index dimension 2/i);
+            expect(await index.add(target('conv-a', 'a'), Float32Array.from([1, 0]))).toSucceed();
+            expect(await index.add(target('conv-a', 'b'), Float32Array.from([1, 0, 0]))).toFailWith(/dimension 3 does not match index dimension 2/i);
         });
     });
     describe('query', () => {
         async function seeded() {
             const index = InMemoryCosineIndex.create().orThrow();
-            (await index.add('a', Float32Array.from([1, 0]))).orThrow();
-            (await index.add('b', Float32Array.from([0, 1]))).orThrow();
-            (await index.add('c', Float32Array.from([1, 1]))).orThrow();
+            (await index.add(target('s', 'a'), Float32Array.from([1, 0]))).orThrow();
+            (await index.add(target('s', 'b'), Float32Array.from([0, 1]))).orThrow();
+            (await index.add(target('s', 'c'), Float32Array.from([1, 1]))).orThrow();
             return index;
         }
         test('returns hits in descending cosine-similarity order', async () => {
             const index = await seeded();
             expect(await index.query(Float32Array.from([1, 0]), 3)).toSucceedAndSatisfy((hits) => {
-                expect(hits.map((h) => h.id)).toEqual(['a', 'c', 'b']);
+                expect(hits.map((h) => h.target.id)).toEqual(['a', 'c', 'b']);
                 expect(hits[0].score).toBeCloseTo(1);
                 expect(hits[1].score).toBeCloseTo(1 / Math.sqrt(2));
                 expect(hits[2].score).toBeCloseTo(0);
@@ -82,7 +111,7 @@ describe('InMemoryCosineIndex', () => {
         test('truncates to topK', async () => {
             const index = await seeded();
             expect(await index.query(Float32Array.from([1, 0]), 2)).toSucceedAndSatisfy((hits) => {
-                expect(hits.map((h) => h.id)).toEqual(['a', 'c']);
+                expect(hits.map((h) => h.target.id)).toEqual(['a', 'c']);
             });
         });
         test('returns empty for a non-positive topK', async () => {
@@ -100,16 +129,16 @@ describe('InMemoryCosineIndex', () => {
         });
         test('scores a degenerate (zero-magnitude) stored vector as 0 rather than NaN', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
-            (await index.add('zero', Float32Array.from([0, 0]))).orThrow();
-            (await index.add('unit', Float32Array.from([1, 0]))).orThrow();
+            (await index.add(target('s', 'zero'), Float32Array.from([0, 0]))).orThrow();
+            (await index.add(target('s', 'unit'), Float32Array.from([1, 0]))).orThrow();
             expect(await index.query(Float32Array.from([1, 0]), 2)).toSucceedAndSatisfy((hits) => {
-                expect(hits.map((h) => h.id)).toEqual(['unit', 'zero']);
+                expect(hits.map((h) => h.target.id)).toEqual(['unit', 'zero']);
                 expect(hits[1].score).toBe(0);
             });
         });
         test('scores against a degenerate (zero-magnitude) query as 0 rather than NaN', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
-            (await index.add('unit', Float32Array.from([1, 0]))).orThrow();
+            (await index.add(target('s', 'unit'), Float32Array.from([1, 0]))).orThrow();
             expect(await index.query(Float32Array.from([0, 0]), 1)).toSucceedAndSatisfy((hits) => {
                 expect(hits[0].score).toBe(0);
             });
@@ -118,17 +147,35 @@ describe('InMemoryCosineIndex', () => {
     describe('remove', () => {
         test('removes a vector and is reflected in subsequent queries', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
-            (await index.add('a', Float32Array.from([1, 0]))).orThrow();
-            (await index.add('b', Float32Array.from([0, 1]))).orThrow();
-            expect(await index.remove('a')).toSucceedWith('a');
+            const a = target('s', 'a');
+            const b = target('s', 'b');
+            (await index.add(a, Float32Array.from([1, 0]))).orThrow();
+            (await index.add(b, Float32Array.from([0, 1]))).orThrow();
+            expect(await index.remove(a)).toSucceedWith(a);
             expect(index.size).toBe(1);
             expect(await index.query(Float32Array.from([1, 0]), 5)).toSucceedAndSatisfy((hits) => {
-                expect(hits.map((h) => h.id)).toEqual(['b']);
+                expect(hits.map((h) => h.target.id)).toEqual(['b']);
+            });
+        });
+        test('removes only the scoped target — a same-stem record in another scope is left intact', async () => {
+            const index = InMemoryCosineIndex.create().orThrow();
+            const a = target('conv-a', 'turn-3');
+            const b = target('conv-b', 'turn-3');
+            (await index.add(a, Float32Array.from([1, 0]))).orThrow();
+            (await index.add(b, Float32Array.from([0, 1]))).orThrow();
+            expect(await index.remove(a)).toSucceedWith(a);
+            expect(index.size).toBe(1);
+            // conv-b's embedding survives untouched.
+            expect(await index.query(Float32Array.from([0, 1]), 5)).toSucceedAndSatisfy((hits) => {
+                expect(hits).toHaveLength(1);
+                expect(hits[0].target).toEqual(b);
+                expect(hits[0].score).toBeCloseTo(1);
             });
         });
-        test('is idempotent — removing an absent id still succeeds', async () => {
+        test('is idempotent — removing an absent target still succeeds', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
-            expect(await index.remove('missing')).toSucceedWith('missing');
+            const t = target('s', 'missing');
+            expect(await index.remove(t)).toSucceedWith(t);
         });
     });
     describe('rebuild', () => {
@@ -137,21 +184,33 @@ describe('InMemoryCosineIndex', () => {
             const code = r.envelope.id.charCodeAt(0);
             return Promise.resolve(succeed(Float32Array.from([code, 1])));
         };
-        test('re-embeds every record from the source and reports the count', async () => {
+        test('re-embeds every scoped record from the source and reports the count', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
-            const source = new FakeSource(succeed([record('a'), record('b'), record('c')]));
+            const source = new FakeSource(succeed([scoped('s', 'a'), scoped('s', 'b'), scoped('s', 'c')]));
             expect(await index.rebuild(source, embed)).toSucceedWith(3);
             expect(index.size).toBe(3);
             expect(await index.query(Float32Array.from([99, 1]), 1)).toSucceedAndSatisfy((hits) => {
                 // 'c' = char code 99, the closest to the query vector.
-                expect(hits[0].id).toBe('c');
+                expect(hits[0].target.id).toBe('c');
+            });
+        });
+        test('keeps same-stem records under different scopes distinct across a rebuild', async () => {
+            const index = InMemoryCosineIndex.create().orThrow();
+            const source = new FakeSource(succeed([scoped('conv-a', 'turn-3'), scoped('conv-b', 'turn-3')]));
+            // The embedder keys only off the (shared) id, so a bare-id rebuild would
+            // index just one entry; the scoped rebuild indexes both.
+            expect(await index.rebuild(source, embed)).toSucceedWith(2);
+            expect(index.size).toBe(2);
+            expect(await index.query(Float32Array.from([116, 1]), 5)).toSucceedAndSatisfy((hits) => {
+                expect(hits).toHaveLength(2);
+                expect(hits.map((h) => h.target.scope).sort()).toEqual(['conv-a', 'conv-b']);
             });
         });
         test('clears prior contents and re-establishes the dimension', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
             // Seed a 3-dim vector, then rebuild with a 2-dim embedder.
-            (await index.add('old', Float32Array.from([1, 2, 3]))).orThrow();
-            const source = new FakeSource(succeed([record('a')]));
+            (await index.add(target('s', 'old'), Float32Array.from([1, 2, 3]))).orThrow();
+            const source = new FakeSource(succeed([scoped('s', 'a')]));
             expect(await index.rebuild(source, embed)).toSucceedWith(1);
             expect(index.size).toBe(1);
             // The 2-dim query now succeeds (dimension was reset by rebuild).
@@ -160,7 +219,7 @@ describe('InMemoryCosineIndex', () => {
         test('fails loudly when the source list fails', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
             // Seed an entry so the reset-before-list rollback is observable, not just "stayed empty".
-            (await index.add('seed', Float32Array.from([1, 1]))).orThrow();
+            (await index.add(target('s', 'seed'), Float32Array.from([1, 1]))).orThrow();
             const source = new FakeSource(fail('disk gone'));
             expect(await index.rebuild(source, embed)).toFailWith(/failed to list records: disk gone/i);
             expect(index.size).toBe(0);
@@ -168,21 +227,22 @@ describe('InMemoryCosineIndex', () => {
         test('fails loudly and rolls back to empty when an embedding fails mid-rebuild', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
             // Seed an existing entry so the rollback (not merely "stayed empty") is observable.
-            (await index.add('seed', Float32Array.from([1, 1]))).orThrow();
+            (await index.add(target('s', 'seed'), Float32Array.from([1, 1]))).orThrow();
             // First record embeds fine, the second fails — so a naive impl would leave
             // record 'a' indexed; the rollback must clear it.
             let calls = 0;
-            const flakyEmbed = (r) => {
+            const flakyEmbed = () => {
                 calls += 1;
                 return Promise.resolve(calls === 1 ? succeed(Float32Array.from([1, 1])) : fail('no model'));
             };
-            const source = new FakeSource(succeed([record('a'), record('b')]));
-            expect(await index.rebuild(source, flakyEmbed)).toFailWith(/embedding 'b' failed: no model/i);
+            const source = new FakeSource(succeed([scoped('conv-a', 'turn-1'), scoped('conv-b', 'turn-1')]));
+            // The failing record's error names its scope-qualified key.
+            expect(await index.rebuild(source, flakyEmbed)).toFailWith(/embedding 'conv-b\0turn-1' failed: no model/i);
             expect(index.size).toBe(0);
         });
         test('fails loudly and rolls back to empty when adding an embedded vector fails', async () => {
             const index = InMemoryCosineIndex.create().orThrow();
-            const source = new FakeSource(succeed([record('a')]));
+            const source = new FakeSource(succeed([scoped('s', 'a')]));
             const emptyEmbed = () => Promise.resolve(succeed(new Float32Array(0)));
             expect(await index.rebuild(source, emptyEmbed)).toFailWith(/empty vector/i);
             expect(index.size).toBe(0);

package/dist/test/unit/vector/inMemoryCosineIndex.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"inMemoryCosineIndex.test.js","sourceRoot":"","sources":["../../../../src/test/unit/vector/inMemoryCosineIndex.test.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,oBAAoB,CAAC;AAC5B,OAAO,EAAU,IAAI,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACtD,OAAO,EAIL,mBAAmB,EAEpB,MAAM,gBAAgB,CAAC;AAExB,gFAAgF;AAChF,SAAS,MAAM,CAAC,EAAU,EAAE,OAAe,QAAQ,EAAE,EAAE;IACrD,OAAO;QACL,0EAA0E;QAC1E,QAAQ,EAAE,EAAE,EAAE,EAAE,EAAc,EAAwC;QACtE,IAAI;KACL,CAAC;AACJ,CAAC;AAED,8CAA8C;AAC9C,MAAM,UAAU;IAEd,YAAmB,MAAqD;QACtE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IACM,IAAI;QACT,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC;CACF;AAED,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;QACnB,IAAI,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;YACvF,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YAChF,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YAChF,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,yEAAyE;YACzE,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YACvC,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;QAC5F,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,2FAA2F,EAAE,KAAK,IAAI,EAAE;YAC3G,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACzC,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YACrD,kFAAkF;YAClF,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACd,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACd,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,yEAAyE;gBACzE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YACvC,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACxF,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YAChF,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAC/E,+CAA+C,CAChD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,KAAK,UAAU,MAAM;YACnB,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YACxE,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YACxE,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YACxE,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,KAAK,GAAG,MAAM,MAAM,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;gBACvD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;gBACrC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;gBACpD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YACvC,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;YACnC,MAAM,KAAK,GAAG,MAAM,MAAM,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YACpD,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,KAAK,GAAG,MAAM,MAAM,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QAClF,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,KAAK,GAAG,MAAM,MAAM,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CACnE,qDAAqD,CACtD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;YACzF,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAkB,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC3E,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAkB,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC3E,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;gBACxD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAChC,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;YACzF,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAkB,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC3E,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAChC,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,IAAI,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YACxE,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YACxE,MAAM,CAAC,MAAM,KAAK,CAAC,MAAM,CAAC,GAAe,CAAC,CAAC,CAAC,aAAa,CAAC,GAAe,CAAC,CAAC;YAC3E,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAC/C,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,MAAM,KAAK,CAAC,MAAM,CAAC,SAAqB,CAAC,CAAC,CAAC,aAAa,CAAC,SAAqB,CAAC,CAAC;QACzF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,MAAM,KAAK,GAAG,CAAC,CAAyB,EAAiC,EAAE;YACzE,kEAAkE;YAClE,MAAM,IAAI,GAAY,CAAC,CAAC,QAAQ,CAAC,EAAa,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAC7D,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC;QAEF,IAAI,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAChF,MAAM,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAC1E,CAAC,IAAoC,EAAE,EAAE;gBACvC,uDAAuD;gBACvD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC/B,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,2DAA2D;YAC3D,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,KAAiB,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC7E,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,iEAAiE;YACjE,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,0FAA0F;YAC1F,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAkB,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC3E,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,oCAAoC,CAAC,CAAC;YAC5F,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;YAC1F,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,oFAAoF;YACpF,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAkB,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC3E,2EAA2E;YAC3E,kDAAkD;YAClD,IAAI,KAAK,GAAW,CAAC,CAAC;YACtB,MAAM,UAAU,GAAG,CAAC,CAAyB,EAAiC,EAAE;gBAC9E,KAAK,IAAI,CAAC,CAAC;gBACX,OAAO,OAAO,CAAC,OAAO,CAAC,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;YAC9F,CAAC,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,iCAAiC,CAAC,CAAC;YAC9F,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YAC3F,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACtD,MAAM,UAAU,GAAG,GAAkC,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACtG,MAAM,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;YAC5E,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["/*\n * Copyright (c) 2026 Erik Fortune\n * SPDX-License-Identifier: MIT\n */\n\nimport '@fgv/ts-utils-jest';\nimport { Result, fail, succeed } from '@fgv/ts-utils';\nimport {\n  IMemoryRecord,\n  IMemoryRecordSource,\n  IVectorQueryHit,\n  InMemoryCosineIndex,\n  MemoryId\n} from '../../../index';\n\n/** A trivial record carrying just the id + a marker body, for rebuild tests. */\nfunction record(id: string, body: string = `body-${id}`): IMemoryRecord<unknown> {\n  return {\n    // The cosine index only reads `envelope.id`; a minimal envelope suffices.\n    envelope: { id: id as MemoryId } as IMemoryRecord<unknown>['envelope'],\n    body\n  };\n}\n\n/** A scripted record source for `rebuild`. */\nclass FakeSource implements IMemoryRecordSource {\n  private readonly _result: Result<ReadonlyArray<IMemoryRecord<unknown>>>;\n  public constructor(result: Result<ReadonlyArray<IMemoryRecord<unknown>>>) {\n    this._result = result;\n  }\n  public list(): Promise<Result<ReadonlyArray<IMemoryRecord<unknown>>>> {\n    return Promise.resolve(this._result);\n  }\n}\n\ndescribe('InMemoryCosineIndex', () => {\n  describe('add', () => {\n    test('returns the id as the embedding ref and tracks size', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      expect(index.size).toBe(0);\n      expect(await index.add('a' as MemoryId, Float32Array.from([1, 0]))).toSucceedWith('a');\n      expect(index.size).toBe(1);\n    });\n\n    test('replaces the vector on a same-id add', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      expect(await index.add('a' as MemoryId, Float32Array.from([1, 0]))).toSucceed();\n      expect(await index.add('a' as MemoryId, Float32Array.from([0, 1]))).toSucceed();\n      expect(index.size).toBe(1);\n      // The query now matches the replacement vector ([0,1]) not the original.\n      expect(await index.query(Float32Array.from([0, 1]), 1)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits[0].score).toBeCloseTo(1);\n        }\n      );\n    });\n\n    test('fails loudly on an empty vector', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      expect(await index.add('a' as MemoryId, new Float32Array(0))).toFailWith(/empty vector/i);\n    });\n\n    test('stores a defensive copy — mutating the caller buffer after add does not corrupt the index', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      const buffer = Float32Array.from([1, 0]);\n      (await index.add('a' as MemoryId, buffer)).orThrow();\n      // Mutate the caller's buffer after the add; the stored vector must be unaffected.\n      buffer[0] = 0;\n      buffer[1] = 1;\n      expect(await index.query(Float32Array.from([1, 0]), 1)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          // Still maximally similar to [1,0] — proves the index kept its own copy.\n          expect(hits[0].score).toBeCloseTo(1);\n        }\n      );\n    });\n\n    test('fails loudly on a dimension mismatch against the established dimension', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      expect(await index.add('a' as MemoryId, Float32Array.from([1, 0]))).toSucceed();\n      expect(await index.add('b' as MemoryId, Float32Array.from([1, 0, 0]))).toFailWith(\n        /dimension 3 does not match index dimension 2/i\n      );\n    });\n  });\n\n  describe('query', () => {\n    async function seeded(): Promise<InMemoryCosineIndex> {\n      const index = InMemoryCosineIndex.create().orThrow();\n      (await index.add('a' as MemoryId, Float32Array.from([1, 0]))).orThrow();\n      (await index.add('b' as MemoryId, Float32Array.from([0, 1]))).orThrow();\n      (await index.add('c' as MemoryId, Float32Array.from([1, 1]))).orThrow();\n      return index;\n    }\n\n    test('returns hits in descending cosine-similarity order', async () => {\n      const index = await seeded();\n      expect(await index.query(Float32Array.from([1, 0]), 3)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits.map((h) => h.id)).toEqual(['a', 'c', 'b']);\n          expect(hits[0].score).toBeCloseTo(1);\n          expect(hits[1].score).toBeCloseTo(1 / Math.sqrt(2));\n          expect(hits[2].score).toBeCloseTo(0);\n        }\n      );\n    });\n\n    test('truncates to topK', async () => {\n      const index = await seeded();\n      expect(await index.query(Float32Array.from([1, 0]), 2)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits.map((h) => h.id)).toEqual(['a', 'c']);\n        }\n      );\n    });\n\n    test('returns empty for a non-positive topK', async () => {\n      const index = await seeded();\n      expect(await index.query(Float32Array.from([1, 0]), 0)).toSucceedWith([]);\n      expect(await index.query(Float32Array.from([1, 0]), -1)).toSucceedWith([]);\n    });\n\n    test('returns empty when the index is empty (no dimension check)', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      expect(await index.query(Float32Array.from([1, 2, 3, 4]), 5)).toSucceedWith([]);\n    });\n\n    test('fails loudly on a query-dimension mismatch', async () => {\n      const index = await seeded();\n      expect(await index.query(Float32Array.from([1, 0, 0]), 3)).toFailWith(\n        /query dimension 3 does not match index dimension 2/i\n      );\n    });\n\n    test('scores a degenerate (zero-magnitude) stored vector as 0 rather than NaN', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      (await index.add('zero' as MemoryId, Float32Array.from([0, 0]))).orThrow();\n      (await index.add('unit' as MemoryId, Float32Array.from([1, 0]))).orThrow();\n      expect(await index.query(Float32Array.from([1, 0]), 2)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits.map((h) => h.id)).toEqual(['unit', 'zero']);\n          expect(hits[1].score).toBe(0);\n        }\n      );\n    });\n\n    test('scores against a degenerate (zero-magnitude) query as 0 rather than NaN', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      (await index.add('unit' as MemoryId, Float32Array.from([1, 0]))).orThrow();\n      expect(await index.query(Float32Array.from([0, 0]), 1)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits[0].score).toBe(0);\n        }\n      );\n    });\n  });\n\n  describe('remove', () => {\n    test('removes a vector and is reflected in subsequent queries', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      (await index.add('a' as MemoryId, Float32Array.from([1, 0]))).orThrow();\n      (await index.add('b' as MemoryId, Float32Array.from([0, 1]))).orThrow();\n      expect(await index.remove('a' as MemoryId)).toSucceedWith('a' as MemoryId);\n      expect(index.size).toBe(1);\n      expect(await index.query(Float32Array.from([1, 0]), 5)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits.map((h) => h.id)).toEqual(['b']);\n        }\n      );\n    });\n\n    test('is idempotent — removing an absent id still succeeds', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      expect(await index.remove('missing' as MemoryId)).toSucceedWith('missing' as MemoryId);\n    });\n  });\n\n  describe('rebuild', () => {\n    const embed = (r: IMemoryRecord<unknown>): Promise<Result<Float32Array>> => {\n      // Deterministic: encode the id's first char code into a 2-vector.\n      const code: number = (r.envelope.id as string).charCodeAt(0);\n      return Promise.resolve(succeed(Float32Array.from([code, 1])));\n    };\n\n    test('re-embeds every record from the source and reports the count', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      const source = new FakeSource(succeed([record('a'), record('b'), record('c')]));\n      expect(await index.rebuild(source, embed)).toSucceedWith(3);\n      expect(index.size).toBe(3);\n      expect(await index.query(Float32Array.from([99, 1]), 1)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          // 'c' = char code 99, the closest to the query vector.\n          expect(hits[0].id).toBe('c');\n        }\n      );\n    });\n\n    test('clears prior contents and re-establishes the dimension', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      // Seed a 3-dim vector, then rebuild with a 2-dim embedder.\n      (await index.add('old' as MemoryId, Float32Array.from([1, 2, 3]))).orThrow();\n      const source = new FakeSource(succeed([record('a')]));\n      expect(await index.rebuild(source, embed)).toSucceedWith(1);\n      expect(index.size).toBe(1);\n      // The 2-dim query now succeeds (dimension was reset by rebuild).\n      expect(await index.query(Float32Array.from([97, 1]), 1)).toSucceed();\n    });\n\n    test('fails loudly when the source list fails', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      // Seed an entry so the reset-before-list rollback is observable, not just \"stayed empty\".\n      (await index.add('seed' as MemoryId, Float32Array.from([1, 1]))).orThrow();\n      const source = new FakeSource(fail('disk gone'));\n      expect(await index.rebuild(source, embed)).toFailWith(/failed to list records: disk gone/i);\n      expect(index.size).toBe(0);\n    });\n\n    test('fails loudly and rolls back to empty when an embedding fails mid-rebuild', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      // Seed an existing entry so the rollback (not merely \"stayed empty\") is observable.\n      (await index.add('seed' as MemoryId, Float32Array.from([1, 1]))).orThrow();\n      // First record embeds fine, the second fails — so a naive impl would leave\n      // record 'a' indexed; the rollback must clear it.\n      let calls: number = 0;\n      const flakyEmbed = (r: IMemoryRecord<unknown>): Promise<Result<Float32Array>> => {\n        calls += 1;\n        return Promise.resolve(calls === 1 ? succeed(Float32Array.from([1, 1])) : fail('no model'));\n      };\n      const source = new FakeSource(succeed([record('a'), record('b')]));\n      expect(await index.rebuild(source, flakyEmbed)).toFailWith(/embedding 'b' failed: no model/i);\n      expect(index.size).toBe(0);\n    });\n\n    test('fails loudly and rolls back to empty when adding an embedded vector fails', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      const source = new FakeSource(succeed([record('a')]));\n      const emptyEmbed = (): Promise<Result<Float32Array>> => Promise.resolve(succeed(new Float32Array(0)));\n      expect(await index.rebuild(source, emptyEmbed)).toFailWith(/empty vector/i);\n      expect(index.size).toBe(0);\n    });\n  });\n});\n"]}
+{"version":3,"file":"inMemoryCosineIndex.test.js","sourceRoot":"","sources":["../../../../src/test/unit/vector/inMemoryCosineIndex.test.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,oBAAoB,CAAC;AAC5B,OAAO,EAAU,IAAI,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACtD,OAAO,EAML,mBAAmB,EAGpB,MAAM,gBAAgB,CAAC;AAExB,0DAA0D;AAC1D,SAAS,MAAM,CAAC,KAAa,EAAE,EAAU;IACvC,OAAO,EAAE,KAAK,EAAE,KAAuB,EAAE,EAAE,EAAE,EAAc,EAAE,CAAC;AAChE,CAAC;AAED,gFAAgF;AAChF,SAAS,MAAM,CAAC,EAAU,EAAE,OAAe,QAAQ,EAAE,EAAE;IACrD,OAAO;QACL,0EAA0E;QAC1E,QAAQ,EAAE,EAAE,EAAE,EAAE,EAAc,EAAwC;QACtE,IAAI;KACL,CAAC;AACJ,CAAC;AAED,2DAA2D;AAC3D,SAAS,MAAM,CAAC,KAAa,EAAE,EAAU,EAAE,IAAa;IACtD,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,CAAC;AACjE,CAAC;AAED,8CAA8C;AAC9C,MAAM,UAAU;IAEd,YAAmB,MAAkD;QACnE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IACM,IAAI;QACT,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC;CACF;AAED,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;QACnB,IAAI,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YACpF,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC;YACtF,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YAClE,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YAClE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,yEAAyE;YACzE,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YACvC,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,gGAAgG,EAAE,KAAK,IAAI,EAAE;YAChH,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACrC,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACrC,8DAA8D;YAC9D,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YAClE,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YAClE,oEAAoE;YACpE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,2EAA2E;YAC3E,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;gBAC7B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC1E,oEAAoE;gBACpE,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACpE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,QAA0B,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9E,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,QAA0B,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YAChF,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;QACvG,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,2FAA2F,EAAE,KAAK,IAAI,EAAE;YAC3G,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACzC,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAChE,kFAAkF;YAClF,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACd,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACd,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,yEAAyE;gBACzE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YACvC,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACxF,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YACtF,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CACrF,+CAA+C,CAChD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,KAAK,UAAU,MAAM;YACnB,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YACzE,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YACzE,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YACzE,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,KAAK,GAAG,MAAM,MAAM,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;gBAC9D,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;gBACrC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;gBACpD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YACvC,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;YACnC,MAAM,KAAK,GAAG,MAAM,MAAM,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;YAC3D,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,KAAK,GAAG,MAAM,MAAM,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QAClF,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,KAAK,GAAG,MAAM,MAAM,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CACnE,qDAAqD,CACtD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;YACzF,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC5E,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC5E,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;gBAC/D,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAChC,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;YACzF,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC5E,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAChC,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,IAAI,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,GAAG,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAC3B,MAAM,CAAC,GAAG,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAC3B,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC1D,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC1D,MAAM,CAAC,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YACtD,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,qFAAqF,EAAE,KAAK,IAAI,EAAE;YACrG,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACrC,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACrC,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC1D,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC1D,MAAM,CAAC,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,yCAAyC;YACzC,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CACzE,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;gBAC7B,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBAClC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YACvC,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,CAAC,GAAG,MAAM,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YACjC,MAAM,CAAC,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,MAAM,KAAK,GAAG,CAAC,CAAyB,EAAiC,EAAE;YACzE,kEAAkE;YAClE,MAAM,IAAI,GAAY,CAAC,CAAC,QAAQ,CAAC,EAAa,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAC7D,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC;QAEF,IAAI,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACrF,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/F,MAAM,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAC1E,CAAC,IAAoC,EAAE,EAAE;gBACvC,uDAAuD;gBACvD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACtC,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;YAC1F,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;YACjG,yEAAyE;YACzE,yDAAyD;YACzD,MAAM,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAC3E,CAAC,IAAoC,EAAE,EAAE;gBACvC,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;gBAC7B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;YAC/E,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,2DAA2D;YAC3D,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC9E,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,iEAAiE;YACjE,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,0FAA0F;YAC1F,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC5E,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,oCAAoC,CAAC,CAAC;YAC5F,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;YAC1F,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,oFAAoF;YACpF,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAC5E,2EAA2E;YAC3E,kDAAkD;YAClD,IAAI,KAAK,GAAW,CAAC,CAAC;YACtB,MAAM,UAAU,GAAG,GAAkC,EAAE;gBACrD,KAAK,IAAI,CAAC,CAAC;gBACX,OAAO,OAAO,CAAC,OAAO,CAAC,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;YAC9F,CAAC,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;YACjG,4DAA4D;YAC5D,MAAM,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,UAAU,CACxD,8CAA8C,CAC/C,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YAC3F,MAAM,KAAK,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACrD,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,UAAU,GAAG,GAAkC,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACtG,MAAM,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;YAC5E,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["/*\n * Copyright (c) 2026 Erik Fortune\n * SPDX-License-Identifier: MIT\n */\n\nimport '@fgv/ts-utils-jest';\nimport { Result, fail, succeed } from '@fgv/ts-utils';\nimport {\n  IEdgeTarget,\n  IMemoryRecord,\n  IMemoryRecordSource,\n  IScopedMemoryRecord,\n  IVectorQueryHit,\n  InMemoryCosineIndex,\n  MemoryId,\n  MemoryScopeKey\n} from '../../../index';\n\n/** A scope-qualified target from a `(scope, id)` pair. */\nfunction target(scope: string, id: string): IEdgeTarget {\n  return { scope: scope as MemoryScopeKey, id: id as MemoryId };\n}\n\n/** A trivial record carrying just the id + a marker body, for rebuild tests. */\nfunction record(id: string, body: string = `body-${id}`): IMemoryRecord<unknown> {\n  return {\n    // The cosine index only reads `envelope.id`; a minimal envelope suffices.\n    envelope: { id: id as MemoryId } as IMemoryRecord<unknown>['envelope'],\n    body\n  };\n}\n\n/** A scope-qualified record entry for a rebuild source. */\nfunction scoped(scope: string, id: string, body?: string): IScopedMemoryRecord {\n  return { target: target(scope, id), record: record(id, body) };\n}\n\n/** A scripted record source for `rebuild`. */\nclass FakeSource implements IMemoryRecordSource {\n  private readonly _result: Result<ReadonlyArray<IScopedMemoryRecord>>;\n  public constructor(result: Result<ReadonlyArray<IScopedMemoryRecord>>) {\n    this._result = result;\n  }\n  public list(): Promise<Result<ReadonlyArray<IScopedMemoryRecord>>> {\n    return Promise.resolve(this._result);\n  }\n}\n\ndescribe('InMemoryCosineIndex', () => {\n  describe('add', () => {\n    test('returns the scoped-target key as the embedding ref and tracks size', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      expect(index.size).toBe(0);\n      const t = target('conv-a', 'turn-1');\n      expect(await index.add(t, Float32Array.from([1, 0]))).toSucceedWith(`conv-a\\0turn-1`);\n      expect(index.size).toBe(1);\n    });\n\n    test('replaces the vector on a same-target add', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      const t = target('conv-a', 'turn-1');\n      expect(await index.add(t, Float32Array.from([1, 0]))).toSucceed();\n      expect(await index.add(t, Float32Array.from([0, 1]))).toSucceed();\n      expect(index.size).toBe(1);\n      // The query now matches the replacement vector ([0,1]) not the original.\n      expect(await index.query(Float32Array.from([0, 1]), 1)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits[0].score).toBeCloseTo(1);\n        }\n      );\n    });\n\n    test('same stem in different scopes are distinct entries — the second add does NOT clobber the first', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      const a = target('conv-a', 'turn-3');\n      const b = target('conv-b', 'turn-3');\n      // Two records with the identical stem, orthogonal embeddings.\n      expect(await index.add(a, Float32Array.from([1, 0]))).toSucceed();\n      expect(await index.add(b, Float32Array.from([0, 1]))).toSucceed();\n      // The bug: a bare-id index would have collapsed these to one entry.\n      expect(index.size).toBe(2);\n      // A query surfaces BOTH as distinct hits carrying distinct scoped targets.\n      expect(await index.query(Float32Array.from([1, 1]), 5)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits).toHaveLength(2);\n          expect(hits.map((h) => h.target)).toEqual(expect.arrayContaining([a, b]));\n          // Each retains its OWN embedding: a matches [1,0], b matches [0,1].\n          const byScope = new Map(hits.map((h) => [h.target.scope, h.score]));\n          expect(byScope.get('conv-a' as MemoryScopeKey)).toBeCloseTo(1 / Math.sqrt(2));\n          expect(byScope.get('conv-b' as MemoryScopeKey)).toBeCloseTo(1 / Math.sqrt(2));\n        }\n      );\n    });\n\n    test('fails loudly on an empty vector', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      expect(await index.add(target('conv-a', 'turn-1'), new Float32Array(0))).toFailWith(/empty vector/i);\n    });\n\n    test('stores a defensive copy — mutating the caller buffer after add does not corrupt the index', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      const buffer = Float32Array.from([1, 0]);\n      (await index.add(target('conv-a', 'turn-1'), buffer)).orThrow();\n      // Mutate the caller's buffer after the add; the stored vector must be unaffected.\n      buffer[0] = 0;\n      buffer[1] = 1;\n      expect(await index.query(Float32Array.from([1, 0]), 1)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          // Still maximally similar to [1,0] — proves the index kept its own copy.\n          expect(hits[0].score).toBeCloseTo(1);\n        }\n      );\n    });\n\n    test('fails loudly on a dimension mismatch against the established dimension', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      expect(await index.add(target('conv-a', 'a'), Float32Array.from([1, 0]))).toSucceed();\n      expect(await index.add(target('conv-a', 'b'), Float32Array.from([1, 0, 0]))).toFailWith(\n        /dimension 3 does not match index dimension 2/i\n      );\n    });\n  });\n\n  describe('query', () => {\n    async function seeded(): Promise<InMemoryCosineIndex> {\n      const index = InMemoryCosineIndex.create().orThrow();\n      (await index.add(target('s', 'a'), Float32Array.from([1, 0]))).orThrow();\n      (await index.add(target('s', 'b'), Float32Array.from([0, 1]))).orThrow();\n      (await index.add(target('s', 'c'), Float32Array.from([1, 1]))).orThrow();\n      return index;\n    }\n\n    test('returns hits in descending cosine-similarity order', async () => {\n      const index = await seeded();\n      expect(await index.query(Float32Array.from([1, 0]), 3)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits.map((h) => h.target.id)).toEqual(['a', 'c', 'b']);\n          expect(hits[0].score).toBeCloseTo(1);\n          expect(hits[1].score).toBeCloseTo(1 / Math.sqrt(2));\n          expect(hits[2].score).toBeCloseTo(0);\n        }\n      );\n    });\n\n    test('truncates to topK', async () => {\n      const index = await seeded();\n      expect(await index.query(Float32Array.from([1, 0]), 2)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits.map((h) => h.target.id)).toEqual(['a', 'c']);\n        }\n      );\n    });\n\n    test('returns empty for a non-positive topK', async () => {\n      const index = await seeded();\n      expect(await index.query(Float32Array.from([1, 0]), 0)).toSucceedWith([]);\n      expect(await index.query(Float32Array.from([1, 0]), -1)).toSucceedWith([]);\n    });\n\n    test('returns empty when the index is empty (no dimension check)', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      expect(await index.query(Float32Array.from([1, 2, 3, 4]), 5)).toSucceedWith([]);\n    });\n\n    test('fails loudly on a query-dimension mismatch', async () => {\n      const index = await seeded();\n      expect(await index.query(Float32Array.from([1, 0, 0]), 3)).toFailWith(\n        /query dimension 3 does not match index dimension 2/i\n      );\n    });\n\n    test('scores a degenerate (zero-magnitude) stored vector as 0 rather than NaN', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      (await index.add(target('s', 'zero'), Float32Array.from([0, 0]))).orThrow();\n      (await index.add(target('s', 'unit'), Float32Array.from([1, 0]))).orThrow();\n      expect(await index.query(Float32Array.from([1, 0]), 2)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits.map((h) => h.target.id)).toEqual(['unit', 'zero']);\n          expect(hits[1].score).toBe(0);\n        }\n      );\n    });\n\n    test('scores against a degenerate (zero-magnitude) query as 0 rather than NaN', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      (await index.add(target('s', 'unit'), Float32Array.from([1, 0]))).orThrow();\n      expect(await index.query(Float32Array.from([0, 0]), 1)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits[0].score).toBe(0);\n        }\n      );\n    });\n  });\n\n  describe('remove', () => {\n    test('removes a vector and is reflected in subsequent queries', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      const a = target('s', 'a');\n      const b = target('s', 'b');\n      (await index.add(a, Float32Array.from([1, 0]))).orThrow();\n      (await index.add(b, Float32Array.from([0, 1]))).orThrow();\n      expect(await index.remove(a)).toSucceedWith(a);\n      expect(index.size).toBe(1);\n      expect(await index.query(Float32Array.from([1, 0]), 5)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits.map((h) => h.target.id)).toEqual(['b']);\n        }\n      );\n    });\n\n    test('removes only the scoped target — a same-stem record in another scope is left intact', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      const a = target('conv-a', 'turn-3');\n      const b = target('conv-b', 'turn-3');\n      (await index.add(a, Float32Array.from([1, 0]))).orThrow();\n      (await index.add(b, Float32Array.from([0, 1]))).orThrow();\n      expect(await index.remove(a)).toSucceedWith(a);\n      expect(index.size).toBe(1);\n      // conv-b's embedding survives untouched.\n      expect(await index.query(Float32Array.from([0, 1]), 5)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits).toHaveLength(1);\n          expect(hits[0].target).toEqual(b);\n          expect(hits[0].score).toBeCloseTo(1);\n        }\n      );\n    });\n\n    test('is idempotent — removing an absent target still succeeds', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      const t = target('s', 'missing');\n      expect(await index.remove(t)).toSucceedWith(t);\n    });\n  });\n\n  describe('rebuild', () => {\n    const embed = (r: IMemoryRecord<unknown>): Promise<Result<Float32Array>> => {\n      // Deterministic: encode the id's first char code into a 2-vector.\n      const code: number = (r.envelope.id as string).charCodeAt(0);\n      return Promise.resolve(succeed(Float32Array.from([code, 1])));\n    };\n\n    test('re-embeds every scoped record from the source and reports the count', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      const source = new FakeSource(succeed([scoped('s', 'a'), scoped('s', 'b'), scoped('s', 'c')]));\n      expect(await index.rebuild(source, embed)).toSucceedWith(3);\n      expect(index.size).toBe(3);\n      expect(await index.query(Float32Array.from([99, 1]), 1)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          // 'c' = char code 99, the closest to the query vector.\n          expect(hits[0].target.id).toBe('c');\n        }\n      );\n    });\n\n    test('keeps same-stem records under different scopes distinct across a rebuild', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      const source = new FakeSource(succeed([scoped('conv-a', 'turn-3'), scoped('conv-b', 'turn-3')]));\n      // The embedder keys only off the (shared) id, so a bare-id rebuild would\n      // index just one entry; the scoped rebuild indexes both.\n      expect(await index.rebuild(source, embed)).toSucceedWith(2);\n      expect(index.size).toBe(2);\n      expect(await index.query(Float32Array.from([116, 1]), 5)).toSucceedAndSatisfy(\n        (hits: ReadonlyArray<IVectorQueryHit>) => {\n          expect(hits).toHaveLength(2);\n          expect(hits.map((h) => h.target.scope).sort()).toEqual(['conv-a', 'conv-b']);\n        }\n      );\n    });\n\n    test('clears prior contents and re-establishes the dimension', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      // Seed a 3-dim vector, then rebuild with a 2-dim embedder.\n      (await index.add(target('s', 'old'), Float32Array.from([1, 2, 3]))).orThrow();\n      const source = new FakeSource(succeed([scoped('s', 'a')]));\n      expect(await index.rebuild(source, embed)).toSucceedWith(1);\n      expect(index.size).toBe(1);\n      // The 2-dim query now succeeds (dimension was reset by rebuild).\n      expect(await index.query(Float32Array.from([97, 1]), 1)).toSucceed();\n    });\n\n    test('fails loudly when the source list fails', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      // Seed an entry so the reset-before-list rollback is observable, not just \"stayed empty\".\n      (await index.add(target('s', 'seed'), Float32Array.from([1, 1]))).orThrow();\n      const source = new FakeSource(fail('disk gone'));\n      expect(await index.rebuild(source, embed)).toFailWith(/failed to list records: disk gone/i);\n      expect(index.size).toBe(0);\n    });\n\n    test('fails loudly and rolls back to empty when an embedding fails mid-rebuild', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      // Seed an existing entry so the rollback (not merely \"stayed empty\") is observable.\n      (await index.add(target('s', 'seed'), Float32Array.from([1, 1]))).orThrow();\n      // First record embeds fine, the second fails — so a naive impl would leave\n      // record 'a' indexed; the rollback must clear it.\n      let calls: number = 0;\n      const flakyEmbed = (): Promise<Result<Float32Array>> => {\n        calls += 1;\n        return Promise.resolve(calls === 1 ? succeed(Float32Array.from([1, 1])) : fail('no model'));\n      };\n      const source = new FakeSource(succeed([scoped('conv-a', 'turn-1'), scoped('conv-b', 'turn-1')]));\n      // The failing record's error names its scope-qualified key.\n      expect(await index.rebuild(source, flakyEmbed)).toFailWith(\n        /embedding 'conv-b\\0turn-1' failed: no model/i\n      );\n      expect(index.size).toBe(0);\n    });\n\n    test('fails loudly and rolls back to empty when adding an embedded vector fails', async () => {\n      const index = InMemoryCosineIndex.create().orThrow();\n      const source = new FakeSource(succeed([scoped('s', 'a')]));\n      const emptyEmbed = (): Promise<Result<Float32Array>> => Promise.resolve(succeed(new Float32Array(0)));\n      expect(await index.rebuild(source, emptyEmbed)).toFailWith(/empty vector/i);\n      expect(index.size).toBe(0);\n    });\n  });\n});\n"]}

package/dist/test/unit/vector/vectorIndex.test.js

@@ -4,39 +4,48 @@
  */
 import '@fgv/ts-utils-jest';
 import { succeed } from '@fgv/ts-utils';
+import { edgeTargetKey } from '../../../index';
+function target(scope, id) {
+    return { scope: scope, id: id };
+}
 /**
  * A trivial conforming implementation, present to verify the {@link IVectorIndex}
- * seam is usable end-to-end (the in-package cosine impl is a post-v1 fast-follow).
+ * seam is usable end-to-end with scope-qualified targets.
  */
 class StubVectorIndex {
     constructor() {
         this._vectors = new Map();
     }
-    add(id, vector) {
-        this._vectors.set(id, vector);
-        return Promise.resolve(succeed(`ref-${id}`));
+    add(t, vector) {
+        const key = edgeTargetKey(t);
+        this._vectors.set(key, t);
+        return Promise.resolve(succeed(`ref-${key}`));
     }
-    remove(id) {
-        this._vectors.delete(id);
-        return Promise.resolve(succeed(id));
+    remove(t) {
+        this._vectors.delete(edgeTargetKey(t));
+        return Promise.resolve(succeed(t));
     }
     query(vector, topK) {
-        const hits = Array.from(this._vectors.keys())
-            .map((id) => ({ id, score: vector.length }))
+        const hits = Array.from(this._vectors.values())
+            .map((t) => ({ target: t, score: vector.length }))
             .slice(0, topK);
         return Promise.resolve(succeed(hits));
     }
 }
 describe('IVectorIndex seam', () => {
-    test('a conforming implementation supports add / query / remove', async () => {
+    test('a conforming implementation supports add / query / remove on scoped targets', async () => {
         const index = new StubVectorIndex();
-        expect(await index.add('a', Float32Array.from([1, 2, 3]))).toSucceedWith('ref-a');
-        expect(await index.add('b', Float32Array.from([4, 5, 6]))).toSucceedWith('ref-b');
-        expect(await index.query(Float32Array.from([0.1, 0.2]), 1)).toSucceedAndSatisfy((hits) => {
-            expect(hits).toHaveLength(1);
+        const a = target('conv-a', 'turn-3');
+        const b = target('conv-b', 'turn-3');
+        expect(await index.add(a, Float32Array.from([1, 2, 3]))).toSucceedWith(`ref-${edgeTargetKey(a)}`);
+        expect(await index.add(b, Float32Array.from([4, 5, 6]))).toSucceedWith(`ref-${edgeTargetKey(b)}`);
+        expect(await index.query(Float32Array.from([0.1, 0.2]), 2)).toSucceedAndSatisfy((hits) => {
+            // Both same-stem-different-scope targets are distinct entries.
+            expect(hits).toHaveLength(2);
+            expect(hits.map((h) => h.target)).toEqual(expect.arrayContaining([a, b]));
             expect(hits[0].score).toBe(2);
         });
-        expect(await index.remove('a')).toSucceedWith('a');
+        expect(await index.remove(a)).toSucceedWith(a);
     });
 });
 //# sourceMappingURL=vectorIndex.test.js.map

package/dist/test/unit/vector/vectorIndex.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"vectorIndex.test.js","sourceRoot":"","sources":["../../../../src/test/unit/vector/vectorIndex.test.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,oBAAoB,CAAC;AAC5B,OAAO,EAAU,OAAO,EAAE,MAAM,eAAe,CAAC;AAGhD;;;GAGG;AACH,MAAM,eAAe;IAArB;QACmB,aAAQ,GAAgC,IAAI,GAAG,EAAE,CAAC;IAkBrE,CAAC;IAhBQ,GAAG,CAAC,EAAY,EAAE,MAAoB;QAC3C,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QAC9B,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;IAC/C,CAAC;IAEM,MAAM,CAAC,EAAY;QACxB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACzB,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;IACtC,CAAC;IAEM,KAAK,CAAC,MAAoB,EAAE,IAAY;QAC7C,MAAM,IAAI,GAAsB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;aAC7D,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;aAC3C,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QAClB,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IACxC,CAAC;CACF;AAED,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,IAAI,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;QACpC,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAC9F,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,GAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAC9F,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAC7E,CAAC,IAAoC,EAAE,EAAE;YACvC,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC,CACF,CAAC;QACF,MAAM,CAAC,MAAM,KAAK,CAAC,MAAM,CAAC,GAAe,CAAC,CAAC,CAAC,aAAa,CAAC,GAAe,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["/*\n * Copyright (c) 2026 Erik Fortune\n * SPDX-License-Identifier: MIT\n */\n\nimport '@fgv/ts-utils-jest';\nimport { Result, succeed } from '@fgv/ts-utils';\nimport { IVectorIndex, IVectorQueryHit, MemoryId } from '../../../index';\n\n/**\n * A trivial conforming implementation, present to verify the {@link IVectorIndex}\n * seam is usable end-to-end (the in-package cosine impl is a post-v1 fast-follow).\n */\nclass StubVectorIndex implements IVectorIndex {\n  private readonly _vectors: Map<MemoryId, Float32Array> = new Map();\n\n  public add(id: MemoryId, vector: Float32Array): Promise<Result<string>> {\n    this._vectors.set(id, vector);\n    return Promise.resolve(succeed(`ref-${id}`));\n  }\n\n  public remove(id: MemoryId): Promise<Result<MemoryId>> {\n    this._vectors.delete(id);\n    return Promise.resolve(succeed(id));\n  }\n\n  public query(vector: Float32Array, topK: number): Promise<Result<ReadonlyArray<IVectorQueryHit>>> {\n    const hits: IVectorQueryHit[] = Array.from(this._vectors.keys())\n      .map((id) => ({ id, score: vector.length }))\n      .slice(0, topK);\n    return Promise.resolve(succeed(hits));\n  }\n}\n\ndescribe('IVectorIndex seam', () => {\n  test('a conforming implementation supports add / query / remove', async () => {\n    const index = new StubVectorIndex();\n    expect(await index.add('a' as MemoryId, Float32Array.from([1, 2, 3]))).toSucceedWith('ref-a');\n    expect(await index.add('b' as MemoryId, Float32Array.from([4, 5, 6]))).toSucceedWith('ref-b');\n    expect(await index.query(Float32Array.from([0.1, 0.2]), 1)).toSucceedAndSatisfy(\n      (hits: ReadonlyArray<IVectorQueryHit>) => {\n        expect(hits).toHaveLength(1);\n        expect(hits[0].score).toBe(2);\n      }\n    );\n    expect(await index.remove('a' as MemoryId)).toSucceedWith('a' as MemoryId);\n  });\n});\n"]}
+{"version":3,"file":"vectorIndex.test.js","sourceRoot":"","sources":["../../../../src/test/unit/vector/vectorIndex.test.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,oBAAoB,CAAC;AAC5B,OAAO,EAAU,OAAO,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,EAML,aAAa,EACd,MAAM,gBAAgB,CAAC;AAExB,SAAS,MAAM,CAAC,KAAa,EAAE,EAAU;IACvC,OAAO,EAAE,KAAK,EAAE,KAAuB,EAAE,EAAE,EAAE,EAAc,EAAE,CAAC;AAChE,CAAC;AAED;;;GAGG;AACH,MAAM,eAAe;IAArB;QACmB,aAAQ,GAA6B,IAAI,GAAG,EAAE,CAAC;IAmBlE,CAAC;IAjBQ,GAAG,CAAC,CAAc,EAAE,MAAoB;QAC7C,MAAM,GAAG,GAAW,aAAa,CAAC,CAAC,CAAC,CAAC;QACrC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAC1B,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,GAAG,EAAE,CAAC,CAAC,CAAC;IAChD,CAAC;IAEM,MAAM,CAAC,CAAc;QAC1B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;QACvC,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC;IAEM,KAAK,CAAC,MAAoB,EAAE,IAAY;QAC7C,MAAM,IAAI,GAAsB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;aAC/D,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;aACjD,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QAClB,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IACxC,CAAC;CACF;AAED,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,IAAI,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC7F,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;QACpC,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAClG,MAAM,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAClG,MAAM,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAC7E,CAAC,IAAoC,EAAE,EAAE;YACvC,+DAA+D;YAC/D,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1E,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC,CACF,CAAC;QACF,MAAM,CAAC,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["/*\n * Copyright (c) 2026 Erik Fortune\n * SPDX-License-Identifier: MIT\n */\n\nimport '@fgv/ts-utils-jest';\nimport { Result, succeed } from '@fgv/ts-utils';\nimport {\n  IEdgeTarget,\n  IVectorIndex,\n  IVectorQueryHit,\n  MemoryId,\n  MemoryScopeKey,\n  edgeTargetKey\n} from '../../../index';\n\nfunction target(scope: string, id: string): IEdgeTarget {\n  return { scope: scope as MemoryScopeKey, id: id as MemoryId };\n}\n\n/**\n * A trivial conforming implementation, present to verify the {@link IVectorIndex}\n * seam is usable end-to-end with scope-qualified targets.\n */\nclass StubVectorIndex implements IVectorIndex {\n  private readonly _vectors: Map<string, IEdgeTarget> = new Map();\n\n  public add(t: IEdgeTarget, vector: Float32Array): Promise<Result<string>> {\n    const key: string = edgeTargetKey(t);\n    this._vectors.set(key, t);\n    return Promise.resolve(succeed(`ref-${key}`));\n  }\n\n  public remove(t: IEdgeTarget): Promise<Result<IEdgeTarget>> {\n    this._vectors.delete(edgeTargetKey(t));\n    return Promise.resolve(succeed(t));\n  }\n\n  public query(vector: Float32Array, topK: number): Promise<Result<ReadonlyArray<IVectorQueryHit>>> {\n    const hits: IVectorQueryHit[] = Array.from(this._vectors.values())\n      .map((t) => ({ target: t, score: vector.length }))\n      .slice(0, topK);\n    return Promise.resolve(succeed(hits));\n  }\n}\n\ndescribe('IVectorIndex seam', () => {\n  test('a conforming implementation supports add / query / remove on scoped targets', async () => {\n    const index = new StubVectorIndex();\n    const a = target('conv-a', 'turn-3');\n    const b = target('conv-b', 'turn-3');\n    expect(await index.add(a, Float32Array.from([1, 2, 3]))).toSucceedWith(`ref-${edgeTargetKey(a)}`);\n    expect(await index.add(b, Float32Array.from([4, 5, 6]))).toSucceedWith(`ref-${edgeTargetKey(b)}`);\n    expect(await index.query(Float32Array.from([0.1, 0.2]), 2)).toSucceedAndSatisfy(\n      (hits: ReadonlyArray<IVectorQueryHit>) => {\n        // Both same-stem-different-scope targets are distinct entries.\n        expect(hits).toHaveLength(2);\n        expect(hits.map((h) => h.target)).toEqual(expect.arrayContaining([a, b]));\n        expect(hits[0].score).toBe(2);\n      }\n    );\n    expect(await index.remove(a)).toSucceedWith(a);\n  });\n});\n"]}