feat: Phase 3 世界书引擎移植 + EJS 支持

- 新增 task-worldinfo.js: 从 EW 移植世界书激活/分桶引擎 - 新增 task-ejs.js: 从 EW 移植 EJS 模板渲染引擎 - 新增 vendor/ejs.js: EJS runtime vendor - prompt-builder.js: 改为异步, 接入 worldInfoBefore/After/atDepth - prompt-profiles.js: 新增内置块 charDescription/userPersona/worldInfoBefore/After - 更新 extractor/retriever/compressor/consolidator 接入新 builder - st-context.js: 扩展 ST 上下文字段兜底 - 新增 tests/task-worldinfo.mjs: 世界书引擎测试
2026-05-15 22:30:38 +08:00 · 2026-03-26 13:57:07 +08:00
parent 2f9524d993
commit d31c0325d3
12 changed files with 3963 additions and 93 deletions
--- a/tests/task-worldinfo.mjs
+++ b/tests/task-worldinfo.mjs
@@ -0,0 +1,207 @@
+import assert from "node:assert/strict";
+
+const originalSillyTavern = globalThis.SillyTavern;
+const originalGetCharWorldbookNames = globalThis.getCharWorldbookNames;
+const originalGetWorldbook = globalThis.getWorldbook;
+const originalGetLorebookEntries = globalThis.getLorebookEntries;
+
+const constantEntry = {
+  uid: 1,
+  name: "常驻设定",
+  comment: "常驻设定",
+  content: "这里是常驻世界设定。",
+  enabled: true,
+  position: {
+    type: "before_character_definition",
+    role: "system",
+    depth: 0,
+    order: 10,
+  },
+  strategy: {
+    type: "constant",
+    keys: [],
+    keys_secondary: { logic: "and_any", keys: [] },
+  },
+  probability: 100,
+  extra: {},
+};
+
+const dynEntry = {
+  uid: 2,
+  name: "Dyn/线索",
+  comment: "线索条目",
+  content: "隐藏线索：<%= charName %> 正在调查。",
+  enabled: false,
+  position: {
+    type: "before_character_definition",
+    role: "system",
+    depth: 0,
+    order: 20,
+  },
+  strategy: {
+    type: "selective",
+    keys: ["调查"],
+    keys_secondary: { logic: "and_any", keys: [] },
+  },
+  probability: 100,
+  extra: {},
+};
+
+const controllerEntry = {
+  uid: 3,
+  name: "EW/Controller/Main",
+  comment: "控制器",
+  content: '<%= await getwi("Dyn/线索") %>',
+  enabled: true,
+  position: {
+    type: "before_character_definition",
+    role: "system",
+    depth: 0,
+    order: 30,
+  },
+  strategy: {
+    type: "constant",
+    keys: [],
+    keys_secondary: { logic: "and_any", keys: [] },
+  },
+  probability: 100,
+  extra: {},
+};
+
+const atDepthEntry = {
+  uid: 4,
+  name: "深度注入",
+  comment: "深度注入",
+  content: "这是一条 atDepth 消息。",
+  enabled: true,
+  position: {
+    type: "at_depth_as_system",
+    role: "system",
+    depth: 2,
+    order: 5,
+  },
+  strategy: {
+    type: "constant",
+    keys: [],
+    keys_secondary: { logic: "and_any", keys: [] },
+  },
+  probability: 100,
+  extra: {},
+};
+
+try {
+  globalThis.SillyTavern = {
+    getContext() {
+      return {
+        name1: "User",
+        name2: "Alice",
+        chat: [{ is_user: true, mes: "我们继续调查那条线索" }],
+        chatMetadata: {},
+        extensionSettings: {},
+      };
+    },
+  };
+  globalThis.getCharWorldbookNames = () => ({
+    primary: "main-book",
+    additional: [],
+  });
+  globalThis.getWorldbook = async () => [
+    constantEntry,
+    dynEntry,
+    controllerEntry,
+    atDepthEntry,
+  ];
+  globalThis.getLorebookEntries = async () => [];
+
+  const { resolveTaskWorldInfo } = await import("../task-worldinfo.js");
+  const { buildTaskPrompt } = await import("../prompt-builder.js");
+
+  const worldInfo = await resolveTaskWorldInfo({
+    templateContext: {
+      recentMessages: "我们继续调查那条线索",
+      charName: "Alice",
+    },
+    userMessage: "继续调查",
+  });
+
+  assert.deepEqual(
+    worldInfo.beforeEntries.map((entry) => entry.name),
+    ["常驻设定", "EW/Controller/Main", "线索条目"],
+  );
+  assert.equal(worldInfo.additionalMessages.length, 1);
+  assert.equal(worldInfo.additionalMessages[0].content, "这是一条 atDepth 消息。");
+
+  const settings = {
+    taskProfiles: {
+      recall: {
+        activeProfileId: "custom",
+        profiles: [
+          {
+            id: "custom",
+            name: "测试预设",
+            taskType: "recall",
+            builtin: false,
+            blocks: [
+              {
+                id: "b1",
+                type: "builtin",
+                sourceKey: "worldInfoBefore",
+                role: "system",
+                enabled: true,
+                order: 0,
+                injectionMode: "append",
+              },
+              {
+                id: "b2",
+                type: "custom",
+                content: "角色: {{charName}}",
+                role: "user",
+                enabled: true,
+                order: 1,
+                injectionMode: "append",
+              },
+            ],
+          },
+        ],
+      },
+    },
+  };
+
+  const promptBuild = await buildTaskPrompt(settings, "recall", {
+    taskName: "recall",
+    userMessage: "继续调查",
+    recentMessages: "我们继续调查那条线索",
+    charName: "Alice",
+  });
+
+  assert.match(promptBuild.systemPrompt, /这里是常驻世界设定/);
+  assert.match(promptBuild.systemPrompt, /隐藏线索：Alice 正在调查/);
+  assert.equal(promptBuild.additionalMessages.length, 1);
+  assert.equal(promptBuild.additionalMessages[0].content, "这是一条 atDepth 消息。");
+
+  console.log("task-worldinfo tests passed");
+} finally {
+  if (originalSillyTavern === undefined) {
+    delete globalThis.SillyTavern;
+  } else {
+    globalThis.SillyTavern = originalSillyTavern;
+  }
+
+  if (originalGetCharWorldbookNames === undefined) {
+    delete globalThis.getCharWorldbookNames;
+  } else {
+    globalThis.getCharWorldbookNames = originalGetCharWorldbookNames;
+  }
+
+  if (originalGetWorldbook === undefined) {
+    delete globalThis.getWorldbook;
+  } else {
+    globalThis.getWorldbook = originalGetWorldbook;
+  }
+
+  if (originalGetLorebookEntries === undefined) {
+    delete globalThis.getLorebookEntries;
+  } else {
+    globalThis.getLorebookEntries = originalGetLorebookEntries;
+  }
+}