- Role: (start, messages) → (ctx: ThreadContext) - AgentFn prompt callbacks: (start) → (ctx) - ModeratorContext → ThreadContext - 13 files updated across knowledge-extraction and solve-issue workflows 小橘 <xiaoju@shazhou.work>
26 lines
771 B
TypeScript
26 lines
771 B
TypeScript
import type { AgentFn, Role, ThreadContext } from "@uncaged/nerve-core";
|
|
import type { LlmExtractorConfig } from "@uncaged/nerve-workflow-utils";
|
|
import { createRole } from "@uncaged/nerve-workflow-utils";
|
|
import { z } from "zod";
|
|
|
|
import { reviewPrompt } from "./prompt.js";
|
|
|
|
export const reviewMetaSchema = z.object({
|
|
approved: z.boolean().describe("true if diff is clean and ready for tests"),
|
|
});
|
|
export type ReviewMeta = z.infer<typeof reviewMetaSchema>;
|
|
|
|
export function createReviewRole(
|
|
adapter: AgentFn,
|
|
extract: LlmExtractorConfig,
|
|
nerveRoot: string,
|
|
): Role<ReviewMeta> {
|
|
return createRole(
|
|
adapter,
|
|
async (ctx: ThreadContext) =>
|
|
reviewPrompt({ threadId: ctx.start.meta.threadId, nerveRoot }),
|
|
reviewMetaSchema,
|
|
extract,
|
|
);
|
|
}
|