멀티 에이전트 오케스트레이션: LangGraph를 활용한 상태 머신 기반 협업 에이전트 구축기

import { StateGraph, END, START } from "@langchain/langgraph";
import { BaseMessage } from "@langchain/core/messages";

// 1. 전역 상태 인터페이스 정의
interface AgentState {
  messages: BaseMessage[];
  code: string;
  testPassed: boolean;
  retryCount: number;
}

// 2. 상태 그래프 인스턴스 생성
const workflow = new StateGraph<AgentState>({
  channels: {
    messages: { value: (x, y) => x.concat(y), default: () => [] },
    code: { value: (x, y) => y ?? x, default: () => "" },
    testPassed: { value: (x, y) => y ?? x, default: () => false },
    retryCount: { value: (x, y) => (y !== undefined ? y : x), default: () => 0 },
  },
});

// 3. 노드 등록 (기획 -> 코드 작성 -> 코드 리뷰)
workflow.addNode("planner", planNode);
workflow.addNode("coder", coderNode);
workflow.addNode("reviewer", reviewNode);

// 4. 조건부 분기 엣지: 리뷰 통과 시 종료, 실패 시 coder로 피드백 루프 (최대 3회)
workflow.addEdge(START, "planner");
workflow.addEdge("planner", "coder");
workflow.addEdge("coder", "reviewer");
workflow.addConditionalEdges("reviewer", (state) => {
  if (state.testPassed) return END;
  if (state.retryCount >= 3) return "human_escalation";
  return "coder";
});

export const multiAgentApp = workflow.compile();