@modusensus/dsh-mneme 0.8.14 → 0.8.16
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/LICENSE +21 -21
- package/README.md +660 -654
- package/assets/icon.svg +3 -3
- package/bin/cli.mjs +691 -689
- package/bin/dsh-mneme-mcp.mjs +510 -510
- package/bin/dsh-mneme-serve.mjs +151 -151
- package/cordis.patch.yml +21 -21
- package/docs/CONFIGURATION.md +275 -0
- package/docs/CONTINUITY.md +124 -0
- package/docs/DAEMON.md +46 -0
- package/docs/DISTILL.md +162 -0
- package/docs/ENTITIES.md +245 -0
- package/docs/LOCAL_MODEL.md +222 -0
- package/docs/SEMANTIC.md +256 -0
- package/docs/SLEEP.md +164 -0
- package/docs/STORAGE.md +98 -0
- package/lib/api-standalone.js +656 -673
- package/lib/api.js +1605 -1433
- package/lib/bootstrap.js +211 -211
- package/lib/client.js +6354 -5296
- package/lib/commands.js +95 -95
- package/lib/config.js +947 -843
- package/lib/content-hash.js +54 -54
- package/lib/continuity.js +220 -220
- package/lib/document-index.js +94 -94
- package/lib/document.js +303 -303
- package/lib/dream/clustering.js +118 -118
- package/lib/dream/decisions.js +694 -674
- package/lib/dream/narratives.js +44 -44
- package/lib/dream/sleep.js +819 -819
- package/lib/dream.js +1503 -1501
- package/lib/embedding.js +177 -177
- package/lib/entities/extractor.js +252 -252
- package/lib/graph/anchoring.js +179 -179
- package/lib/guide.js +74 -74
- package/lib/heat.js +144 -144
- package/lib/hot-memory.js +108 -46
- package/lib/index.js +552 -503
- package/lib/inject.js +428 -450
- package/lib/lang.js +248 -229
- package/lib/local-embedder.js +358 -358
- package/lib/maintenance.js +162 -142
- package/lib/mirror.js +330 -330
- package/lib/organize.js +450 -450
- package/lib/peak-hours.js +117 -117
- package/lib/placement.js +13 -0
- package/lib/pre-inject-gate.js +340 -0
- package/lib/quality-filter.js +126 -126
- package/lib/recall-stats.js +190 -175
- package/lib/reclaim-http.js +150 -0
- package/lib/reranker.js +269 -269
- package/lib/runtime/adopt.js +176 -176
- package/lib/runtime/closure.js +172 -172
- package/lib/runtime/download.js +247 -247
- package/lib/runtime/layout.js +265 -265
- package/lib/runtime/loader.js +252 -252
- package/lib/runtime/model-fetch.js +361 -361
- package/lib/runtime/provision.js +237 -237
- package/lib/runtime/tarball.js +130 -130
- package/lib/runtime/verify.js +223 -204
- package/lib/safe-slice.js +18 -0
- package/lib/scope.js +146 -146
- package/lib/search/adaptive.js +60 -60
- package/lib/search/bm25.js +96 -96
- package/lib/semantic.js +229 -229
- package/lib/sensitive-scan.js +209 -209
- package/lib/serve.js +215 -215
- package/lib/service.js +2688 -2605
- package/lib/settings.js +528 -511
- package/lib/store.js +3196 -3007
- package/lib/summarize.js +1039 -856
- package/lib/tools.js +743 -737
- package/lib/vector-index.js +110 -110
- package/lib/version-check.js +122 -122
- package/lib/write-admission.js +402 -402
- package/locale/en.json +6 -6
- package/locale/zh.json +6 -6
- package/package.json +118 -118
- package/runtime-manifest.json +526 -526
- package/scripts/benchmark-embed.js +201 -201
- package/scripts/benchmark-recall.js +249 -249
- package/scripts/benchmark-rerank.js +166 -166
- package/scripts/build-runtime-manifest.mjs +224 -224
- package/scripts/changelog-prep.mjs +36 -36
- package/scripts/check-sync.js +42 -42
- package/scripts/e2e-dsh.js +220 -220
- package/scripts/mneme-runtime.mjs +277 -277
- package/scripts/stress-dsh.js +255 -255
- package/scripts/sync-lib.js +47 -47
- package/scripts/sync-test-badge.mjs +69 -69
- package/scripts/test-count-sync.mjs +44 -44
- package/src/api-standalone.js +656 -673
- package/src/api.js +1605 -1433
- package/src/bootstrap.js +211 -211
- package/src/commands.js +95 -95
- package/src/config.js +947 -843
- package/src/content-hash.js +54 -54
- package/src/continuity.js +220 -220
- package/src/document-index.js +94 -94
- package/src/document.js +303 -303
- package/src/dream/clustering.js +118 -118
- package/src/dream/decisions.js +694 -674
- package/src/dream/narratives.js +44 -44
- package/src/dream/sleep.js +819 -819
- package/src/dream.js +1503 -1501
- package/src/embedding.js +177 -177
- package/src/entities/extractor.js +252 -252
- package/src/graph/anchoring.js +179 -179
- package/src/guide.js +74 -74
- package/src/heat.js +144 -144
- package/src/hot-memory.js +108 -46
- package/src/index.js +552 -503
- package/src/inject.js +428 -450
- package/src/lang.js +248 -229
- package/src/local-embedder.js +358 -358
- package/src/maintenance.js +162 -142
- package/src/mirror.js +330 -330
- package/src/organize.js +450 -450
- package/src/peak-hours.js +117 -117
- package/src/placement.js +13 -0
- package/src/pre-inject-gate.js +340 -0
- package/src/quality-filter.js +126 -126
- package/src/recall-stats.js +190 -175
- package/src/reclaim-http.js +150 -0
- package/src/reranker.js +269 -269
- package/src/runtime/adopt.js +176 -176
- package/src/runtime/closure.js +172 -172
- package/src/runtime/download.js +247 -247
- package/src/runtime/layout.js +265 -265
- package/src/runtime/loader.js +252 -252
- package/src/runtime/model-fetch.js +361 -361
- package/src/runtime/provision.js +237 -237
- package/src/runtime/tarball.js +130 -130
- package/src/runtime/verify.js +223 -204
- package/src/safe-slice.js +18 -0
- package/src/scope.js +146 -146
- package/src/search/adaptive.js +60 -60
- package/src/search/bm25.js +96 -96
- package/src/semantic.js +229 -229
- package/src/sensitive-scan.js +209 -209
- package/src/serve.js +215 -215
- package/src/service.js +2688 -2605
- package/src/settings.js +528 -511
- package/src/store.js +3196 -3007
- package/src/summarize.js +1039 -856
- package/src/tools.js +743 -737
- package/src/vector-index.js +110 -110
- package/src/version-check.js +122 -122
- package/src/write-admission.js +402 -402
- package/test/adaptive-inject-budget.test.js +85 -85
- package/test/api-routes-gaps.test.js +899 -218
- package/test/api.test.js +1725 -1690
- package/test/audit.test.js +448 -448
- package/test/baseline-persist.test.js +248 -248
- package/test/benchmark.test.js +94 -94
- package/test/bootstrap.test.js +220 -220
- package/test/changelog-prep.test.js +60 -60
- package/test/client.test.js +1582 -1169
- package/test/clustering.test.js +100 -100
- package/test/commands.test.js +106 -106
- package/test/config.test.js +66 -66
- package/test/conflict-freeze.test.js +300 -300
- package/test/content-hash.test.js +204 -204
- package/test/continuity.test.js +333 -333
- package/test/contrast.test.js +314 -0
- package/test/document-dir.test.js +233 -233
- package/test/document.test.js +671 -671
- package/test/dream-archive-guardrail.test.js +116 -116
- package/test/dream-candidate-hybrid.test.js +281 -281
- package/test/dream-coverage-degrade.test.js +86 -86
- package/test/dream-failure-backoff.test.js +173 -173
- package/test/dream-merge-guardrail.test.js +219 -0
- package/test/dream-peak-hours.test.js +194 -194
- package/test/dream.test.js +1385 -1383
- package/test/embedding-ready.test.js +196 -196
- package/test/entities.test.js +627 -547
- package/test/entity-recall.test.js +179 -179
- package/test/epistemic.test.js +325 -325
- package/test/fnew-0112.test.js +311 -311
- package/test/fnew-03.test.js +422 -422
- package/test/graph-anchoring.test.js +184 -184
- package/test/graph-api.test.js +175 -175
- package/test/graph-inject-budget.test.js +103 -103
- package/test/graph-passive-confirm.test.js +91 -91
- package/test/graph-weight-evolution.test.js +104 -104
- package/test/heat.test.js +175 -175
- package/test/helpers/dream-mock.js +82 -82
- package/test/helpers/peer-worker.mjs +25 -25
- package/test/helpers/runtime-source.js +60 -60
- package/test/helpers/tar-builder.js +80 -80
- package/test/helpers/write-admission-samples.js +308 -308
- package/test/hot-memory.test.js +211 -145
- package/test/inject-content-cap.test.js +91 -91
- package/test/inject-first-round-bm25.test.js +89 -89
- package/test/inject-heat.test.js +77 -77
- package/test/inject-parent-gate.test.js +106 -106
- package/test/inject-pools.test.js +194 -194
- package/test/inject-recall.test.js +144 -105
- package/test/inject-rotation.test.js +161 -161
- package/test/inject.test.js +443 -420
- package/test/lang.test.js +87 -87
- package/test/lastrunat-persist.test.js +101 -101
- package/test/lib-smoke.test.js +96 -96
- package/test/llm-audit.test.js +696 -598
- package/test/llm-message-source.test.js +207 -207
- package/test/local-embedder.test.js +292 -292
- package/test/maintenance.test.js +501 -391
- package/test/mcp-stdio.test.js +385 -385
- package/test/mirror-dirty.test.js +424 -424
- package/test/mirror-edit-digest.test.js +187 -187
- package/test/mirror-generation.test.js +499 -499
- package/test/mirror-types.test.js +208 -208
- package/test/mirror.test.js +249 -249
- package/test/narrative-bars.test.js +162 -162
- package/test/organize.test.js +418 -418
- package/test/peer-blockers.test.js +193 -193
- package/test/placement.test.js +36 -0
- package/test/policy-epoch.test.js +259 -259
- package/test/pre-inject-gate.test.js +444 -0
- package/test/quality-filter.test.js +186 -186
- package/test/reasoning-effort.test.js +752 -752
- package/test/recall-anti-update.test.js +82 -82
- package/test/recall-evals.test.js +253 -253
- package/test/recall-layer.test.js +326 -326
- package/test/recall-stats.test.js +274 -189
- package/test/receipt-chain.test.js +451 -451
- package/test/reclaim-http.test.js +209 -0
- package/test/reflection.test.js +226 -226
- package/test/reindex-backfill.test.js +62 -62
- package/test/reranker-single-logit.test.js +73 -73
- package/test/reranker.test.js +197 -197
- package/test/resident-status.test.js +83 -83
- package/test/runtime-adopt.test.js +206 -206
- package/test/runtime-cli.test.js +89 -89
- package/test/runtime-closure.test.js +193 -193
- package/test/runtime-download.test.js +302 -302
- package/test/runtime-layout.test.js +163 -163
- package/test/runtime-loader.test.js +255 -255
- package/test/runtime-manifest.test.js +159 -159
- package/test/runtime-model-fetch.test.js +348 -348
- package/test/runtime-provision.test.js +101 -101
- package/test/runtime-tarball.test.js +81 -81
- package/test/runtime-verify.test.js +319 -279
- package/test/scope-candidates.test.js +239 -239
- package/test/scope-explicit.test.js +452 -452
- package/test/scope-inject-soft.test.js +106 -106
- package/test/scope-retrieval.test.js +197 -197
- package/test/scope-storage.test.js +255 -255
- package/test/scope-strict.test.js +367 -367
- package/test/search-fusion.test.js +90 -90
- package/test/semantic.test.js +124 -124
- package/test/sensitive-scan.test.js +207 -207
- package/test/serve-bin.test.js +121 -121
- package/test/serve.test.js +196 -196
- package/test/service-search.test.js +174 -174
- package/test/service.test.js +410 -405
- package/test/settings.test.js +269 -261
- package/test/skip-invalid-restore.test.js +262 -262
- package/test/sleep-action-set.test.js +422 -422
- package/test/sleep-heat.test.js +126 -126
- package/test/sleep-scheduler-rearm.test.js +89 -89
- package/test/sleep.test.js +527 -527
- package/test/standalone-api.test.js +751 -751
- package/test/store-vector-cache.test.js +98 -98
- package/test/store.test.js +504 -396
- package/test/stress.test.js +209 -209
- package/test/summarize-occurred-at.test.js +127 -0
- package/test/summarize-reasoning-effort.test.js +189 -189
- package/test/summarize.test.js +1639 -1465
- package/test/supersede-state-tracking.test.js +394 -0
- package/test/test-count-sync.test.js +60 -60
- package/test/tools-dto-schema-matrix.test.js +325 -325
- package/test/tools-memory-get-schema.test.js +65 -65
- package/test/tools.test.js +428 -374
- package/test/updated-at-semantics.test.js +113 -113
- package/test/vector-index.test.js +221 -221
- package/test/version-check.test.js +152 -152
- package/test/webServer-optional.test.js +82 -82
- package/test/write-admission.test.js +668 -668
package/test/graph-api.test.js
CHANGED
|
@@ -1,175 +1,175 @@
|
|
|
1
|
-
import test from "node:test";
|
|
2
|
-
import assert from "node:assert/strict";
|
|
3
|
-
import { EventEmitter } from "node:events";
|
|
4
|
-
import { createStore } from "../src/store.js";
|
|
5
|
-
import { createService } from "../src/service.js";
|
|
6
|
-
import { createApi } from "../src/api.js";
|
|
7
|
-
import { createSettings } from "../src/settings.js";
|
|
8
|
-
|
|
9
|
-
// Ego-graph read API (graph panel P1). The routes under test:
|
|
10
|
-
// GET /api/dsh-mneme/semantic/graph/ego?entity=&depth=&limit=
|
|
11
|
-
// GET /api/dsh-mneme/semantic/graph/entity-attrs?entity=
|
|
12
|
-
// Both are read-only: they must stay reachable without an apiToken, exactly
|
|
13
|
-
// like list/search/semantic.
|
|
14
|
-
|
|
15
|
-
class FakeRes extends EventEmitter {
|
|
16
|
-
constructor() { super(); this.statusCode = 200; this.body = ""; }
|
|
17
|
-
writeHead(code, headers) { this.statusCode = code; this.headers = headers; return this; }
|
|
18
|
-
end(text) { this.body = text ?? ""; this.emit("end"); return this; }
|
|
19
|
-
}
|
|
20
|
-
|
|
21
|
-
function req(path) {
|
|
22
|
-
const r = new EventEmitter();
|
|
23
|
-
r.url = path;
|
|
24
|
-
r.method = "GET";
|
|
25
|
-
r.headers = {};
|
|
26
|
-
return r;
|
|
27
|
-
}
|
|
28
|
-
|
|
29
|
-
function setup(apiToken = "") {
|
|
30
|
-
const store = createStore(":memory:");
|
|
31
|
-
const service = createService({ store, mirror: null, config: {} });
|
|
32
|
-
const settings = createSettings(store.db);
|
|
33
|
-
const commands = { add: () => {}, remove: () => {}, list: () => [] };
|
|
34
|
-
const routes = [];
|
|
35
|
-
const ctx = { webServer: { register(route) { routes.push(route); return () => {}; } } };
|
|
36
|
-
createApi(ctx, service, settings, commands, null, undefined, apiToken);
|
|
37
|
-
return { store, service, routes };
|
|
38
|
-
}
|
|
39
|
-
|
|
40
|
-
function graphRoute(routes, path) {
|
|
41
|
-
return routes.find((r) => r.path === path);
|
|
42
|
-
}
|
|
43
|
-
|
|
44
|
-
function buildChain(service, names, linkType = "related_to") {
|
|
45
|
-
// names → entities; each consecutive pair gets one relation. Returns a
|
|
46
|
-
// name→entity map so tests can assert on ids.
|
|
47
|
-
const map = new Map();
|
|
48
|
-
for (const n of names) map.set(n, service.createEntity({ name: n, type: "concept" }));
|
|
49
|
-
for (let i = 0; i < names.length - 1; i++) {
|
|
50
|
-
service.saveRelation({
|
|
51
|
-
from_entity: map.get(names[i]).id,
|
|
52
|
-
to_entity: map.get(names[i + 1]).id,
|
|
53
|
-
relation_type: linkType
|
|
54
|
-
});
|
|
55
|
-
}
|
|
56
|
-
return map;
|
|
57
|
-
}
|
|
58
|
-
|
|
59
|
-
test("ego route is registered and open without apiToken", async () => {
|
|
60
|
-
const { routes } = setup("secret-token");
|
|
61
|
-
const ego = graphRoute(routes, "/api/dsh-mneme/semantic/graph/ego");
|
|
62
|
-
const attrs = graphRoute(routes, "/api/dsh-mneme/semantic/graph/entity-attrs");
|
|
63
|
-
assert.ok(ego, "ego route must be registered");
|
|
64
|
-
assert.ok(attrs, "entity-attrs route must be registered");
|
|
65
|
-
// read-only: no 401 even with a token configured
|
|
66
|
-
const res = new FakeRes();
|
|
67
|
-
await ego.handler(req("/api/dsh-mneme/semantic/graph/ego"), res);
|
|
68
|
-
assert.equal(res.statusCode, 400, "missing param is a 400, not a 401");
|
|
69
|
-
});
|
|
70
|
-
|
|
71
|
-
test("ego depth 1 returns root + direct neighbors with distances", async () => {
|
|
72
|
-
const { routes, service } = setup();
|
|
73
|
-
const map = buildChain(service, ["A", "B", "C"]);
|
|
74
|
-
const route = graphRoute(routes, "/api/dsh-mneme/semantic/graph/ego");
|
|
75
|
-
const res = new FakeRes();
|
|
76
|
-
await route.handler(req("/api/dsh-mneme/semantic/graph/ego?entity=A&depth=1"), res);
|
|
77
|
-
assert.equal(res.statusCode, 200);
|
|
78
|
-
const data = JSON.parse(res.body);
|
|
79
|
-
assert.equal(data.root.name, "A");
|
|
80
|
-
assert.equal(data.nodes.length, 2, "root + B only (C is 2 hops)");
|
|
81
|
-
assert.equal(data.edges.length, 1);
|
|
82
|
-
const b = data.nodes.find((n) => n.name === "B");
|
|
83
|
-
assert.equal(b.distance, 1);
|
|
84
|
-
assert.equal(data.nodes.find((n) => n.name === "A").distance, 0);
|
|
85
|
-
// edge shape: ids + relation_type, memory_id nullable
|
|
86
|
-
const edge = data.edges[0];
|
|
87
|
-
assert.ok(edge.id && edge.from && edge.to);
|
|
88
|
-
assert.equal(edge.relation_type, "related_to");
|
|
89
|
-
assert.equal(edge.memory_id, null);
|
|
90
|
-
});
|
|
91
|
-
|
|
92
|
-
test("ego depth 2 includes 2-hop nodes, not 3-hop", async () => {
|
|
93
|
-
const { routes, service } = setup();
|
|
94
|
-
buildChain(service, ["A", "B", "C", "D"]);
|
|
95
|
-
const route = graphRoute(routes, "/api/dsh-mneme/semantic/graph/ego");
|
|
96
|
-
const res = new FakeRes();
|
|
97
|
-
await route.handler(req("/api/dsh-mneme/semantic/graph/ego?entity=A&depth=2"), res);
|
|
98
|
-
const data = JSON.parse(res.body);
|
|
99
|
-
const names = data.nodes.map((n) => n.name).sort();
|
|
100
|
-
assert.deepEqual(names, ["A", "B", "C"], "D is 3 hops away and must be excluded");
|
|
101
|
-
assert.equal(data.nodes.find((n) => n.name === "C").distance, 2);
|
|
102
|
-
assert.equal(data.edges.length, 2);
|
|
103
|
-
});
|
|
104
|
-
|
|
105
|
-
test("ego respects limit and keeps only in-graph edges", async () => {
|
|
106
|
-
const { routes, service } = setup();
|
|
107
|
-
const map = buildChain(service, ["A", "B", "C"]);
|
|
108
|
-
const route = graphRoute(routes, "/api/dsh-mneme/semantic/graph/ego");
|
|
109
|
-
const res = new FakeRes();
|
|
110
|
-
await route.handler(req("/api/dsh-mneme/semantic/graph/ego?entity=A&depth=2&limit=2"), res);
|
|
111
|
-
const data = JSON.parse(res.body);
|
|
112
|
-
assert.equal(data.nodes.length, 2, "limit caps node count");
|
|
113
|
-
assert.equal(data.edges.length, 1, "edges to cut nodes are dropped");
|
|
114
|
-
});
|
|
115
|
-
|
|
116
|
-
test("ego reports 404 for unknown entity and 400 without entity", async () => {
|
|
117
|
-
const { routes } = setup();
|
|
118
|
-
const route = graphRoute(routes, "/api/dsh-mneme/semantic/graph/ego");
|
|
119
|
-
let res = new FakeRes();
|
|
120
|
-
await route.handler(req("/api/dsh-mneme/semantic/graph/ego?entity=ghost"), res);
|
|
121
|
-
assert.equal(res.statusCode, 404);
|
|
122
|
-
assert.equal(JSON.parse(res.body).error, "entity-not-found");
|
|
123
|
-
res = new FakeRes();
|
|
124
|
-
await route.handler(req("/api/dsh-mneme/semantic/graph/ego"), res);
|
|
125
|
-
assert.equal(res.statusCode, 400);
|
|
126
|
-
assert.equal(JSON.parse(res.body).error, "missing-entity");
|
|
127
|
-
});
|
|
128
|
-
|
|
129
|
-
test("entity-attrs returns current valid attrs only", async () => {
|
|
130
|
-
const { routes, service } = setup();
|
|
131
|
-
const mem = service.saveWithDedupe({ type: "project", title: "t", content: "c" });
|
|
132
|
-
const entity = service.createEntity({ name: "X", type: "technology" });
|
|
133
|
-
service.saveAttr({ entity_id: entity.id, attr_key: "version", attr_value: "1", memory_id: mem.id, confidence: 0.9 });
|
|
134
|
-
// a second save of the same key invalidates the first: only one row stays valid
|
|
135
|
-
service.saveAttr({ entity_id: entity.id, attr_key: "version", attr_value: "2", memory_id: mem.id, confidence: 0.8 });
|
|
136
|
-
const route = graphRoute(routes, "/api/dsh-mneme/semantic/graph/entity-attrs");
|
|
137
|
-
const res = new FakeRes();
|
|
138
|
-
await route.handler(req("/api/dsh-mneme/semantic/graph/entity-attrs?entity=X"), res);
|
|
139
|
-
assert.equal(res.statusCode, 200);
|
|
140
|
-
const data = JSON.parse(res.body);
|
|
141
|
-
assert.equal(data.entity.name, "X");
|
|
142
|
-
assert.equal(data.attrs.length, 1);
|
|
143
|
-
assert.equal(data.attrs[0].value, "2", "latest valid attr wins");
|
|
144
|
-
assert.equal(data.attrs[0].confidence, 0.8);
|
|
145
|
-
// unknown / missing entity
|
|
146
|
-
let r2 = new FakeRes();
|
|
147
|
-
await route.handler(req("/api/dsh-mneme/semantic/graph/entity-attrs?entity=nope"), r2);
|
|
148
|
-
assert.equal(r2.statusCode, 404);
|
|
149
|
-
r2 = new FakeRes();
|
|
150
|
-
await route.handler(req("/api/dsh-mneme/semantic/graph/entity-attrs"), r2);
|
|
151
|
-
assert.equal(r2.statusCode, 400);
|
|
152
|
-
});
|
|
153
|
-
|
|
154
|
-
test("ego 2-hop over 100+ nodes stays under 50ms (spec §7)", async () => {
|
|
155
|
-
const { routes, service } = setup();
|
|
156
|
-
// star-of-stars: hub → 20 spokes, each spoke → 5 leaves = 121 nodes, 120 edges
|
|
157
|
-
const hub = service.createEntity({ name: "hub", type: "project" });
|
|
158
|
-
for (let i = 0; i < 20; i++) {
|
|
159
|
-
const spoke = service.createEntity({ name: `spoke-${i}`, type: "technology" });
|
|
160
|
-
service.saveRelation({ from_entity: hub.id, to_entity: spoke.id, relation_type: "uses" });
|
|
161
|
-
for (let j = 0; j < 5; j++) {
|
|
162
|
-
const leaf = service.createEntity({ name: `leaf-${i}-${j}`, type: "concept" });
|
|
163
|
-
service.saveRelation({ from_entity: spoke.id, to_entity: leaf.id, relation_type: "related_to" });
|
|
164
|
-
}
|
|
165
|
-
}
|
|
166
|
-
const route = graphRoute(routes, "/api/dsh-mneme/semantic/graph/ego");
|
|
167
|
-
const res = new FakeRes();
|
|
168
|
-
const start = process.hrtime.bigint();
|
|
169
|
-
await route.handler(req("/api/dsh-mneme/semantic/graph/ego?entity=hub&depth=2&limit=100"), res);
|
|
170
|
-
const ms = Number(process.hrtime.bigint() - start) / 1e6;
|
|
171
|
-
const data = JSON.parse(res.body);
|
|
172
|
-
assert.equal(data.nodes.length, 100, "limit caps the traversal");
|
|
173
|
-
assert.ok(data.edges.length > 0);
|
|
174
|
-
assert.ok(ms < 50, `2-hop query took ${ms.toFixed(1)}ms, spec budget is 50ms`);
|
|
175
|
-
});
|
|
1
|
+
import test from "node:test";
|
|
2
|
+
import assert from "node:assert/strict";
|
|
3
|
+
import { EventEmitter } from "node:events";
|
|
4
|
+
import { createStore } from "../src/store.js";
|
|
5
|
+
import { createService } from "../src/service.js";
|
|
6
|
+
import { createApi } from "../src/api.js";
|
|
7
|
+
import { createSettings } from "../src/settings.js";
|
|
8
|
+
|
|
9
|
+
// Ego-graph read API (graph panel P1). The routes under test:
|
|
10
|
+
// GET /api/dsh-mneme/semantic/graph/ego?entity=&depth=&limit=
|
|
11
|
+
// GET /api/dsh-mneme/semantic/graph/entity-attrs?entity=
|
|
12
|
+
// Both are read-only: they must stay reachable without an apiToken, exactly
|
|
13
|
+
// like list/search/semantic.
|
|
14
|
+
|
|
15
|
+
class FakeRes extends EventEmitter {
|
|
16
|
+
constructor() { super(); this.statusCode = 200; this.body = ""; }
|
|
17
|
+
writeHead(code, headers) { this.statusCode = code; this.headers = headers; return this; }
|
|
18
|
+
end(text) { this.body = text ?? ""; this.emit("end"); return this; }
|
|
19
|
+
}
|
|
20
|
+
|
|
21
|
+
function req(path) {
|
|
22
|
+
const r = new EventEmitter();
|
|
23
|
+
r.url = path;
|
|
24
|
+
r.method = "GET";
|
|
25
|
+
r.headers = {};
|
|
26
|
+
return r;
|
|
27
|
+
}
|
|
28
|
+
|
|
29
|
+
function setup(apiToken = "") {
|
|
30
|
+
const store = createStore(":memory:");
|
|
31
|
+
const service = createService({ store, mirror: null, config: {} });
|
|
32
|
+
const settings = createSettings(store.db);
|
|
33
|
+
const commands = { add: () => {}, remove: () => {}, list: () => [] };
|
|
34
|
+
const routes = [];
|
|
35
|
+
const ctx = { webServer: { register(route) { routes.push(route); return () => {}; } } };
|
|
36
|
+
createApi(ctx, service, settings, commands, null, undefined, apiToken);
|
|
37
|
+
return { store, service, routes };
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
function graphRoute(routes, path) {
|
|
41
|
+
return routes.find((r) => r.path === path);
|
|
42
|
+
}
|
|
43
|
+
|
|
44
|
+
function buildChain(service, names, linkType = "related_to") {
|
|
45
|
+
// names → entities; each consecutive pair gets one relation. Returns a
|
|
46
|
+
// name→entity map so tests can assert on ids.
|
|
47
|
+
const map = new Map();
|
|
48
|
+
for (const n of names) map.set(n, service.createEntity({ name: n, type: "concept" }));
|
|
49
|
+
for (let i = 0; i < names.length - 1; i++) {
|
|
50
|
+
service.saveRelation({
|
|
51
|
+
from_entity: map.get(names[i]).id,
|
|
52
|
+
to_entity: map.get(names[i + 1]).id,
|
|
53
|
+
relation_type: linkType
|
|
54
|
+
});
|
|
55
|
+
}
|
|
56
|
+
return map;
|
|
57
|
+
}
|
|
58
|
+
|
|
59
|
+
test("ego route is registered and open without apiToken", async () => {
|
|
60
|
+
const { routes } = setup("secret-token");
|
|
61
|
+
const ego = graphRoute(routes, "/api/dsh-mneme/semantic/graph/ego");
|
|
62
|
+
const attrs = graphRoute(routes, "/api/dsh-mneme/semantic/graph/entity-attrs");
|
|
63
|
+
assert.ok(ego, "ego route must be registered");
|
|
64
|
+
assert.ok(attrs, "entity-attrs route must be registered");
|
|
65
|
+
// read-only: no 401 even with a token configured
|
|
66
|
+
const res = new FakeRes();
|
|
67
|
+
await ego.handler(req("/api/dsh-mneme/semantic/graph/ego"), res);
|
|
68
|
+
assert.equal(res.statusCode, 400, "missing param is a 400, not a 401");
|
|
69
|
+
});
|
|
70
|
+
|
|
71
|
+
test("ego depth 1 returns root + direct neighbors with distances", async () => {
|
|
72
|
+
const { routes, service } = setup();
|
|
73
|
+
const map = buildChain(service, ["A", "B", "C"]);
|
|
74
|
+
const route = graphRoute(routes, "/api/dsh-mneme/semantic/graph/ego");
|
|
75
|
+
const res = new FakeRes();
|
|
76
|
+
await route.handler(req("/api/dsh-mneme/semantic/graph/ego?entity=A&depth=1"), res);
|
|
77
|
+
assert.equal(res.statusCode, 200);
|
|
78
|
+
const data = JSON.parse(res.body);
|
|
79
|
+
assert.equal(data.root.name, "A");
|
|
80
|
+
assert.equal(data.nodes.length, 2, "root + B only (C is 2 hops)");
|
|
81
|
+
assert.equal(data.edges.length, 1);
|
|
82
|
+
const b = data.nodes.find((n) => n.name === "B");
|
|
83
|
+
assert.equal(b.distance, 1);
|
|
84
|
+
assert.equal(data.nodes.find((n) => n.name === "A").distance, 0);
|
|
85
|
+
// edge shape: ids + relation_type, memory_id nullable
|
|
86
|
+
const edge = data.edges[0];
|
|
87
|
+
assert.ok(edge.id && edge.from && edge.to);
|
|
88
|
+
assert.equal(edge.relation_type, "related_to");
|
|
89
|
+
assert.equal(edge.memory_id, null);
|
|
90
|
+
});
|
|
91
|
+
|
|
92
|
+
test("ego depth 2 includes 2-hop nodes, not 3-hop", async () => {
|
|
93
|
+
const { routes, service } = setup();
|
|
94
|
+
buildChain(service, ["A", "B", "C", "D"]);
|
|
95
|
+
const route = graphRoute(routes, "/api/dsh-mneme/semantic/graph/ego");
|
|
96
|
+
const res = new FakeRes();
|
|
97
|
+
await route.handler(req("/api/dsh-mneme/semantic/graph/ego?entity=A&depth=2"), res);
|
|
98
|
+
const data = JSON.parse(res.body);
|
|
99
|
+
const names = data.nodes.map((n) => n.name).sort();
|
|
100
|
+
assert.deepEqual(names, ["A", "B", "C"], "D is 3 hops away and must be excluded");
|
|
101
|
+
assert.equal(data.nodes.find((n) => n.name === "C").distance, 2);
|
|
102
|
+
assert.equal(data.edges.length, 2);
|
|
103
|
+
});
|
|
104
|
+
|
|
105
|
+
test("ego respects limit and keeps only in-graph edges", async () => {
|
|
106
|
+
const { routes, service } = setup();
|
|
107
|
+
const map = buildChain(service, ["A", "B", "C"]);
|
|
108
|
+
const route = graphRoute(routes, "/api/dsh-mneme/semantic/graph/ego");
|
|
109
|
+
const res = new FakeRes();
|
|
110
|
+
await route.handler(req("/api/dsh-mneme/semantic/graph/ego?entity=A&depth=2&limit=2"), res);
|
|
111
|
+
const data = JSON.parse(res.body);
|
|
112
|
+
assert.equal(data.nodes.length, 2, "limit caps node count");
|
|
113
|
+
assert.equal(data.edges.length, 1, "edges to cut nodes are dropped");
|
|
114
|
+
});
|
|
115
|
+
|
|
116
|
+
test("ego reports 404 for unknown entity and 400 without entity", async () => {
|
|
117
|
+
const { routes } = setup();
|
|
118
|
+
const route = graphRoute(routes, "/api/dsh-mneme/semantic/graph/ego");
|
|
119
|
+
let res = new FakeRes();
|
|
120
|
+
await route.handler(req("/api/dsh-mneme/semantic/graph/ego?entity=ghost"), res);
|
|
121
|
+
assert.equal(res.statusCode, 404);
|
|
122
|
+
assert.equal(JSON.parse(res.body).error, "entity-not-found");
|
|
123
|
+
res = new FakeRes();
|
|
124
|
+
await route.handler(req("/api/dsh-mneme/semantic/graph/ego"), res);
|
|
125
|
+
assert.equal(res.statusCode, 400);
|
|
126
|
+
assert.equal(JSON.parse(res.body).error, "missing-entity");
|
|
127
|
+
});
|
|
128
|
+
|
|
129
|
+
test("entity-attrs returns current valid attrs only", async () => {
|
|
130
|
+
const { routes, service } = setup();
|
|
131
|
+
const mem = service.saveWithDedupe({ type: "project", title: "t", content: "c" });
|
|
132
|
+
const entity = service.createEntity({ name: "X", type: "technology" });
|
|
133
|
+
service.saveAttr({ entity_id: entity.id, attr_key: "version", attr_value: "1", memory_id: mem.id, confidence: 0.9 });
|
|
134
|
+
// a second save of the same key invalidates the first: only one row stays valid
|
|
135
|
+
service.saveAttr({ entity_id: entity.id, attr_key: "version", attr_value: "2", memory_id: mem.id, confidence: 0.8 });
|
|
136
|
+
const route = graphRoute(routes, "/api/dsh-mneme/semantic/graph/entity-attrs");
|
|
137
|
+
const res = new FakeRes();
|
|
138
|
+
await route.handler(req("/api/dsh-mneme/semantic/graph/entity-attrs?entity=X"), res);
|
|
139
|
+
assert.equal(res.statusCode, 200);
|
|
140
|
+
const data = JSON.parse(res.body);
|
|
141
|
+
assert.equal(data.entity.name, "X");
|
|
142
|
+
assert.equal(data.attrs.length, 1);
|
|
143
|
+
assert.equal(data.attrs[0].value, "2", "latest valid attr wins");
|
|
144
|
+
assert.equal(data.attrs[0].confidence, 0.8);
|
|
145
|
+
// unknown / missing entity
|
|
146
|
+
let r2 = new FakeRes();
|
|
147
|
+
await route.handler(req("/api/dsh-mneme/semantic/graph/entity-attrs?entity=nope"), r2);
|
|
148
|
+
assert.equal(r2.statusCode, 404);
|
|
149
|
+
r2 = new FakeRes();
|
|
150
|
+
await route.handler(req("/api/dsh-mneme/semantic/graph/entity-attrs"), r2);
|
|
151
|
+
assert.equal(r2.statusCode, 400);
|
|
152
|
+
});
|
|
153
|
+
|
|
154
|
+
test("ego 2-hop over 100+ nodes stays under 50ms (spec §7)", async () => {
|
|
155
|
+
const { routes, service } = setup();
|
|
156
|
+
// star-of-stars: hub → 20 spokes, each spoke → 5 leaves = 121 nodes, 120 edges
|
|
157
|
+
const hub = service.createEntity({ name: "hub", type: "project" });
|
|
158
|
+
for (let i = 0; i < 20; i++) {
|
|
159
|
+
const spoke = service.createEntity({ name: `spoke-${i}`, type: "technology" });
|
|
160
|
+
service.saveRelation({ from_entity: hub.id, to_entity: spoke.id, relation_type: "uses" });
|
|
161
|
+
for (let j = 0; j < 5; j++) {
|
|
162
|
+
const leaf = service.createEntity({ name: `leaf-${i}-${j}`, type: "concept" });
|
|
163
|
+
service.saveRelation({ from_entity: spoke.id, to_entity: leaf.id, relation_type: "related_to" });
|
|
164
|
+
}
|
|
165
|
+
}
|
|
166
|
+
const route = graphRoute(routes, "/api/dsh-mneme/semantic/graph/ego");
|
|
167
|
+
const res = new FakeRes();
|
|
168
|
+
const start = process.hrtime.bigint();
|
|
169
|
+
await route.handler(req("/api/dsh-mneme/semantic/graph/ego?entity=hub&depth=2&limit=100"), res);
|
|
170
|
+
const ms = Number(process.hrtime.bigint() - start) / 1e6;
|
|
171
|
+
const data = JSON.parse(res.body);
|
|
172
|
+
assert.equal(data.nodes.length, 100, "limit caps the traversal");
|
|
173
|
+
assert.ok(data.edges.length > 0);
|
|
174
|
+
assert.ok(ms < 50, `2-hop query took ${ms.toFixed(1)}ms, spec budget is 50ms`);
|
|
175
|
+
});
|
|
@@ -1,104 +1,104 @@
|
|
|
1
|
-
// issue #24 块3 关联提示防幻觉(graphInjectHint / graphInjectBudget)测试。
|
|
2
|
-
// 覆盖:图候选打标 graphHint / 保守档(默认关)线索行不进注入块 / 开档受独立
|
|
3
|
-
// 预算 graphInjectBudget 约束且不挤占 maxItems / [检索线索] 引导语与线索行文本。
|
|
4
|
-
import test from "node:test";
|
|
5
|
-
import assert from "node:assert/strict";
|
|
6
|
-
import { createStore } from "../src/store.js";
|
|
7
|
-
import { createService } from "../src/service.js";
|
|
8
|
-
import { STR, langOf } from "../src/lang.js";
|
|
9
|
-
|
|
10
|
-
function setup(config = {}) {
|
|
11
|
-
const store = createStore(":memory:");
|
|
12
|
-
const service = createService({ store, mirror: null, config });
|
|
13
|
-
return { store, service };
|
|
14
|
-
}
|
|
15
|
-
|
|
16
|
-
function linkEntityMem(store, service, { name, title, content, key = "k", value = "v", type = "technology" }) {
|
|
17
|
-
const mem = service.saveWithDedupe({ type: "project", title, content, importance: 3 }).memory;
|
|
18
|
-
let ent = store.findEntityByName(name);
|
|
19
|
-
if (!ent) ent = store.createEntity({ name, type });
|
|
20
|
-
store.saveAttr({ entity_id: ent.id, attr_key: key, attr_value: value, memory_id: mem.id });
|
|
21
|
-
return mem;
|
|
22
|
-
}
|
|
23
|
-
|
|
24
|
-
test("graphHint candidates are tagged when entityRecallEnabled, and do NOT enter the block by default (conservative)", () => {
|
|
25
|
-
const { store, service } = setup({ entityRecallEnabled: true, maxInjectedItems: 5 });
|
|
26
|
-
// 一条与查询无关的普通记忆(避免规则路占用),一条实体挂联记忆。
|
|
27
|
-
linkEntityMem(store, service, {
|
|
28
|
-
name: "PostgreSQL", title: "数据库迁移", content: "主库切换完成,复制延迟归零"
|
|
29
|
-
});
|
|
30
|
-
service.saveWithDedupe({ type: "decision", title: "无关决策", content: "今天天气不错", importance: 4 });
|
|
31
|
-
const selected = service.injectCandidates({ maxItems: 5, query: "PostgreSQL" });
|
|
32
|
-
// 保守档默认 graphInjectHint=false:图线索行被 general 选取跳过——
|
|
33
|
-
// 候选池里能查到 graphHint 标记,但注入终集会排除它。
|
|
34
|
-
assert.ok(selected.every((m) => m.graphHint !== true), "conservative: no graph hint rows in the injected set");
|
|
35
|
-
});
|
|
36
|
-
|
|
37
|
-
test("graphInjectHint=true admits graph hint rows under independent budget", () => {
|
|
38
|
-
const { store, service } = setup({
|
|
39
|
-
entityRecallEnabled: true, maxInjectedItems: 5, graphInjectHint: true, graphInjectBudget: 2
|
|
40
|
-
});
|
|
41
|
-
// 两条实体挂联记忆(都命中查询实体),一条普通规则记忆。
|
|
42
|
-
linkEntityMem(store, service, { name: "PostgreSQL", title: "数据库迁移", content: "主库切换完成", key: "k1" });
|
|
43
|
-
linkEntityMem(store, service, { name: "PostgreSQL", title: "连接池调优", content: "PgBouncer 配置", key: "k2" });
|
|
44
|
-
service.saveWithDedupe({ type: "decision", title: "无关决策", content: "其他内容", importance: 5 });
|
|
45
|
-
const selected = service.injectCandidates({ maxItems: 5, query: "PostgreSQL" });
|
|
46
|
-
const hints = selected.filter((m) => m.graphHint === true);
|
|
47
|
-
assert.equal(hints.length, 2, "both graph hint rows fit within budget 2");
|
|
48
|
-
// 普通规则记忆仍在(graph 行不挤占 maxItems 槽)。
|
|
49
|
-
assert.ok(selected.some((m) => !m.graphHint), "non-graph rows still present");
|
|
50
|
-
store.close();
|
|
51
|
-
});
|
|
52
|
-
|
|
53
|
-
test("graphInjectBudget caps graph hint rows even when more match the query", () => {
|
|
54
|
-
const { store, service } = setup({
|
|
55
|
-
entityRecallEnabled: true, maxInjectedItems: 10, graphInjectHint: true, graphInjectBudget: 1
|
|
56
|
-
});
|
|
57
|
-
// 三条候选实体挂联记忆,预算只有 1。
|
|
58
|
-
for (const i of [1, 2, 3]) {
|
|
59
|
-
linkEntityMem(store, service, { name: "PostgreSQL", title: `PG 记录 ${i}`, content: `内容 ${i}`, key: `k${i}` });
|
|
60
|
-
}
|
|
61
|
-
const selected = service.injectCandidates({ maxItems: 10 });
|
|
62
|
-
const hints = selected.filter((m) => m.graphHint === true);
|
|
63
|
-
assert.ok(hints.length <= 1, `budget 1 caps at 1, got ${hints.length}`);
|
|
64
|
-
store.close();
|
|
65
|
-
});
|
|
66
|
-
|
|
67
|
-
test("graphHint=false (default) + entityRecallEnabled returns zero candidate set change (conservative = sorting only)", () => {
|
|
68
|
-
const { store, service } = setup({ entityRecallEnabled: true, maxInjectedItems: 5 });
|
|
69
|
-
// 仅实体挂联记忆,保守档下不该有任何候选进块。
|
|
70
|
-
linkEntityMem(store, service, { name: "PostgreSQL", title: "数据库迁移", content: "主库切换完成" });
|
|
71
|
-
const selected = service.injectCandidates({ maxItems: 5, query: "PostgreSQL 现状" });
|
|
72
|
-
assert.equal(selected.filter((m) => m.graphHint).length, 0);
|
|
73
|
-
store.close();
|
|
74
|
-
});
|
|
75
|
-
|
|
76
|
-
test("[检索线索] 引导语与线索行文本渲染正确(zh/en 双语)", () => {
|
|
77
|
-
assert.ok(STR.graphHintHeader.zh.includes("检索线索"));
|
|
78
|
-
assert.ok(STR.graphHintHeader.zh.includes("非事实断言"));
|
|
79
|
-
assert.ok(STR.graphHintHeader.en.includes("Retrieval hints"));
|
|
80
|
-
assert.ok(STR.graphHintHeader.en.toLowerCase().includes("not factual"));
|
|
81
|
-
const lineZh = STR.graphHintLine.zh("entity", "某某", "内容");
|
|
82
|
-
assert.ok(lineZh.startsWith("- [线索/entity]"));
|
|
83
|
-
const lineEn = STR.graphHintLine.en("technology", "Foo", "bar");
|
|
84
|
-
assert.ok(lineEn.startsWith("- [hint/technology]"));
|
|
85
|
-
});
|
|
86
|
-
|
|
87
|
-
// ---- 复审回归 ---------------------------------------------------------------
|
|
88
|
-
|
|
89
|
-
// 保守档(graphInjectHint 关)下线索行不许从 pin 池侧漏进注入块:pin 池取自打标
|
|
90
|
-
// 之后、又不占 maxItems 名额,pin 若不排除 graphHint 行,「图召回 + 约束型记忆」
|
|
91
|
-
// 就会在默认关档下照常进常驻注入文本,把线索开关整个绕过去。
|
|
92
|
-
test("conservative mode: a graph hint row cannot leak in via the pin pool", () => {
|
|
93
|
-
const { store, service } = setup({ entityRecallEnabled: true, maxInjectedItems: 5, pinnedInjectBudget: 1 });
|
|
94
|
-
// 约束型记忆 + 命中查询实体:同时是 pin 候选与图线索候选(最坏交叉)。
|
|
95
|
-
const mem = service.saveWithDedupe({
|
|
96
|
-
type: "constraint", title: "部署约束", content: "生产环境不许直接改库", importance: 4
|
|
97
|
-
}).memory;
|
|
98
|
-
const ent = store.createEntity({ name: "PostgreSQL", type: "technology" });
|
|
99
|
-
store.saveAttr({ entity_id: ent.id, attr_key: "k", attr_value: "v", memory_id: mem.id });
|
|
100
|
-
|
|
101
|
-
const selected = service.injectCandidates({ maxItems: 5, query: "PostgreSQL" });
|
|
102
|
-
assert.equal(selected.filter((m) => m.graphHint === true).length, 0, "hint rows stay out while graphInjectHint is off");
|
|
103
|
-
store.close();
|
|
1
|
+
// issue #24 块3 关联提示防幻觉(graphInjectHint / graphInjectBudget)测试。
|
|
2
|
+
// 覆盖:图候选打标 graphHint / 保守档(默认关)线索行不进注入块 / 开档受独立
|
|
3
|
+
// 预算 graphInjectBudget 约束且不挤占 maxItems / [检索线索] 引导语与线索行文本。
|
|
4
|
+
import test from "node:test";
|
|
5
|
+
import assert from "node:assert/strict";
|
|
6
|
+
import { createStore } from "../src/store.js";
|
|
7
|
+
import { createService } from "../src/service.js";
|
|
8
|
+
import { STR, langOf } from "../src/lang.js";
|
|
9
|
+
|
|
10
|
+
function setup(config = {}) {
|
|
11
|
+
const store = createStore(":memory:");
|
|
12
|
+
const service = createService({ store, mirror: null, config });
|
|
13
|
+
return { store, service };
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
function linkEntityMem(store, service, { name, title, content, key = "k", value = "v", type = "technology" }) {
|
|
17
|
+
const mem = service.saveWithDedupe({ type: "project", title, content, importance: 3 }).memory;
|
|
18
|
+
let ent = store.findEntityByName(name);
|
|
19
|
+
if (!ent) ent = store.createEntity({ name, type });
|
|
20
|
+
store.saveAttr({ entity_id: ent.id, attr_key: key, attr_value: value, memory_id: mem.id });
|
|
21
|
+
return mem;
|
|
22
|
+
}
|
|
23
|
+
|
|
24
|
+
test("graphHint candidates are tagged when entityRecallEnabled, and do NOT enter the block by default (conservative)", () => {
|
|
25
|
+
const { store, service } = setup({ entityRecallEnabled: true, maxInjectedItems: 5 });
|
|
26
|
+
// 一条与查询无关的普通记忆(避免规则路占用),一条实体挂联记忆。
|
|
27
|
+
linkEntityMem(store, service, {
|
|
28
|
+
name: "PostgreSQL", title: "数据库迁移", content: "主库切换完成,复制延迟归零"
|
|
29
|
+
});
|
|
30
|
+
service.saveWithDedupe({ type: "decision", title: "无关决策", content: "今天天气不错", importance: 4 });
|
|
31
|
+
const selected = service.injectCandidates({ maxItems: 5, query: "PostgreSQL" });
|
|
32
|
+
// 保守档默认 graphInjectHint=false:图线索行被 general 选取跳过——
|
|
33
|
+
// 候选池里能查到 graphHint 标记,但注入终集会排除它。
|
|
34
|
+
assert.ok(selected.every((m) => m.graphHint !== true), "conservative: no graph hint rows in the injected set");
|
|
35
|
+
});
|
|
36
|
+
|
|
37
|
+
test("graphInjectHint=true admits graph hint rows under independent budget", () => {
|
|
38
|
+
const { store, service } = setup({
|
|
39
|
+
entityRecallEnabled: true, maxInjectedItems: 5, graphInjectHint: true, graphInjectBudget: 2
|
|
40
|
+
});
|
|
41
|
+
// 两条实体挂联记忆(都命中查询实体),一条普通规则记忆。
|
|
42
|
+
linkEntityMem(store, service, { name: "PostgreSQL", title: "数据库迁移", content: "主库切换完成", key: "k1" });
|
|
43
|
+
linkEntityMem(store, service, { name: "PostgreSQL", title: "连接池调优", content: "PgBouncer 配置", key: "k2" });
|
|
44
|
+
service.saveWithDedupe({ type: "decision", title: "无关决策", content: "其他内容", importance: 5 });
|
|
45
|
+
const selected = service.injectCandidates({ maxItems: 5, query: "PostgreSQL" });
|
|
46
|
+
const hints = selected.filter((m) => m.graphHint === true);
|
|
47
|
+
assert.equal(hints.length, 2, "both graph hint rows fit within budget 2");
|
|
48
|
+
// 普通规则记忆仍在(graph 行不挤占 maxItems 槽)。
|
|
49
|
+
assert.ok(selected.some((m) => !m.graphHint), "non-graph rows still present");
|
|
50
|
+
store.close();
|
|
51
|
+
});
|
|
52
|
+
|
|
53
|
+
test("graphInjectBudget caps graph hint rows even when more match the query", () => {
|
|
54
|
+
const { store, service } = setup({
|
|
55
|
+
entityRecallEnabled: true, maxInjectedItems: 10, graphInjectHint: true, graphInjectBudget: 1
|
|
56
|
+
});
|
|
57
|
+
// 三条候选实体挂联记忆,预算只有 1。
|
|
58
|
+
for (const i of [1, 2, 3]) {
|
|
59
|
+
linkEntityMem(store, service, { name: "PostgreSQL", title: `PG 记录 ${i}`, content: `内容 ${i}`, key: `k${i}` });
|
|
60
|
+
}
|
|
61
|
+
const selected = service.injectCandidates({ maxItems: 10 });
|
|
62
|
+
const hints = selected.filter((m) => m.graphHint === true);
|
|
63
|
+
assert.ok(hints.length <= 1, `budget 1 caps at 1, got ${hints.length}`);
|
|
64
|
+
store.close();
|
|
65
|
+
});
|
|
66
|
+
|
|
67
|
+
test("graphHint=false (default) + entityRecallEnabled returns zero candidate set change (conservative = sorting only)", () => {
|
|
68
|
+
const { store, service } = setup({ entityRecallEnabled: true, maxInjectedItems: 5 });
|
|
69
|
+
// 仅实体挂联记忆,保守档下不该有任何候选进块。
|
|
70
|
+
linkEntityMem(store, service, { name: "PostgreSQL", title: "数据库迁移", content: "主库切换完成" });
|
|
71
|
+
const selected = service.injectCandidates({ maxItems: 5, query: "PostgreSQL 现状" });
|
|
72
|
+
assert.equal(selected.filter((m) => m.graphHint).length, 0);
|
|
73
|
+
store.close();
|
|
74
|
+
});
|
|
75
|
+
|
|
76
|
+
test("[检索线索] 引导语与线索行文本渲染正确(zh/en 双语)", () => {
|
|
77
|
+
assert.ok(STR.graphHintHeader.zh.includes("检索线索"));
|
|
78
|
+
assert.ok(STR.graphHintHeader.zh.includes("非事实断言"));
|
|
79
|
+
assert.ok(STR.graphHintHeader.en.includes("Retrieval hints"));
|
|
80
|
+
assert.ok(STR.graphHintHeader.en.toLowerCase().includes("not factual"));
|
|
81
|
+
const lineZh = STR.graphHintLine.zh("entity", "某某", "内容");
|
|
82
|
+
assert.ok(lineZh.startsWith("- [线索/entity]"));
|
|
83
|
+
const lineEn = STR.graphHintLine.en("technology", "Foo", "bar");
|
|
84
|
+
assert.ok(lineEn.startsWith("- [hint/technology]"));
|
|
85
|
+
});
|
|
86
|
+
|
|
87
|
+
// ---- 复审回归 ---------------------------------------------------------------
|
|
88
|
+
|
|
89
|
+
// 保守档(graphInjectHint 关)下线索行不许从 pin 池侧漏进注入块:pin 池取自打标
|
|
90
|
+
// 之后、又不占 maxItems 名额,pin 若不排除 graphHint 行,「图召回 + 约束型记忆」
|
|
91
|
+
// 就会在默认关档下照常进常驻注入文本,把线索开关整个绕过去。
|
|
92
|
+
test("conservative mode: a graph hint row cannot leak in via the pin pool", () => {
|
|
93
|
+
const { store, service } = setup({ entityRecallEnabled: true, maxInjectedItems: 5, pinnedInjectBudget: 1 });
|
|
94
|
+
// 约束型记忆 + 命中查询实体:同时是 pin 候选与图线索候选(最坏交叉)。
|
|
95
|
+
const mem = service.saveWithDedupe({
|
|
96
|
+
type: "constraint", title: "部署约束", content: "生产环境不许直接改库", importance: 4
|
|
97
|
+
}).memory;
|
|
98
|
+
const ent = store.createEntity({ name: "PostgreSQL", type: "technology" });
|
|
99
|
+
store.saveAttr({ entity_id: ent.id, attr_key: "k", attr_value: "v", memory_id: mem.id });
|
|
100
|
+
|
|
101
|
+
const selected = service.injectCandidates({ maxItems: 5, query: "PostgreSQL" });
|
|
102
|
+
assert.equal(selected.filter((m) => m.graphHint === true).length, 0, "hint rows stay out while graphInjectHint is off");
|
|
103
|
+
store.close();
|
|
104
104
|
});
|