239 lines
7.9 KiB
TypeScript
239 lines
7.9 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import { mkdtemp, rm } from "node:fs/promises";
|
|
|
|
let runtimeRoot = "";
|
|
let saveAiAccount: (typeof import("../src/lib/boss-data"))["saveAiAccount"];
|
|
let updateProjectAgentControls: (typeof import("../src/lib/boss-data"))["updateProjectAgentControls"];
|
|
let updateMasterAgentPromptPolicy: (typeof import("../src/lib/boss-data"))["updateMasterAgentPromptPolicy"];
|
|
let updateUserMasterPrompt: (typeof import("../src/lib/boss-data"))["updateUserMasterPrompt"];
|
|
let createUserMasterMemory: (typeof import("../src/lib/boss-data"))["createUserMasterMemory"];
|
|
let resolveMasterAgentExecutionConfig: (typeof import("../src/lib/boss-master-agent"))["resolveMasterAgentExecutionConfig"];
|
|
let stateFile = "";
|
|
let stateBackupFile = "";
|
|
|
|
async function setup() {
|
|
if (runtimeRoot) return;
|
|
|
|
runtimeRoot = await mkdtemp(path.join(os.tmpdir(), "boss-master-agent-config-"));
|
|
process.env.BOSS_RUNTIME_ROOT = runtimeRoot;
|
|
process.env.BOSS_STATE_FILE = path.join(runtimeRoot, "boss-state.json");
|
|
stateFile = process.env.BOSS_STATE_FILE;
|
|
stateBackupFile = `${stateFile}.bak`;
|
|
|
|
const [data, masterAgent] = await Promise.all([
|
|
import("../src/lib/boss-data.ts"),
|
|
import("../src/lib/boss-master-agent.ts"),
|
|
]);
|
|
|
|
saveAiAccount = data.saveAiAccount;
|
|
updateProjectAgentControls = data.updateProjectAgentControls;
|
|
updateMasterAgentPromptPolicy = data.updateMasterAgentPromptPolicy;
|
|
updateUserMasterPrompt = data.updateUserMasterPrompt;
|
|
createUserMasterMemory = data.createUserMasterMemory;
|
|
resolveMasterAgentExecutionConfig = masterAgent.resolveMasterAgentExecutionConfig;
|
|
}
|
|
|
|
async function resetState() {
|
|
if (!stateFile) {
|
|
return;
|
|
}
|
|
await Promise.all([
|
|
rm(stateFile, { force: true }),
|
|
rm(stateBackupFile, { force: true }),
|
|
]);
|
|
}
|
|
|
|
test.beforeEach(async () => {
|
|
await setup();
|
|
await resetState();
|
|
});
|
|
|
|
test.after(async () => {
|
|
if (runtimeRoot) {
|
|
await rm(runtimeRoot, { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
test("当前对话 override 优先于主控账号默认值", async () => {
|
|
await saveAiAccount({
|
|
accountId: "master-codex-primary",
|
|
label: "主 GPT",
|
|
role: "primary",
|
|
provider: "master_codex_node",
|
|
displayName: "Mac 上的 Master Codex Node",
|
|
nodeId: "local-codex-node",
|
|
nodeLabel: "本机 Codex",
|
|
model: "gpt-4.1-mini",
|
|
enabled: true,
|
|
setActive: true,
|
|
loginStatusNote: "通过绑定的 Master Codex Node 对话。",
|
|
});
|
|
|
|
await updateProjectAgentControls("master-agent", {
|
|
modelOverride: "gpt-5.4",
|
|
reasoningEffortOverride: "high",
|
|
});
|
|
|
|
assert.equal(typeof resolveMasterAgentExecutionConfig, "function");
|
|
|
|
const resolved = await resolveMasterAgentExecutionConfig("master-agent");
|
|
|
|
assert.equal(resolved.model, "gpt-5.4");
|
|
assert.equal(resolved.reasoningEffort, "high");
|
|
assert.equal(resolved.account.accountId, "master-codex-primary");
|
|
assert.equal(resolved.account.model, "gpt-4.1-mini");
|
|
});
|
|
|
|
test("主 Agent 模型策略会按聊天与深度任务选择不同默认模型", async () => {
|
|
await saveAiAccount({
|
|
accountId: "master-codex-primary",
|
|
label: "主 GPT",
|
|
role: "primary",
|
|
provider: "openai_api",
|
|
displayName: "OpenAI 主控",
|
|
model: "gpt-5.4",
|
|
apiKey: "sk-test-master-agent-policy",
|
|
enabled: true,
|
|
setActive: true,
|
|
loginStatusNote: "用于模型策略测试。",
|
|
});
|
|
|
|
await updateProjectAgentControls("master-agent", {
|
|
fastModelOverride: "gpt-5.4-mini",
|
|
fastReasoningEffortOverride: "low",
|
|
smartModelOverride: "gpt-5.4",
|
|
smartReasoningEffortOverride: "high",
|
|
});
|
|
|
|
const chatResolved = await resolveMasterAgentExecutionConfig(
|
|
"master-agent",
|
|
"17600003315",
|
|
"帮我看一下当前状态",
|
|
"chat",
|
|
);
|
|
assert.equal(chatResolved.model, "gpt-5.4-mini");
|
|
assert.equal(chatResolved.reasoningEffort, "low");
|
|
assert.equal(chatResolved.modelPolicy.mode, "fast");
|
|
|
|
const deepResolved = await resolveMasterAgentExecutionConfig(
|
|
"master-agent",
|
|
"17600003315",
|
|
"深度理解当前项目进度",
|
|
"deep_task",
|
|
);
|
|
assert.equal(deepResolved.model, "gpt-5.4");
|
|
assert.equal(deepResolved.reasoningEffort, "high");
|
|
assert.equal(deepResolved.modelPolicy.mode, "smart");
|
|
|
|
await updateProjectAgentControls("master-agent", {
|
|
modelOverride: "gpt-4.1",
|
|
reasoningEffortOverride: "medium",
|
|
});
|
|
const forcedResolved = await resolveMasterAgentExecutionConfig(
|
|
"master-agent",
|
|
"17600003315",
|
|
"深度理解当前项目进度",
|
|
"deep_task",
|
|
);
|
|
assert.equal(forcedResolved.model, "gpt-4.1");
|
|
assert.equal(forcedResolved.reasoningEffort, "medium");
|
|
assert.equal(forcedResolved.modelPolicy.mode, "manual_override");
|
|
});
|
|
|
|
test("主 Agent 执行配置会合成管理员提示词、用户提示词和当前对话提示词", async () => {
|
|
await saveAiAccount({
|
|
accountId: "master-codex-primary",
|
|
label: "主 GPT",
|
|
role: "primary",
|
|
provider: "master_codex_node",
|
|
displayName: "Mac 上的 Master Codex Node",
|
|
nodeId: "local-codex-node",
|
|
nodeLabel: "本机 Codex",
|
|
model: "gpt-4.1-mini",
|
|
enabled: true,
|
|
setActive: true,
|
|
loginStatusNote: "通过绑定的 Master Codex Node 对话。",
|
|
});
|
|
|
|
await updateMasterAgentPromptPolicy({
|
|
globalPrompt: "全局主提示词",
|
|
updatedBy: "17600003315",
|
|
});
|
|
await updateUserMasterPrompt("17600003315", "用户私有主提示词");
|
|
await updateProjectAgentControls("master-agent", {
|
|
modelOverride: "gpt-5.4",
|
|
reasoningEffortOverride: "high",
|
|
promptOverride: "当前对话提示词",
|
|
});
|
|
|
|
const resolved = await resolveMasterAgentExecutionConfig("master-agent");
|
|
|
|
assert.equal(resolved.promptPolicy?.globalPrompt, "全局主提示词");
|
|
assert.equal(resolved.userPrompt?.content, "用户私有主提示词");
|
|
assert.equal(resolved.projectPromptOverride, "当前对话提示词");
|
|
assert.equal(resolved.promptPolicy?.updatedBy, "17600003315");
|
|
});
|
|
|
|
test("主 Agent 执行 prompt 会明确声明管理员全局提示词不可覆盖，并带出项目记忆来源", async () => {
|
|
await saveAiAccount({
|
|
accountId: "master-codex-primary",
|
|
label: "主 GPT",
|
|
role: "primary",
|
|
provider: "master_codex_node",
|
|
displayName: "Mac 上的 Master Codex Node",
|
|
nodeId: "local-codex-node",
|
|
nodeLabel: "本机 Codex",
|
|
model: "gpt-4.1-mini",
|
|
enabled: true,
|
|
setActive: true,
|
|
loginStatusNote: "通过绑定的 Master Codex Node 对话。",
|
|
});
|
|
|
|
await updateMasterAgentPromptPolicy({
|
|
globalPrompt: "系统级主提示词",
|
|
updatedBy: "17600003315",
|
|
});
|
|
await updateUserMasterPrompt("17600003315", "用户私有主提示词");
|
|
await updateProjectAgentControls("master-agent", {
|
|
promptOverride: "当前对话提示词",
|
|
});
|
|
await createUserMasterMemory({
|
|
account: "17600003315",
|
|
scope: "project",
|
|
projectId: "boss-console",
|
|
title: "boss 项目进度",
|
|
content: "boss 项目当前按项目聚合加线程下钻展示。",
|
|
memoryType: "project_progress",
|
|
tags: ["boss", "会话"],
|
|
});
|
|
await createUserMasterMemory({
|
|
account: "17600003315",
|
|
scope: "project",
|
|
projectId: "project-wenshenapp",
|
|
title: "wenshenapp 项目进度",
|
|
content: "wenshenapp 当前只有一个主线程。",
|
|
memoryType: "project_progress",
|
|
tags: ["wenshenapp"],
|
|
});
|
|
|
|
const resolved = await resolveMasterAgentExecutionConfig(
|
|
"master-agent",
|
|
"17600003315",
|
|
"继续推进 boss 项目的会话归档逻辑",
|
|
);
|
|
|
|
assert.equal(
|
|
resolved.executionPrompt,
|
|
[
|
|
"管理员全局主提示词：\n系统级主提示词",
|
|
"用户私有主提示词：\n用户私有主提示词",
|
|
"当前对话附加提示词：\n当前对话提示词",
|
|
"项目记忆：\n- [boss-console] boss 项目进度: boss 项目当前按项目聚合加线程下钻展示。",
|
|
"当前消息：\n继续推进 boss 项目的会话归档逻辑",
|
|
].join("\n\n"),
|
|
);
|
|
});
|