@modusensus/dsh-mneme 0.6.6 → 0.6.8
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 +463 -457
- package/cordis.patch.yml +15 -15
- package/lib/api.js +783 -693
- package/lib/client.js +1757 -1656
- package/lib/commands.js +64 -64
- package/lib/config.js +288 -288
- package/lib/dream/clustering.js +118 -118
- package/lib/dream/decisions.js +439 -439
- package/lib/dream/sleep.js +561 -554
- package/lib/dream/tag-extractor.js +156 -156
- package/lib/dream.js +935 -929
- package/lib/embedding.js +154 -154
- package/lib/entities/extractor.js +242 -242
- package/lib/hot-memory.js +53 -53
- package/lib/index.js +361 -361
- package/lib/inject.js +208 -208
- package/lib/local-embedder.js +282 -282
- package/lib/mirror.js +170 -170
- package/lib/parser/tag.js +59 -59
- package/lib/parser/wiki-link.js +38 -38
- package/lib/quality-filter.js +123 -123
- package/lib/reranker.js +218 -218
- package/lib/search/adaptive.js +22 -22
- package/lib/search/bm25.js +96 -96
- package/lib/search/tag-boost.js +61 -61
- package/lib/service.js +1726 -1726
- package/lib/settings.js +172 -142
- package/lib/store.js +2238 -2238
- package/lib/summarize.js +236 -236
- package/lib/tools.js +290 -290
- package/lib/vector-index.js +116 -116
- package/package.json +80 -75
- package/scripts/benchmark-embed.js +201 -201
- package/scripts/benchmark-recall.js +133 -133
- package/scripts/benchmark-rerank.js +166 -166
- package/scripts/e2e-dsh.js +218 -218
- package/scripts/stress-dsh.js +255 -255
- package/scripts/sync-lib.js +52 -52
- package/src/api.js +783 -693
- package/src/commands.js +64 -64
- package/src/config.js +288 -288
- package/src/dream/clustering.js +118 -118
- package/src/dream/decisions.js +439 -439
- package/src/dream/sleep.js +561 -554
- package/src/dream/tag-extractor.js +156 -156
- package/src/dream.js +935 -929
- package/src/embedding.js +154 -154
- package/src/entities/extractor.js +242 -242
- package/src/hot-memory.js +53 -53
- package/src/index.js +361 -361
- package/src/inject.js +208 -208
- package/src/local-embedder.js +282 -282
- package/src/mirror.js +170 -170
- package/src/parser/tag.js +59 -59
- package/src/parser/wiki-link.js +38 -38
- package/src/quality-filter.js +123 -123
- package/src/reranker.js +218 -218
- package/src/search/adaptive.js +22 -22
- package/src/search/bm25.js +96 -96
- package/src/search/tag-boost.js +61 -61
- package/src/service.js +1726 -1726
- package/src/settings.js +172 -142
- package/src/store.js +2238 -2238
- package/src/summarize.js +236 -236
- package/src/tools.js +290 -290
- package/src/vector-index.js +116 -116
- package/test/api.test.js +594 -549
- package/test/audit.test.js +448 -448
- package/test/benchmark.test.js +35 -35
- package/test/boundary-v0625.test.js +82 -82
- package/test/client.test.js +368 -368
- package/test/clustering.test.js +100 -100
- package/test/commands.test.js +69 -69
- package/test/config.test.js +50 -50
- package/test/conflict-freeze.test.js +290 -290
- package/test/directory.test.js +134 -134
- package/test/dream.test.js +903 -901
- package/test/entities.test.js +522 -522
- package/test/epistemic.test.js +298 -298
- package/test/fnew-0112.test.js +311 -311
- package/test/fnew-03.test.js +422 -422
- package/test/graph-api.test.js +175 -175
- package/test/helpers/dream-mock.js +82 -82
- package/test/hot-memory.test.js +174 -174
- package/test/inject.test.js +103 -103
- package/test/llm-audit.test.js +279 -279
- package/test/local-embedder.test.js +227 -227
- 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.test.js +249 -249
- package/test/normalize-decisions.test.js +120 -120
- package/test/peer-blockers.test.js +190 -190
- package/test/policy-epoch.test.js +259 -259
- package/test/provenance.test.js +103 -103
- package/test/quality-filter.test.js +118 -118
- package/test/reasoning-effort.test.js +199 -199
- package/test/recall-evals.test.js +235 -235
- package/test/recall-layer.test.js +315 -315
- package/test/receipt-chain.test.js +451 -451
- package/test/reflection.test.js +226 -226
- package/test/reranker.test.js +240 -240
- package/test/search-fusion.test.js +90 -90
- package/test/semantic.test.js +124 -124
- package/test/service-search.test.js +199 -199
- package/test/service.test.js +435 -435
- package/test/settings.test.js +118 -101
- package/test/sleep.test.js +365 -365
- package/test/store.test.js +436 -436
- package/test/stress.test.js +209 -209
- package/test/summarize.test.js +191 -191
- package/test/tag-boost.test.js +125 -125
- package/test/tag.test.js +312 -312
- package/test/tools.test.js +285 -285
- package/test/vector-index.test.js +221 -221
- package/test/wiki-link.test.js +332 -332
package/test/tools.test.js
CHANGED
|
@@ -1,285 +1,285 @@
|
|
|
1
|
-
import test from "node:test";
|
|
2
|
-
import assert from "node:assert/strict";
|
|
3
|
-
import { assertSupportedJsonSchema, validateJsonSchemaValue } from "@deepseek-ai/dsh-tools";
|
|
4
|
-
import { createStore } from "../src/store.js";
|
|
5
|
-
import { createService } from "../src/service.js";
|
|
6
|
-
import { createTools } from "../src/tools.js";
|
|
7
|
-
|
|
8
|
-
function setup(embedder) {
|
|
9
|
-
const store = createStore(":memory:");
|
|
10
|
-
const service = createService({ store, mirror: null, config: {} });
|
|
11
|
-
const registered = [];
|
|
12
|
-
const ctx = {
|
|
13
|
-
tools: {
|
|
14
|
-
register(def) {
|
|
15
|
-
registered.push(def);
|
|
16
|
-
return () => {};
|
|
17
|
-
}
|
|
18
|
-
}
|
|
19
|
-
};
|
|
20
|
-
const tools = createTools(ctx, service, {}, embedder);
|
|
21
|
-
return { store, service, tools, registered };
|
|
22
|
-
}
|
|
23
|
-
|
|
24
|
-
// Collect authoring-DSL regressions in a compiled schema: property-level
|
|
25
|
-
// `required: true` (must be projected to a top-level required array by
|
|
26
|
-
// defineTool) and `minimum`/`maximum` (outside the enforced subset) are
|
|
27
|
-
// rejected by the real harness, so they must never appear post-compilation.
|
|
28
|
-
function walkSchema(node, path, problems) {
|
|
29
|
-
if (typeof node !== "object" || node === null || Array.isArray(node)) return;
|
|
30
|
-
if (typeof node.required === "boolean") {
|
|
31
|
-
problems.push(`${path}.required must be an array (property-level required leaks through)`);
|
|
32
|
-
}
|
|
33
|
-
if (Object.hasOwn(node, "minimum") || Object.hasOwn(node, "maximum")) {
|
|
34
|
-
problems.push(`${path} uses minimum/maximum, which the enforced JSON Schema subset rejects`);
|
|
35
|
-
}
|
|
36
|
-
if (node.properties && typeof node.properties === "object") {
|
|
37
|
-
for (const [key, value] of Object.entries(node.properties)) {
|
|
38
|
-
walkSchema(value, `${path}.properties.${key}`, problems);
|
|
39
|
-
}
|
|
40
|
-
}
|
|
41
|
-
if (node.items) walkSchema(node.items, `${path}.items`, problems);
|
|
42
|
-
if (Array.isArray(node.oneOf)) {
|
|
43
|
-
node.oneOf.forEach((branch, index) => walkSchema(branch, `${path}.oneOf[${index}]`, problems));
|
|
44
|
-
}
|
|
45
|
-
}
|
|
46
|
-
|
|
47
|
-
test("registers seven tools with correct names", () => {
|
|
48
|
-
const { registered } = setup();
|
|
49
|
-
const names = registered.map((t) => t.name).sort();
|
|
50
|
-
assert.deepEqual(names, ["memory_archive", "memory_delete", "memory_forget", "memory_list", "memory_save", "memory_search", "memory_update"]);
|
|
51
|
-
});
|
|
52
|
-
|
|
53
|
-
test("compiled schemas pass the enforced DSH subset (defineTool projection)", () => {
|
|
54
|
-
const { registered } = setup();
|
|
55
|
-
assert.equal(registered.length, 7);
|
|
56
|
-
for (const tool of registered) {
|
|
57
|
-
assertSupportedJsonSchema(tool.parameters);
|
|
58
|
-
assertSupportedJsonSchema(tool.output.schema);
|
|
59
|
-
const problems = [];
|
|
60
|
-
walkSchema(tool.parameters, `parameters(${tool.name})`, problems);
|
|
61
|
-
walkSchema(tool.output.schema, `output.schema(${tool.name})`, problems);
|
|
62
|
-
assert.deepEqual(problems, []);
|
|
63
|
-
}
|
|
64
|
-
});
|
|
65
|
-
|
|
66
|
-
test("memory_save executes and stores", async () => {
|
|
67
|
-
const { registered, store } = setup();
|
|
68
|
-
const save = registered.find((t) => t.name === "memory_save");
|
|
69
|
-
const result = await save.execute({ type: "preference", title: "语言", content: "中文", importance: 4 });
|
|
70
|
-
assert.equal(result.action, "created");
|
|
71
|
-
assert.equal(store.count(), 1);
|
|
72
|
-
});
|
|
73
|
-
|
|
74
|
-
test("memory_save merges on matching title within type", async () => {
|
|
75
|
-
const { registered, store } = setup();
|
|
76
|
-
const save = registered.find((t) => t.name === "memory_save");
|
|
77
|
-
const first = await save.execute({ type: "preference", title: "语言", content: "中文" });
|
|
78
|
-
const second = await save.execute({ type: "preference", title: "语言", content: "简体中文" });
|
|
79
|
-
assert.equal(first.action, "created");
|
|
80
|
-
assert.equal(second.action, "merged");
|
|
81
|
-
assert.equal(first.id, second.id);
|
|
82
|
-
assert.equal(store.count(), 1);
|
|
83
|
-
});
|
|
84
|
-
|
|
85
|
-
test("memory_search finds by CJK substring", async () => {
|
|
86
|
-
const { registered, store, service } = setup();
|
|
87
|
-
service.saveWithDedupe({ type: "project", title: "记忆插件", content: "SQLite 存储中文记忆", importance: 3 });
|
|
88
|
-
const search = registered.find((t) => t.name === "memory_search");
|
|
89
|
-
const result = await search.execute({ query: "中文" });
|
|
90
|
-
assert.ok(result.items.length >= 1);
|
|
91
|
-
assert.equal(result.items[0].title, "记忆插件");
|
|
92
|
-
});
|
|
93
|
-
|
|
94
|
-
test("memory_list filters by type", async () => {
|
|
95
|
-
const { registered, service } = setup();
|
|
96
|
-
service.saveWithDedupe({ type: "preference", title: "a", content: "x" });
|
|
97
|
-
service.saveWithDedupe({ type: "project", title: "b", content: "y" });
|
|
98
|
-
const list = registered.find((t) => t.name === "memory_list");
|
|
99
|
-
const result = await list.execute({ type: "project" });
|
|
100
|
-
assert.equal(result.items.length, 1);
|
|
101
|
-
assert.equal(result.items[0].title, "b");
|
|
102
|
-
});
|
|
103
|
-
|
|
104
|
-
test("memory_list total reflects the type filter", async () => {
|
|
105
|
-
const { registered, service } = setup();
|
|
106
|
-
service.saveWithDedupe({ type: "preference", title: "a", content: "x" });
|
|
107
|
-
service.saveWithDedupe({ type: "preference", title: "b", content: "y" });
|
|
108
|
-
service.saveWithDedupe({ type: "project", title: "c", content: "z" });
|
|
109
|
-
const list = registered.find((t) => t.name === "memory_list");
|
|
110
|
-
const projectRes = await list.execute({ type: "project" });
|
|
111
|
-
assert.equal(projectRes.total, 1);
|
|
112
|
-
const allRes = await list.execute({});
|
|
113
|
-
assert.equal(allRes.total, 3);
|
|
114
|
-
});
|
|
115
|
-
|
|
116
|
-
test("memory_update modifies an entry", async () => {
|
|
117
|
-
const { registered, service } = setup();
|
|
118
|
-
const { memory } = service.saveWithDedupe({ type: "decision", title: "t", content: "c" });
|
|
119
|
-
const update = registered.find((t) => t.name === "memory_update");
|
|
120
|
-
const result = await update.execute({ id: memory.id, content: "updated" });
|
|
121
|
-
assert.equal(result.memory.content, "updated");
|
|
122
|
-
});
|
|
123
|
-
|
|
124
|
-
test("memory_update on missing id rejects", async () => {
|
|
125
|
-
const { registered } = setup();
|
|
126
|
-
const update = registered.find((t) => t.name === "memory_update");
|
|
127
|
-
await assert.rejects(() => update.execute({ id: "missing", content: "x" }), /memory not found/);
|
|
128
|
-
});
|
|
129
|
-
|
|
130
|
-
test("memory_delete removes an entry", async () => {
|
|
131
|
-
const { registered, service, store } = setup();
|
|
132
|
-
const { memory } = service.saveWithDedupe({ type: "decision", title: "t", content: "c" });
|
|
133
|
-
const del = registered.find((t) => t.name === "memory_delete");
|
|
134
|
-
await del.execute({ id: memory.id });
|
|
135
|
-
assert.equal(store.count(), 0);
|
|
136
|
-
});
|
|
137
|
-
|
|
138
|
-
test("memory_delete on missing id returns deleted:false", async () => {
|
|
139
|
-
const { registered } = setup();
|
|
140
|
-
const del = registered.find((t) => t.name === "memory_delete");
|
|
141
|
-
const result = await del.execute({ id: "missing" });
|
|
142
|
-
assert.equal(result.deleted, false);
|
|
143
|
-
});
|
|
144
|
-
|
|
145
|
-
test("memory_delete by query deletes the best match (no id round-trip)", async () => {
|
|
146
|
-
const { registered, service, store } = setup();
|
|
147
|
-
const { memory } = service.saveWithDedupe({ type: "preference", title: "喜欢 Rust", content: "用户偏好 Rust 优先于 Go", importance: 4 });
|
|
148
|
-
service.saveWithDedupe({ type: "preference", title: "喜欢 Python", content: "用户偏好 Python 写脚本", importance: 3 });
|
|
149
|
-
const del = registered.find((t) => t.name === "memory_delete");
|
|
150
|
-
const result = await del.execute({ query: "Rust" });
|
|
151
|
-
assert.equal(result.deleted, true);
|
|
152
|
-
assert.equal(service.getById(memory.id), undefined, "best-matching Rust entry removed");
|
|
153
|
-
assert.equal(store.count(), 1, "non-matching entry untouched");
|
|
154
|
-
});
|
|
155
|
-
|
|
156
|
-
test("memory_delete by query with no match returns deleted:false", async () => {
|
|
157
|
-
const { registered, service, store } = setup();
|
|
158
|
-
service.saveWithDedupe({ type: "preference", title: "喜欢 Rust", content: "用户偏好 Rust", importance: 4 });
|
|
159
|
-
const del = registered.find((t) => t.name === "memory_delete");
|
|
160
|
-
const result = await del.execute({ query: "完全不存在的关键词xyz" });
|
|
161
|
-
assert.equal(result.deleted, false);
|
|
162
|
-
assert.equal(store.count(), 1, "nothing removed when no match");
|
|
163
|
-
});
|
|
164
|
-
|
|
165
|
-
test("memory_forget suppresses injection without deleting", async () => {
|
|
166
|
-
const { registered, service, store } = setup();
|
|
167
|
-
const { memory } = service.saveWithDedupe({ type: "project", title: "t", content: "c", importance: 5 });
|
|
168
|
-
const forget = registered.find((t) => t.name === "memory_forget");
|
|
169
|
-
const result = await forget.execute({ id: memory.id });
|
|
170
|
-
assert.equal(result.memory.forgotten, true);
|
|
171
|
-
assert.equal(store.count(undefined, { includeForgotten: true }), 1, "still stored but suppressed");
|
|
172
|
-
});
|
|
173
|
-
|
|
174
|
-
test("memory_forget on missing id rejects", async () => {
|
|
175
|
-
const { registered } = setup();
|
|
176
|
-
const forget = registered.find((t) => t.name === "memory_forget");
|
|
177
|
-
await assert.rejects(() => forget.execute({ id: "missing" }), /memory not found/);
|
|
178
|
-
});
|
|
179
|
-
|
|
180
|
-
test("memory_forget restores with forgotten:false", async () => {
|
|
181
|
-
const { registered, service } = setup();
|
|
182
|
-
const { memory } = service.saveWithDedupe({ type: "project", title: "t", content: "c", importance: 5 });
|
|
183
|
-
const forget = registered.find((t) => t.name === "memory_forget");
|
|
184
|
-
await forget.execute({ id: memory.id });
|
|
185
|
-
const restored = await forget.execute({ id: memory.id, forgotten: false });
|
|
186
|
-
assert.equal(restored.memory.forgotten, false);
|
|
187
|
-
});
|
|
188
|
-
|
|
189
|
-
test("execution results validate against their declared output schemas", async () => {
|
|
190
|
-
const { registered, service } = setup();
|
|
191
|
-
const byName = (n) => registered.find((t) => t.name === n);
|
|
192
|
-
|
|
193
|
-
const saved = await byName("memory_save").execute({ type: "decision", title: "x", content: "y", tags: ["k"], source: "test" });
|
|
194
|
-
assert.deepEqual(validateJsonSchemaValue(byName("memory_save").output.schema, saved), []);
|
|
195
|
-
|
|
196
|
-
service.saveWithDedupe({ type: "project", title: "p", content: "q" });
|
|
197
|
-
const searchRes = await byName("memory_search").execute({ query: "q" });
|
|
198
|
-
assert.deepEqual(validateJsonSchemaValue(byName("memory_search").output.schema, searchRes), []);
|
|
199
|
-
|
|
200
|
-
const listRes = await byName("memory_list").execute({});
|
|
201
|
-
assert.deepEqual(validateJsonSchemaValue(byName("memory_list").output.schema, listRes), []);
|
|
202
|
-
|
|
203
|
-
const updRes = await byName("memory_update").execute({ id: saved.id, content: "z" });
|
|
204
|
-
assert.deepEqual(validateJsonSchemaValue(byName("memory_update").output.schema, updRes), []);
|
|
205
|
-
|
|
206
|
-
const forgetRes = await byName("memory_forget").execute({ id: saved.id });
|
|
207
|
-
assert.deepEqual(validateJsonSchemaValue(byName("memory_forget").output.schema, forgetRes), []);
|
|
208
|
-
|
|
209
|
-
const delRes = await byName("memory_delete").execute({ id: saved.id });
|
|
210
|
-
assert.deepEqual(validateJsonSchemaValue(byName("memory_delete").output.schema, delRes), []);
|
|
211
|
-
});
|
|
212
|
-
|
|
213
|
-
test("memory_search semantic mode merges vector recalls", async () => {
|
|
214
|
-
const store = createStore(":memory:");
|
|
215
|
-
const service = createService({ store, mirror: null, config: {} });
|
|
216
|
-
const registered = [];
|
|
217
|
-
const ctx = { tools: { register(def) { registered.push(def); return () => {}; } } };
|
|
218
|
-
const embedder = { embed: async () => [1, 0, 0] };
|
|
219
|
-
createTools(ctx, service, {}, embedder);
|
|
220
|
-
|
|
221
|
-
const v = service.saveWithDedupe({ type: "preference", title: "猫", content: "喜欢猫" });
|
|
222
|
-
store.setEmbedding(v.memory.id, [1, 0, 0]);
|
|
223
|
-
service.saveWithDedupe({ type: "preference", title: "狗", content: "喜欢狗" });
|
|
224
|
-
|
|
225
|
-
const search = registered.find((t) => t.name === "memory_search");
|
|
226
|
-
// literal query matches only 猫; vector recall also surfaces it via embedding
|
|
227
|
-
const res = await search.execute({ query: "猫", semantic: true });
|
|
228
|
-
assert.ok(res.items.some((m) => m.title === "猫"), "vector + keyword merged result contains the hit");
|
|
229
|
-
// mode=vector keeps keyword items too (dedup merge)
|
|
230
|
-
const res2 = await search.execute({ query: "猫", mode: "vector" });
|
|
231
|
-
assert.ok(res2.items.some((m) => m.title === "猫"));
|
|
232
|
-
});
|
|
233
|
-
|
|
234
|
-
test("memory_search falls back to keyword when embedder unavailable or returns null", async () => {
|
|
235
|
-
// No embedder passed → plain keyword search still works.
|
|
236
|
-
const { registered, service } = setup();
|
|
237
|
-
service.saveWithDedupe({ type: "preference", title: "语言", content: "中文交流" });
|
|
238
|
-
const search = registered.find((t) => t.name === "memory_search");
|
|
239
|
-
const res = await search.execute({ query: "中文", semantic: true });
|
|
240
|
-
assert.equal(res.items.length, 1);
|
|
241
|
-
assert.equal(res.items[0].title, "语言");
|
|
242
|
-
|
|
243
|
-
// Embedder that resolves null (provider disabled) → keyword fallback.
|
|
244
|
-
const store = createStore(":memory:");
|
|
245
|
-
const service2 = createService({ store, mirror: null, config: {} });
|
|
246
|
-
const registered2 = [];
|
|
247
|
-
const ctx = { tools: { register(def) { registered2.push(def); return () => {}; } } };
|
|
248
|
-
createTools(ctx, service2, {}, { embed: async () => null });
|
|
249
|
-
service2.saveWithDedupe({ type: "preference", title: "语言", content: "中文交流" });
|
|
250
|
-
const search2 = registered2.find((t) => t.name === "memory_search");
|
|
251
|
-
const res2 = await search2.execute({ query: "中文", semantic: true });
|
|
252
|
-
assert.equal(res2.items.length, 1);
|
|
253
|
-
});
|
|
254
|
-
|
|
255
|
-
// --- memory_archive (item ⑤: archive + verifiable recovery) ----------------
|
|
256
|
-
|
|
257
|
-
test("memory_archive hides an entry and memory_unarchive restores it", async () => {
|
|
258
|
-
const { registered, service } = setup();
|
|
259
|
-
const { memory: m } = service.saveWithDedupe({ type: "project", title: "t", content: "c" });
|
|
260
|
-
const archive = registered.find((t) => t.name === "memory_archive");
|
|
261
|
-
const archived = await archive.execute({ id: m.id });
|
|
262
|
-
assert.equal(archived.memory.archived, true);
|
|
263
|
-
assert.ok(!service.list({ type: "project" }).some((x) => x.id === m.id), "archived hidden from default list");
|
|
264
|
-
const restored = await archive.execute({ id: m.id, archived: false });
|
|
265
|
-
assert.equal(restored.memory.archived, false);
|
|
266
|
-
assert.ok(service.list({ type: "project" }).some((x) => x.id === m.id), "restored entry visible again");
|
|
267
|
-
});
|
|
268
|
-
|
|
269
|
-
test("memory_list include_archived lists hidden entries so they can be restored", async () => {
|
|
270
|
-
const { registered, service } = setup();
|
|
271
|
-
const { memory: m } = service.saveWithDedupe({ type: "preference", title: "a", content: "x" });
|
|
272
|
-
service.setArchived(m.id, true);
|
|
273
|
-
const list = registered.find((t) => t.name === "memory_list");
|
|
274
|
-
const hidden = await list.execute({ type: "preference", include_archived: true });
|
|
275
|
-
assert.ok(hidden.items.some((x) => x.id === m.id), "archived entry discoverable");
|
|
276
|
-
assert.equal(hidden.total, 1, "total counts the archived entry");
|
|
277
|
-
const visible = await list.execute({ type: "preference" });
|
|
278
|
-
assert.ok(!visible.items.some((x) => x.id === m.id), "archived entry hidden by default");
|
|
279
|
-
});
|
|
280
|
-
|
|
281
|
-
test("memory_archive on missing id rejects", async () => {
|
|
282
|
-
const { registered } = setup();
|
|
283
|
-
const archive = registered.find((t) => t.name === "memory_archive");
|
|
284
|
-
await assert.rejects(() => archive.execute({ id: "missing" }), /memory not found/);
|
|
285
|
-
});
|
|
1
|
+
import test from "node:test";
|
|
2
|
+
import assert from "node:assert/strict";
|
|
3
|
+
import { assertSupportedJsonSchema, validateJsonSchemaValue } from "@deepseek-ai/dsh-tools";
|
|
4
|
+
import { createStore } from "../src/store.js";
|
|
5
|
+
import { createService } from "../src/service.js";
|
|
6
|
+
import { createTools } from "../src/tools.js";
|
|
7
|
+
|
|
8
|
+
function setup(embedder) {
|
|
9
|
+
const store = createStore(":memory:");
|
|
10
|
+
const service = createService({ store, mirror: null, config: {} });
|
|
11
|
+
const registered = [];
|
|
12
|
+
const ctx = {
|
|
13
|
+
tools: {
|
|
14
|
+
register(def) {
|
|
15
|
+
registered.push(def);
|
|
16
|
+
return () => {};
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
};
|
|
20
|
+
const tools = createTools(ctx, service, {}, embedder);
|
|
21
|
+
return { store, service, tools, registered };
|
|
22
|
+
}
|
|
23
|
+
|
|
24
|
+
// Collect authoring-DSL regressions in a compiled schema: property-level
|
|
25
|
+
// `required: true` (must be projected to a top-level required array by
|
|
26
|
+
// defineTool) and `minimum`/`maximum` (outside the enforced subset) are
|
|
27
|
+
// rejected by the real harness, so they must never appear post-compilation.
|
|
28
|
+
function walkSchema(node, path, problems) {
|
|
29
|
+
if (typeof node !== "object" || node === null || Array.isArray(node)) return;
|
|
30
|
+
if (typeof node.required === "boolean") {
|
|
31
|
+
problems.push(`${path}.required must be an array (property-level required leaks through)`);
|
|
32
|
+
}
|
|
33
|
+
if (Object.hasOwn(node, "minimum") || Object.hasOwn(node, "maximum")) {
|
|
34
|
+
problems.push(`${path} uses minimum/maximum, which the enforced JSON Schema subset rejects`);
|
|
35
|
+
}
|
|
36
|
+
if (node.properties && typeof node.properties === "object") {
|
|
37
|
+
for (const [key, value] of Object.entries(node.properties)) {
|
|
38
|
+
walkSchema(value, `${path}.properties.${key}`, problems);
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
if (node.items) walkSchema(node.items, `${path}.items`, problems);
|
|
42
|
+
if (Array.isArray(node.oneOf)) {
|
|
43
|
+
node.oneOf.forEach((branch, index) => walkSchema(branch, `${path}.oneOf[${index}]`, problems));
|
|
44
|
+
}
|
|
45
|
+
}
|
|
46
|
+
|
|
47
|
+
test("registers seven tools with correct names", () => {
|
|
48
|
+
const { registered } = setup();
|
|
49
|
+
const names = registered.map((t) => t.name).sort();
|
|
50
|
+
assert.deepEqual(names, ["memory_archive", "memory_delete", "memory_forget", "memory_list", "memory_save", "memory_search", "memory_update"]);
|
|
51
|
+
});
|
|
52
|
+
|
|
53
|
+
test("compiled schemas pass the enforced DSH subset (defineTool projection)", () => {
|
|
54
|
+
const { registered } = setup();
|
|
55
|
+
assert.equal(registered.length, 7);
|
|
56
|
+
for (const tool of registered) {
|
|
57
|
+
assertSupportedJsonSchema(tool.parameters);
|
|
58
|
+
assertSupportedJsonSchema(tool.output.schema);
|
|
59
|
+
const problems = [];
|
|
60
|
+
walkSchema(tool.parameters, `parameters(${tool.name})`, problems);
|
|
61
|
+
walkSchema(tool.output.schema, `output.schema(${tool.name})`, problems);
|
|
62
|
+
assert.deepEqual(problems, []);
|
|
63
|
+
}
|
|
64
|
+
});
|
|
65
|
+
|
|
66
|
+
test("memory_save executes and stores", async () => {
|
|
67
|
+
const { registered, store } = setup();
|
|
68
|
+
const save = registered.find((t) => t.name === "memory_save");
|
|
69
|
+
const result = await save.execute({ type: "preference", title: "语言", content: "中文", importance: 4 });
|
|
70
|
+
assert.equal(result.action, "created");
|
|
71
|
+
assert.equal(store.count(), 1);
|
|
72
|
+
});
|
|
73
|
+
|
|
74
|
+
test("memory_save merges on matching title within type", async () => {
|
|
75
|
+
const { registered, store } = setup();
|
|
76
|
+
const save = registered.find((t) => t.name === "memory_save");
|
|
77
|
+
const first = await save.execute({ type: "preference", title: "语言", content: "中文" });
|
|
78
|
+
const second = await save.execute({ type: "preference", title: "语言", content: "简体中文" });
|
|
79
|
+
assert.equal(first.action, "created");
|
|
80
|
+
assert.equal(second.action, "merged");
|
|
81
|
+
assert.equal(first.id, second.id);
|
|
82
|
+
assert.equal(store.count(), 1);
|
|
83
|
+
});
|
|
84
|
+
|
|
85
|
+
test("memory_search finds by CJK substring", async () => {
|
|
86
|
+
const { registered, store, service } = setup();
|
|
87
|
+
service.saveWithDedupe({ type: "project", title: "记忆插件", content: "SQLite 存储中文记忆", importance: 3 });
|
|
88
|
+
const search = registered.find((t) => t.name === "memory_search");
|
|
89
|
+
const result = await search.execute({ query: "中文" });
|
|
90
|
+
assert.ok(result.items.length >= 1);
|
|
91
|
+
assert.equal(result.items[0].title, "记忆插件");
|
|
92
|
+
});
|
|
93
|
+
|
|
94
|
+
test("memory_list filters by type", async () => {
|
|
95
|
+
const { registered, service } = setup();
|
|
96
|
+
service.saveWithDedupe({ type: "preference", title: "a", content: "x" });
|
|
97
|
+
service.saveWithDedupe({ type: "project", title: "b", content: "y" });
|
|
98
|
+
const list = registered.find((t) => t.name === "memory_list");
|
|
99
|
+
const result = await list.execute({ type: "project" });
|
|
100
|
+
assert.equal(result.items.length, 1);
|
|
101
|
+
assert.equal(result.items[0].title, "b");
|
|
102
|
+
});
|
|
103
|
+
|
|
104
|
+
test("memory_list total reflects the type filter", async () => {
|
|
105
|
+
const { registered, service } = setup();
|
|
106
|
+
service.saveWithDedupe({ type: "preference", title: "a", content: "x" });
|
|
107
|
+
service.saveWithDedupe({ type: "preference", title: "b", content: "y" });
|
|
108
|
+
service.saveWithDedupe({ type: "project", title: "c", content: "z" });
|
|
109
|
+
const list = registered.find((t) => t.name === "memory_list");
|
|
110
|
+
const projectRes = await list.execute({ type: "project" });
|
|
111
|
+
assert.equal(projectRes.total, 1);
|
|
112
|
+
const allRes = await list.execute({});
|
|
113
|
+
assert.equal(allRes.total, 3);
|
|
114
|
+
});
|
|
115
|
+
|
|
116
|
+
test("memory_update modifies an entry", async () => {
|
|
117
|
+
const { registered, service } = setup();
|
|
118
|
+
const { memory } = service.saveWithDedupe({ type: "decision", title: "t", content: "c" });
|
|
119
|
+
const update = registered.find((t) => t.name === "memory_update");
|
|
120
|
+
const result = await update.execute({ id: memory.id, content: "updated" });
|
|
121
|
+
assert.equal(result.memory.content, "updated");
|
|
122
|
+
});
|
|
123
|
+
|
|
124
|
+
test("memory_update on missing id rejects", async () => {
|
|
125
|
+
const { registered } = setup();
|
|
126
|
+
const update = registered.find((t) => t.name === "memory_update");
|
|
127
|
+
await assert.rejects(() => update.execute({ id: "missing", content: "x" }), /memory not found/);
|
|
128
|
+
});
|
|
129
|
+
|
|
130
|
+
test("memory_delete removes an entry", async () => {
|
|
131
|
+
const { registered, service, store } = setup();
|
|
132
|
+
const { memory } = service.saveWithDedupe({ type: "decision", title: "t", content: "c" });
|
|
133
|
+
const del = registered.find((t) => t.name === "memory_delete");
|
|
134
|
+
await del.execute({ id: memory.id });
|
|
135
|
+
assert.equal(store.count(), 0);
|
|
136
|
+
});
|
|
137
|
+
|
|
138
|
+
test("memory_delete on missing id returns deleted:false", async () => {
|
|
139
|
+
const { registered } = setup();
|
|
140
|
+
const del = registered.find((t) => t.name === "memory_delete");
|
|
141
|
+
const result = await del.execute({ id: "missing" });
|
|
142
|
+
assert.equal(result.deleted, false);
|
|
143
|
+
});
|
|
144
|
+
|
|
145
|
+
test("memory_delete by query deletes the best match (no id round-trip)", async () => {
|
|
146
|
+
const { registered, service, store } = setup();
|
|
147
|
+
const { memory } = service.saveWithDedupe({ type: "preference", title: "喜欢 Rust", content: "用户偏好 Rust 优先于 Go", importance: 4 });
|
|
148
|
+
service.saveWithDedupe({ type: "preference", title: "喜欢 Python", content: "用户偏好 Python 写脚本", importance: 3 });
|
|
149
|
+
const del = registered.find((t) => t.name === "memory_delete");
|
|
150
|
+
const result = await del.execute({ query: "Rust" });
|
|
151
|
+
assert.equal(result.deleted, true);
|
|
152
|
+
assert.equal(service.getById(memory.id), undefined, "best-matching Rust entry removed");
|
|
153
|
+
assert.equal(store.count(), 1, "non-matching entry untouched");
|
|
154
|
+
});
|
|
155
|
+
|
|
156
|
+
test("memory_delete by query with no match returns deleted:false", async () => {
|
|
157
|
+
const { registered, service, store } = setup();
|
|
158
|
+
service.saveWithDedupe({ type: "preference", title: "喜欢 Rust", content: "用户偏好 Rust", importance: 4 });
|
|
159
|
+
const del = registered.find((t) => t.name === "memory_delete");
|
|
160
|
+
const result = await del.execute({ query: "完全不存在的关键词xyz" });
|
|
161
|
+
assert.equal(result.deleted, false);
|
|
162
|
+
assert.equal(store.count(), 1, "nothing removed when no match");
|
|
163
|
+
});
|
|
164
|
+
|
|
165
|
+
test("memory_forget suppresses injection without deleting", async () => {
|
|
166
|
+
const { registered, service, store } = setup();
|
|
167
|
+
const { memory } = service.saveWithDedupe({ type: "project", title: "t", content: "c", importance: 5 });
|
|
168
|
+
const forget = registered.find((t) => t.name === "memory_forget");
|
|
169
|
+
const result = await forget.execute({ id: memory.id });
|
|
170
|
+
assert.equal(result.memory.forgotten, true);
|
|
171
|
+
assert.equal(store.count(undefined, { includeForgotten: true }), 1, "still stored but suppressed");
|
|
172
|
+
});
|
|
173
|
+
|
|
174
|
+
test("memory_forget on missing id rejects", async () => {
|
|
175
|
+
const { registered } = setup();
|
|
176
|
+
const forget = registered.find((t) => t.name === "memory_forget");
|
|
177
|
+
await assert.rejects(() => forget.execute({ id: "missing" }), /memory not found/);
|
|
178
|
+
});
|
|
179
|
+
|
|
180
|
+
test("memory_forget restores with forgotten:false", async () => {
|
|
181
|
+
const { registered, service } = setup();
|
|
182
|
+
const { memory } = service.saveWithDedupe({ type: "project", title: "t", content: "c", importance: 5 });
|
|
183
|
+
const forget = registered.find((t) => t.name === "memory_forget");
|
|
184
|
+
await forget.execute({ id: memory.id });
|
|
185
|
+
const restored = await forget.execute({ id: memory.id, forgotten: false });
|
|
186
|
+
assert.equal(restored.memory.forgotten, false);
|
|
187
|
+
});
|
|
188
|
+
|
|
189
|
+
test("execution results validate against their declared output schemas", async () => {
|
|
190
|
+
const { registered, service } = setup();
|
|
191
|
+
const byName = (n) => registered.find((t) => t.name === n);
|
|
192
|
+
|
|
193
|
+
const saved = await byName("memory_save").execute({ type: "decision", title: "x", content: "y", tags: ["k"], source: "test" });
|
|
194
|
+
assert.deepEqual(validateJsonSchemaValue(byName("memory_save").output.schema, saved), []);
|
|
195
|
+
|
|
196
|
+
service.saveWithDedupe({ type: "project", title: "p", content: "q" });
|
|
197
|
+
const searchRes = await byName("memory_search").execute({ query: "q" });
|
|
198
|
+
assert.deepEqual(validateJsonSchemaValue(byName("memory_search").output.schema, searchRes), []);
|
|
199
|
+
|
|
200
|
+
const listRes = await byName("memory_list").execute({});
|
|
201
|
+
assert.deepEqual(validateJsonSchemaValue(byName("memory_list").output.schema, listRes), []);
|
|
202
|
+
|
|
203
|
+
const updRes = await byName("memory_update").execute({ id: saved.id, content: "z" });
|
|
204
|
+
assert.deepEqual(validateJsonSchemaValue(byName("memory_update").output.schema, updRes), []);
|
|
205
|
+
|
|
206
|
+
const forgetRes = await byName("memory_forget").execute({ id: saved.id });
|
|
207
|
+
assert.deepEqual(validateJsonSchemaValue(byName("memory_forget").output.schema, forgetRes), []);
|
|
208
|
+
|
|
209
|
+
const delRes = await byName("memory_delete").execute({ id: saved.id });
|
|
210
|
+
assert.deepEqual(validateJsonSchemaValue(byName("memory_delete").output.schema, delRes), []);
|
|
211
|
+
});
|
|
212
|
+
|
|
213
|
+
test("memory_search semantic mode merges vector recalls", async () => {
|
|
214
|
+
const store = createStore(":memory:");
|
|
215
|
+
const service = createService({ store, mirror: null, config: {} });
|
|
216
|
+
const registered = [];
|
|
217
|
+
const ctx = { tools: { register(def) { registered.push(def); return () => {}; } } };
|
|
218
|
+
const embedder = { embed: async () => [1, 0, 0] };
|
|
219
|
+
createTools(ctx, service, {}, embedder);
|
|
220
|
+
|
|
221
|
+
const v = service.saveWithDedupe({ type: "preference", title: "猫", content: "喜欢猫" });
|
|
222
|
+
store.setEmbedding(v.memory.id, [1, 0, 0]);
|
|
223
|
+
service.saveWithDedupe({ type: "preference", title: "狗", content: "喜欢狗" });
|
|
224
|
+
|
|
225
|
+
const search = registered.find((t) => t.name === "memory_search");
|
|
226
|
+
// literal query matches only 猫; vector recall also surfaces it via embedding
|
|
227
|
+
const res = await search.execute({ query: "猫", semantic: true });
|
|
228
|
+
assert.ok(res.items.some((m) => m.title === "猫"), "vector + keyword merged result contains the hit");
|
|
229
|
+
// mode=vector keeps keyword items too (dedup merge)
|
|
230
|
+
const res2 = await search.execute({ query: "猫", mode: "vector" });
|
|
231
|
+
assert.ok(res2.items.some((m) => m.title === "猫"));
|
|
232
|
+
});
|
|
233
|
+
|
|
234
|
+
test("memory_search falls back to keyword when embedder unavailable or returns null", async () => {
|
|
235
|
+
// No embedder passed → plain keyword search still works.
|
|
236
|
+
const { registered, service } = setup();
|
|
237
|
+
service.saveWithDedupe({ type: "preference", title: "语言", content: "中文交流" });
|
|
238
|
+
const search = registered.find((t) => t.name === "memory_search");
|
|
239
|
+
const res = await search.execute({ query: "中文", semantic: true });
|
|
240
|
+
assert.equal(res.items.length, 1);
|
|
241
|
+
assert.equal(res.items[0].title, "语言");
|
|
242
|
+
|
|
243
|
+
// Embedder that resolves null (provider disabled) → keyword fallback.
|
|
244
|
+
const store = createStore(":memory:");
|
|
245
|
+
const service2 = createService({ store, mirror: null, config: {} });
|
|
246
|
+
const registered2 = [];
|
|
247
|
+
const ctx = { tools: { register(def) { registered2.push(def); return () => {}; } } };
|
|
248
|
+
createTools(ctx, service2, {}, { embed: async () => null });
|
|
249
|
+
service2.saveWithDedupe({ type: "preference", title: "语言", content: "中文交流" });
|
|
250
|
+
const search2 = registered2.find((t) => t.name === "memory_search");
|
|
251
|
+
const res2 = await search2.execute({ query: "中文", semantic: true });
|
|
252
|
+
assert.equal(res2.items.length, 1);
|
|
253
|
+
});
|
|
254
|
+
|
|
255
|
+
// --- memory_archive (item ⑤: archive + verifiable recovery) ----------------
|
|
256
|
+
|
|
257
|
+
test("memory_archive hides an entry and memory_unarchive restores it", async () => {
|
|
258
|
+
const { registered, service } = setup();
|
|
259
|
+
const { memory: m } = service.saveWithDedupe({ type: "project", title: "t", content: "c" });
|
|
260
|
+
const archive = registered.find((t) => t.name === "memory_archive");
|
|
261
|
+
const archived = await archive.execute({ id: m.id });
|
|
262
|
+
assert.equal(archived.memory.archived, true);
|
|
263
|
+
assert.ok(!service.list({ type: "project" }).some((x) => x.id === m.id), "archived hidden from default list");
|
|
264
|
+
const restored = await archive.execute({ id: m.id, archived: false });
|
|
265
|
+
assert.equal(restored.memory.archived, false);
|
|
266
|
+
assert.ok(service.list({ type: "project" }).some((x) => x.id === m.id), "restored entry visible again");
|
|
267
|
+
});
|
|
268
|
+
|
|
269
|
+
test("memory_list include_archived lists hidden entries so they can be restored", async () => {
|
|
270
|
+
const { registered, service } = setup();
|
|
271
|
+
const { memory: m } = service.saveWithDedupe({ type: "preference", title: "a", content: "x" });
|
|
272
|
+
service.setArchived(m.id, true);
|
|
273
|
+
const list = registered.find((t) => t.name === "memory_list");
|
|
274
|
+
const hidden = await list.execute({ type: "preference", include_archived: true });
|
|
275
|
+
assert.ok(hidden.items.some((x) => x.id === m.id), "archived entry discoverable");
|
|
276
|
+
assert.equal(hidden.total, 1, "total counts the archived entry");
|
|
277
|
+
const visible = await list.execute({ type: "preference" });
|
|
278
|
+
assert.ok(!visible.items.some((x) => x.id === m.id), "archived entry hidden by default");
|
|
279
|
+
});
|
|
280
|
+
|
|
281
|
+
test("memory_archive on missing id rejects", async () => {
|
|
282
|
+
const { registered } = setup();
|
|
283
|
+
const archive = registered.find((t) => t.name === "memory_archive");
|
|
284
|
+
await assert.rejects(() => archive.execute({ id: "missing" }), /memory not found/);
|
|
285
|
+
});
|