recallnest 2.6.1 → 3.0.1

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.
Files changed (207) hide show
  1. package/CHANGELOG.md +292 -0
  2. package/README.md +280 -195
  3. package/README_CN.md +266 -199
  4. package/config.json.example +8 -1
  5. package/docs/api-reference.md +1 -1
  6. package/integrations/README.md +1 -1
  7. package/integrations/agy/README.md +73 -0
  8. package/integrations/agy/setup.sh +75 -0
  9. package/integrations/setup-all.sh +1 -1
  10. package/package.json +7 -5
  11. package/src/access-tracker.ts +10 -1
  12. package/src/activity-counter.ts +69 -11
  13. package/src/age-label.ts +29 -0
  14. package/src/audit-log.ts +35 -0
  15. package/src/auto-gc.ts +97 -30
  16. package/src/capture-engine.ts +26 -4
  17. package/src/cli.ts +224 -10
  18. package/src/consolidation-engine.ts +317 -123
  19. package/src/context-composer-stable.ts +5 -1
  20. package/src/data-checkup.ts +17 -3
  21. package/src/dream-pipeline.ts +90 -12
  22. package/src/embedder.ts +44 -5
  23. package/src/env-config.ts +71 -0
  24. package/src/forget-engine.ts +8 -2
  25. package/src/gateway-server.ts +272 -0
  26. package/src/ingest.ts +248 -22
  27. package/src/kg-extractor.ts +17 -0
  28. package/src/llm-client.ts +86 -25
  29. package/src/mcp-server.ts +2 -1
  30. package/src/mcp-tools-advanced.ts +7 -2
  31. package/src/mcp-tools-core.ts +11 -1
  32. package/src/mcp-tools-governance.ts +45 -3
  33. package/src/memory-evolution.ts +16 -0
  34. package/src/memory-output.ts +47 -8
  35. package/src/memory-promotion.ts +280 -0
  36. package/src/memory-store-port.ts +32 -3
  37. package/src/memory-utility.ts +314 -0
  38. package/src/pivot-apply.ts +52 -1
  39. package/src/pivot-evidence.ts +128 -0
  40. package/src/ppr-traversal.ts +25 -1
  41. package/src/recall-ledger.ts +76 -0
  42. package/src/retriever.ts +82 -9
  43. package/src/runtime-config.ts +9 -0
  44. package/src/scope-policy.ts +22 -1
  45. package/src/session-engine.ts +11 -4
  46. package/src/session-output.ts +13 -4
  47. package/src/session-store.ts +2 -3
  48. package/src/skill-promotion.ts +218 -1
  49. package/src/store.ts +231 -71
  50. package/src/synthesis-contract.ts +286 -0
  51. package/src/version-manager.ts +83 -0
  52. package/src/workflow-observation-schema.ts +26 -0
  53. package/integrations/gemini-cli/README.md +0 -57
  54. package/integrations/gemini-cli/setup.sh +0 -64
  55. package/src/__tests__/access-tracker-novelty.test.ts +0 -304
  56. package/src/__tests__/activity-counter.test.ts +0 -125
  57. package/src/__tests__/adaptive-fidelity.test.ts +0 -102
  58. package/src/__tests__/admission-control.test.ts +0 -206
  59. package/src/__tests__/anchor-generator.test.ts +0 -67
  60. package/src/__tests__/api-server-security.test.ts +0 -102
  61. package/src/__tests__/auto-consolidation.test.ts +0 -168
  62. package/src/__tests__/auto-recall.test.ts +0 -131
  63. package/src/__tests__/babel-integration.test.ts +0 -144
  64. package/src/__tests__/batch-dedup.test.ts +0 -65
  65. package/src/__tests__/belief-history.test.ts +0 -220
  66. package/src/__tests__/canary-eval.test.ts +0 -114
  67. package/src/__tests__/capture-engine.test.ts +0 -1328
  68. package/src/__tests__/capture-heuristic.test.ts +0 -171
  69. package/src/__tests__/cascade-forget.test.ts +0 -210
  70. package/src/__tests__/category-min-scores.test.ts +0 -116
  71. package/src/__tests__/cc8-cc1-essential-context.test.ts +0 -298
  72. package/src/__tests__/circuit-breaker.test.ts +0 -99
  73. package/src/__tests__/cluster-consolidation.test.ts +0 -945
  74. package/src/__tests__/confidence-metadata.test.ts +0 -125
  75. package/src/__tests__/confidence-tracker.test.ts +0 -261
  76. package/src/__tests__/conflict-advisor.test.ts +0 -206
  77. package/src/__tests__/conflict-escalation.test.ts +0 -149
  78. package/src/__tests__/conflict-lifecycle.test.ts +0 -83
  79. package/src/__tests__/conflict-output.test.ts +0 -85
  80. package/src/__tests__/conflict-store.test.ts +0 -418
  81. package/src/__tests__/connector-v1.test.ts +0 -181
  82. package/src/__tests__/consolidation-engine.test.ts +0 -430
  83. package/src/__tests__/constructive-retrieval.test.ts +0 -509
  84. package/src/__tests__/context-collapse-renderer.test.ts +0 -214
  85. package/src/__tests__/context-composer-pins.test.ts +0 -64
  86. package/src/__tests__/context-composer-task-selection.test.ts +0 -24
  87. package/src/__tests__/context-reconstructor.test.ts +0 -65
  88. package/src/__tests__/context-renderer.test.ts +0 -91
  89. package/src/__tests__/continuity-eval.test.ts +0 -275
  90. package/src/__tests__/continuity-schema.test.ts +0 -97
  91. package/src/__tests__/contracts.test.ts +0 -262
  92. package/src/__tests__/conversation-importer.test.ts +0 -692
  93. package/src/__tests__/core-recall-eval.test.ts +0 -235
  94. package/src/__tests__/core-summary.test.ts +0 -109
  95. package/src/__tests__/data-checkup.test.ts +0 -276
  96. package/src/__tests__/decay-exemption.test.ts +0 -129
  97. package/src/__tests__/dedup-actions.test.ts +0 -58
  98. package/src/__tests__/dedup-guard.test.ts +0 -39
  99. package/src/__tests__/deterministic-id.test.ts +0 -39
  100. package/src/__tests__/distill-lock.test.ts +0 -305
  101. package/src/__tests__/distill-session-argshape.test.ts +0 -79
  102. package/src/__tests__/doctor.test.ts +0 -319
  103. package/src/__tests__/dream-pipeline.test.ts +0 -780
  104. package/src/__tests__/drill-down.test.ts +0 -108
  105. package/src/__tests__/embedder.test.ts +0 -122
  106. package/src/__tests__/emotion-decay.test.ts +0 -46
  107. package/src/__tests__/emotion-detector.test.ts +0 -56
  108. package/src/__tests__/emotion-phase1-integration.test.ts +0 -153
  109. package/src/__tests__/emotion-retrieval.test.ts +0 -56
  110. package/src/__tests__/entity-resolver.test.ts +0 -58
  111. package/src/__tests__/env-config.test.ts +0 -151
  112. package/src/__tests__/error-signature-boost.test.ts +0 -87
  113. package/src/__tests__/error-signature.test.ts +0 -95
  114. package/src/__tests__/error-taxonomy.test.ts +0 -129
  115. package/src/__tests__/ethics-layer.test.ts +0 -476
  116. package/src/__tests__/event-segmenter.test.ts +0 -153
  117. package/src/__tests__/feature-flags-integration.test.ts +0 -54
  118. package/src/__tests__/frequency-tracker.test.ts +0 -113
  119. package/src/__tests__/freshness.test.ts +0 -186
  120. package/src/__tests__/graph-export.test.ts +0 -489
  121. package/src/__tests__/hotness-blend.test.ts +0 -283
  122. package/src/__tests__/ingest-chunking.test.ts +0 -44
  123. package/src/__tests__/ingest.test.ts +0 -360
  124. package/src/__tests__/interference-detector.test.ts +0 -203
  125. package/src/__tests__/kg-extractor.test.ts +0 -192
  126. package/src/__tests__/kg-store.test.ts +0 -300
  127. package/src/__tests__/llm-client-pivot.test.ts +0 -628
  128. package/src/__tests__/llm-consolidation.test.ts +0 -226
  129. package/src/__tests__/maintenance-locks.test.ts +0 -189
  130. package/src/__tests__/mcp-efficiency.test.ts +0 -288
  131. package/src/__tests__/mcp-related-scopes.test.ts +0 -151
  132. package/src/__tests__/memory-boundaries.test.ts +0 -282
  133. package/src/__tests__/memory-evolution-b2-b3-e1.test.ts +0 -345
  134. package/src/__tests__/memory-evolution.test.ts +0 -271
  135. package/src/__tests__/memory-health-rebalance.test.ts +0 -144
  136. package/src/__tests__/memory-lint.test.ts +0 -421
  137. package/src/__tests__/memory-output.test.ts +0 -215
  138. package/src/__tests__/memory-promotion.test.ts +0 -315
  139. package/src/__tests__/migrate-dedup-ids.test.ts +0 -65
  140. package/src/__tests__/multi-vector.test.ts +0 -242
  141. package/src/__tests__/narrative-architecture.test.ts +0 -456
  142. package/src/__tests__/noise-filter.test.ts +0 -205
  143. package/src/__tests__/noise-prototype-bank.test.ts +0 -280
  144. package/src/__tests__/obsidian-connector.test.ts +0 -363
  145. package/src/__tests__/omitted-reporting.test.ts +0 -98
  146. package/src/__tests__/pending-review.test.ts +0 -77
  147. package/src/__tests__/pii-redact.test.ts +0 -333
  148. package/src/__tests__/pivot-apply.test.ts +0 -373
  149. package/src/__tests__/pivot-distill-supervisor.test.ts +0 -457
  150. package/src/__tests__/pivot-distill.test.ts +0 -1178
  151. package/src/__tests__/post-retrieval-filter.test.ts +0 -178
  152. package/src/__tests__/ppr-traversal.test.ts +0 -256
  153. package/src/__tests__/prediction-engine.test.ts +0 -349
  154. package/src/__tests__/predictive-prospective-memory.test.ts +0 -416
  155. package/src/__tests__/preference-guard-smoke.test.ts +0 -211
  156. package/src/__tests__/preference-matcher.test.ts +0 -282
  157. package/src/__tests__/preference-slots.test.ts +0 -205
  158. package/src/__tests__/prospective-memory.test.ts +0 -283
  159. package/src/__tests__/pull-from-macbook-sync-only.test.ts +0 -679
  160. package/src/__tests__/query-entity-detector.test.ts +0 -171
  161. package/src/__tests__/query-expander-alias.test.ts +0 -146
  162. package/src/__tests__/query-expander.test.ts +0 -34
  163. package/src/__tests__/recall-gate.test.ts +0 -266
  164. package/src/__tests__/recall-governor.test.ts +0 -227
  165. package/src/__tests__/recall-mode.test.ts +0 -64
  166. package/src/__tests__/resolve-db-path.test.ts +0 -130
  167. package/src/__tests__/resolve-tier.test.ts +0 -64
  168. package/src/__tests__/result-synthesizer.test.ts +0 -198
  169. package/src/__tests__/resume-context.test.ts +0 -4898
  170. package/src/__tests__/retrieval-eval.test.ts +0 -163
  171. package/src/__tests__/retrieval-trace.test.ts +0 -213
  172. package/src/__tests__/retriever-boundary-weight.test.ts +0 -136
  173. package/src/__tests__/retriever-lazy-init-dispatch.test.ts +0 -123
  174. package/src/__tests__/retroactive-boost.test.ts +0 -223
  175. package/src/__tests__/salience-scoring.test.ts +0 -106
  176. package/src/__tests__/scope-inventory.test.ts +0 -200
  177. package/src/__tests__/scope-policy.test.ts +0 -69
  178. package/src/__tests__/scope-suggester.test.ts +0 -80
  179. package/src/__tests__/session-checkpoint.test.ts +0 -291
  180. package/src/__tests__/session-distiller.test.ts +0 -640
  181. package/src/__tests__/session-gc.test.ts +0 -215
  182. package/src/__tests__/skill-engine.test.ts +0 -552
  183. package/src/__tests__/skill-promotion.test.ts +0 -654
  184. package/src/__tests__/skill-verifier.test.ts +0 -122
  185. package/src/__tests__/source-heartbeat.test.ts +0 -250
  186. package/src/__tests__/stable-context-length-contract.test.ts +0 -62
  187. package/src/__tests__/store-patch-metadata.test.ts +0 -205
  188. package/src/__tests__/store-read-consistency.test.ts +0 -122
  189. package/src/__tests__/store-schema-migration.test.ts +0 -87
  190. package/src/__tests__/store-scope-audit.test.ts +0 -81
  191. package/src/__tests__/store-upsert.test.ts +0 -201
  192. package/src/__tests__/supersede-chain.test.ts +0 -154
  193. package/src/__tests__/temporal-parser.test.ts +0 -241
  194. package/src/__tests__/temporal-validity.test.ts +0 -160
  195. package/src/__tests__/term-registry-terminal-roster.test.ts +0 -127
  196. package/src/__tests__/term-registry.test.ts +0 -121
  197. package/src/__tests__/tool-error-surfacing.test.ts +0 -98
  198. package/src/__tests__/tool-output-compressor.test.ts +0 -160
  199. package/src/__tests__/topic-tag.test.ts +0 -113
  200. package/src/__tests__/trusted-memory.test.ts +0 -839
  201. package/src/__tests__/ui-server-security.test.ts +0 -93
  202. package/src/__tests__/usage-tracker.test.ts +0 -287
  203. package/src/__tests__/version-manager.test.ts +0 -213
  204. package/src/__tests__/workflow-observation-skill-outcome.test.ts +0 -352
  205. package/src/__tests__/workflow-observation.test.ts +0 -433
  206. package/src/__tests__/write-verifier.test.ts +0 -122
  207. /package/integrations/{gemini-cli/gemini-md-snippet.md → agy/agy-md-snippet.md} +0 -0
@@ -1,163 +0,0 @@
1
- import { readFileSync } from "node:fs";
2
- import { describe, expect, it } from "bun:test";
3
-
4
- import { scoreRetrievalCase, type RetrievalEvalCase } from "../eval.js";
5
-
6
- // 对齐 eval/cases.json 的真实形状:20 个 case 全部只声明 expectAny +
7
- // expectScopePrefixes,没有一个声明 expectAll 或 forbid。
8
- function makeCase(overrides: Partial<RetrievalEvalCase> = {}): RetrievalEvalCase {
9
- return {
10
- name: "case",
11
- query: "q",
12
- expectAny: ["alpha", "beta", "gamma", "delta"],
13
- expectScopePrefixes: ["project:"],
14
- ...overrides,
15
- };
16
- }
17
-
18
- function hits(texts: string[], scope = "project:test") {
19
- return texts.map((text) => ({ entry: { text, scope } }));
20
- }
21
-
22
- describe("scoreRetrievalCase", () => {
23
- it("case 没声明的期望不再白送权重", () => {
24
- // 这条守的就是那个漏洞:expectAll 在所有真实 case 里都没声明,
25
- // 白送它的 0.3 意味着 expectAny 命中 1 个再加 scope 前缀命中,
26
- // 就正好压在 0.7 及格线上——检索质量其实很差。
27
- const weak = scoreRetrievalCase(makeCase(), hits(["alpha only"]));
28
-
29
- expect(weak.matchedAny).toEqual(["alpha"]);
30
- expect(weak.matchedScopes).toEqual(["project:"]);
31
- // (0.4×1/4 + 0.2×1/1) / 0.6 × 0.9 + 0.1 = 0.55
32
- expect(weak.score).toBeCloseTo(0.55, 5);
33
- expect(weak.passed).toBe(false);
34
- });
35
-
36
- it("同样形状下 expectAny 要命中一半才刚好摸到及格线", () => {
37
- const report = scoreRetrievalCase(makeCase(), hits(["alpha beta"]));
38
-
39
- expect(report.matchedAny).toEqual(["alpha", "beta"]);
40
- // (0.4×2/4 + 0.2) / 0.6 × 0.9 + 0.1 = 0.7
41
- expect(report.score).toBeCloseTo(0.7, 5);
42
- expect(report.passed).toBe(true);
43
- });
44
-
45
- it("声明的期望全部命中给满分", () => {
46
- const report = scoreRetrievalCase(makeCase(), hits(["alpha beta gamma delta"]));
47
-
48
- expect(report.score).toBeCloseTo(1.0, 5);
49
- expect(report.passed).toBe(true);
50
- });
51
-
52
- it("多声明一个维度不会稀释满分——归一化是重分配不是膨胀", () => {
53
- const withAll = scoreRetrievalCase(
54
- makeCase({ expectAll: ["alpha", "beta"] }),
55
- hits(["alpha beta gamma delta"]),
56
- );
57
-
58
- expect(withAll.score).toBeCloseTo(1.0, 5);
59
- expect(withAll.passed).toBe(true);
60
- });
61
-
62
- it("声明了 expectAll 却全 miss 会把分数拉到及格线上", () => {
63
- const report = scoreRetrievalCase(
64
- makeCase({ expectAll: ["epsilon", "zeta"] }),
65
- hits(["alpha beta gamma delta"]),
66
- );
67
-
68
- expect(report.matchedAll).toEqual([]);
69
- // 声明总权重 0.9 里挣到 0.4 + 0 + 0.2,归一化后加非空奖励正好 0.7——
70
- // 这是"仍能通过"的最弱一档:其余全中、只丢掉整个 expectAll。
71
- expect(report.score).toBeCloseTo((0.6 / 0.9) * 0.9 + 0.1, 5);
72
- expect(report.passed).toBe(true);
73
- });
74
-
75
- it("再滑掉一个维度就掉到线下", () => {
76
- const report = scoreRetrievalCase(
77
- makeCase({ expectAll: ["epsilon", "zeta"] }),
78
- hits(["alpha beta gamma"]), // expectAny 4 中 3,expectAll 全 miss
79
- );
80
-
81
- expect(report.score).toBeLessThan(0.7);
82
- expect(report.passed).toBe(false);
83
- });
84
-
85
- it("只声明一个维度时该维度独占全部权重", () => {
86
- const single = makeCase({ expectScopePrefixes: undefined });
87
-
88
- const full = scoreRetrievalCase(single, hits(["alpha beta gamma delta"]));
89
- expect(full.score).toBeCloseTo(1.0, 5);
90
-
91
- const half = scoreRetrievalCase(single, hits(["alpha beta"]));
92
- // 4 个词命中 2 个 → 拿走整个信封的一半,再加非空奖励
93
- expect(half.score).toBeCloseTo(0.5 * 0.9 + 0.1, 5);
94
- expect(half.passed).toBe(false);
95
- });
96
-
97
- it("expectAll 按命中比例计分,不是全有或全无", () => {
98
- const report = scoreRetrievalCase(
99
- makeCase({ expectAny: ["alpha"], expectScopePrefixes: undefined, expectAll: ["beta", "epsilon"] }),
100
- hits(["alpha beta"]),
101
- );
102
-
103
- expect(report.matchedAll).toEqual(["beta"]);
104
- // (0.4×1 + 0.3×1/2) / 0.7 × 0.9 + 0.1
105
- expect(report.score).toBeCloseTo((0.55 / 0.7) * 0.9 + 0.1, 5);
106
- });
107
-
108
- it("scope 前缀也按命中比例计分", () => {
109
- const report = scoreRetrievalCase(
110
- makeCase({ expectAny: undefined, expectScopePrefixes: ["project:", "hippo:"] }),
111
- hits(["随便什么内容"]),
112
- );
113
-
114
- expect(report.matchedScopes).toEqual(["project:"]);
115
- // 2 个前缀命中 1 个 → 0.5 × 0.9 + 0.1
116
- expect(report.score).toBeCloseTo(0.55, 5);
117
- expect(report.passed).toBe(false);
118
- });
119
-
120
- it("命中 forbid 既扣分又直接判 fail", () => {
121
- const report = scoreRetrievalCase(
122
- makeCase({ forbid: ["secret"] }),
123
- hits(["alpha beta gamma delta secret"]),
124
- );
125
-
126
- expect(report.forbiddenMatches).toEqual(["secret"]);
127
- expect(report.score).toBeCloseTo(0.7, 5);
128
- // 分数够线,但命中禁词一律判不通过
129
- expect(report.passed).toBe(false);
130
- });
131
-
132
- it("空召回拿不到非空奖励,得 0 分", () => {
133
- const report = scoreRetrievalCase(makeCase(), []);
134
-
135
- expect(report.hitCount).toBe(0);
136
- expect(report.score).toBe(0);
137
- expect(report.passed).toBe(false);
138
- });
139
-
140
- it("一个期望都没声明的畸形 case 拿不到及格分", () => {
141
- const report = scoreRetrievalCase({ name: "empty", query: "q" }, hits(["anything at all"]));
142
-
143
- // 只剩非空奖励 0.1;改之前这里是三个维度全额白送 = 满分 1.0
144
- expect(report.score).toBeCloseTo(0.1, 5);
145
- expect(report.passed).toBe(false);
146
- });
147
-
148
- it("生产 case 集:完全不相关的召回不得有任何一个 case 过线", () => {
149
- // 守住评分器不再退回白送状态——不依赖 case 集当前的写法,
150
- // 以后无论 eval/cases.json 怎么增删,零匹配都必须是零通过。
151
- const cases = JSON.parse(
152
- readFileSync(new URL("../../eval/cases.json", import.meta.url), "utf-8"),
153
- ) as RetrievalEvalCase[];
154
- expect(cases.length).toBeGreaterThan(0);
155
-
156
- const irrelevant = [{ entry: { text: "完全无关的内容", scope: "unrelated/scope" } }];
157
- const leaked = cases
158
- .map((c) => scoreRetrievalCase(c, irrelevant))
159
- .filter((r) => r.passed)
160
- .map((r) => r.name);
161
- expect(leaked).toEqual([]);
162
- });
163
- });
@@ -1,213 +0,0 @@
1
- import { describe, expect, it } from "bun:test";
2
- import { TraceCollector } from "../retrieval-trace.js";
3
- import { createRetriever } from "../retriever.js";
4
-
5
- describe("TraceCollector", () => {
6
- it("accumulates stages correctly", () => {
7
- const trace = new TraceCollector();
8
- trace.startStage("vector_search", 0);
9
- trace.endStage(20, [0.3, 0.9]);
10
-
11
- trace.startStage("min_score_filter", 20);
12
- trace.endStage(15, [0.35, 0.9]);
13
-
14
- const result = trace.finalize("test query", "hybrid");
15
- expect(result.query).toBe("test query");
16
- expect(result.mode).toBe("hybrid");
17
- expect(result.stages).toHaveLength(2);
18
- expect(result.finalCount).toBe(15);
19
- expect(result.totalMs).toBeGreaterThanOrEqual(0);
20
- });
21
-
22
- it("computes droppedCount as input - output", () => {
23
- const trace = new TraceCollector();
24
- trace.startStage("hard_min_score", 20);
25
- trace.endStage(12);
26
-
27
- const result = trace.finalize("q", "vector");
28
- expect(result.stages[0].droppedCount).toBe(8);
29
- });
30
-
31
- it("computes scoreRange from surviving scores", () => {
32
- const trace = new TraceCollector();
33
- trace.startStage("rerank", 10);
34
- trace.endStage(8, [0.2, 0.5, 0.8, 0.3, 0.6, 0.7, 0.4, 0.9]);
35
-
36
- const stage = trace.finalize("q", "hybrid").stages[0];
37
- expect(stage.scoreRange).not.toBeNull();
38
- expect(stage.scoreRange![0]).toBeCloseTo(0.2);
39
- expect(stage.scoreRange![1]).toBeCloseTo(0.9);
40
- });
41
-
42
- it("returns null scoreRange when no scores provided", () => {
43
- const trace = new TraceCollector();
44
- trace.startStage("noise_filter", 5);
45
- trace.endStage(3);
46
-
47
- expect(trace.finalize("q", "hybrid").stages[0].scoreRange).toBeNull();
48
- });
49
-
50
- it("summarize() produces readable multi-line output", () => {
51
- const trace = new TraceCollector();
52
- trace.startStage("vector_search", 0);
53
- trace.endStage(20, [0.31, 0.89]);
54
- trace.startStage("min_score_filter", 20);
55
- trace.endStage(18, [0.35, 0.89]);
56
-
57
- const summary = trace.summarize("auth middleware", "hybrid");
58
- expect(summary).toContain('[retrieve] query="auth middleware" mode=hybrid');
59
- expect(summary).toContain("vector_search: 0 → 20");
60
- expect(summary).toContain("min_score_filter: 20 → 18 (2 dropped)");
61
- expect(summary).toContain("final: 18 results");
62
- });
63
- });
64
-
65
- describe("retriever with trace", () => {
66
- it("populates trace stages without changing retrieval behavior", async () => {
67
- const trace = new TraceCollector();
68
-
69
- const retriever = createRetriever(
70
- {
71
- hasFtsSupport: false,
72
- async vectorSearch() {
73
- return [
74
- {
75
- entry: {
76
- id: "m1",
77
- text: "test memory",
78
- vector: [1, 0, 0],
79
- category: "entities",
80
- scope: "project:test",
81
- importance: 0.8,
82
- timestamp: Date.now(),
83
- metadata: "{}",
84
- },
85
- score: 0.8,
86
- },
87
- ];
88
- },
89
- } as any,
90
- {
91
- async embedQuery() { return [1, 0, 0]; },
92
- async embedPassage() { return [1, 0, 0]; },
93
- } as any,
94
- {
95
- mode: "vector",
96
- rerank: "none",
97
- filterNoise: false,
98
- hardMinScore: 0,
99
- minScore: 0,
100
- recencyWeight: 0,
101
- timeDecayHalfLifeDays: 0,
102
- lengthNormAnchor: 0,
103
- },
104
- );
105
-
106
- const results = await retriever.retrieve({
107
- query: "authentication middleware config",
108
- limit: 5,
109
- trace,
110
- });
111
-
112
- // Results unchanged
113
- expect(results).toHaveLength(1);
114
- expect(results[0].entry.id).toBe("m1");
115
-
116
- // Trace populated
117
- const traceResult = trace.finalize("test", "vector");
118
- expect(traceResult.stages.length).toBeGreaterThan(0);
119
- expect(traceResult.stages[0].name).toBe("vector_search");
120
- expect(traceResult.finalCount).toBe(1);
121
- });
122
-
123
- it("works identically without trace (no overhead)", async () => {
124
- const retriever = createRetriever(
125
- {
126
- hasFtsSupport: false,
127
- async vectorSearch() {
128
- return [
129
- {
130
- entry: {
131
- id: "m1",
132
- text: "test memory",
133
- vector: [1, 0, 0],
134
- category: "entities",
135
- scope: "project:test",
136
- importance: 0.8,
137
- timestamp: Date.now(),
138
- metadata: "{}",
139
- },
140
- score: 0.8,
141
- },
142
- ];
143
- },
144
- } as any,
145
- {
146
- async embedQuery() { return [1, 0, 0]; },
147
- async embedPassage() { return [1, 0, 0]; },
148
- } as any,
149
- {
150
- mode: "vector",
151
- rerank: "none",
152
- filterNoise: false,
153
- hardMinScore: 0,
154
- minScore: 0,
155
- recencyWeight: 0,
156
- timeDecayHalfLifeDays: 0,
157
- lengthNormAnchor: 0,
158
- },
159
- );
160
-
161
- // No trace passed — should work exactly as before
162
- const results = await retriever.retrieve({
163
- query: "authentication middleware config",
164
- limit: 5,
165
- });
166
-
167
- expect(results).toHaveLength(1);
168
- expect(results[0].entry.id).toBe("m1");
169
- });
170
-
171
- it("trace captures hard_min_score drops", async () => {
172
- const trace = new TraceCollector();
173
-
174
- const retriever = createRetriever(
175
- {
176
- hasFtsSupport: false,
177
- async vectorSearch() {
178
- return [
179
- {
180
- entry: { id: "high", text: "good", vector: [1, 0, 0], category: "entities", scope: "p:t", importance: 0.8, timestamp: Date.now(), metadata: "{}" },
181
- score: 0.9,
182
- },
183
- {
184
- entry: { id: "low", text: "weak", vector: [0, 1, 0], category: "entities", scope: "p:t", importance: 0.5, timestamp: Date.now(), metadata: "{}" },
185
- score: 0.2,
186
- },
187
- ];
188
- },
189
- } as any,
190
- {
191
- async embedQuery() { return [1, 0, 0]; },
192
- async embedPassage() { return [1, 0, 0]; },
193
- } as any,
194
- {
195
- mode: "vector",
196
- rerank: "none",
197
- filterNoise: false,
198
- hardMinScore: 0.35,
199
- minScore: 0,
200
- recencyWeight: 0,
201
- timeDecayHalfLifeDays: 0,
202
- lengthNormAnchor: 0,
203
- },
204
- );
205
-
206
- await retriever.retrieve({ query: "authentication middleware config", limit: 5, trace });
207
-
208
- const result = trace.finalize("test", "vector");
209
- const hardStage = result.stages.find(s => s.name === "hard_min_score");
210
- expect(hardStage).toBeDefined();
211
- expect(hardStage!.droppedCount).toBeGreaterThan(0);
212
- });
213
- });
@@ -1,136 +0,0 @@
1
- import { describe, expect, it } from "bun:test";
2
-
3
- import { createRetriever } from "../retriever.js";
4
-
5
- function buildEntry(id: string, scoreText: string, scope: string, metadata: Record<string, unknown>) {
6
- return {
7
- entry: {
8
- id,
9
- text: scoreText,
10
- vector: [1, 0, 0],
11
- category: "preferences",
12
- scope,
13
- importance: 0.8,
14
- timestamp: Date.parse("2026-03-16T00:00:00.000Z"),
15
- metadata: JSON.stringify(metadata),
16
- },
17
- score: 0.8,
18
- };
19
- }
20
-
21
- describe("retriever provenance weighting", () => {
22
- it("prefers durable structured memories over evidence when scores are close", async () => {
23
- const retriever = createRetriever({
24
- hasFtsSupport: false,
25
- async vectorSearch() {
26
- return [
27
- {
28
- ...buildEntry("evidence-1", "Transcript hint", "cc:session1", {
29
- source: "cc",
30
- boundary: {
31
- layer: "evidence",
32
- authority: "transcript-ingest",
33
- conflictPolicy: "append-only",
34
- originalCategory: "preferences",
35
- },
36
- }),
37
- score: 0.8,
38
- },
39
- {
40
- ...buildEntry("durable-1", "Curated durable preference", "memory:agent", {
41
- source: "agent",
42
- boundary: {
43
- layer: "durable",
44
- authority: "structured-memory",
45
- conflictPolicy: "latest-wins",
46
- originalCategory: "preferences",
47
- },
48
- }),
49
- score: 0.79,
50
- },
51
- ];
52
- },
53
- } as any, {
54
- async embedQuery() {
55
- return [1, 0, 0];
56
- },
57
- async embedPassage() {
58
- return [1, 0, 0];
59
- },
60
- } as any, {
61
- mode: "vector",
62
- rerank: "none",
63
- filterNoise: false,
64
- hardMinScore: 0,
65
- minScore: 0,
66
- recencyWeight: 0,
67
- timeDecayHalfLifeDays: 0,
68
- });
69
-
70
- const results = await retriever.retrieve({
71
- query: "user reply style preference",
72
- limit: 5,
73
- category: "preferences",
74
- });
75
-
76
- expect(results[0]?.entry.id).toBe("durable-1");
77
- expect(results[1]?.entry.id).toBe("evidence-1");
78
- });
79
-
80
- it("does not override clearly stronger relevance with provenance weighting", async () => {
81
- const retriever = createRetriever({
82
- hasFtsSupport: false,
83
- async vectorSearch() {
84
- return [
85
- {
86
- ...buildEntry("evidence-strong", "Very strong transcript evidence", "cc:session1", {
87
- source: "cc",
88
- boundary: {
89
- layer: "evidence",
90
- authority: "transcript-ingest",
91
- conflictPolicy: "append-only",
92
- originalCategory: "preferences",
93
- },
94
- }),
95
- score: 0.95,
96
- },
97
- {
98
- ...buildEntry("durable-weaker", "Weaker durable preference", "memory:agent", {
99
- source: "agent",
100
- boundary: {
101
- layer: "durable",
102
- authority: "structured-memory",
103
- conflictPolicy: "latest-wins",
104
- originalCategory: "preferences",
105
- },
106
- }),
107
- score: 0.7,
108
- },
109
- ];
110
- },
111
- } as any, {
112
- async embedQuery() {
113
- return [1, 0, 0];
114
- },
115
- async embedPassage() {
116
- return [1, 0, 0];
117
- },
118
- } as any, {
119
- mode: "vector",
120
- rerank: "none",
121
- filterNoise: false,
122
- hardMinScore: 0,
123
- minScore: 0,
124
- recencyWeight: 0,
125
- timeDecayHalfLifeDays: 0,
126
- });
127
-
128
- const results = await retriever.retrieve({
129
- query: "user reply style preference",
130
- limit: 5,
131
- category: "preferences",
132
- });
133
-
134
- expect(results[0]?.entry.id).toBe("evidence-strong");
135
- });
136
- });
@@ -1,123 +0,0 @@
1
- import { describe, expect, it } from "bun:test";
2
-
3
- import { createRetriever } from "../retriever.js";
4
-
5
- /**
6
- * 回归:retrieve() 在分派前必须等 store 完成惰性初始化。
7
- *
8
- * 真实 MemoryStore 的 hasFtsSupport 在 ensureInitialized() 跑完前恒为 false
9
- * (ftsIndexCreated 初值 false,建索引发生在 doInitialize 里)。retriever.ts 的
10
- * 分派条件是 `mode === "vector" || !this.store.hasFtsSupport`,若不先 await
11
- * store.ready(),hybrid 模式的**首次**检索会读到初始化前的 false 并静默落回
12
- * vector 路径——同一实例第二次才正常。
13
- *
14
- * 该缺陷自 c10607a(2026-03-03 初始版本)就在,长期未暴露:生产 mode 一直是
15
- * "vector",第一个条件短路;而 eval 每 case 新建组件,每个 case 都停在"首次",
16
- * 于是 eval 从未测到过 hybrid 路径。
17
- *
18
- * 下面的 fake store 用 getter + ready() 精确复现这个时序:不修复则 bm25Search
19
- * 不会被调用。
20
- */
21
- function buildResult(id: string, text: string) {
22
- return {
23
- entry: {
24
- id,
25
- text,
26
- vector: [1, 0, 0],
27
- category: "preferences",
28
- scope: "memory:test",
29
- importance: 0.8,
30
- timestamp: Date.parse("2026-08-07T00:00:00.000Z"),
31
- metadata: JSON.stringify({}),
32
- },
33
- score: 0.8,
34
- };
35
- }
36
-
37
- const fakeEmbedder = {
38
- async embedQuery() {
39
- return [1, 0, 0];
40
- },
41
- async embedPassage() {
42
- return [1, 0, 0];
43
- },
44
- } as any;
45
-
46
- function buildLazyStore() {
47
- const calls = { bm25: 0, vector: 0, ready: 0 };
48
- let initialized = false;
49
-
50
- const store = {
51
- // 复现真实时序:初始化完成前恒 false
52
- get hasFtsSupport() {
53
- return initialized;
54
- },
55
- async ready() {
56
- calls.ready += 1;
57
- initialized = true;
58
- },
59
- async vectorSearch() {
60
- calls.vector += 1;
61
- return [buildResult("vec-1", "vector hit")];
62
- },
63
- async bm25Search() {
64
- calls.bm25 += 1;
65
- return [buildResult("bm25-1", "bm25 hit")];
66
- },
67
- };
68
-
69
- return { store, calls };
70
- }
71
-
72
- const LOOSE_CONFIG = {
73
- rerank: "none" as const,
74
- filterNoise: false,
75
- hardMinScore: 0,
76
- minScore: 0,
77
- recencyWeight: 0,
78
- timeDecayHalfLifeDays: 0,
79
- };
80
-
81
- describe("retrieve() 分派与 store 惰性初始化的时序", () => {
82
- it("hybrid 模式的首次检索不因惰性初始化落回 vector 路径", async () => {
83
- const { store, calls } = buildLazyStore();
84
- const retriever = createRetriever(store as any, fakeEmbedder, {
85
- ...LOOSE_CONFIG,
86
- mode: "hybrid",
87
- });
88
-
89
- await retriever.retrieve({ query: "双机契约 同步", limit: 5 });
90
-
91
- // 修复前:hasFtsSupport 读到初始化前的 false → 落回 vectorOnly → bm25 为 0。
92
- expect(calls.ready).toBeGreaterThan(0);
93
- expect(calls.bm25).toBeGreaterThan(0);
94
- });
95
-
96
- it("vector 模式不受影响——短路在第一个条件,不走 BM25", async () => {
97
- const { store, calls } = buildLazyStore();
98
- const retriever = createRetriever(store as any, fakeEmbedder, {
99
- ...LOOSE_CONFIG,
100
- mode: "vector",
101
- });
102
-
103
- await retriever.retrieve({ query: "双机契约 同步", limit: 5 });
104
-
105
- // 这条守的是"修复不得改变生产行为":生产 mode 就是 vector。
106
- expect(calls.bm25).toBe(0);
107
- expect(calls.vector).toBeGreaterThan(0);
108
- });
109
-
110
- it("琐碎 query 走 skip 分支时不触发初始化", async () => {
111
- const { store, calls } = buildLazyStore();
112
- const retriever = createRetriever(store as any, fakeEmbedder, {
113
- ...LOOSE_CONFIG,
114
- mode: "hybrid",
115
- });
116
-
117
- // shouldSkipRetrieval 在分派之前提前 return,await ready() 不该被执行到。
118
- await retriever.retrieve({ query: "ok", limit: 5 });
119
-
120
- expect(calls.ready).toBe(0);
121
- expect(calls.vector).toBe(0);
122
- });
123
- });