64 lines
2.0 KiB
TypeScript
64 lines
2.0 KiB
TypeScript
import assert from "node:assert/strict";
|
|
import test from "node:test";
|
|
import {
|
|
buildExecutionPromptForTesting,
|
|
resolveRelevantMemoriesForTesting,
|
|
} from "@/lib/execution/prompt-assembler";
|
|
|
|
test("PromptAssembler 会按固定顺序拼管理员提示词、用户提示词、对话提示词和记忆", () => {
|
|
const prompt = buildExecutionPromptForTesting({
|
|
globalPrompt: "GLOBAL",
|
|
userPrompt: "USER",
|
|
conversationPrompt: "CONVERSATION",
|
|
projectMemories: [{ title: "项目记忆", content: "PROJECT", projectId: "boss-console", tags: [] }],
|
|
userMemories: [{ title: "用户记忆", content: "USER_MEMORY", tags: [] }],
|
|
requestText: "继续",
|
|
});
|
|
|
|
assert.equal(
|
|
prompt,
|
|
[
|
|
"管理员全局主提示词：\nGLOBAL",
|
|
"用户私有主提示词：\nUSER",
|
|
"当前对话附加提示词：\nCONVERSATION",
|
|
"项目记忆：\n- [boss-console] 项目记忆: PROJECT",
|
|
"用户通用记忆：\n- 用户记忆: USER_MEMORY",
|
|
"当前消息：\n继续",
|
|
].join("\n\n"),
|
|
);
|
|
});
|
|
|
|
test("PromptAssembler 会透出可测试的记忆筛选器", () => {
|
|
const resolved = resolveRelevantMemoriesForTesting({
|
|
projectId: "master-agent",
|
|
requestText: "boss-console 的审批流",
|
|
memories: [
|
|
{
|
|
memoryId: "m1",
|
|
scope: "project",
|
|
projectId: "boss-console",
|
|
title: "审批流",
|
|
content: "boss-console approval",
|
|
tags: ["approval"],
|
|
memoryType: "project_progress",
|
|
createdAt: "2026-01-01T00:00:00.000Z",
|
|
updatedAt: "2026-01-01T00:00:00.000Z",
|
|
},
|
|
{
|
|
memoryId: "m2",
|
|
scope: "project",
|
|
projectId: "wenshenapp",
|
|
title: "UI",
|
|
content: "wechat ui",
|
|
tags: ["ui"],
|
|
memoryType: "project_progress",
|
|
createdAt: "2026-01-02T00:00:00.000Z",
|
|
updatedAt: "2026-01-02T00:00:00.000Z",
|
|
},
|
|
],
|
|
});
|
|
|
|
assert.equal(resolved.projectMemories.length, 1);
|
|
assert.equal(resolved.projectMemories[0]?.projectId, "boss-console");
|
|
});
|