小橘
6a2d6b0627
attempt is a deterministic fact (count of tester steps), not something
an LLM should guess. Removed from tester meta schema, moderator counts
it directly from context.steps.
小橘 🍊（NEKO Team）
34 lines
1.1 KiB
TypeScript
34 lines
1.1 KiB
TypeScript
import type { WorkflowDefinition } from "@uncaged/nerve-core";
|
|
import { END } from "@uncaged/nerve-core";
|
|
import { buildPlannerRole } from "./roles/planner/index.js";
|
|
import { buildCoderRole } from "./roles/coder/index.js";
|
|
import { buildTesterRole } from "./roles/tester/index.js";
|
|
|
|
import type { SenseMeta } from "./roles/types.js";
|
|
|
|
function countRole(steps: { role: string }[], name: string): number {
|
|
return steps.filter((s) => s.role === name).length;
|
|
}
|
|
|
|
const workflow: WorkflowDefinition<SenseMeta> = {
|
|
name: "sense-generator",
|
|
roles: {
|
|
planner: await buildPlannerRole(),
|
|
coder: await buildCoderRole(),
|
|
tester: await buildTesterRole(),
|
|
},
|
|
moderator(context) {
|
|
if (context.steps.length === 0) return "planner";
|
|
const last = context.steps[context.steps.length - 1];
|
|
if (last.role === "planner") return "coder";
|
|
if (last.role === "coder") return "tester";
|
|
if (last.role === "tester") {
|
|
if (last.meta.passed) return END;
|
|
return countRole(context.steps, "tester") < 3 ? "coder" : END;
|
|
}
|
|
return END;
|
|
},
|
|
};
|
|
|
|
export default workflow;
|