@modusensus/dsh-mneme 0.2.4 → 0.2.5

package/test/client.test.js

@@ -0,0 +1,26 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+import { readFileSync, existsSync } from "node:fs";
+import { fileURLToPath } from "node:url";
+import { dirname, join } from "node:path";
+
+const root = join(dirname(fileURLToPath(import.meta.url)), "..");
+const clientSource = readFileSync(join(root, "lib/client.js"), "utf8");
+const pkg = JSON.parse(readFileSync(join(root, "package.json"), "utf8"));
+
+// The Web client bundle registers itself via __ModuleLoader__.load. DSH
+// resolves the bundle by plugin package name, so the registered id must match
+// package.json `name` exactly (a mismatch surfaces as "loaded without
+// registering '@modusensus/dsh-mneme'" in the DSH client).
+test("client bundle registers under the package name", () => {
+  const match = clientSource.match(/__ModuleLoader__\.load\(\{\s*id:\s*"([^"]+)"/);
+  assert.ok(match, "client bundle must call __ModuleLoader__.load with an id");
+  assert.equal(match[1], pkg.name, "registered id must equal package.json name");
+});
+
+// client.js is hand-authored under lib/ only (no src/ counterpart), so the
+// src->lib sync must never prune it.
+test("client bundle is lib-only with no src counterpart", () => {
+  assert.equal(existsSync(join(root, "src/client.js")), false, "src/ must not contain client.js");
+  assert.equal(existsSync(join(root, "lib/client.js")), true, "lib/client.js must exist");
+});

package/test/clustering.test.js

@@ -0,0 +1,100 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+import { cosineSimilarity, kMeans, clusterMemories, findPotentialConflicts } from "../src/dream/clustering.js";
+
+test("cosineSimilarity: identical vectors => 1", () => {
+  assert.equal(cosineSimilarity([1, 2, 3], [1, 2, 3]), 1);
+});
+
+test("cosineSimilarity: orthogonal vectors => 0 (approx)", () => {
+  assert.ok(Math.abs(cosineSimilarity([1, 0], [0, 1])) < 1e-9);
+});
+
+test("cosineSimilarity: mismatched or empty lengths => 0", () => {
+  assert.equal(cosineSimilarity([1, 2], [1, 2, 3]), 0);
+  assert.equal(cosineSimilarity([], []), 0);
+});
+
+test("cosineSimilarity: zero-norm vector => 0", () => {
+  assert.equal(cosineSimilarity([0, 0], [1, 1]), 0);
+});
+
+// Three well-separated point groups in the plane.
+const GROUP1 = [[0, 0], [0, 0.1], [0.1, 0], [0.1, 0.1]];
+const GROUP2 = [[10, 10], [10, 10.1], [10.1, 10], [10.1, 10.1]];
+const GROUP3 = [[-10, -10], [-10, -9.9], [-9.9, -10], [-9.9, -9.9]];
+const SEPARATED = [...GROUP1, ...GROUP2, ...GROUP3];
+
+test("kMeans: three separated groups cluster correctly (labels may permute)", () => {
+  const clusters = kMeans(SEPARATED, 3, { maxIter: 100 });
+  assert.equal(clusters.length, 3);
+  // Content-based check: each cluster must fit exactly one expected group.
+  const expectedGroups = [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]];
+  for (const c of clusters) {
+    assert.ok(c.length >= 1);
+    const fits = expectedGroups.filter((g) => c.every((idx) => g.includes(idx)));
+    assert.equal(fits.length, 1, `cluster ${JSON.stringify(c)} spans multiple groups`);
+  }
+  // Every index assigned exactly once.
+  assert.deepEqual(clusters.flat().sort((a, b) => a - b), [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]);
+});
+
+test("kMeans: k >= n returns a single cluster of all indices", () => {
+  assert.deepEqual(kMeans(SEPARATED, 12), [[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]]);
+  assert.deepEqual(kMeans(SEPARATED, 0), [[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]]);
+});
+
+test("kMeans: empty input => no clusters", () => {
+  assert.deepEqual(kMeans([], 3), []);
+});
+
+test("kMeans: empty clusters get reseeded and still terminate", () => {
+  // Many points near two poles with k=3 forces one cluster empty on seed.
+  const vectors = [
+    ...Array.from({ length: 30 }, () => [0, 0]),
+    ...Array.from({ length: 30 }, () => [1, 1])
+  ];
+  const clusters = kMeans(vectors, 3, { maxIter: 50 });
+  assert.equal(clusters.length, 3);
+  assert.equal(clusters.flat().length, vectors.length);
+});
+
+test("clusterMemories: preserves object references and group count", () => {
+  const memories = SEPARATED.map((v, i) => ({ id: `m${i}`, type: "project", title: `t${i}` }));
+  const groups = clusterMemories(memories, SEPARATED, 3);
+  assert.equal(groups.length, 3);
+  assert.equal(groups.flat().length, memories.length);
+  // Reference identity preserved: every group member is an object from input.
+  const all = groups.flat();
+  for (const m of all) assert.ok(memories.includes(m), "cluster contains a foreign object");
+  // Full coverage: each source object appears exactly once across groups.
+  assert.equal(new Set(all).size, memories.length);
+  assert.equal(groups.flat().find((m) => m.id === "m5"), memories[5]);
+});
+
+test("findPotentialConflicts: only same-type highly-similar pairs reported", () => {
+  const memories = [
+    { id: "a", type: "preference", content: "like cold coffee" },
+    { id: "b", type: "preference", content: "hate cold coffee" },
+    { id: "c", type: "project", content: "ship the plugin" }
+  ];
+  const vectors = [
+    [1, 0, 0],
+    [0.99, 0.02, 0],   // nearly parallel to a, same type => conflict
+    [0, 1, 0]          // different type, ignored even if near
+  ];
+  const pairs = findPotentialConflicts(memories, vectors, 0.85);
+  assert.equal(pairs.length, 1);
+  assert.equal(pairs[0].a, memories[0]);
+  assert.equal(pairs[0].b, memories[1]);
+  assert.ok(pairs[0].similarity > 0.85);
+});
+
+test("findPotentialConflicts: below threshold or different type => no pairs", () => {
+  const memories = [
+    { id: "x", type: "preference" },
+    { id: "y", type: "project" }
+  ];
+  assert.deepEqual(findPotentialConflicts(memories, [[1, 0], [0, 1]]), []);
+  assert.deepEqual(findPotentialConflicts(memories, [[1, 0], [0.9, 0.1]], 0.95), []);
+});

package/test/commands.test.js

@@ -0,0 +1,69 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+import { createStore } from "../src/store.js";
+import { createSettings } from "../src/settings.js";
+import { createCommandManager } from "../src/commands.js";
+
+function setup() {
+  const store = createStore(":memory:");
+  const settings = createSettings(store.db);
+  const registered = new Map(); // name -> definition
+  const ctx = {
+    commands: {
+      register(def) {
+        registered.set(def.name, def);
+        return () => registered.delete(def.name);
+      }
+    }
+  };
+  const manager = createCommandManager({ ctx, settings, logger: null });
+  return { store, settings, manager, registered };
+}
+
+test("sync registers all stored commands", () => {
+  const { store, settings, manager, registered } = setup();
+  settings.addCommand({ name: "agenda", description: "d", instruction: "列议程" });
+  settings.addCommand({ name: "banner", instruction: "打印横幅" });
+  manager.sync();
+  assert.equal(registered.size, 2);
+  assert.ok(registered.has("agenda") && registered.has("banner"));
+  store.close();
+});
+
+test("add registers live; remove unregisters live", () => {
+  const { store, manager, registered } = setup();
+  const cmd = manager.add({ name: "review", description: "r", instruction: "审查代码" });
+  assert.ok(registered.has("review"), "registered after add");
+  assert.equal(registered.get("review").description, "r");
+  assert.equal(manager.remove(cmd.id), true);
+  assert.ok(!registered.has("review"), "unregistered after remove");
+  store.close();
+});
+
+test("handler returns the instruction as success text", async () => {
+  const { store, manager, registered } = setup();
+  manager.add({ name: "fmt", description: "格式化", instruction: "请按项目的 lint 规则格式化当前文件" });
+  const def = registered.get("fmt");
+  assert.ok(def, "definition registered");
+  const result = await def.handler({ agent: {}, rawInput: "", signal: null });
+  assert.equal(result.kind, "success");
+  assert.equal(result.text, "请按项目的 lint 规则格式化当前文件");
+  store.close();
+});
+
+test("remove of missing id returns false", () => {
+  const { store, manager } = setup();
+  assert.equal(manager.remove("nope"), false);
+  store.close();
+});
+
+test("dispose unregisters everything", () => {
+  const { store, settings, manager, registered } = setup();
+  settings.addCommand({ name: "a", instruction: "x" });
+  settings.addCommand({ name: "b", instruction: "y" });
+  manager.sync();
+  assert.equal(registered.size, 2);
+  manager.dispose();
+  assert.equal(registered.size, 0);
+  store.close();
+});

package/test/config.test.js

@@ -0,0 +1,31 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+import { readFileSync } from "node:fs";
+import { fileURLToPath } from "node:url";
+import { Config } from "../src/config.js";
+
+// ---------------------------------------------------------------- rerank opt-in (item ⑥)
+
+test("rerank is opt-in: default config does not enable the local reranker", () => {
+  const cfg = Config({});
+  assert.equal(cfg.rerankEnabled, false, "rerankEnabled defaults to false");
+  assert.equal(cfg.rerankProvider, "none", "rerankProvider defaults to none");
+  // The plugin gate in index.js: LocalReranker is only constructed when both
+  // hold — under defaults the gate is closed, so onnxruntime is never loaded.
+  assert.equal(cfg.rerankEnabled && cfg.rerankProvider === "local", false, "gate closed by default");
+  const enabled = Config({ rerankEnabled: true, rerankProvider: "local" });
+  assert.equal(enabled.rerankEnabled && enabled.rerankProvider === "local", true, "explicit opt-in opens the gate");
+});
+
+test("startup probe: the reranker module never statically imports transformers/onnxruntime", () => {
+  // LocalReranker is imported eagerly by index.js, so a bare install must not
+  // pull onnxruntime in at module load. The heavy load is a lazy dynamic import
+  // that only runs inside init(), which index.js calls only when the opt-in gate
+  // is open (rerankEnabled && rerankProvider === "local").
+  const src = readFileSync(new URL("../src/reranker.js", import.meta.url), "utf8");
+  assert.ok(
+    !src.includes('from "@huggingface/transformers"'),
+    "no static transformers.js import in reranker.js"
+  );
+  assert.match(src, /await import\("@huggingface\/transformers"\)/, "transformers.js loads lazily via dynamic import");
+});