xiaoju
5ed6f68e4b
CI / check (pull_request) Successful in 1m42s
Agents now receive a Thread Progress section showing current step number and role visit count, eliminating tool calls to count turns. - util-agent: new buildThreadProgress() helper - agent-hermes: inject before continuation/first-visit prompt - agent-claude-code: same injection point Fixes #127
60 lines
1.8 KiB
TypeScript
60 lines
1.8 KiB
TypeScript
import type { StepContext } from "@united-workforce/protocol";
|
|
import { describe, expect, test } from "vitest";
|
|
import { buildThreadProgress } from "../src/build-thread-progress.js";
|
|
|
|
function makeStep(role: string): StepContext {
|
|
return {
|
|
role,
|
|
output: {},
|
|
detail: "0000000000000" as string,
|
|
agent: "uwf-mock",
|
|
edgePrompt: "",
|
|
startedAtMs: 0,
|
|
completedAtMs: 0,
|
|
cwd: "",
|
|
assembledPrompt: null,
|
|
usage: null,
|
|
content: null,
|
|
};
|
|
}
|
|
|
|
describe("buildThreadProgress", () => {
|
|
test("first step of thread", () => {
|
|
const result = buildThreadProgress([], "proponent");
|
|
expect(result).toContain("## Thread Progress");
|
|
expect(result).toContain("first step");
|
|
expect(result).toContain("first time");
|
|
expect(result).toContain("proponent");
|
|
});
|
|
|
|
test("second step, role not seen before", () => {
|
|
const steps = [makeStep("opponent")];
|
|
const result = buildThreadProgress(steps, "proponent");
|
|
expect(result).toContain("Thread step 2");
|
|
expect(result).toContain("spoken 0 times");
|
|
});
|
|
|
|
test("role has spoken once before", () => {
|
|
const steps = [makeStep("proponent"), makeStep("opponent")];
|
|
const result = buildThreadProgress(steps, "proponent");
|
|
expect(result).toContain("Thread step 3");
|
|
expect(result).toContain("spoken 1 time before");
|
|
// singular "time" not "times"
|
|
expect(result).not.toContain("1 times");
|
|
});
|
|
|
|
test("role has spoken multiple times", () => {
|
|
const steps = [
|
|
makeStep("proponent"),
|
|
makeStep("opponent"),
|
|
makeStep("proponent"),
|
|
makeStep("opponent"),
|
|
makeStep("proponent"),
|
|
makeStep("opponent"),
|
|
];
|
|
const result = buildThreadProgress(steps, "proponent");
|
|
expect(result).toContain("Thread step 7");
|
|
expect(result).toContain("spoken 3 times");
|
|
});
|
|
});
|