Initial commit

tests/worker-llm.test.ts

@@ -0,0 +1,100 @@
+import { describe, expect, it } from "vitest";
+import {
+  filterInputsForRolePerspective,
+  isRoleUserOnlyTag,
+} from "../src/skills/worker-llm.js";
+import type { ParsedWorkerSkill } from "../src/skills/types.js";
+import { resolveWorkerLlmProvider } from "../src/skills/worker-llm.js";
+import { MockLlmProvider } from "../src/llm/client.js";
+
+describe("filterInputsForRolePerspective", () => {
+  it("keeps only current role tags when 世界.当前角色.id is set", () => {
+    const inputs = {
+      "世界.当前角色.id": "A",
+      "角色.A.可见信息": "hand A",
+      "角色.B.可见信息": "hand B",
+      "场景.公开叙述": "public",
+    };
+    const filtered = filterInputsForRolePerspective(inputs, {
+      "世界.当前角色.id": "A",
+    });
+    expect(filtered["角色.A.可见信息"]).toBe("hand A");
+    expect(filtered["角色.B.可见信息"]).toBeUndefined();
+    expect(filtered["场景.公开叙述"]).toBe("public");
+  });
+
+  it("strips thinking tags even for current role", () => {
+    const inputs = {
+      "角色.A.可见信息": "hand A",
+      "角色.A.思考": "inner monologue",
+      "角色.A.推理.候选": "legacy thinking",
+      "角色.B.行动": "should not see",
+    };
+    const filtered = filterInputsForRolePerspective(inputs, {
+      "世界.当前角色.id": "A",
+    });
+    expect(filtered["角色.A.可见信息"]).toBe("hand A");
+    expect(filtered["角色.A.思考"]).toBeUndefined();
+    expect(filtered["角色.A.推理.候选"]).toBeUndefined();
+    expect(filtered["角色.B.行动"]).toBeUndefined();
+  });
+
+  it("isRoleUserOnlyTag detects thinking suffixes", () => {
+    expect(isRoleUserOnlyTag("角色.A.思考")).toBe(true);
+    expect(isRoleUserOnlyTag("角色.A.推理.候选")).toBe(true);
+    expect(isRoleUserOnlyTag("角色.A.行动")).toBe(false);
+  });
+});
+
+describe("resolveWorkerLlmProvider", () => {
+  const fallback = new MockLlmProvider([]);
+
+  it("returns fallback when no binding", () => {
+    const worker: ParsedWorkerSkill = {
+      id: "role-decide",
+      skill: "roleplay-game-theory",
+      name: "x",
+      description: "",
+      version: 1,
+      inputTags: [],
+      outputTags: [],
+      path: "",
+      body: "",
+    };
+    expect(
+      resolveWorkerLlmProvider({
+        worker,
+        slots: { "世界.当前角色.id": "A" },
+        fallbackLlm: fallback,
+      }),
+    ).toBe(fallback);
+  });
+
+  it("uses byRole when bindings match", () => {
+    const worker: ParsedWorkerSkill = {
+      id: "role-decide",
+      skill: "roleplay-game-theory",
+      name: "x",
+      description: "",
+      version: 1,
+      inputTags: [],
+      outputTags: [],
+      path: "",
+      body: "",
+    };
+    const resolved = resolveWorkerLlmProvider({
+      worker,
+      bindings: {
+        workers: {
+          "role-decide": {
+            byRole: { A: "nonexistent-profile-id" },
+          },
+        },
+      },
+      slots: { "世界.当前角色.id": "A" },
+      fallbackLlm: fallback,
+    });
+    // profile 不存在时回退 fallback
+    expect(resolved).toBe(fallback);
+  });
+});