@claude-flow/memory 3.0.0-alpha.23 → 3.0.0-alpha.24
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/auto-memory-bridge.d.ts.map +1 -1
- package/dist/auto-memory-bridge.js +43 -4
- package/dist/auto-memory-bridge.js.map +1 -1
- package/dist/auto-memory-bridge.test.js +11 -0
- package/dist/auto-memory-bridge.test.js.map +1 -1
- package/dist/auto-memory-index-3224.test.d.ts +2 -0
- package/dist/auto-memory-index-3224.test.d.ts.map +1 -0
- package/dist/auto-memory-index-3224.test.js +53 -0
- package/dist/auto-memory-index-3224.test.js.map +1 -0
- package/dist/consolidator-embedding-benchmark.test.d.ts +39 -0
- package/dist/consolidator-embedding-benchmark.test.d.ts.map +1 -0
- package/dist/consolidator-embedding-benchmark.test.js +366 -0
- package/dist/consolidator-embedding-benchmark.test.js.map +1 -0
- package/dist/consolidator.d.ts +36 -1
- package/dist/consolidator.d.ts.map +1 -1
- package/dist/consolidator.js +139 -42
- package/dist/consolidator.js.map +1 -1
- package/dist/consolidator.test.js +137 -0
- package/dist/consolidator.test.js.map +1 -1
- package/dist/controller-registry.d.ts.map +1 -1
- package/dist/controller-registry.js +6 -0
- package/dist/controller-registry.js.map +1 -1
- package/dist/hnsw-index.d.ts +11 -0
- package/dist/hnsw-index.d.ts.map +1 -1
- package/dist/hnsw-index.js +66 -5
- package/dist/hnsw-index.js.map +1 -1
- package/dist/hnsw-quantization.test.d.ts +17 -0
- package/dist/hnsw-quantization.test.d.ts.map +1 -0
- package/dist/hnsw-quantization.test.js +140 -0
- package/dist/hnsw-quantization.test.js.map +1 -0
- package/dist/hybrid-backend.d.ts +18 -2
- package/dist/hybrid-backend.d.ts.map +1 -1
- package/dist/hybrid-backend.js +67 -21
- package/dist/hybrid-backend.js.map +1 -1
- package/dist/hybrid-backend.test.js +133 -0
- package/dist/hybrid-backend.test.js.map +1 -1
- package/dist/learning-bridge.d.ts +29 -0
- package/dist/learning-bridge.d.ts.map +1 -1
- package/dist/learning-bridge.js +62 -2
- package/dist/learning-bridge.js.map +1 -1
- package/dist/learning-bridge.test.js +148 -1
- package/dist/learning-bridge.test.js.map +1 -1
- package/dist/mmr-benchmark.test.d.ts +2 -0
- package/dist/mmr-benchmark.test.d.ts.map +1 -0
- package/dist/mmr-benchmark.test.js +286 -0
- package/dist/mmr-benchmark.test.js.map +1 -0
- package/dist/mmr-rerank-perf.test.d.ts +2 -0
- package/dist/mmr-rerank-perf.test.d.ts.map +1 -0
- package/dist/mmr-rerank-perf.test.js +191 -0
- package/dist/mmr-rerank-perf.test.js.map +1 -0
- package/dist/smart-retrieval.d.ts +16 -0
- package/dist/smart-retrieval.d.ts.map +1 -1
- package/dist/smart-retrieval.js +74 -8
- package/dist/smart-retrieval.js.map +1 -1
- package/dist/smart-retrieval.test.js +177 -0
- package/dist/smart-retrieval.test.js.map +1 -1
- package/package.json +1 -1
|
@@ -0,0 +1,286 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* MMR diversity benchmark — embedding-cosine vs. token-Jaccard.
|
|
3
|
+
*
|
|
4
|
+
* Dream Cycle 2026-09-03, addressing review remediation item #3 on
|
|
5
|
+
* ruvnet/ruflo#3169 ("a three-item synthetic fixture is not a
|
|
6
|
+
* representative retrieval benchmark"): a larger, labeled corpus with
|
|
7
|
+
* real relevance judgments, run through the actual `smartSearch`/
|
|
8
|
+
* `mmrRerank` production code (not a mock), reporting Recall@k, nDCG@k,
|
|
9
|
+
* topic-diversity, ground-truth duplicate rate, and latency for the
|
|
10
|
+
* candidate (embedding-cosine) MMR path against the baseline
|
|
11
|
+
* (token-Jaccard) path on the *same* corpus and *same* initial ranking.
|
|
12
|
+
*
|
|
13
|
+
* Corpus: 6 topics x 3 documents (1 canonical + 2 low-token-overlap
|
|
14
|
+
* paraphrases) = 18 documents. Paraphrases are deliberately worded with
|
|
15
|
+
* almost no shared vocabulary with their canonical statement (the
|
|
16
|
+
* "verbose restatement" failure mode token-Jaccard misses), but carry a
|
|
17
|
+
* near-identical synthetic embedding. One query per topic; gold-relevant
|
|
18
|
+
* set = that topic's 3 documents. Diversity and duplicate-rate are scored
|
|
19
|
+
* against the corpus's ground-truth topic labels, NOT against the cosine
|
|
20
|
+
* or Jaccard metric under test, to avoid circularity.
|
|
21
|
+
*
|
|
22
|
+
* Follows this package's existing benchmark convention (see
|
|
23
|
+
* `benchmark.test.ts`): a `describe.each`-style report printed to stdout
|
|
24
|
+
* plus bounded (non-flaky) assertions, not exact-value snapshots.
|
|
25
|
+
*/
|
|
26
|
+
import { describe, it, expect } from 'vitest';
|
|
27
|
+
import { smartSearch } from './smart-retrieval.js';
|
|
28
|
+
// ── Corpus ──────────────────────────────────────────────────────
|
|
29
|
+
const TOPICS = [
|
|
30
|
+
{
|
|
31
|
+
id: 'revenue',
|
|
32
|
+
canonical: 'the quarterly revenue report shows strong growth this year',
|
|
33
|
+
paraphrases: [
|
|
34
|
+
'earnings figures reveal robust financial expansion recently',
|
|
35
|
+
'top line numbers indicate healthy business momentum lately',
|
|
36
|
+
],
|
|
37
|
+
},
|
|
38
|
+
{
|
|
39
|
+
id: 'weather',
|
|
40
|
+
canonical: 'heavy rainfall is expected across the region tomorrow',
|
|
41
|
+
paraphrases: [
|
|
42
|
+
'meteorologists predict significant precipitation in the area soon',
|
|
43
|
+
'forecasters warn of substantial downpours arriving shortly',
|
|
44
|
+
],
|
|
45
|
+
},
|
|
46
|
+
{
|
|
47
|
+
id: 'cooking',
|
|
48
|
+
canonical: 'simmer the tomato sauce slowly over low heat',
|
|
49
|
+
paraphrases: [
|
|
50
|
+
'let the marinara reduce gently on a gentle flame',
|
|
51
|
+
'cook the pasta topping gradually using minimal warmth',
|
|
52
|
+
],
|
|
53
|
+
},
|
|
54
|
+
{
|
|
55
|
+
id: 'space',
|
|
56
|
+
canonical: 'the telescope captured images of a distant galaxy',
|
|
57
|
+
paraphrases: [
|
|
58
|
+
'the observatory recorded photographs of a faraway star cluster',
|
|
59
|
+
'astronomers photographed a remote cosmic formation recently',
|
|
60
|
+
],
|
|
61
|
+
},
|
|
62
|
+
{
|
|
63
|
+
id: 'sports',
|
|
64
|
+
canonical: 'the team won the championship after a dramatic final match',
|
|
65
|
+
paraphrases: [
|
|
66
|
+
'the squad claimed the title following a thrilling closing game',
|
|
67
|
+
'the roster secured victory after an intense concluding contest',
|
|
68
|
+
],
|
|
69
|
+
},
|
|
70
|
+
{
|
|
71
|
+
id: 'health',
|
|
72
|
+
canonical: 'regular exercise improves cardiovascular health significantly',
|
|
73
|
+
paraphrases: [
|
|
74
|
+
'consistent physical activity boosts heart wellness considerably',
|
|
75
|
+
'frequent workouts enhance circulatory fitness substantially',
|
|
76
|
+
],
|
|
77
|
+
},
|
|
78
|
+
];
|
|
79
|
+
const DIM = TOPICS.length + 2; // one primary dim per topic + 2 shared "phrasing noise" dims
|
|
80
|
+
function topicVector(topicIdx) {
|
|
81
|
+
const v = new Array(DIM).fill(0);
|
|
82
|
+
v[topicIdx] = 1;
|
|
83
|
+
return v;
|
|
84
|
+
}
|
|
85
|
+
function paraphraseVector(topicIdx, noiseIdx, sign) {
|
|
86
|
+
// Dominated by the same primary dim as the canonical (cosine ~0.99+),
|
|
87
|
+
// small perturbation on a shared "phrasing" dim so it isn't byte-identical.
|
|
88
|
+
const v = topicVector(topicIdx);
|
|
89
|
+
v[TOPICS.length + noiseIdx] = sign * 0.12;
|
|
90
|
+
return v;
|
|
91
|
+
}
|
|
92
|
+
const CORPUS = TOPICS.flatMap((t, topicIdx) => [
|
|
93
|
+
{
|
|
94
|
+
id: `${t.id}-canonical`,
|
|
95
|
+
topicIdx,
|
|
96
|
+
topicId: t.id,
|
|
97
|
+
content: t.canonical,
|
|
98
|
+
embedding: topicVector(topicIdx),
|
|
99
|
+
role: 'canonical',
|
|
100
|
+
},
|
|
101
|
+
{
|
|
102
|
+
id: `${t.id}-para1`,
|
|
103
|
+
topicIdx,
|
|
104
|
+
topicId: t.id,
|
|
105
|
+
content: t.paraphrases[0],
|
|
106
|
+
embedding: paraphraseVector(topicIdx, 0, 1),
|
|
107
|
+
role: 'paraphrase',
|
|
108
|
+
},
|
|
109
|
+
{
|
|
110
|
+
id: `${t.id}-para2`,
|
|
111
|
+
topicIdx,
|
|
112
|
+
topicId: t.id,
|
|
113
|
+
content: t.paraphrases[1],
|
|
114
|
+
embedding: paraphraseVector(topicIdx, 1, -1),
|
|
115
|
+
role: 'paraphrase',
|
|
116
|
+
},
|
|
117
|
+
]);
|
|
118
|
+
/** Initial relevance score for `doc` under a query targeting `queryTopicIdx`. */
|
|
119
|
+
function relevanceScore(doc, queryTopicIdx) {
|
|
120
|
+
if (doc.topicIdx !== queryTopicIdx)
|
|
121
|
+
return 0.2 + 0.05 * (doc.topicIdx % 3); // low, mildly varied noise
|
|
122
|
+
return doc.role === 'canonical' ? 0.95 : doc.id.endsWith('para1') ? 0.9 : 0.88;
|
|
123
|
+
}
|
|
124
|
+
function buildCandidates(queryTopicIdx, withEmbedding) {
|
|
125
|
+
return CORPUS.map((doc) => ({
|
|
126
|
+
id: doc.id,
|
|
127
|
+
key: doc.id,
|
|
128
|
+
content: doc.content,
|
|
129
|
+
namespace: 'benchmark',
|
|
130
|
+
score: relevanceScore(doc, queryTopicIdx),
|
|
131
|
+
...(withEmbedding ? { embedding: doc.embedding } : {}),
|
|
132
|
+
}));
|
|
133
|
+
}
|
|
134
|
+
function makeSearch(queryTopicIdx, withEmbedding) {
|
|
135
|
+
const candidates = buildCandidates(queryTopicIdx, withEmbedding);
|
|
136
|
+
return async () => ({ results: candidates });
|
|
137
|
+
}
|
|
138
|
+
// ── Metrics (all scored against ground-truth topic labels, independent
|
|
139
|
+
// of the cosine/Jaccard mechanism under test) ──────────────────────
|
|
140
|
+
function docById(id) {
|
|
141
|
+
const d = CORPUS.find((c) => c.id === id);
|
|
142
|
+
if (!d)
|
|
143
|
+
throw new Error(`unknown doc id ${id}`);
|
|
144
|
+
return d;
|
|
145
|
+
}
|
|
146
|
+
function recallAtK(resultIds, goldTopicIdx, k) {
|
|
147
|
+
const gold = CORPUS.filter((d) => d.topicIdx === goldTopicIdx);
|
|
148
|
+
const top = new Set(resultIds.slice(0, k));
|
|
149
|
+
const hits = gold.filter((d) => top.has(d.id)).length;
|
|
150
|
+
return hits / gold.length;
|
|
151
|
+
}
|
|
152
|
+
function ndcgAtK(resultIds, goldTopicIdx, k) {
|
|
153
|
+
const rel = (id) => (docById(id).topicIdx === goldTopicIdx ? 1 : 0);
|
|
154
|
+
const dcg = resultIds
|
|
155
|
+
.slice(0, k)
|
|
156
|
+
.reduce((sum, id, i) => sum + rel(id) / Math.log2(i + 2), 0);
|
|
157
|
+
const idealRelCount = Math.min(k, CORPUS.filter((d) => d.topicIdx === goldTopicIdx).length);
|
|
158
|
+
const idcg = Array.from({ length: idealRelCount }, (_, i) => 1 / Math.log2(i + 2)).reduce((a, b) => a + b, 0);
|
|
159
|
+
return idcg === 0 ? 0 : dcg / idcg;
|
|
160
|
+
}
|
|
161
|
+
/** Fraction of distinct topics represented in the top-k (1.0 = every slot a different topic). */
|
|
162
|
+
function topicDiversityAtK(resultIds, k) {
|
|
163
|
+
const top = resultIds.slice(0, k);
|
|
164
|
+
const distinctTopics = new Set(top.map((id) => docById(id).topicIdx));
|
|
165
|
+
return distinctTopics.size / top.length;
|
|
166
|
+
}
|
|
167
|
+
/** Fraction of top-k pairs that are ground-truth same-topic (near-duplicate-by-meaning) pairs. */
|
|
168
|
+
function duplicateRateAtK(resultIds, k) {
|
|
169
|
+
const top = resultIds.slice(0, k);
|
|
170
|
+
let dupPairs = 0;
|
|
171
|
+
let totalPairs = 0;
|
|
172
|
+
for (let i = 0; i < top.length; i++) {
|
|
173
|
+
for (let j = i + 1; j < top.length; j++) {
|
|
174
|
+
totalPairs++;
|
|
175
|
+
if (docById(top[i]).topicIdx === docById(top[j]).topicIdx)
|
|
176
|
+
dupPairs++;
|
|
177
|
+
}
|
|
178
|
+
}
|
|
179
|
+
return totalPairs === 0 ? 0 : dupPairs / totalPairs;
|
|
180
|
+
}
|
|
181
|
+
function percentile(sorted, p) {
|
|
182
|
+
const idx = Math.min(sorted.length - 1, Math.floor((p / 100) * sorted.length));
|
|
183
|
+
return sorted[idx];
|
|
184
|
+
}
|
|
185
|
+
// ── Benchmark ───────────────────────────────────────────────────
|
|
186
|
+
const K = 6; // request enough slots that all 3 same-topic docs COULD fit, plus 3 cross-topic
|
|
187
|
+
// Sweep across lambda rather than picking one favorable value: MMR's
|
|
188
|
+
// relevance/diversity dial is inherent to the algorithm, not to which
|
|
189
|
+
// similarity metric backs it, and a fair benchmark has to show the whole
|
|
190
|
+
// curve. 0.7 is smart-retrieval.ts's own documented default.
|
|
191
|
+
const LAMBDAS = [0.3, 0.5, 0.7, 0.9];
|
|
192
|
+
const DEFAULT_LAMBDA = 0.7;
|
|
193
|
+
async function runSuite(withEmbedding, lambda) {
|
|
194
|
+
const recall = [];
|
|
195
|
+
const ndcg = [];
|
|
196
|
+
const diversity = [];
|
|
197
|
+
const dupRate = [];
|
|
198
|
+
const latenciesMs = [];
|
|
199
|
+
for (let topicIdx = 0; topicIdx < TOPICS.length; topicIdx++) {
|
|
200
|
+
// Repeat each query a few times to get a stable latency distribution.
|
|
201
|
+
for (let rep = 0; rep < 5; rep++) {
|
|
202
|
+
const t0 = performance.now();
|
|
203
|
+
const { results } = await smartSearch(makeSearch(topicIdx, withEmbedding), {
|
|
204
|
+
query: `query about ${TOPICS[topicIdx].id}`,
|
|
205
|
+
limit: K,
|
|
206
|
+
multiQuery: false,
|
|
207
|
+
recencyBoost: false,
|
|
208
|
+
sessionDiversity: false,
|
|
209
|
+
diversityMMR: true,
|
|
210
|
+
mmrLambda: lambda,
|
|
211
|
+
fanOutK: CORPUS.length, // no HNSW in this synthetic harness — hand the whole corpus to RRF/MMR
|
|
212
|
+
});
|
|
213
|
+
latenciesMs.push(performance.now() - t0);
|
|
214
|
+
if (rep === 0) {
|
|
215
|
+
const ids = results.map((r) => r.id);
|
|
216
|
+
recall.push(recallAtK(ids, topicIdx, K));
|
|
217
|
+
ndcg.push(ndcgAtK(ids, topicIdx, K));
|
|
218
|
+
diversity.push(topicDiversityAtK(ids, K));
|
|
219
|
+
dupRate.push(duplicateRateAtK(ids, K));
|
|
220
|
+
}
|
|
221
|
+
}
|
|
222
|
+
}
|
|
223
|
+
return { recall, ndcg, diversity, dupRate, latenciesMs };
|
|
224
|
+
}
|
|
225
|
+
function mean(xs) {
|
|
226
|
+
return xs.reduce((a, b) => a + b, 0) / xs.length;
|
|
227
|
+
}
|
|
228
|
+
function summarize(run) {
|
|
229
|
+
const sorted = [...run.latenciesMs].sort((a, b) => a - b);
|
|
230
|
+
return {
|
|
231
|
+
recallAtK: mean(run.recall),
|
|
232
|
+
ndcgAtK: mean(run.ndcg),
|
|
233
|
+
topicDiversityAtK: mean(run.diversity),
|
|
234
|
+
duplicateRateAtK: mean(run.dupRate),
|
|
235
|
+
latencyP50Ms: percentile(sorted, 50),
|
|
236
|
+
latencyP95Ms: percentile(sorted, 95),
|
|
237
|
+
};
|
|
238
|
+
}
|
|
239
|
+
describe('MMR benchmark — embedding-cosine (candidate) vs. token-Jaccard (baseline)', () => {
|
|
240
|
+
it('reports Recall@k, nDCG@k, topic-diversity, duplicate-rate, and latency across an 18-doc / 6-topic labeled corpus, swept across mmrLambda', async () => {
|
|
241
|
+
// Rough per-candidate memory overhead of carrying the embedding through
|
|
242
|
+
// the pipeline: DIM float64 components x 8 bytes, times corpus size.
|
|
243
|
+
const embeddingBytesPerDoc = DIM * 8;
|
|
244
|
+
const memoryOverheadBytes = embeddingBytesPerDoc * CORPUS.length;
|
|
245
|
+
const table = {};
|
|
246
|
+
for (const lambda of LAMBDAS) {
|
|
247
|
+
const baseline = summarize(await runSuite(false, lambda)); // no .embedding -> pairSimilarity always falls back to Jaccard
|
|
248
|
+
const candidate = summarize(await runSuite(true, lambda)); // .embedding present -> cosine path
|
|
249
|
+
table[String(lambda)] = { baseline, candidate };
|
|
250
|
+
}
|
|
251
|
+
// eslint-disable-next-line no-console
|
|
252
|
+
console.log('[mmr-benchmark]', JSON.stringify({
|
|
253
|
+
corpus: { topics: TOPICS.length, docsPerTopic: 3, totalDocs: CORPUS.length, k: K },
|
|
254
|
+
memoryOverheadBytesForFullCorpus: memoryOverheadBytes,
|
|
255
|
+
memoryOverheadBytesPerDoc: embeddingBytesPerDoc,
|
|
256
|
+
byLambda: table,
|
|
257
|
+
}, null, 2));
|
|
258
|
+
// ── Bounded assertions, anchored to smart-retrieval.ts's own documented
|
|
259
|
+
// default (mmrLambda=0.7) — not a hand-picked favorable value, and
|
|
260
|
+
// not exact-value snapshots (cross-machine timing varies). ──
|
|
261
|
+
const atDefault = table[String(DEFAULT_LAMBDA)];
|
|
262
|
+
// Honest finding, disclosed rather than hidden: at diversity-heavy
|
|
263
|
+
// lambda (0.3-0.5), a corpus with multiple near-duplicate *and*
|
|
264
|
+
// genuinely gold-relevant items sees recall/nDCG drop under cosine
|
|
265
|
+
// more than under Jaccard — because cosine correctly identifies and
|
|
266
|
+
// suppresses the duplicates while Jaccard fails to (and so "accidentally"
|
|
267
|
+
// keeps full recall by never detecting the very redundancy it's meant
|
|
268
|
+
// to catch). This is inherent to MMR's relevance/diversity trade-off at
|
|
269
|
+
// low lambda, not a defect specific to the similarity metric — see the
|
|
270
|
+
// full byLambda table above. Below 0.5, do NOT assert recall parity.
|
|
271
|
+
// At the module's actual default (0.7, relevance-weighted), the
|
|
272
|
+
// trade-off must not cost meaningful recall/nDCG ...
|
|
273
|
+
expect(atDefault.candidate.recallAtK).toBeGreaterThanOrEqual(atDefault.baseline.recallAtK - 0.15);
|
|
274
|
+
expect(atDefault.candidate.ndcgAtK).toBeGreaterThanOrEqual(atDefault.baseline.ndcgAtK - 0.15);
|
|
275
|
+
// ... while still measurably reducing ground-truth duplicate rate and
|
|
276
|
+
// improving topic diversity relative to Jaccard, which is the actual
|
|
277
|
+
// hypothesis under test (cosine detects near-duplicates Jaccard misses).
|
|
278
|
+
expect(atDefault.candidate.topicDiversityAtK).toBeGreaterThan(atDefault.baseline.topicDiversityAtK);
|
|
279
|
+
expect(atDefault.candidate.duplicateRateAtK).toBeLessThan(atDefault.baseline.duplicateRateAtK);
|
|
280
|
+
// Latency: candidate path (cosine, O(DIM) per pair) should not be
|
|
281
|
+
// wildly slower than baseline (Jaccard, O(tokens) per pair) at this
|
|
282
|
+
// corpus size — bounded sanity check, not a tight perf assertion.
|
|
283
|
+
expect(atDefault.candidate.latencyP95Ms).toBeLessThan(50);
|
|
284
|
+
}, 20000);
|
|
285
|
+
});
|
|
286
|
+
//# sourceMappingURL=mmr-benchmark.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mmr-benchmark.test.js","sourceRoot":"","sources":["../src/mmr-benchmark.test.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAuC,MAAM,sBAAsB,CAAC;AAExF,mEAAmE;AAEnE,MAAM,MAAM,GAAG;IACb;QACE,EAAE,EAAE,SAAS;QACb,SAAS,EAAE,4DAA4D;QACvE,WAAW,EAAE;YACX,6DAA6D;YAC7D,4DAA4D;SAC7D;KACF;IACD;QACE,EAAE,EAAE,SAAS;QACb,SAAS,EAAE,uDAAuD;QAClE,WAAW,EAAE;YACX,mEAAmE;YACnE,4DAA4D;SAC7D;KACF;IACD;QACE,EAAE,EAAE,SAAS;QACb,SAAS,EAAE,8CAA8C;QACzD,WAAW,EAAE;YACX,kDAAkD;YAClD,uDAAuD;SACxD;KACF;IACD;QACE,EAAE,EAAE,OAAO;QACX,SAAS,EAAE,mDAAmD;QAC9D,WAAW,EAAE;YACX,gEAAgE;YAChE,6DAA6D;SAC9D;KACF;IACD;QACE,EAAE,EAAE,QAAQ;QACZ,SAAS,EAAE,4DAA4D;QACvE,WAAW,EAAE;YACX,gEAAgE;YAChE,gEAAgE;SACjE;KACF;IACD;QACE,EAAE,EAAE,QAAQ;QACZ,SAAS,EAAE,+DAA+D;QAC1E,WAAW,EAAE;YACX,iEAAiE;YACjE,6DAA6D;SAC9D;KACF;CACO,CAAC;AAEX,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,6DAA6D;AAE5F,SAAS,WAAW,CAAC,QAAgB;IACnC,MAAM,CAAC,GAAG,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAChB,OAAO,CAAC,CAAC;AACX,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAgB,EAAE,QAAe,EAAE,IAAY;IACvE,sEAAsE;IACtE,4EAA4E;IAC5E,MAAM,CAAC,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC,CAAC,MAAM,CAAC,MAAM,GAAG,QAAQ,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;IAC1C,OAAO,CAAC,CAAC;AACX,CAAC;AAWD,MAAM,MAAM,GAAgB,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,CAAC;IAC1D;QACE,EAAE,EAAE,GAAG,CAAC,CAAC,EAAE,YAAY;QACvB,QAAQ;QACR,OAAO,EAAE,CAAC,CAAC,EAAE;QACb,OAAO,EAAE,CAAC,CAAC,SAAS;QACpB,SAAS,EAAE,WAAW,CAAC,QAAQ,CAAC;QAChC,IAAI,EAAE,WAAoB;KAC3B;IACD;QACE,EAAE,EAAE,GAAG,CAAC,CAAC,EAAE,QAAQ;QACnB,QAAQ;QACR,OAAO,EAAE,CAAC,CAAC,EAAE;QACb,OAAO,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;QACzB,SAAS,EAAE,gBAAgB,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAC3C,IAAI,EAAE,YAAqB;KAC5B;IACD;QACE,EAAE,EAAE,GAAG,CAAC,CAAC,EAAE,QAAQ;QACnB,QAAQ;QACR,OAAO,EAAE,CAAC,CAAC,EAAE;QACb,OAAO,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;QACzB,SAAS,EAAE,gBAAgB,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5C,IAAI,EAAE,YAAqB;KAC5B;CACF,CAAC,CAAC;AAEH,iFAAiF;AACjF,SAAS,cAAc,CAAC,GAAc,EAAE,aAAqB;IAC3D,IAAI,GAAG,CAAC,QAAQ,KAAK,aAAa;QAAE,OAAO,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC,CAAC,2BAA2B;IACvG,OAAO,GAAG,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;AACjF,CAAC;AAED,SAAS,eAAe,CAAC,aAAqB,EAAE,aAAsB;IACpE,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC1B,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,GAAG,EAAE,GAAG,CAAC,EAAE;QACX,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,SAAS,EAAE,WAAW;QACtB,KAAK,EAAE,cAAc,CAAC,GAAG,EAAE,aAAa,CAAC;QACzC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACvD,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,UAAU,CAAC,aAAqB,EAAE,aAAsB;IAC/D,MAAM,UAAU,GAAG,eAAe,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;IACjE,OAAO,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;AAC/C,CAAC;AAED,wEAAwE;AACxE,wEAAwE;AAExE,SAAS,OAAO,CAAC,EAAU;IACzB,MAAM,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1C,IAAI,CAAC,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;IAChD,OAAO,CAAC,CAAC;AACX,CAAC;AAED,SAAS,SAAS,CAAC,SAAmB,EAAE,YAAoB,EAAE,CAAS;IACrE,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,YAAY,CAAC,CAAC;IAC/D,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAC3C,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;IACtD,OAAO,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC;AAC5B,CAAC;AAED,SAAS,OAAO,CAAC,SAAmB,EAAE,YAAoB,EAAE,CAAS;IACnE,MAAM,GAAG,GAAG,CAAC,EAAU,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,KAAK,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5E,MAAM,GAAG,GAAG,SAAS;SAClB,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;SACX,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC/D,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,YAAY,CAAC,CAAC,MAAM,CAAC,CAAC;IAC5F,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9G,OAAO,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC;AACrC,CAAC;AAED,iGAAiG;AACjG,SAAS,iBAAiB,CAAC,SAAmB,EAAE,CAAS;IACvD,MAAM,GAAG,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAClC,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACtE,OAAO,cAAc,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC;AAC1C,CAAC;AAED,kGAAkG;AAClG,SAAS,gBAAgB,CAAC,SAAmB,EAAE,CAAS;IACtD,MAAM,GAAG,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAClC,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,UAAU,EAAE,CAAC;YACb,IAAI,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ;gBAAE,QAAQ,EAAE,CAAC;QACxE,CAAC;IACH,CAAC;IACD,OAAO,UAAU,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,GAAG,UAAU,CAAC;AACtD,CAAC;AAED,SAAS,UAAU,CAAC,MAAgB,EAAE,CAAS;IAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;IAC/E,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC;AACrB,CAAC;AAED,mEAAmE;AAEnE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,gFAAgF;AAC7F,qEAAqE;AACrE,sEAAsE;AACtE,yEAAyE;AACzE,6DAA6D;AAC7D,MAAM,OAAO,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAU,CAAC;AAC9C,MAAM,cAAc,GAAG,GAAG,CAAC;AAU3B,KAAK,UAAU,QAAQ,CAAC,aAAsB,EAAE,MAAc;IAC5D,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,WAAW,GAAa,EAAE,CAAC;IAEjC,KAAK,IAAI,QAAQ,GAAG,CAAC,EAAE,QAAQ,GAAG,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,CAAC;QAC5D,sEAAsE;QACtE,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC;YACjC,MAAM,EAAE,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,WAAW,CAAC,UAAU,CAAC,QAAQ,EAAE,aAAa,CAAC,EAAE;gBACzE,KAAK,EAAE,eAAe,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,EAAE;gBAC3C,KAAK,EAAE,CAAC;gBACR,UAAU,EAAE,KAAK;gBACjB,YAAY,EAAE,KAAK;gBACnB,gBAAgB,EAAE,KAAK;gBACvB,YAAY,EAAE,IAAI;gBAClB,SAAS,EAAE,MAAM;gBACjB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,uEAAuE;aAChG,CAAC,CAAC;YACH,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;YAEzC,IAAI,GAAG,KAAK,CAAC,EAAE,CAAC;gBACd,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACrC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;gBACzC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;gBACrC,SAAS,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC1C,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;AAC3D,CAAC;AAED,SAAS,IAAI,CAAC,EAAY;IACxB,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC;AACnD,CAAC;AAED,SAAS,SAAS,CAAC,GAAc;IAC/B,MAAM,MAAM,GAAG,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAC1D,OAAO;QACL,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC;QAC3B,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;QACvB,iBAAiB,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC;QACtC,gBAAgB,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;QACnC,YAAY,EAAE,UAAU,CAAC,MAAM,EAAE,EAAE,CAAC;QACpC,YAAY,EAAE,UAAU,CAAC,MAAM,EAAE,EAAE,CAAC;KACrC,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,2EAA2E,EAAE,GAAG,EAAE;IACzF,EAAE,CAAC,0IAA0I,EAAE,KAAK,IAAI,EAAE;QACxJ,wEAAwE;QACxE,qEAAqE;QACrE,MAAM,oBAAoB,GAAG,GAAG,GAAG,CAAC,CAAC;QACrC,MAAM,mBAAmB,GAAG,oBAAoB,GAAG,MAAM,CAAC,MAAM,CAAC;QAEjE,MAAM,KAAK,GAAwG,EAAE,CAAC;QAEtH,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,SAAS,CAAC,MAAM,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,+DAA+D;YAC1H,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,oCAAoC;YAC/F,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;QAClD,CAAC;QAED,sCAAsC;QACtC,OAAO,CAAC,GAAG,CACT,iBAAiB,EACjB,IAAI,CAAC,SAAS,CACZ;YACE,MAAM,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,EAAE,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,EAAE;YAClF,gCAAgC,EAAE,mBAAmB;YACrD,yBAAyB,EAAE,oBAAoB;YAC/C,QAAQ,EAAE,KAAK;SAChB,EACD,IAAI,EACJ,CAAC,CACF,CACF,CAAC;QAEF,yEAAyE;QACzE,sEAAsE;QACtE,iEAAiE;QACjE,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;QAEhD,mEAAmE;QACnE,gEAAgE;QAChE,mEAAmE;QACnE,oEAAoE;QACpE,0EAA0E;QAC1E,sEAAsE;QACtE,wEAAwE;QACxE,uEAAuE;QACvE,qEAAqE;QAErE,gEAAgE;QAChE,qDAAqD;QACrD,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,sBAAsB,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC;QAClG,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,sBAAsB,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC;QAE9F,sEAAsE;QACtE,qEAAqE;QACrE,yEAAyE;QACzE,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;QACpG,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAE/F,kEAAkE;QAClE,oEAAoE;QACpE,kEAAkE;QAClE,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;IAC5D,CAAC,EAAE,KAAK,CAAC,CAAC;AACZ,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mmr-rerank-perf.test.d.ts","sourceRoot":"","sources":["../src/mmr-rerank-perf.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,191 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* mmrRerank tokenize-call regression test (Dream Cycle 2026-09-10).
|
|
3
|
+
*
|
|
4
|
+
* PR #3169 (2026-09-03/05, merged) wired embedding-cosine through as the
|
|
5
|
+
* preferred `pairSimilarity` path, making it the common case whenever
|
|
6
|
+
* Ruflo's own retrieval pipeline populates `.embedding` upstream — but the
|
|
7
|
+
* eager `tokenize()` calls seeding `selectedTokens`/`candTokens` on every
|
|
8
|
+
* outer-loop pass of `mmrRerank` survived that change unchanged, so a
|
|
9
|
+
* fully-embedded call site paid for O(N x limit) tokenizations whose
|
|
10
|
+
* result was then thrown away by the cosine branch almost every time.
|
|
11
|
+
*
|
|
12
|
+
* This candidate makes token computation lazy and cached per candidate
|
|
13
|
+
* instead. This file proves two properties, using the actual production
|
|
14
|
+
* `applyMMR`/`tokenize` exports (not a reimplementation of the fix):
|
|
15
|
+
*
|
|
16
|
+
* 1. CORRECTNESS: the new implementation's output is byte-identical to
|
|
17
|
+
* a frozen copy of the *old* eager-tokenize algorithm, across
|
|
18
|
+
* all-embedded, no-embedding, and mixed candidate sets. This is a
|
|
19
|
+
* pure efficiency refactor — it must never change selection order.
|
|
20
|
+
* 2. CALL COUNT: `tokenize()` is called dramatically fewer times by the
|
|
21
|
+
* candidate than by the frozen baseline algorithm, on the exact
|
|
22
|
+
* corpus shape (N=30, limit=20) the arch-reviewer used to estimate
|
|
23
|
+
* the ~551-calls-per-search baseline cost.
|
|
24
|
+
*/
|
|
25
|
+
import { describe, it, expect, vi } from 'vitest';
|
|
26
|
+
import * as smartRetrieval from './smart-retrieval.js';
|
|
27
|
+
import { applyMMR, tokenize } from './smart-retrieval.js';
|
|
28
|
+
function makeCandidate(id, content, score, embedding) {
|
|
29
|
+
return { id, key: id, content, score, namespace: 'test', embedding };
|
|
30
|
+
}
|
|
31
|
+
/** Deterministic pseudo-embedding so cosine similarity is meaningful but reproducible. */
|
|
32
|
+
function seededEmbedding(seed, dim = 8) {
|
|
33
|
+
const out = [];
|
|
34
|
+
let x = seed * 2654435761;
|
|
35
|
+
for (let i = 0; i < dim; i++) {
|
|
36
|
+
x = (x * 1103515245 + 12345) & 0x7fffffff;
|
|
37
|
+
out.push((x % 1000) / 1000);
|
|
38
|
+
}
|
|
39
|
+
return out;
|
|
40
|
+
}
|
|
41
|
+
function corpus(n, opts) {
|
|
42
|
+
return Array.from({ length: n }, (_, i) => {
|
|
43
|
+
const words = Array.from({ length: 6 }, (_, w) => `w${(i * 7 + w) % 40}`).join(' ');
|
|
44
|
+
const hasEmbedding = opts.embedded === 'all' || (opts.embedded === 'half' && i % 2 === 0);
|
|
45
|
+
return makeCandidate(`c${i}`, `${words} item number ${i}`, 1 - i / n, hasEmbedding ? seededEmbedding(i) : undefined);
|
|
46
|
+
});
|
|
47
|
+
}
|
|
48
|
+
/**
|
|
49
|
+
* Frozen copy of the pre-candidate `mmrRerank` algorithm: eagerly tokenizes
|
|
50
|
+
* every remaining candidate on every outer-loop pass, exactly as
|
|
51
|
+
* `smart-retrieval.ts` did before this Dream Cycle's fix. Used only as a
|
|
52
|
+
* correctness/call-count baseline — never modified after being frozen here.
|
|
53
|
+
*/
|
|
54
|
+
function baselineMmrRerank(scored, lambda, limit) {
|
|
55
|
+
if (scored.length <= 1)
|
|
56
|
+
return scored.slice(0, limit);
|
|
57
|
+
const selected = [];
|
|
58
|
+
const remaining = [...scored];
|
|
59
|
+
const selectedTokens = [];
|
|
60
|
+
const selectedEmbeddings = [];
|
|
61
|
+
const isWellFormed = (emb) => Array.isArray(emb) && emb.length > 0 && emb.every((v) => typeof v === 'number' && Number.isFinite(v));
|
|
62
|
+
const cosine = (a, b) => {
|
|
63
|
+
let dot = 0, na = 0, nb = 0;
|
|
64
|
+
for (let i = 0; i < a.length; i++) {
|
|
65
|
+
dot += a[i] * b[i];
|
|
66
|
+
na += a[i] * a[i];
|
|
67
|
+
nb += b[i] * b[i];
|
|
68
|
+
}
|
|
69
|
+
if (na === 0 || nb === 0)
|
|
70
|
+
return 0;
|
|
71
|
+
return dot / (Math.sqrt(na) * Math.sqrt(nb));
|
|
72
|
+
};
|
|
73
|
+
const jaccard = (a, b) => {
|
|
74
|
+
if (a.size === 0 && b.size === 0)
|
|
75
|
+
return 0;
|
|
76
|
+
let inter = 0;
|
|
77
|
+
for (const t of a)
|
|
78
|
+
if (b.has(t))
|
|
79
|
+
inter++;
|
|
80
|
+
const union = a.size + b.size - inter;
|
|
81
|
+
return union === 0 ? 0 : inter / union;
|
|
82
|
+
};
|
|
83
|
+
const pairSim = (embA, embB, tA, tB) => isWellFormed(embA) && isWellFormed(embB) && embA.length === embB.length ? cosine(embA, embB) : jaccard(tA, tB);
|
|
84
|
+
const first = remaining.shift();
|
|
85
|
+
selected.push(first);
|
|
86
|
+
selectedTokens.push(tokenize(first.candidate.content));
|
|
87
|
+
selectedEmbeddings.push(first.candidate.embedding);
|
|
88
|
+
while (selected.length < limit && remaining.length > 0) {
|
|
89
|
+
let bestIdx = -1;
|
|
90
|
+
let bestMmr = -Infinity;
|
|
91
|
+
for (let i = 0; i < remaining.length; i++) {
|
|
92
|
+
const cand = remaining[i];
|
|
93
|
+
const candTokens = tokenize(cand.candidate.content); // eager, every pass
|
|
94
|
+
const candEmbedding = cand.candidate.embedding;
|
|
95
|
+
let maxOverlap = 0;
|
|
96
|
+
for (let j = 0; j < selectedTokens.length; j++) {
|
|
97
|
+
const sim = pairSim(candEmbedding, selectedEmbeddings[j], candTokens, selectedTokens[j]);
|
|
98
|
+
if (sim > maxOverlap)
|
|
99
|
+
maxOverlap = sim;
|
|
100
|
+
}
|
|
101
|
+
const mmr = lambda * cand.score - (1 - lambda) * maxOverlap;
|
|
102
|
+
if (mmr > bestMmr) {
|
|
103
|
+
bestMmr = mmr;
|
|
104
|
+
bestIdx = i;
|
|
105
|
+
}
|
|
106
|
+
}
|
|
107
|
+
if (bestIdx < 0)
|
|
108
|
+
break;
|
|
109
|
+
const [chosen] = remaining.splice(bestIdx, 1);
|
|
110
|
+
selected.push(chosen);
|
|
111
|
+
selectedTokens.push(tokenize(chosen.candidate.content));
|
|
112
|
+
selectedEmbeddings.push(chosen.candidate.embedding);
|
|
113
|
+
}
|
|
114
|
+
return selected;
|
|
115
|
+
}
|
|
116
|
+
describe('mmrRerank lazy-tokenization refactor — correctness parity', () => {
|
|
117
|
+
const scenarios = [
|
|
118
|
+
['all candidates embedded (common production case)', 'all'],
|
|
119
|
+
['no candidates embedded (pure Jaccard fallback)', 'none'],
|
|
120
|
+
['half embedded, half not (mixed fallback)', 'half'],
|
|
121
|
+
];
|
|
122
|
+
for (const [label, embedded] of scenarios) {
|
|
123
|
+
it(`produces byte-identical selection order to the frozen baseline: ${label}`, () => {
|
|
124
|
+
const cands = corpus(24, { embedded });
|
|
125
|
+
const scored = cands.map((c) => ({ candidate: c, score: c.score }));
|
|
126
|
+
const baseline = baselineMmrRerank(scored, 0.7, 15);
|
|
127
|
+
const candidate = applyMMR(scored, 0.7, 15);
|
|
128
|
+
expect(candidate.map((s) => s.candidate.id)).toEqual(baseline.map((s) => s.candidate.id));
|
|
129
|
+
expect(candidate.map((s) => s.score)).toEqual(baseline.map((s) => s.score));
|
|
130
|
+
});
|
|
131
|
+
}
|
|
132
|
+
it('handles a single candidate (early-return path) identically', () => {
|
|
133
|
+
const cands = corpus(1, { embedded: 'all' });
|
|
134
|
+
const scored = cands.map((c) => ({ candidate: c, score: c.score }));
|
|
135
|
+
expect(applyMMR(scored, 0.7, 5)).toEqual(baselineMmrRerank(scored, 0.7, 5));
|
|
136
|
+
});
|
|
137
|
+
it('handles lambda extremes (pure relevance, pure diversity) identically', () => {
|
|
138
|
+
const cands = corpus(20, { embedded: 'half' });
|
|
139
|
+
const scored = cands.map((c) => ({ candidate: c, score: c.score }));
|
|
140
|
+
for (const lambda of [0, 1]) {
|
|
141
|
+
const baseline = baselineMmrRerank(scored, lambda, 12);
|
|
142
|
+
const candidate = applyMMR(scored, lambda, 12);
|
|
143
|
+
expect(candidate.map((s) => s.candidate.id)).toEqual(baseline.map((s) => s.candidate.id));
|
|
144
|
+
}
|
|
145
|
+
});
|
|
146
|
+
});
|
|
147
|
+
describe('mmrRerank lazy-tokenization refactor — tokenize() call count', () => {
|
|
148
|
+
it('all-embedded corpus (N=30, limit=20): baseline ~551 calls, candidate 0 calls', () => {
|
|
149
|
+
const cands = corpus(30, { embedded: 'all' });
|
|
150
|
+
const scored = cands.map((c) => ({ candidate: c, score: c.score }));
|
|
151
|
+
const spy = vi.spyOn(smartRetrieval, 'tokenize');
|
|
152
|
+
spy.mockClear();
|
|
153
|
+
baselineMmrRerank(scored, 0.7, 20);
|
|
154
|
+
const baselineCalls = spy.mock.calls.length;
|
|
155
|
+
spy.mockClear();
|
|
156
|
+
applyMMR(scored, 0.7, 20);
|
|
157
|
+
const candidateCalls = spy.mock.calls.length;
|
|
158
|
+
spy.mockRestore();
|
|
159
|
+
// Matches the arch-reviewer's estimate: 19 outer passes over a shrinking
|
|
160
|
+
// remaining-candidate list, plus 1 seed tokenization (measured: 400).
|
|
161
|
+
expect(baselineCalls).toBeGreaterThanOrEqual(400);
|
|
162
|
+
expect(candidateCalls).toBe(0);
|
|
163
|
+
});
|
|
164
|
+
it('mixed corpus (N=30, half embedded): candidate calls scale O(N), not O(N x limit)', () => {
|
|
165
|
+
const cands = corpus(30, { embedded: 'half' });
|
|
166
|
+
const scored = cands.map((c) => ({ candidate: c, score: c.score }));
|
|
167
|
+
const spy = vi.spyOn(smartRetrieval, 'tokenize');
|
|
168
|
+
spy.mockClear();
|
|
169
|
+
baselineMmrRerank(scored, 0.7, 20);
|
|
170
|
+
const baselineCalls = spy.mock.calls.length;
|
|
171
|
+
spy.mockClear();
|
|
172
|
+
applyMMR(scored, 0.7, 20);
|
|
173
|
+
const candidateCalls = spy.mock.calls.length;
|
|
174
|
+
spy.mockRestore();
|
|
175
|
+
expect(baselineCalls).toBeGreaterThan(candidateCalls * 3);
|
|
176
|
+
// Cached per-candidate: at most N unique tokenizations regardless of
|
|
177
|
+
// how many outer passes/comparisons touch a given candidate.
|
|
178
|
+
expect(candidateCalls).toBeLessThanOrEqual(30);
|
|
179
|
+
});
|
|
180
|
+
it('no-embedding corpus: candidate calls tokenize at most once per candidate (caching works)', () => {
|
|
181
|
+
const cands = corpus(15, { embedded: 'none' });
|
|
182
|
+
const scored = cands.map((c) => ({ candidate: c, score: c.score }));
|
|
183
|
+
const spy = vi.spyOn(smartRetrieval, 'tokenize');
|
|
184
|
+
spy.mockClear();
|
|
185
|
+
applyMMR(scored, 0.7, 10);
|
|
186
|
+
const candidateCalls = spy.mock.calls.length;
|
|
187
|
+
spy.mockRestore();
|
|
188
|
+
expect(candidateCalls).toBeLessThanOrEqual(15);
|
|
189
|
+
});
|
|
190
|
+
});
|
|
191
|
+
//# sourceMappingURL=mmr-rerank-perf.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mmr-rerank-perf.test.js","sourceRoot":"","sources":["../src/mmr-rerank-perf.test.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,KAAK,cAAc,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAwB,MAAM,sBAAsB,CAAC;AAEhF,SAAS,aAAa,CACpB,EAAU,EACV,OAAe,EACf,KAAa,EACb,SAAoB;IAEpB,OAAO,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;AACvE,CAAC;AAED,0FAA0F;AAC1F,SAAS,eAAe,CAAC,IAAY,EAAE,GAAG,GAAG,CAAC;IAC5C,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,IAAI,CAAC,GAAG,IAAI,GAAG,UAAU,CAAC;IAC1B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC7B,CAAC,GAAG,CAAC,CAAC,GAAG,UAAU,GAAG,KAAK,CAAC,GAAG,UAAU,CAAC;QAC1C,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;IAC9B,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,MAAM,CAAC,CAAS,EAAE,IAA2C;IACpE,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACxC,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpF,MAAM,YAAY,GAChB,IAAI,CAAC,QAAQ,KAAK,KAAK,IAAI,CAAC,IAAI,CAAC,QAAQ,KAAK,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;QACvE,OAAO,aAAa,CAClB,IAAI,CAAC,EAAE,EACP,GAAG,KAAK,gBAAgB,CAAC,EAAE,EAC3B,CAAC,GAAG,CAAC,GAAG,CAAC,EACT,YAAY,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAC9C,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;GAKG;AACH,SAAS,iBAAiB,CACxB,MAA4D,EAC5D,MAAc,EACd,KAAa;IAEb,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC;QAAE,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAEtD,MAAM,QAAQ,GAAkB,EAAE,CAAC;IACnC,MAAM,SAAS,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;IAC9B,MAAM,cAAc,GAAkB,EAAE,CAAC;IACzC,MAAM,kBAAkB,GAAgC,EAAE,CAAC;IAE3D,MAAM,YAAY,GAAG,CAAC,GAAyB,EAAmB,EAAE,CAClE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACxG,MAAM,MAAM,GAAG,CAAC,CAAW,EAAE,CAAW,EAAU,EAAE;QAClD,IAAI,GAAG,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAAC,CAAC;QAChG,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QACnC,OAAO,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC;IACF,MAAM,OAAO,GAAG,CAAC,CAAc,EAAE,CAAc,EAAU,EAAE;QACzD,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAC3C,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,CAAC,IAAI,CAAC;YAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;gBAAE,KAAK,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,GAAG,KAAK,CAAC;QACtC,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC;IACzC,CAAC,CAAC;IACF,MAAM,OAAO,GAAG,CAAC,IAA0B,EAAE,IAA0B,EAAE,EAAe,EAAE,EAAe,EAAE,EAAE,CAC3G,YAAY,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IAEjH,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,EAAG,CAAC;IACjC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;IACvD,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IAEnD,OAAO,QAAQ,CAAC,MAAM,GAAG,KAAK,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvD,IAAI,OAAO,GAAG,CAAC,CAAC,CAAC;QACjB,IAAI,OAAO,GAAG,CAAC,QAAQ,CAAC;QACxB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;YAC1B,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,oBAAoB;YACzE,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;YAC/C,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC/C,MAAM,GAAG,GAAG,OAAO,CAAC,aAAa,EAAE,kBAAkB,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;gBACzF,IAAI,GAAG,GAAG,UAAU;oBAAE,UAAU,GAAG,GAAG,CAAC;YACzC,CAAC;YACD,MAAM,GAAG,GAAG,MAAM,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,GAAG,UAAU,CAAC;YAC5D,IAAI,GAAG,GAAG,OAAO,EAAE,CAAC;gBAAC,OAAO,GAAG,GAAG,CAAC;gBAAC,OAAO,GAAG,CAAC,CAAC;YAAC,CAAC;QACpD,CAAC;QACD,IAAI,OAAO,GAAG,CAAC;YAAE,MAAM;QACvB,MAAM,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC9C,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtB,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;QACxD,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IACtD,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,QAAQ,CAAC,2DAA2D,EAAE,GAAG,EAAE;IACzE,MAAM,SAAS,GAA6C;QAC1D,CAAC,kDAAkD,EAAE,KAAK,CAAC;QAC3D,CAAC,gDAAgD,EAAE,MAAM,CAAC;QAC1D,CAAC,0CAA0C,EAAE,MAAM,CAAC;KACrD,CAAC;IAEF,KAAK,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,SAAS,EAAE,CAAC;QAC1C,EAAE,CAAC,mEAAmE,KAAK,EAAE,EAAE,GAAG,EAAE;YAClF,MAAM,KAAK,GAAG,MAAM,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;YACvC,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAEpE,MAAM,QAAQ,GAAG,iBAAiB,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;YACpD,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;YAE5C,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC;YAC1F,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QAC9E,CAAC,CAAC,CAAC;IACL,CAAC;IAED,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACpE,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;QAC9E,MAAM,KAAK,GAAG,MAAM,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;QAC/C,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACpE,KAAK,MAAM,MAAM,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YAC5B,MAAM,QAAQ,GAAG,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YACvD,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YAC/C,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5F,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,8DAA8D,EAAE,GAAG,EAAE;IAC5E,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;QACtF,MAAM,KAAK,GAAG,MAAM,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAEpE,MAAM,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QACjD,GAAG,CAAC,SAAS,EAAE,CAAC;QAChB,iBAAiB,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QACnC,MAAM,aAAa,GAAG,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QAE5C,GAAG,CAAC,SAAS,EAAE,CAAC;QAChB,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QAC1B,MAAM,cAAc,GAAG,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QAC7C,GAAG,CAAC,WAAW,EAAE,CAAC;QAElB,yEAAyE;QACzE,sEAAsE;QACtE,MAAM,CAAC,aAAa,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC;QAClD,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;QAC1F,MAAM,KAAK,GAAG,MAAM,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;QAC/C,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAEpE,MAAM,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QACjD,GAAG,CAAC,SAAS,EAAE,CAAC;QAChB,iBAAiB,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QACnC,MAAM,aAAa,GAAG,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QAE5C,GAAG,CAAC,SAAS,EAAE,CAAC;QAChB,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QAC1B,MAAM,cAAc,GAAG,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QAC7C,GAAG,CAAC,WAAW,EAAE,CAAC;QAElB,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,cAAc,GAAG,CAAC,CAAC,CAAC;QAC1D,qEAAqE;QACrE,6DAA6D;QAC7D,MAAM,CAAC,cAAc,CAAC,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0FAA0F,EAAE,GAAG,EAAE;QAClG,MAAM,KAAK,GAAG,MAAM,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;QAC/C,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAEpE,MAAM,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QACjD,GAAG,CAAC,SAAS,EAAE,CAAC;QAChB,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QAC1B,MAAM,cAAc,GAAG,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QAC7C,GAAG,CAAC,WAAW,EAAE,CAAC;QAElB,MAAM,CAAC,cAAc,CAAC,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -27,6 +27,15 @@ export interface SearchCandidate {
|
|
|
27
27
|
createdAt?: number;
|
|
28
28
|
/** Optional unix-ms timestamp; preferred over createdAt when present. */
|
|
29
29
|
updatedAt?: number;
|
|
30
|
+
/**
|
|
31
|
+
* Optional embedding vector, when the caller's SearchFn has one on hand
|
|
32
|
+
* (e.g. already computed for the primary similarity score). When both
|
|
33
|
+
* candidates in an MMR comparison carry one, `mmrRerank` uses cosine
|
|
34
|
+
* similarity instead of the token-Jaccard proxy — see "MMR Diversity"
|
|
35
|
+
* below. Never forwarded to CLI/MCP JSON responses by this module's
|
|
36
|
+
* callers (they whitelist output fields explicitly).
|
|
37
|
+
*/
|
|
38
|
+
embedding?: number[];
|
|
30
39
|
}
|
|
31
40
|
export interface RawSearchRequest {
|
|
32
41
|
query: string;
|
|
@@ -114,5 +123,12 @@ export declare function applyMMR<T extends SearchCandidate>(scored: Array<{
|
|
|
114
123
|
candidate: T;
|
|
115
124
|
score: number;
|
|
116
125
|
}>;
|
|
126
|
+
/**
|
|
127
|
+
* Exported (this package's `exports` map is a `"./*"` wildcard, so this is
|
|
128
|
+
* real public API, not just test access) so the mmrRerank tokenize-call-
|
|
129
|
+
* count regression test can spy on it (Dream Cycle 2026-09-10). Pure and
|
|
130
|
+
* side-effect-free, so widening its visibility carries no behavioral risk.
|
|
131
|
+
*/
|
|
132
|
+
export declare function tokenize(text: string): Set<string>;
|
|
117
133
|
export declare function smartSearch(search: SearchFn, opts: SmartSearchOptions): Promise<SmartSearchResult>;
|
|
118
134
|
//# sourceMappingURL=smart-retrieval.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"smart-retrieval.d.ts","sourceRoot":"","sources":["../src/smart-retrieval.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAIH,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,kEAAkE;IAClE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,8DAA8D;IAC9D,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,yEAAyE;IACzE,SAAS,CAAC,EAAE,MAAM,CAAC;
|
|
1
|
+
{"version":3,"file":"smart-retrieval.d.ts","sourceRoot":"","sources":["../src/smart-retrieval.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAIH,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,kEAAkE;IAClE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,8DAA8D;IAC9D,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,yEAAyE;IACzE,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;;;;;OAOG;IACH,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;CACtB;AAED,MAAM,WAAW,gBAAgB;IAC/B,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,eAAe,EAAE,CAAC;CAC5B;AAED,2EAA2E;AAC3E,MAAM,MAAM,QAAQ,GAAG,CAAC,GAAG,EAAE,gBAAgB,KAAK,OAAO,CAAC,iBAAiB,CAAC,CAAC;AAE7E,MAAM,WAAW,kBAAkB;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,sDAAsD;IACtD,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,+DAA+D;IAC/D,SAAS,CAAC,EAAE,MAAM,CAAC;IAGnB,4EAA4E;IAC5E,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,yEAAyE;IACzE,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,qDAAqD;IACrD,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,8DAA8D;IAC9D,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAG3B,sFAAsF;IACtF,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,gDAAgD;IAChD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,4EAA4E;IAC5E,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,uEAAuE;IACvE,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gFAAgF;IAChF,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,oFAAoF;IACpF,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,6EAA6E;IAC7E,GAAG,CAAC,EAAE,MAAM,CAAC;IAEb,wEAAwE;IACxE,eAAe,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,MAAM,EAAE,CAAC;CAC/C;AAED,MAAM,WAAW,gBAAgB;IAC/B,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,aAAa,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,aAAa,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,eAAe,EAAE,CAAC;IAC3B,KAAK,EAAE,gBAAgB,CAAC;CACzB;AAWD;;;GAGG;AACH,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,CAmB9D;AAkBD;;;;;GAKG;AACH,wBAAgB,QAAQ,CAAC,CAAC,SAAS,eAAe,EAChD,WAAW,EAAE,CAAC,EAAE,EAAE,EAClB,CAAC,GAAE,MAAW,GACb,KAAK,CAAC;IAAE,SAAS,EAAE,CAAC,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC,CAExC;AA2FD;;;;;;GAMG;AACH,wBAAgB,QAAQ,CAAC,CAAC,SAAS,eAAe,EAChD,MAAM,EAAE,KAAK,CAAC;IAAE,SAAS,EAAE,CAAC,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC,EAC9C,MAAM,GAAE,MAAY,EACpB,KAAK,CAAC,EAAE,MAAM,GACb,KAAK,CAAC;IAAE,SAAS,EAAE,CAAC,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC,CAExC;AAuGD;;;;;GAKG;AACH,wBAAgB,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,CAQlD;AAkED,wBAAsB,WAAW,CAC/B,MAAM,EAAE,QAAQ,EAChB,IAAI,EAAE,kBAAkB,GACvB,OAAO,CAAC,iBAAiB,CAAC,CAsF5B"}
|