代码生成与辅助

import { Annotation } from '@langchain/langgraph';

const CodingState = Annotation.Root({
  requirements: Annotation<string>(),
  plan: Annotation<string>(),
  files: Annotation<Record<string, string>>({
    // Reducer: 合并文件更新
    reducer: (state, update) => ({ ...state, ...update }),
    default: () => ({}),
  }),
  reviewComments: Annotation<string[]>(),
});

import { z } from 'zod';

const fileSchema = z.object({
  filename: z.string(),
  content: z.string(),
});

const multiFileSchema = z.object({
  files: z.array(fileSchema).min(1),
});

const coderNode = async (state) => {
  const out = await coderModel
    .withStructuredOutput(multiFileSchema)
    .invoke(state.messages);
  const patch: Record<string, string> = {};
  for (const f of out.files) patch[f.filename] = f.content;
  return { files: patch };
};

import { z } from 'zod';

const fileSchema = z.object({
  filename: z.string(),
  content: z.string(),
});

const coderNode = async (state) => {
  const result = await coderModel
    .withStructuredOutput(fileSchema)
    .invoke(state.messages);
  // 更新虚拟文件系统
  return { files: { [result.filename]: result.content } };
};

import { tool } from '@langchain/core/tools';
import { z } from 'zod';

const runCommand = tool(
  async ({ cmd }) => {
    // 伪代码：真实项目请用隔离沙箱，不要直接 exec
    return { ok: true, stdout: '...', stderr: '' };
  },
  {
    name: 'run_command',
    description: 'Run a shell command in a sandbox and return stdout/stderr.',
    schema: z.object({ cmd: z.string() }),
  },
);

const plannerNode = async (state) => ({ plan: '拆分为 UI / API / Tests' });
const coderNode = async (state) => ({ files: { 'app.tsx': '...' } });
const reviewerNode = async (state) => ({
  approved: false,
  comments: ['缺少测试'],
});

const reviewSchema = z.object({
  approved: z.boolean(),
  comments: z.array(z.string()),
});

const reviewerNode = async (state) => {
  return await reviewerModel
    .withStructuredOutput(reviewSchema)
    .invoke(state.messages);
};

const routeAfterReview = (state) => (state.approved ? END : 'coder');

const fileSchema = z.object({
  filename: z.string(),
  content: z.string(),
});

const outputSchema = z.object({
  files: z.array(fileSchema).min(1),
});

const coderNode = async (state) => {
  const out = await coderModel
    .withStructuredOutput(outputSchema)
    .invoke(state.messages);
  const patch: Record<string, string> = {};
  for (const f of out.files) patch[f.filename] = f.content;
  return { files: patch };
};

const refactorNode = async (state) => {
  const prompt = `请基于以下代码进行可读性优化：\n${state.files['app.tsx']}`;
  const response = await model.invoke([new HumanMessage(prompt)]);
  return { files: { 'app.tsx': String(response.content) } };
};

const system = new SystemMessage(
  '代码需使用 TypeScript，禁止使用 any，函数需添加注释。',
);

const runCommand = tool(
  async ({ cmd }) => ({ ok: true, stdout: '...', stderr: '' }),
  {
    name: 'run_command',
    description: 'Run in sandbox',
    schema: z.object({ cmd: z.string() }),
  },
);

const validateNode = async () => ({
  report: await runCommand.invoke({ cmd: 'pnpm test' }),
});