小橘
bbcaf1eba5
- coder self-iterations (done=false): max 5 - reviewer rejections: max 3 - tester rejections: max 3 - committer rejections: max 2 Each budget is independent, no longer starved by coder's own passes.
63 lines
2.2 KiB
TypeScript
63 lines
2.2 KiB
TypeScript
import { END } from "@uncaged/nerve-core";
|
|
import type { Moderator } from "@uncaged/nerve-core";
|
|
import type { PlannerMeta } from "./roles/planner/index.js";
|
|
import type { CoderMeta } from "./roles/coder/index.js";
|
|
import type { ReviewerMeta } from "./roles/reviewer/index.js";
|
|
import type { TesterMeta } from "./roles/tester/index.js";
|
|
import type { CommitterMeta } from "./roles/committer/index.js";
|
|
|
|
export type WorkflowMeta = {
|
|
planner: PlannerMeta;
|
|
coder: CoderMeta;
|
|
reviewer: ReviewerMeta;
|
|
tester: TesterMeta;
|
|
committer: CommitterMeta;
|
|
};
|
|
|
|
const MAX_CODER_SELF_ITERATIONS = 5;
|
|
const MAX_REVIEWER_REJECTIONS = 3;
|
|
const MAX_TESTER_REJECTIONS = 3;
|
|
const MAX_COMMITTER_REJECTIONS = 2;
|
|
|
|
function countRejections(steps: { role: string; meta: unknown }[], rejector: string, metaKey: string): number {
|
|
return steps.filter((s) => s.role === rejector && !(s.meta as Record<string, boolean>)[metaKey]).length;
|
|
}
|
|
|
|
function countCoderSelfIterations(steps: { role: string; meta: unknown }[]): number {
|
|
// coder rounds where done=false (self-iteration, not a rejection retry)
|
|
return steps.filter((s) => s.role === "coder" && !(s.meta as Record<string, boolean>).done).length;
|
|
}
|
|
|
|
export const moderator: Moderator<WorkflowMeta> = (context) => {
|
|
if (context.steps.length === 0) return "planner";
|
|
|
|
const last = context.steps[context.steps.length - 1];
|
|
|
|
if (last.role === "planner") {
|
|
return last.meta.ready ? "coder" : END;
|
|
}
|
|
|
|
if (last.role === "coder") {
|
|
if (last.meta.done) return "reviewer";
|
|
// coder says not done yet — allow self-iteration
|
|
return countCoderSelfIterations(context.steps) < MAX_CODER_SELF_ITERATIONS ? "coder" : END;
|
|
}
|
|
|
|
if (last.role === "reviewer") {
|
|
if (last.meta.approved) return "tester";
|
|
return countRejections(context.steps, "reviewer", "approved") < MAX_REVIEWER_REJECTIONS ? "coder" : END;
|
|
}
|
|
|
|
if (last.role === "tester") {
|
|
if (last.meta.passed) return "committer";
|
|
return countRejections(context.steps, "tester", "passed") < MAX_TESTER_REJECTIONS ? "coder" : END;
|
|
}
|
|
|
|
if (last.role === "committer") {
|
|
if (last.meta.success) return END;
|
|
return countRejections(context.steps, "committer", "success") < MAX_COMMITTER_REJECTIONS ? "coder" : END;
|
|
}
|
|
|
|
return END;
|
|
};
|