Files
boss/tests/master-agent-thread-status-prompt.test.ts

380 lines
15 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 readState: (typeof import("../src/lib/boss-data"))["readState"];
let writeState: (typeof import("../src/lib/boss-data"))["writeState"];
let saveAiAccount: (typeof import("../src/lib/boss-data"))["saveAiAccount"];
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 updateProjectAgentControls: (typeof import("../src/lib/boss-data"))["updateProjectAgentControls"];
let resolveMasterAgentExecutionConfig: (typeof import("../src/lib/boss-master-agent"))["resolveMasterAgentExecutionConfig"];
let replyToMasterAgentUserMessage: (typeof import("../src/lib/boss-master-agent"))["replyToMasterAgentUserMessage"];
async function setup() {
if (runtimeRoot) return;
runtimeRoot = await mkdtemp(path.join(os.tmpdir(), "boss-master-agent-thread-status-"));
process.env.BOSS_RUNTIME_ROOT = runtimeRoot;
process.env.BOSS_STATE_FILE = path.join(runtimeRoot, "boss-state.json");
const [data, masterAgent] = await Promise.all([
import("../src/lib/boss-data.ts"),
import("../src/lib/boss-master-agent.ts"),
]);
readState = data.readState;
writeState = data.writeState;
saveAiAccount = data.saveAiAccount;
updateMasterAgentPromptPolicy = data.updateMasterAgentPromptPolicy;
updateUserMasterPrompt = data.updateUserMasterPrompt;
createUserMasterMemory = data.createUserMasterMemory;
updateProjectAgentControls = data.updateProjectAgentControls;
resolveMasterAgentExecutionConfig = masterAgent.resolveMasterAgentExecutionConfig;
replyToMasterAgentUserMessage = masterAgent.replyToMasterAgentUserMessage;
}
test.after(async () => {
if (runtimeRoot) {
await rm(runtimeRoot, { recursive: true, force: true });
}
});
test("主 Agent 执行 prompt 命中线程时只读取相关状态文档和最近进展事件,不再常态注入深拉兜底", async () => {
await setup();
await saveAiAccount({
accountId: "master-codex-primary",
label: "主 GPT",
role: "primary",
provider: "master_codex_node",
displayName: "17600003315 · Master Codex Node",
nodeId: "mac-studio",
nodeLabel: "Mac Studio",
enabled: true,
setActive: true,
status: "ready",
loginStatusNote: "主节点可用。",
});
await updateMasterAgentPromptPolicy({
globalPrompt: "管理员全局主提示词",
updatedBy: "17600003315",
});
await updateUserMasterPrompt("17600003315", "用户私有主提示词");
await updateProjectAgentControls("master-agent", {
promptOverride: "当前对话提示词",
});
await createUserMasterMemory({
account: "17600003315",
scope: "project",
projectId: "master-agent",
title: "项目记忆",
content: "项目记忆正文",
memoryType: "project_progress",
tags: ["线程状态"],
});
await createUserMasterMemory({
account: "master-codex-primary",
scope: "project",
projectId: "master-agent",
title: "项目记忆",
content: "项目记忆正文",
memoryType: "project_progress",
tags: ["线程状态"],
});
const state = await readState();
const auditProject = state.projects.find((project) => project.id === "audit-collab");
assert.ok(auditProject, "expected seeded audit-collab project");
const masterProject = state.projects.find((project) => project.id === "master-agent");
assert.ok(masterProject, "expected seeded master-agent project");
auditProject!.projectUnderstanding = {
projectGoal: "深拉兜底目标",
currentProgress: "深拉兜底进度",
technicalArchitecture: "深拉兜底架构",
currentBlockers: "深拉兜底阻塞",
recommendedNextStep: "深拉兜底下一步",
sourceTaskId: "task-deep-pull",
updatedAt: "2026-04-04T18:00:00+08:00",
sourceKind: "thread_sync",
};
masterProject!.projectUnderstanding = {
projectGoal: "主 Agent 旧目标",
currentProgress: "主 Agent 旧进度",
technicalArchitecture: "主 Agent 旧架构",
currentBlockers: "主 Agent 旧阻塞",
recommendedNextStep: "主 Agent 旧下一步",
sourceTaskId: "task-master-legacy",
updatedAt: "2026-04-04T17:50:00+08:00",
sourceKind: "thread_sync",
};
state.threadStatusDocuments = [
{
documentId: "thread-status-doc-master",
projectId: "master-agent",
threadId: "thread-master-main",
threadDisplayName: "主 Agent 汇总",
folderName: "主控线程",
deviceId: "mac-studio",
projectGoal: "主 Agent 额外状态目标",
currentPhase: "主 Agent 额外阶段",
currentProgress: "主 Agent 额外进度",
technicalArchitecture: "主 Agent 额外架构",
currentBlockers: "主 Agent 额外阻塞",
recommendedNextStep: "主 Agent 额外下一步",
keyFiles: ["src/lib/boss-master-agent.ts"],
keyCommands: ["npm run lint"],
updatedAt: "2026-04-04T18:03:00+08:00",
sourceTaskId: "task-master-status",
sourceKind: "incremental_sync",
},
{
documentId: "thread-status-doc-1",
projectId: "audit-collab",
threadId: "thread-audit-chief",
threadDisplayName: "审计对话",
folderName: "审计群聊",
deviceId: "mac-studio",
projectGoal: "线程状态目标",
currentPhase: "线程状态阶段",
currentProgress: "线程状态进度",
technicalArchitecture: "线程状态架构",
currentBlockers: "线程状态阻塞",
recommendedNextStep: "线程状态下一步",
keyFiles: ["src/lib/boss-master-agent.ts"],
keyCommands: ["npm run build"],
updatedAt: "2026-04-04T18:01:00+08:00",
sourceTaskId: "task-thread-status",
sourceKind: "incremental_sync",
},
];
state.threadProgressEvents = [
{
eventId: "thread-progress-event-master",
projectId: "master-agent",
threadId: "thread-master-main",
threadDisplayName: "主 Agent 汇总",
deviceId: "mac-studio",
eventType: "progress_updated",
summary: "主 Agent 额外进展摘要",
phase: "主 Agent 额外阶段",
blockerDelta: "主 Agent 额外阻塞",
nextStepDelta: "主 Agent 额外下一步",
createdAt: "2026-04-04T18:03:30+08:00",
sourceTaskId: "task-master-progress",
},
{
eventId: "thread-progress-event-1",
projectId: "audit-collab",
threadId: "thread-audit-chief",
threadDisplayName: "审计对话",
deviceId: "mac-studio",
eventType: "progress_updated",
summary: "最近进展事件摘要",
phase: "线程状态阶段",
blockerDelta: "线程状态阻塞",
nextStepDelta: "线程状态下一步",
createdAt: "2026-04-04T18:02:00+08:00",
sourceTaskId: "task-thread-progress",
},
];
await writeState(state);
const resolved = await resolveMasterAgentExecutionConfig(
"master-agent",
"17600003315",
"审计对话,请继续推进线程状态同步",
);
assert.ok(resolved.projectMemories.length > 0);
assert.equal(resolved.projectMemories[0]?.content, "项目记忆正文");
assert.ok(resolved.executionPrompt.includes("当前对话提示词"));
assert.ok(
resolved.executionPrompt.indexOf("管理员全局主提示词:") <
resolved.executionPrompt.indexOf("用户私有主提示词:") &&
resolved.executionPrompt.indexOf("用户私有主提示词:") <
resolved.executionPrompt.indexOf("当前对话附加提示词:") &&
resolved.executionPrompt.indexOf("当前对话附加提示词:") <
resolved.executionPrompt.indexOf("当前消息:"),
);
const reply = await replyToMasterAgentUserMessage({
requestText: "审计对话,请继续推进线程状态同步",
requestedBy: "Boss 超级管理员",
requestedByAccount: "17600003315",
mode: "enqueue",
});
assert.equal(reply.ok, true);
assert.equal(reply.masterReplyState, "queued");
const queuedTask = (await readState()).masterAgentTasks.find(
(task) => task.projectId === "master-agent" && task.requestText === "审计对话,请继续推进线程状态同步",
);
assert.ok(queuedTask, "expected master-agent task to be queued");
assert.ok(queuedTask?.executionPrompt.includes("线程状态文档:"));
assert.ok(queuedTask?.executionPrompt.includes("线程状态目标"));
assert.ok(!queuedTask?.executionPrompt.includes("主 Agent 额外状态目标"));
assert.ok(queuedTask?.executionPrompt.includes("最近进展事件:"));
assert.ok(queuedTask?.executionPrompt.includes("最近进展事件摘要"));
assert.ok(!queuedTask?.executionPrompt.includes("主 Agent 额外进展摘要"));
assert.ok(!queuedTask?.executionPrompt.includes("深拉兜底目标"));
assert.ok(!queuedTask?.executionPrompt.includes("关键时刻深拉线程兜底:"));
});
test("主 Agent 执行 prompt 在未命中时退回最近活跃项目,且不会常态注入深拉兜底", async () => {
await setup();
const state = await readState();
const auditProject = state.projects.find((project) => project.id === "audit-collab");
const masterProject = state.projects.find((project) => project.id === "master-agent");
assert.ok(auditProject, "expected seeded audit-collab project");
assert.ok(masterProject, "expected seeded master-agent project");
auditProject!.projectUnderstanding = {
projectGoal: "审计兜底目标",
currentProgress: "审计兜底进度",
technicalArchitecture: "审计兜底架构",
currentBlockers: "审计兜底阻塞",
recommendedNextStep: "审计兜底下一步",
sourceTaskId: "task-audit-fallback",
updatedAt: "2026-04-04T17:55:00+08:00",
sourceKind: "thread_sync",
};
masterProject!.projectUnderstanding = {
projectGoal: "最近活跃目标",
currentProgress: "最近活跃进度",
technicalArchitecture: "最近活跃架构",
currentBlockers: "最近活跃阻塞",
recommendedNextStep: "最近活跃下一步",
sourceTaskId: "task-master-active",
updatedAt: "2026-04-04T18:05:00+08:00",
sourceKind: "thread_sync",
};
state.threadStatusDocuments = [
{
documentId: "thread-status-doc-audit-fallback",
projectId: "audit-collab",
threadId: "thread-audit-chief",
threadDisplayName: "审计对话",
folderName: "审计群聊",
deviceId: "mac-studio",
projectGoal: "审计兜底状态",
currentPhase: "审计兜底阶段",
currentProgress: "审计兜底进度",
technicalArchitecture: "审计兜底架构",
currentBlockers: "审计兜底阻塞",
recommendedNextStep: "审计兜底下一步",
keyFiles: ["src/lib/boss-master-agent.ts"],
keyCommands: ["npm run build"],
updatedAt: "2026-04-04T17:56:00+08:00",
sourceTaskId: "task-audit-status",
sourceKind: "incremental_sync",
},
{
documentId: "thread-status-doc-master-fallback",
projectId: "master-agent",
threadId: "thread-master-main",
threadDisplayName: "主 Agent 汇总",
folderName: "主控线程",
deviceId: "mac-studio",
projectGoal: "最近活跃状态",
currentPhase: "最近活跃阶段",
currentProgress: "最近活跃进度",
technicalArchitecture: "最近活跃架构",
currentBlockers: "最近活跃阻塞",
recommendedNextStep: "最近活跃下一步",
keyFiles: ["src/lib/boss-data.ts"],
keyCommands: ["npm run lint"],
updatedAt: "2026-04-04T18:06:00+08:00",
sourceTaskId: "task-master-status-fallback",
sourceKind: "incremental_sync",
},
];
state.threadProgressEvents = [
{
eventId: "thread-progress-event-audit-fallback",
projectId: "audit-collab",
threadId: "thread-audit-chief",
threadDisplayName: "审计对话",
deviceId: "mac-studio",
eventType: "progress_updated",
summary: "审计兜底进展摘要",
phase: "审计兜底阶段",
blockerDelta: "审计兜底阻塞",
nextStepDelta: "审计兜底下一步",
createdAt: "2026-04-04T17:56:30+08:00",
sourceTaskId: "task-audit-progress",
},
{
eventId: "thread-progress-event-master-fallback",
projectId: "master-agent",
threadId: "thread-master-main",
threadDisplayName: "主 Agent 汇总",
deviceId: "mac-studio",
eventType: "progress_updated",
summary: "最近活跃进展摘要",
phase: "最近活跃阶段",
blockerDelta: "最近活跃阻塞",
nextStepDelta: "最近活跃下一步",
createdAt: "2026-04-04T18:06:30+08:00",
sourceTaskId: "task-master-progress-fallback",
},
];
await writeState(state);
const reply = await replyToMasterAgentUserMessage({
requestText: "请继续推进线程状态同步(仅深拉兜底)",
requestedBy: "Boss 超级管理员",
requestedByAccount: "17600003315",
mode: "enqueue",
});
assert.equal(reply.ok, true);
const queuedTask = (await readState()).masterAgentTasks.find(
(task) => task.projectId === "master-agent" && task.requestText === "请继续推进线程状态同步(仅深拉兜底)",
);
assert.ok(queuedTask, "expected master-agent task to be queued");
assert.ok(queuedTask?.executionPrompt.includes("最近活跃状态"));
assert.ok(queuedTask?.executionPrompt.includes("最近活跃进展摘要"));
assert.ok(!queuedTask?.executionPrompt.includes("关键时刻深拉线程兜底:"));
});
test("主 Agent 执行 prompt 在没有线程状态文档和进展事件时才会注入深拉兜底", async () => {
await setup();
const state = await readState();
const auditProject = state.projects.find((project) => project.id === "audit-collab");
assert.ok(auditProject, "expected seeded audit-collab project");
auditProject!.projectUnderstanding = {
projectGoal: "深拉兜底目标",
currentProgress: "深拉兜底进度",
technicalArchitecture: "深拉兜底架构",
currentBlockers: "深拉兜底阻塞",
recommendedNextStep: "深拉兜底下一步",
sourceTaskId: "task-deep-pull",
updatedAt: "2026-04-04T18:00:00+08:00",
sourceKind: "thread_sync",
};
state.threadStatusDocuments = [];
state.threadProgressEvents = [];
await writeState(state);
const reply = await replyToMasterAgentUserMessage({
requestText: "请继续推进线程状态同步",
requestedBy: "Boss 超级管理员",
requestedByAccount: "17600003315",
mode: "enqueue",
});
assert.equal(reply.ok, true);
const queuedTask = (await readState()).masterAgentTasks.find(
(task) => task.projectId === "master-agent" && task.requestText === "请继续推进线程状态同步",
);
assert.ok(queuedTask, "expected master-agent task to be queued");
assert.ok(queuedTask?.executionPrompt.includes("关键时刻深拉线程兜底:"));
assert.ok(queuedTask?.executionPrompt.includes("深拉兜底目标"));
});