xiaoju
fce2bf7441
- CreateRoleArgs.systemPrompt simplified to string (no more ctx callback) - buildAgentPrompt injects real ctx.threadId in Tools section - All four roles compute prompt at construction time from config only - Removed duplicate ctx.threadId injection from reviewer/committer prompts 小橘 <xiaoju@shazhou.work>
59 lines
1.8 KiB
TypeScript
59 lines
1.8 KiB
TypeScript
import type { AgentFn, Role } from "@uncaged/workflow";
|
|
import { createRole } from "@uncaged/workflow-agent-llm";
|
|
import type { LlmProvider } from "@uncaged/workflow-util-role";
|
|
import * as z from "zod/v4";
|
|
|
|
export const reviewerMetaSchema = z.discriminatedUnion("status", [
|
|
z.object({
|
|
status: z.literal("approved"),
|
|
}),
|
|
z.object({
|
|
status: z.literal("rejected"),
|
|
issues: z.array(z.string()).describe("blocking issues that must be fixed"),
|
|
}),
|
|
]);
|
|
export type ReviewerMeta = z.infer<typeof reviewerMetaSchema>;
|
|
|
|
export type ReviewerConfig = {
|
|
cwd: string;
|
|
};
|
|
|
|
export const DEFAULT_REVIEWER_CONFIG: ReviewerConfig = {
|
|
cwd: ".",
|
|
};
|
|
|
|
function reviewerPrompt(config: ReviewerConfig): string {
|
|
const { cwd } = config;
|
|
|
|
return `You are a code reviewer. The project is at \`${cwd}\`.
|
|
|
|
## Task
|
|
|
|
Review the current git diff in \`${cwd}\`. Give a clear **approve** or **reject** verdict.
|
|
|
|
Only reject for **blocking issues** — things that must be fixed before merge. Do not mention minor style preferences or non-blocking suggestions; they will be ignored.
|
|
|
|
End with your verdict — clearly state whether the code is approved or rejected, and if rejected, list the blocking issues.`;
|
|
}
|
|
|
|
/**
|
|
* Code review role: agent inspects git diffs; structured extract yields approve/reject verdict.
|
|
*/
|
|
export function createReviewerRole(
|
|
adapter: AgentFn,
|
|
extract: { provider: LlmProvider; dryRun: boolean | null; dryRunMeta: ReviewerMeta },
|
|
config: ReviewerConfig = DEFAULT_REVIEWER_CONFIG,
|
|
): Role<ReviewerMeta> {
|
|
return createRole({
|
|
name: "reviewer",
|
|
schema: reviewerMetaSchema,
|
|
systemPrompt: reviewerPrompt(config),
|
|
agent: adapter,
|
|
extract: {
|
|
provider: extract.provider,
|
|
dryRun: extract.dryRun,
|
|
dryRunMeta: extract.dryRunMeta,
|
|
},
|
|
});
|
|
}
|