에이전트 개발의 다음 단계: 제어 흐름(Control Flow) 도입하기

// 1. 에이전트의 상태와 실행 컨텍스트 정의
#[derive(Debug, Clone)]
struct AgentContext {
    current_data: String,
    step_count: usize,
}

// 2. 워크플로우의 각 단계(Step) 정의
enum WorkflowStep {
    Initialize,
    ProcessData,
    ValidateResult, // 결과에 따라 다음 단계가 결정됨
    Finalize,
    Complete,
}

// 3. 각 단계별 실행 로직 구제화
impl WorkflowStep {
    fn execute(&self, ctx: &mut AgentContext) -> WorkflowStep {
        match self {
            WorkflowStep::Initialize => {
                println!("[Step 1] 데이터 초기화 중...");
                ctx.current_data = "Sample Input Data".to_string();
                ctx.step_count += 1;
                WorkflowStep::ProcessData
            },
            WorkflowStep::ProcessData => {
                println!("[Step 2] LLM을 통해 데이터 처리 중: {}", ctx.current_data);
                // 실제로는 여기서 MCP 클라이언트 등을 호출하여 LLM 추론을 수행
                ctx.current_data = format!("Processed: {}", ctx.current_data);
                ctx.step_count += 1;
                WorkflowStep::ValidateResult
            },
            WorkflowStep::ValidateResult => {
                println!("[Step 3] 결과 검증 중...");
                // 비즈니스 로직: 예를 들어 결과 길이가 10보다 작으면 재시작(혹은 에러 처리)
                if ctx.current_data.len() < 10 {
                    println!("데이터가 부족하여 초기화 단계로 돌아갑니다.");
                    WorkflowStep::Initialize // 루프 구조
                } else {
                    println!("검증 통과. 완료 단계로 이동합니다.");
                    WorkflowStep::Finalize
                }
            },
            WorkflowStep::Finalize => {
                println!("[Step 4] 최종 결과 저장: {}", ctx.current_data);
                WorkflowStep::Complete
            },
            WorkflowStep::Complete => {
                println!("워크플로우 완료.");
                WorkflowStep::Complete // 종료 상태 유지
            }
        }
    }
}

// 4. 메인 실행 루프 (Runner)
fn run_agent_workflow() {
    let mut ctx = AgentContext {
        current_data: String::new(),
        step_count: 0,
    };
    
    let mut current_step = WorkflowStep::Initialize;
    
    // 최대 10회 반복 방지 (Safety Breaker)
    for _ in 0..10 {
        if matches!(current_step, WorkflowStep::Complete) {
            break;
        }
        current_step = current_step.execute(&mut ctx);
    }
}

fn main() {
    run_agent_workflow();
}