xiaoju
6b7636b088
CI / check (pull_request) Failing after 3m6s
Breaking changes: - UWF_STORAGE_ROOT → UWF_HOME - WORKFLOW_STORAGE_ROOT removed (no fallback) - OCAS_DIR → OCAS_HOME (aligned with ocas CLI) Library functions no longer read process.env: - util-agent/storage.ts: resolveStorageRoot(override), getGlobalCasDir(override) - agent-hermes: isResumeDisabled(flag) pure function, CLI reads env - agent-claude-code: CLI reads CLAUDE_MODEL and passes to agent Fixes #37
106 lines
3.5 KiB
TypeScript
106 lines
3.5 KiB
TypeScript
import type { ThreadId } from "@united-workforce/protocol";
|
|
import type { AgentContext } from "@united-workforce/util-agent";
|
|
import { describe, expect, test } from "vitest";
|
|
import { buildClaudeCodePrompt } from "../src/claude-code.js";
|
|
|
|
function makeCtx(overrides: Partial<AgentContext> = {}): AgentContext {
|
|
return {
|
|
threadId: "01JTEST0000000000000000000" as ThreadId,
|
|
edgePrompt: "Proceed with the assigned role.",
|
|
isFirstVisit: true,
|
|
workflow: {
|
|
roles: {
|
|
developer: {
|
|
description: "TDD implementation per test spec",
|
|
goal: "Write code",
|
|
capabilities: ["coding"],
|
|
procedure: "1. Read spec\n2. Write code",
|
|
output: "List files changed",
|
|
frontmatter: "",
|
|
},
|
|
},
|
|
conditions: {},
|
|
graph: {},
|
|
},
|
|
role: "developer",
|
|
start: { prompt: "Fix the bug", workflowHash: "abc123", threadId: "t1" },
|
|
steps: [],
|
|
store: {} as AgentContext["store"],
|
|
outputFormatInstruction: "Use YAML frontmatter",
|
|
storageRoot: "/tmp/uwf-test",
|
|
casDir: "/tmp/ocas-test",
|
|
...overrides,
|
|
};
|
|
}
|
|
|
|
describe("buildClaudeCodePrompt", () => {
|
|
test("assembles outputFormatInstruction + role prompt + task prompt", () => {
|
|
const result = buildClaudeCodePrompt(makeCtx());
|
|
expect(result).toMatch(/^Use YAML frontmatter/);
|
|
expect(result).toContain("Write code");
|
|
expect(result).toContain("## Task\nFix the bug");
|
|
});
|
|
|
|
test("includes previous steps with content on first visit", () => {
|
|
const ctx = makeCtx({
|
|
steps: [
|
|
{
|
|
role: "planner",
|
|
output: '{"plan":"do X"}',
|
|
agent: "hermes",
|
|
detail: "detail-1",
|
|
edgePrompt: "Create a plan.",
|
|
content: "Here is my detailed plan for doing X.",
|
|
},
|
|
],
|
|
});
|
|
const result = buildClaudeCodePrompt(ctx);
|
|
expect(result).toContain("## What Happened Since Your Last Turn");
|
|
expect(result).toContain("Step 1: planner");
|
|
expect(result).toContain("do X");
|
|
// First visit should include step content
|
|
expect(result).toContain("Here is my detailed plan for doing X.");
|
|
});
|
|
|
|
test("re-entry shows steps since last visit without content", () => {
|
|
const ctx = makeCtx({
|
|
isFirstVisit: false,
|
|
steps: [
|
|
{
|
|
role: "developer",
|
|
output: '{"status":"done"}',
|
|
agent: "claude-code",
|
|
detail: "detail-1",
|
|
edgePrompt: "Implement.",
|
|
content: "I implemented everything.",
|
|
},
|
|
{
|
|
role: "reviewer",
|
|
output: '{"approved":false}',
|
|
agent: "claude-code",
|
|
detail: "detail-2",
|
|
edgePrompt: "Review.",
|
|
content: "Rejected: complexity too high, refactor cmdStepRead.",
|
|
},
|
|
],
|
|
});
|
|
const result = buildClaudeCodePrompt(ctx);
|
|
expect(result).toContain("## What Happened Since Your Last Turn");
|
|
expect(result).toContain("reviewer");
|
|
expect(result).toContain("approved");
|
|
});
|
|
|
|
test("omits history section when steps array is empty", () => {
|
|
const result = buildClaudeCodePrompt(makeCtx({ steps: [] }));
|
|
expect(result).not.toContain("## What Happened Since Your Last Turn");
|
|
expect(result).toContain("## Current Instruction");
|
|
});
|
|
|
|
test("works without outputFormatInstruction", () => {
|
|
const result = buildClaudeCodePrompt(makeCtx({ outputFormatInstruction: "" }));
|
|
expect(result).not.toMatch(/^\s*\n/);
|
|
expect(result).toContain("Write code");
|
|
expect(result).toContain("## Task");
|
|
});
|
|
});
|