@modusensus/dsh-mneme 0.2.4 → 0.2.5

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 (45) hide show
  1. package/README.md +10 -9
  2. package/lib/config.js +5 -2
  3. package/lib/dream/decisions.js +174 -62
  4. package/lib/dream.js +30 -5
  5. package/lib/index.js +3 -1
  6. package/lib/mirror.js +7 -1
  7. package/lib/service.js +117 -3
  8. package/lib/store.js +40 -0
  9. package/lib/tools.js +47 -4
  10. package/package.json +3 -1
  11. package/scripts/benchmark-embed.js +201 -0
  12. package/scripts/benchmark-rerank.js +166 -0
  13. package/scripts/e2e-dsh.js +216 -0
  14. package/scripts/stress-dsh.js +255 -0
  15. package/scripts/sync-lib.js +47 -0
  16. package/src/config.js +5 -2
  17. package/src/dream/decisions.js +174 -62
  18. package/src/dream.js +30 -5
  19. package/src/index.js +3 -1
  20. package/src/mirror.js +7 -1
  21. package/src/service.js +117 -3
  22. package/src/store.js +40 -0
  23. package/src/tools.js +47 -4
  24. package/test/api.test.js +385 -0
  25. package/test/audit.test.js +290 -0
  26. package/test/client.test.js +26 -0
  27. package/test/clustering.test.js +100 -0
  28. package/test/commands.test.js +69 -0
  29. package/test/config.test.js +31 -0
  30. package/test/dream.test.js +526 -0
  31. package/test/helpers/dream-mock.js +82 -0
  32. package/test/inject.test.js +82 -0
  33. package/test/local-embedder.test.js +227 -0
  34. package/test/mirror.test.js +249 -0
  35. package/test/reflection.test.js +226 -0
  36. package/test/reranker.test.js +197 -0
  37. package/test/semantic.test.js +123 -0
  38. package/test/service-search.test.js +169 -0
  39. package/test/service.test.js +198 -0
  40. package/test/settings.test.js +101 -0
  41. package/test/store.test.js +293 -0
  42. package/test/stress.test.js +209 -0
  43. package/test/summarize.test.js +156 -0
  44. package/test/tools.test.js +265 -0
  45. package/test/vector-index.test.js +205 -0
@@ -0,0 +1,216 @@
1
+ // scripts/e2e-dsh.js
2
+ // 端到端演示:用 cordis 搭建最小 DSH 环境,真实装载 dsh-mneme 插件,
3
+ // 走一遍完整用户流程(工具注册 → 保存/搜索记忆 → 上下文注入 → 会话摘要 → autoDream)。
4
+ //
5
+ // 运行:node scripts/e2e-dsh.js
6
+ import { Context } from "@deepseek-ai/cordis";
7
+ import { mkdtempSync } from "node:fs";
8
+ import { tmpdir } from "node:os";
9
+ import { join } from "node:path";
10
+ import * as mneme from "../lib/index.js";
11
+
12
+ const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
13
+
14
+ // ---------------------------------------------------------------- 环境
15
+ const memDir = mkdtempSync(join(tmpdir(), "dsh-mneme-e2e-"));
16
+ const ctx = new Context();
17
+
18
+ // 插件声明的 5 个依赖服务(tools / systemPrompt / webServer / llm / agentDefaultModel)
19
+ const registeredTools = [];
20
+ const injectContexts = [];
21
+ const apiRoutes = [];
22
+ const registeredCommands = new Map();
23
+ let llmCalls = [];
24
+
25
+ ctx.provide("tools", {
26
+ register(def) { registeredTools.push(def); return () => {}; }
27
+ });
28
+ ctx.provide("commands", {
29
+ register(def) { registeredCommands.set(def.name, def); return () => registeredCommands.delete(def.name); }
30
+ });
31
+ ctx.provide("systemPrompt", {
32
+ context(def) { injectContexts.push(def); return () => {}; }
33
+ });
34
+ ctx.provide("webServer", {
35
+ register(route) { apiRoutes.push(route); return () => {}; }
36
+ });
37
+ ctx.provide("agentDefaultModel", {
38
+ currentSelection() { return { provider: "mock", model: "mock-model" }; }
39
+ });
40
+
41
+ // LLM mock:按 purpose / 消息内容返回合理结果(模拟真实 LLM 的巩固行为)
42
+ ctx.provide("llm", {
43
+ async *stream(options) {
44
+ llmCalls.push(options.purpose);
45
+ const userText = options.messages.find((m) => m.role === "user")?.content?.[0]?.text ?? "";
46
+ let text = "";
47
+ if (options.purpose === "summarization") {
48
+ text = JSON.stringify([
49
+ { type: "preference", title: "语言偏好", content: "用户偏好使用中文交流", importance: 4 }
50
+ ]);
51
+ } else if (options.purpose === "compaction" && userText.includes("id=")) {
52
+ // 从传入的记忆清单解析 id 与类型,动态构造有效决策:优先演示 merge,其余 keep
53
+ const entries = [...userText.matchAll(/id=([^\s|]+).*?type=(\w+)/g)].map((m) => ({ id: m[1], type: m[2] }));
54
+ const byType = {};
55
+ for (const e of entries) (byType[e.type] ??= []).push(e.id);
56
+ const decisions = [];
57
+ const claimed = new Set();
58
+ const mergeType = Object.keys(byType).find((t) => byType[t].length >= 2);
59
+ if (mergeType) {
60
+ const ids = byType[mergeType];
61
+ decisions.push({
62
+ action: "merge", ids, keepSource: ids[0],
63
+ title: "合并条目", content: "两条主题相近的记忆合并后的精炼摘要", importance: 4
64
+ });
65
+ ids.forEach((id) => claimed.add(id));
66
+ }
67
+ for (const e of entries) {
68
+ if (!claimed.has(e.id)) decisions.push({ action: "keep", ids: [e.id] });
69
+ }
70
+ text = JSON.stringify(decisions);
71
+ } else if (options.purpose === "compaction") {
72
+ text = "记忆库总览:用户偏好中文交流;工作偏好(上午深度工作);项目采用 SQLite 存储。";
73
+ }
74
+ yield { type: "block-start", index: 0, blockType: "text" };
75
+ yield { type: "text-delta", index: 0, text };
76
+ yield { type: "block-end", index: 0, block: { type: "text" } };
77
+ yield { type: "finish", reason: { kind: "stop" } };
78
+ }
79
+ });
80
+
81
+ // ---------------------------------------------------------------- 装载
82
+ const fiber = ctx.plugin(
83
+ { name: mneme.name, inject: mneme.inject, Config: mneme.Config, apply: mneme.apply },
84
+ {
85
+ memoryDir: memDir,
86
+ autoInject: true,
87
+ autoSummarize: true,
88
+ maxInjectedItems: 5,
89
+ importanceThreshold: 3,
90
+ autoDream: true,
91
+ dreamThresholdCount: 3,
92
+ dreamThresholdChars: 100,
93
+ dreamDelayMs: 80
94
+ }
95
+ );
96
+ await fiber;
97
+
98
+ // ---------------------------------------------------------------- 工具
99
+ const tool = (name) => registeredTools.find((t) => t.name === name);
100
+
101
+ console.log("══════ dsh-mneme 端到端演示 ══════");
102
+ console.log(`记忆目录:${memDir}\n`);
103
+
104
+ // 1. 装载检查
105
+ console.log("【1】插件装载");
106
+ const checks = [];
107
+ checks.push(["注册 6 个模型工具", registeredTools.length === 6]);
108
+ checks.push(["注册 2 个注入上下文", injectContexts.length === 2 && injectContexts[0].name === "memory"]);
109
+ checks.push(["注册 9 条 API 路由", apiRoutes.length === 9]);
110
+ for (const [label, ok] of checks) console.log(` ${ok ? "✅" : "❌"} ${label}`);
111
+ if (!checks.every(([, ok]) => ok)) { console.log("\n装载检查失败,中止。"); process.exit(1); }
112
+ console.log(` 工具:${registeredTools.map((t) => t.name).join(", ")}\n`);
113
+
114
+ // 2. 保存记忆(工具执行)
115
+ console.log("【2】memory_save 保存记忆");
116
+ const saves = [];
117
+ saves.push(await tool("memory_save").execute({ type: "preference", title: "语言", content: "用户用中文交流", importance: 5 }));
118
+ saves.push(await tool("memory_save").execute({ type: "preference", title: "工作时段", content: "9-18 点在线,上午深度工作", importance: 4 }));
119
+ saves.push(await tool("memory_save").execute({ type: "preference", title: "工作习惯", content: "习惯上午处理复杂任务", importance: 3 }));
120
+ saves.push(await tool("memory_save").execute({ type: "project", title: "记忆插件", content: "dsh-mneme 用 SQLite + Markdown 镜像", importance: 4 }));
121
+ saves.push(await tool("memory_save").execute({ type: "decision", title: "存储选型", content: "确定使用 node:sqlite", importance: 5 }));
122
+ // 同标题再存一次 → 应触发去重合并
123
+ saves.push(await tool("memory_save").execute({ type: "preference", title: "语言", content: "用户用中文交流,偶尔英文", importance: 5 }));
124
+ for (const s of saves) console.log(` memory ${s.action}: ${s.id}`);
125
+ console.log(" 同标题二次保存 → 触发 merge ✓\n");
126
+
127
+ // 3. 搜索
128
+ console.log("【3】memory_search / memory_list");
129
+ const search = await tool("memory_search").execute({ query: "SQLite" });
130
+ console.log(` 搜索 "SQLite" → ${search.items.length} 条`);
131
+ const list = await tool("memory_list").execute({ type: "preference" });
132
+ console.log(` preference 共 ${list.total} 条\n`);
133
+
134
+ // 4. 上下文注入
135
+ console.log("【4】自动注入(新会话上下文)");
136
+ const injected = injectContexts[0].text({});
137
+ console.log(injected ? ` ${injected.split("\n").filter(Boolean).map((l) => " " + l).join("\n")}` : " (空)");
138
+ console.log("");
139
+
140
+ // 5. 会话摘要(turn/end 事件)
141
+ console.log("【5】会话摘要(模拟 turn/end)");
142
+ const session = {
143
+ id: "e2e-session-1",
144
+ events: [
145
+ { type: "user/message", data: { source: { kind: "user" }, content: [{ type: "text", text: "我偏好中文交流,以后都用中文。" }] } }
146
+ ],
147
+ requestHeader() { return { config: { provider: "mock", model: "mock-model" } }; }
148
+ };
149
+ await ctx.emit("session/event", session, { type: "turn/end" });
150
+ await sleep(50);
151
+ const afterSummary = await tool("memory_search").execute({ query: "中文交流" });
152
+ console.log(` 摘要入库:${afterSummary.items.length} 条命中(source: session)\n`);
153
+
154
+ // 6. autoDream
155
+ console.log("【6】autoDream(阈值触发 → LLM 决策 → merge + 摘要)");
156
+ console.log(" 等待调度器触发(delay 80ms)…");
157
+ await sleep(400);
158
+ const injectedAfter = injectContexts[0].text({});
159
+ const hasSummary = injectedAfter.includes("记忆库总览");
160
+ console.log(` 注入中出现"记忆库总览":${hasSummary ? "是 ✅" : "否 ❌"}`);
161
+ const prefs = await tool("memory_list").execute({ type: "preference" });
162
+ console.log(` dream merge 后 preference 条目:${prefs.items.length}(dream 时 4 条相近项,merge 成 1 条)`);
163
+
164
+ // 7. Web API 模拟调用
165
+ console.log("\n【7】Web 面板 API(/api/dsh-mneme/list)");
166
+ const listRoute = apiRoutes.find((r) => r.path === "/api/dsh-mneme/list");
167
+ const res = { writeHead(code, h) { this.statusCode = code; this.headers = h; return this; }, end(body) { this.body = body; } };
168
+ await listRoute.handler({ url: "/api/dsh-mneme/list?type=project" }, res);
169
+ const payload = JSON.parse(res.body);
170
+ console.log(` HTTP ${res.statusCode},project 条目 ${payload.items.length} 条\n`);
171
+
172
+ // 8. 用户画像 / 规则 / 自定义命令
173
+ console.log("【8】用户画像 / 规则 / 自定义命令");
174
+ function apiReq(path, method = "GET", body = null) {
175
+ const listeners = {};
176
+ const r = {
177
+ url: path, method, headers: {},
178
+ on(ev, fn) { listeners[ev] = fn; return r; },
179
+ emit(ev, data) { listeners[ev]?.(data); return r; }
180
+ };
181
+ if (body !== null) {
182
+ process.nextTick(() => { r.emit("data", Buffer.from(JSON.stringify(body))); r.emit("end"); });
183
+ }
184
+ return r;
185
+ }
186
+ function apiRes() { return { statusCode: 0, body: "", writeHead(code) { this.statusCode = code; return this; }, end(body) { this.body = body; } }; }
187
+
188
+ const profileRoute = apiRoutes.find((r) => r.path === "/api/dsh-mneme/profile");
189
+ const rp = apiRes();
190
+ await profileRoute.handler(apiReq("/api/dsh-mneme/profile", "PUT", { profile: "我是后端开发者,擅长 Node.js" }), rp);
191
+ const rulesRoute = apiRoutes.find((r) => r.path === "/api/dsh-mneme/rules");
192
+ const rr = apiRes();
193
+ await rulesRoute.handler(apiReq("/api/dsh-mneme/rules", "PUT", { rules: ["回答时先给结论", "使用简体中文"] }), rr);
194
+ const userSettings = injectContexts.find((c) => c.name === "user-settings").text({});
195
+ console.log(` 注入含用户画像:${userSettings.includes("后端开发者") ? "✅" : "❌"}`);
196
+ console.log(` 注入含规则:${userSettings.includes("先给结论") ? "✅" : "❌"}`);
197
+ const cmdRoute = apiRoutes.find((r) => r.path === "/api/dsh-mneme/commands");
198
+ const rc = apiRes();
199
+ await cmdRoute.handler(apiReq("/api/dsh-mneme/commands", "POST", { name: "review", description: "审查代码", instruction: "请按项目规范审查当前代码" }), rc);
200
+ const cmdRegistered = registeredCommands.has("review");
201
+ console.log(` 自定义命令 /review 已注册:${cmdRegistered ? "✅" : "❌"}`);
202
+ if (cmdRegistered) {
203
+ const handler = registeredCommands.get("review").handler;
204
+ const result = await handler({ agent: {}, rawInput: "", signal: null });
205
+ console.log(` 命令触发返回:${result.text.slice(0, 24)}…`);
206
+ }
207
+ console.log("");
208
+
209
+ // ---------------------------------------------------------------- 汇总
210
+ console.log("══════ 汇总 ══════");
211
+ console.log(` LLM 调用:${llmCalls.join(" → ")}`);
212
+ console.log(` 记忆文件:${memDir}`);
213
+ const all = await tool("memory_list").execute({});
214
+ console.log(` 当前可注入条目:${all.total}`);
215
+ console.log("══════ 演示结束 ══════");
216
+ fiber.dispose?.();
@@ -0,0 +1,255 @@
1
+ // scripts/stress-dsh.js
2
+ // 压测:沿着三条轴线检验 dsh-mneme 的 autoDream 与存储在长会话、冲突、
3
+ // 并发压力下的表现。LLM 全部用确定性的 mock 决策,离线可跑、无需 API Key。
4
+ //
5
+ // 轴线 1 · 长会话检索:Recall@k(规范记忆能否被召回)与陈旧残留率
6
+ // 轴线 2 · 冲突裁决:可重放仲裁集 —— 仲裁正确率、确定性、审计/回放
7
+ // 轴线 3 · 多 Agent 并发:丢更新、重复合并、事务/崩溃恢复
8
+ //
9
+ // 运行:node scripts/stress-dsh.js
10
+ import { mkdtempSync } from "node:fs";
11
+ import { tmpdir } from "node:os";
12
+ import { join } from "node:path";
13
+ import { createStore } from "../src/store.js";
14
+ import { createService } from "../src/service.js";
15
+ import {
16
+ createDreamScheduler,
17
+ hashSnapshot,
18
+ parseReceipt
19
+ } from "../src/dream.js";
20
+ import { applyDecisions } from "../src/dream/decisions.js";
21
+ import {
22
+ sessionDecisions,
23
+ arbitrationDecisions,
24
+ mockCtx
25
+ } from "../test/helpers/dream-mock.js";
26
+
27
+ const pass = (ok) => (ok ? "✅" : "❌");
28
+
29
+ // ---------------------------------------------------------------- 轴线 1
30
+
31
+ async function axis1LongSessionRetrieval() {
32
+ console.log("【轴线 1】长会话检索:Recall@k 与陈旧残留率");
33
+ const topics = 20;
34
+ const rounds = 8;
35
+ const dir = mkdtempSync(join(tmpdir(), "dsh-mneme-stress-a1-"));
36
+ const store = createStore(join(dir, "memory.db"));
37
+ const service = createService({ store, mirror: null, config: {} });
38
+ const dream = createDreamScheduler({ onRun: () => Promise.resolve({ ok: true, skipped: true }) });
39
+ const ctx = mockCtx({ onConsolidation: sessionDecisions });
40
+
41
+ // 每个主题一条规范记忆(ground truth,importance 5)
42
+ const gt = [];
43
+ for (let t = 1; t <= topics; t++) {
44
+ const title = `主题${String(t).padStart(2, "0")}`;
45
+ const { memory } = service.saveWithDedupe({ type: "project", title, content: `${title} 的规范内容`, importance: 5 });
46
+ gt.push({ title, id: memory.id });
47
+ }
48
+ // 长会话:每轮追加同主题旧变体(importance 3)并触发一次 autoDream
49
+ for (let round = 1; round <= rounds; round++) {
50
+ for (let t = 1; t <= topics; t++) {
51
+ service.saveWithDedupe({
52
+ type: "project",
53
+ title: `主题${String(t).padStart(2, "0")}·变体${round}`,
54
+ content: `主题${t} 第${round}轮旧变体`,
55
+ importance: 3
56
+ });
57
+ }
58
+ await dream.runDream(ctx, service, {});
59
+ }
60
+
61
+ let hits5 = 0;
62
+ let hits10 = 0;
63
+ for (const { title, id } of gt) {
64
+ const k5 = service.search(title, { limit: 5 });
65
+ const k10 = service.search(title, { limit: 10 });
66
+ if (k5.some((m) => m.id === id)) hits5++;
67
+ if (k10.some((m) => m.id === id)) hits10++;
68
+ }
69
+ const recall5 = hits5 / topics;
70
+ const recall10 = hits10 / topics;
71
+
72
+ // 陈旧残留率 = 仍活跃的旧变体 / 全部旧变体(dream 应全部清掉 → 0)
73
+ const all = store.all();
74
+ const variants = all.filter((m) => m.type === "project" && m.title.includes("变体"));
75
+ const staleResidual = variants.filter((m) => !m.archived).length / variants.length;
76
+ const runs = store.listDreamRuns().length;
77
+
78
+ console.log(` 主题 ${topics} 个 × 变体 ${rounds} 轮 = ${topics * rounds} 条旧记忆,autoDream 触发 ${runs} 次`);
79
+ console.log(` ${pass(recall5 >= 0.95)} Recall@5 = ${(recall5 * 100).toFixed(1)}%(规范记忆 top5 召回)`);
80
+ console.log(` ${pass(recall10 >= 0.95)} Recall@10 = ${(recall10 * 100).toFixed(1)}%`);
81
+ console.log(` ${pass(staleResidual === 0)} 陈旧残留率 = ${(staleResidual * 100).toFixed(1)}%(理想 0%)`);
82
+ store.close();
83
+ return { recall5, recall10, staleResidual, runs };
84
+ }
85
+
86
+ // ---------------------------------------------------------------- 轴线 2
87
+
88
+ async function axis2ConflictArbitration() {
89
+ console.log("【轴线 2】冲突裁决:可重放仲裁集");
90
+ const sets = [
91
+ { key: "项目截止日期", winner: { type: "decision", title: "项目截止日期", content: "8月20日交付", importance: 5 }, loser: { type: "decision", title: "项目截止日期(旧)", content: "8月15日交付", importance: 3 } },
92
+ { key: "语言偏好", winner: { type: "preference", title: "语言偏好", content: "使用简体中文", importance: 5 }, loser: { type: "preference", title: "语言偏好(旧)", content: "使用繁体中文", importance: 2 } },
93
+ { key: "存储选型", winner: { type: "decision", title: "存储选型", content: "node:sqlite(零依赖)", importance: 5 }, loser: { type: "decision", title: "存储选型(旧)", content: "better-sqlite3(需编译)", importance: 3 } },
94
+ { key: "部署环境", winner: { type: "project", title: "部署环境", content: "生产环境 Node 24 + Windows Server", importance: 4 }, loser: { type: "project", title: "部署环境(旧)", content: "生产环境 Node 20 + Linux", importance: 2 } }
95
+ ];
96
+
97
+ const dir = mkdtempSync(join(tmpdir(), "dsh-mneme-stress-a2-"));
98
+ const store = createStore(join(dir, "memory.db"));
99
+ const service = createService({ store, mirror: null, config: {} });
100
+ const dream = createDreamScheduler({ onRun: () => Promise.resolve({ ok: true, skipped: true }) });
101
+ const ctx = mockCtx({ onConsolidation: arbitrationDecisions });
102
+
103
+ // 仲裁集入库(先败者后胜者,模拟旧信息先存、新信息后到)
104
+ for (const set of sets) {
105
+ service.saveWithDedupe({ ...set.loser });
106
+ service.saveWithDedupe({ ...set.winner });
107
+ }
108
+ const before = hashSnapshot(service.all().filter((m) => !m.archived && m.type !== "summary"));
109
+
110
+ // 跑一次裁决 → 审计 + receipt
111
+ const run1 = await dream.runDream(ctx, service, {});
112
+ console.log(` 裁决运行:${pass(run1.ok)}(applied=${run1.applied},receipt=${run1.receipt.slice(0, 40)}…)`);
113
+
114
+ // 正确性:胜者保留 + 败者归档 + 来源链注释
115
+ let arbOk = 0;
116
+ for (const set of sets) {
117
+ const winner = service.list({ type: set.winner.type }).find((m) => m.title === set.winner.title);
118
+ const loser = store.all().find((m) => m.title === set.loser.title);
119
+ if (winner && loser && loser.archived && winner.content.includes("已否决旧信息")) arbOk++;
120
+ }
121
+ const arbRate = arbOk / sets.length;
122
+ console.log(` ${pass(arbRate === 1)} 仲裁正确率 = ${(arbRate * 100).toFixed(0)}%(胜者保留 + 败者归档 + 来源链注释)`);
123
+
124
+ // 确定性:同一输入快照 → 同一决策。重跑前先把败者"复活"为原始状态。
125
+ const audit = store.getDreamRun(run1.runId);
126
+ const receipt = parseReceipt(audit.receipt);
127
+ console.log(` ${pass(!!receipt && receipt.status === "ok")} receipt 可解析,snapshot=${receipt.snapshotHash},输入 ${receipt.inputCount} 条`);
128
+ console.log(` ${pass(audit.outcome && Object.keys(audit.outcome.byId).length === sets.length * 2)} 审计 outcome 覆盖全部 ${sets.length * 2} 个仲裁对象`);
129
+
130
+ // 可重放:审计中的决策清单在同一 store 上重放 → 幂等(无副作用)
131
+ const winnerBefore = store.list({ type: sets[0].winner.type }).find((m) => m.title === sets[0].winner.title).content;
132
+ const replayed = applyDecisions(audit.decisions, service);
133
+ const winnerAfter = store.list({ type: sets[0].winner.type }).find((m) => m.title === sets[0].winner.title).content;
134
+ const idempotent = replayed.applied === 0 && winnerBefore === winnerAfter;
135
+ console.log(` ${pass(idempotent)} 决策清单重放幂等(repeat 无副作用,来源链不重复追加)`);
136
+ console.log(` ${pass(before.length === 64)} 快照哈希稳定(${before.length} hex 位)`);
137
+
138
+ const runs = store.listDreamRuns().length;
139
+ store.close();
140
+ return { arbRate, runs, idempotent };
141
+ }
142
+
143
+ // ---------------------------------------------------------------- 轴线 3
144
+
145
+ function bump(content) {
146
+ const m = content.match(/count=(\d+)/);
147
+ return `count=${m ? Number(m[1]) + 1 : 1}`;
148
+ }
149
+
150
+ async function axis3Concurrency() {
151
+ console.log("【轴线 3】多 Agent 并发:丢更新 / 重复合并 / 事务恢复");
152
+ const dir = mkdtempSync(join(tmpdir(), "dsh-mneme-stress-a3-"));
153
+ const path = join(dir, "memory.db");
154
+
155
+ // --- 重复合并:20 个 agent 并发保存同标题 → 最终只 1 条活跃
156
+ const sA = createStore(path);
157
+ const svA = createService({ store: sA, mirror: null, config: {} });
158
+ const agents = 20;
159
+ const writes = [];
160
+ for (let i = 0; i < agents; i++) {
161
+ writes.push(Promise.resolve().then(() => svA.saveWithDedupe({ type: "preference", title: "并发任务", content: `agent-${i}` })));
162
+ }
163
+ await Promise.all(writes);
164
+ const dupes = svA.list({ type: "preference", limit: 100 }).filter((m) => m.title === "并发任务");
165
+ const noDupes = dupes.length === 1;
166
+ console.log(` ${pass(noDupes)} 重复合并:${agents} 个 agent 同标题并发保存 → 活跃 ${dupes.length} 条(期望 1)`);
167
+
168
+ // --- 丢更新:CAS 原子递增(硬断言:陈旧版本必须被拒,增量不得丢失)
169
+ const sB = createStore(path);
170
+ const svB = createService({ store: sB, mirror: null, config: {} });
171
+ svA.saveWithDedupe({ type: "history", title: "计数器", content: "count=0", importance: 3 });
172
+ const counterId = svA.list({ type: "history" })[0].id;
173
+ svA.update(counterId, { content: "count=0" });
174
+ const baseline = svA.getById(counterId); // updated_at=T, count=0
175
+ // Agent A 持最新版本 → CAS 成功(count=1)
176
+ const aOk = svA.compareAndUpdate(counterId, baseline.updated_at, { content: bump(baseline.content) });
177
+ // Agent B 仍持过期版本 T → CAS 必须被拒;若被接受即丢更新
178
+ const bStale = svB.compareAndUpdate(counterId, baseline.updated_at, { content: bump("count=0") });
179
+ const casRejected = aOk !== undefined && bStale === undefined;
180
+ // B 重读最新值重试 → 两个 +1 都落地
181
+ if (bStale === undefined) {
182
+ const cur = svB.getById(counterId);
183
+ svB.compareAndUpdate(counterId, cur.updated_at, { content: bump(cur.content) });
184
+ }
185
+ const casResult = svB.getById(counterId).content;
186
+ const noLostUpdate = casRejected && casResult === "count=2";
187
+ console.log(` ${pass(noLostUpdate)} CAS 并发递增:陈旧版本被拒=${casRejected},重读重试后 → ${casResult}(期望 count=2,硬断言)`);
188
+ // 串行化修复(无 CAS 时的人工约定):写前重读最新值
189
+ svA.update(counterId, { content: "count=0" });
190
+ svA.update(counterId, { content: bump(svA.getById(counterId).content) });
191
+ svB.update(counterId, { content: bump(svB.getById(counterId).content) });
192
+ const fixedResult = svB.getById(counterId).content;
193
+ const fixed = fixedResult === "count=2";
194
+ console.log(` ${pass(fixed)} 串行重读修复:写前重读最新 → ${fixedResult}(期望 count=2)`);
195
+
196
+ // --- 多步原子性:事务中途抛错 → 全部回滚(硬断言,无半成品)
197
+ let txThrew = false;
198
+ try {
199
+ svB.transaction(() => {
200
+ svB.saveWithDedupe({ type: "project", title: "原子A", content: "x" });
201
+ svB.saveWithDedupe({ type: "project", title: "原子B", content: "y" });
202
+ throw new Error("boom");
203
+ });
204
+ } catch { txThrew = true; }
205
+ const allNow = svB.all();
206
+ const atomic = txThrew
207
+ && !allNow.some((m) => m.title === "原子A")
208
+ && !allNow.some((m) => m.title === "原子B");
209
+ console.log(` ${pass(atomic)} 多步原子性:事务中途抛错 → 原子A/原子B 全部回滚(期望均不存在,硬断言)`);
210
+
211
+ // --- 事务/崩溃恢复:已提交写入在异常后 + reopen 后完整保留
212
+ const svC = createService({ store: sB, mirror: null, config: {} }); // 复用同一文件连接
213
+ svC.saveWithDedupe({ type: "project", title: "已提交A", content: "x" });
214
+ svC.saveWithDedupe({ type: "project", title: "已提交B", content: "y" });
215
+ let threw = false;
216
+ try {
217
+ svC.update("nonexistent-id", { content: "boom" }); // store.update 对未知 id 抛错
218
+ } catch {
219
+ threw = true;
220
+ }
221
+ sA.close();
222
+ sB.close();
223
+ const sR = createStore(path); // 重新打开(模拟进程重启)
224
+ const recovered = sR.all();
225
+ const recoveryOk = threw
226
+ && recovered.some((m) => m.title === "已提交A")
227
+ && recovered.some((m) => m.title === "已提交B")
228
+ && recovered.some((m) => m.title === "计数器");
229
+ console.log(` ${pass(recoveryOk)} 事务恢复:中途异常后已提交写入不丢,reopen 后完整可读`);
230
+ console.log(` ${pass(recovered.filter((m) => m.title === "已提交A").length === 1)} 无半成品残留(已提交A 仅 1 条)`);
231
+
232
+ sR.close();
233
+ return { noDupes, noLostUpdate, atomic, fixed, recoveryOk };
234
+ }
235
+
236
+ // ---------------------------------------------------------------- 汇总
237
+
238
+ console.log("══════ dsh-mneme 压测:三条轴线 ══════");
239
+ console.log("");
240
+ const a1 = await axis1LongSessionRetrieval();
241
+ console.log("");
242
+ const a2 = await axis2ConflictArbitration();
243
+ console.log("");
244
+ const a3 = await axis3Concurrency();
245
+ console.log("");
246
+ console.log("══════ 压测汇总 ══════");
247
+ console.log(` 轴线1 长会话检索:Recall@5=${(a1.recall5 * 100).toFixed(1)}% Recall@10=${(a1.recall10 * 100).toFixed(1)}% 陈旧残留=${(a1.staleResidual * 100).toFixed(1)}% (${a1.runs} 次 autoDream 全部入审计)`);
248
+ console.log(` 轴线2 冲突裁决:仲裁正确率=${(a2.arbRate * 100).toFixed(0)}% 重放幂等=${a2.idempotent ? "是" : "否"}`);
249
+ console.log(` 轴线3 多 Agent 并发:去重=${a3.noDupes ? "✅" : "❌"} CAS无丢更新=${a3.noLostUpdate ? "✅" : "❌"} 多步原子=${a3.atomic ? "✅" : "❌"} 串行修复=${a3.fixed ? "✅" : "❌"} 崩溃恢复=${a3.recoveryOk ? "✅" : "❌"}`);
250
+ // 丢更新与多步原子性都是硬断言:任一失败 → 非零退出,绝不把风险标成通过
251
+ const allOk = a1.recall5 >= 0.95 && a1.staleResidual === 0 && a2.arbRate === 1 && a2.idempotent
252
+ && a3.noDupes && a3.noLostUpdate && a3.atomic && a3.fixed && a3.recoveryOk;
253
+ console.log(` 结论:${allOk ? "全部通过 ✅" : "存在失败项 ❌"}`);
254
+ console.log("══════ 压测结束 ══════");
255
+ process.exit(allOk ? 0 : 1);
@@ -0,0 +1,47 @@
1
+ // Sync the src/ tree into lib/ — the distributable consumed by DSH (package
2
+ // main is lib/index.js). lib/client.js (the Web panel bundle) is authored
3
+ // independently under lib/ and is left untouched: the sync only copies
4
+ // src -> lib, it never prunes lib-only files.
5
+ //
6
+ // Usage: npm run sync (also run automatically by `npm pack`/`npm publish`
7
+ // via the prepack hook, so a published tarball always ships a fresh lib/).
8
+ import { cpSync, existsSync, mkdirSync, readdirSync } from "node:fs";
9
+ import { join, relative } from "node:path";
10
+ import { fileURLToPath } from "node:url";
11
+
12
+ const root = join(fileURLToPath(new URL("..", import.meta.url)));
13
+ const srcDir = join(root, "src");
14
+ const libDir = join(root, "lib");
15
+
16
+ /** Recursively collect all files under a directory (sorted for stable output). */
17
+ function walk(dir) {
18
+ const out = [];
19
+ for (const entry of readdirSync(dir, { withFileTypes: true }).sort((a, b) => a.name.localeCompare(b.name))) {
20
+ const full = join(dir, entry.name);
21
+ if (entry.isDirectory()) out.push(...walk(full));
22
+ else if (entry.isFile()) out.push(full);
23
+ }
24
+ return out;
25
+ }
26
+
27
+ let copied = 0;
28
+ mkdirSync(libDir, { recursive: true });
29
+
30
+ for (const file of walk(srcDir)) {
31
+ const rel = relative(srcDir, file);
32
+ const dest = join(libDir, rel);
33
+ mkdirSync(join(dest, ".."), { recursive: true });
34
+ cpSync(file, dest);
35
+ copied++;
36
+ console.log(`synced ${rel}`);
37
+ }
38
+
39
+ // Report lib-only files (authored separately, e.g. client.js) so drift is
40
+ // visible but never clobbered.
41
+ for (const file of walk(libDir)) {
42
+ const rel = relative(libDir, file);
43
+ if (!existsSync(join(srcDir, rel))) console.log(`kept ${rel} (lib-only, not pruned)`);
44
+ }
45
+
46
+ console.log(`\nSynced ${copied} file(s) from src/ to lib/.`);
47
+ if (!copied) process.exitCode = 1;
package/src/config.js CHANGED
@@ -49,8 +49,11 @@ export const Config = z.object({
49
49
  hybridSearchKeywordWeight: z.number().min(0).max(1).default(0.4),
50
50
 
51
51
  // --- semantic: rerank layer (v0.2) --------------------------------------
52
- rerankEnabled: z.boolean().default(true),
53
- rerankProvider: z.union([z.const("local"), z.const("none")]).default("local"),
52
+ // Opt-in by default (item ⑥): the local cross-encoder pulls in onnxruntime
53
+ // (transformers.js) at init, so a bare install must not load it. Only an
54
+ // explicit rerankEnabled=true + rerankProvider="local" constructs LocalReranker.
55
+ rerankEnabled: z.boolean().default(false),
56
+ rerankProvider: z.union([z.const("local"), z.const("none")]).default("none"),
54
57
  rerankModel: z.string().default("Xenova/bge-reranker-base"),
55
58
  rerankBatchSize: z.natural().min(1).max(64).default(8),
56
59
  rerankMaxCandidates: z.natural().min(5).max(100).default(30),