67 lines
2.3 KiB
TypeScript
67 lines
2.3 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import {
|
|
createMockMainAgentResponse,
|
|
createMockToolCall,
|
|
MockLlmProvider,
|
|
} from "../src/llm/client.js";
|
|
import { runMainAgentToolLoop } from "../src/main-agent/tool-loop.js";
|
|
import { toolCallToDecision } from "../src/runtime/tool-registry.js";
|
|
import { createSession } from "../src/runtime/phase-machine.js";
|
|
|
|
describe("main agent tool loop", () => {
|
|
const baseContext = {
|
|
session: createSession("default"),
|
|
blackboardIndex: [{ id: "1", tag: "book.brief", source: "user" }],
|
|
availableWorkers: [{ id: "outline", description: "生成大纲" }],
|
|
};
|
|
|
|
const handlers = {
|
|
readBlackboard: (tags: string[]) =>
|
|
Object.fromEntries(tags.map((t) => [t, t === "book.brief" ? "科幻中篇" : ""])),
|
|
listWorkers: () => [{ id: "outline", description: "生成大纲" }],
|
|
listArtifacts: () => [],
|
|
};
|
|
|
|
it("runs loop tools then returns terminal decision", async () => {
|
|
const llm = new MockLlmProvider([
|
|
createMockToolCall("read_blackboard", { tags: ["book.brief"] }),
|
|
createMockToolCall("run_worker", {
|
|
workerId: "outline",
|
|
reason: "需求已齐，生成大纲",
|
|
requiresApproval: true,
|
|
}),
|
|
]);
|
|
|
|
const result = await runMainAgentToolLoop(llm, baseContext, handlers);
|
|
expect(result.iterations).toBe(2);
|
|
expect(result.toolTrace).toEqual(["read_blackboard", "run_worker (terminal)"]);
|
|
expect(result.decision.action).toBe("run_worker");
|
|
expect(result.decision.workerId).toBe("outline");
|
|
expect(result.decision.requiresApproval).toBe(true);
|
|
});
|
|
|
|
it("falls back to JSON content when no tool calls", async () => {
|
|
const llm = new MockLlmProvider([
|
|
createMockMainAgentResponse({
|
|
action: "ask_user",
|
|
reason: "请补充篇幅",
|
|
requiresApproval: false,
|
|
}),
|
|
]);
|
|
|
|
const result = await runMainAgentToolLoop(llm, baseContext, handlers);
|
|
expect(result.decision.action).toBe("ask_user");
|
|
expect(result.decision.reason).toContain("篇幅");
|
|
});
|
|
|
|
it("maps finish tool to decision", () => {
|
|
const decision = toolCallToDecision({
|
|
id: "c1",
|
|
name: "finish",
|
|
arguments: JSON.stringify({ reason: "完成" }),
|
|
});
|
|
expect(decision.action).toBe("finish");
|
|
expect(decision.reason).toBe("完成");
|
|
});
|
|
});
|