주말 프로젝트: 자가 수정 코더

TEST_CASES = [
    ("3+2*2", 7),       # 10이 아님! 우선순위가 중요
    (" 3/2 ", 1),       # 0 방향으로 버림
    (" 3+5 / 2 ", 5),   # 공백과 혼합 연산자
    ("42", 42),         # 단일 숫자
    ("100*2+12", 212),  # 여러 자릿수 숫자
    ("1-1+1", 1),       # 같은 우선순위는 왼쪽에서 오른쪽
    ("2*3*4", 24),      # 연쇄 연산자
]

def naive_solve(model_name, problem):
    response = client.chat.completions.create(
        model=model_name,
        messages=[
            {"role": "system", "content": "당신은 Python 전문가입니다. 코드만 출력하세요."},
            {"role": "user", "content": problem}
        ]
    )
    return response.choices[0].message.content
 
# 실행
code_gpt4o = naive_solve("gpt-4o", PROBLEM_DESCRIPTION)

def run_and_test(code_str, test_cases):
    try:
        # 격리된 네임스페이스에서 실행
        local_env = {}
        exec(code_str, {}, local_env)
 
        solve_func = local_env.get('calculate')
        if not solve_func:
            return False, "'calculate' 함수를 찾을 수 없습니다."
 
        # 테스트 케이스 실행
        for inp, expected in test_cases:
            result = solve_func(inp)
            if result != expected:
                return False, f"'{inp}'에서 실패. 예상: {expected}, 결과: {result}"
 
        return True, "모든 테스트 통과!"
 
    except Exception as e:
        return False, f"에러: {e}"

def agentic_solve(model_name, problem, max_retries=3):
    history = [
        {"role": "system", "content": "당신은 Python 전문가입니다. 유효한 Python 코드만 출력하세요. eval() 사용 금지."},
        {"role": "user", "content": problem}
    ]
 
    for i in range(max_retries):
        print(f"🔄 시도 {i+1} ({model_name})...")
 
        # 1. 생성
        response = client.chat.completions.create(
            model=model_name,
            messages=history
        )
        code = clean_code(response.choices[0].message.content)
 
        # 2. 테스트
        success, feedback = run_and_test(code, TEST_CASES)
        print(f"   결과: {feedback}")
 
        if success:
            return code, i+1  # 코드와 시도 횟수 반환
 
        # 3. 리플렉션 - 에러를 히스토리에 추가
        history.append({"role": "assistant", "content": code})
        history.append({
            "role": "user",
            "content": f"코드가 다음 에러로 실패했습니다: {feedback}\n\n수정해주세요."
        })
 
    return None, max_retries

--- Naive GPT-4o ---
통과? True (모든 테스트 통과!)

--- Agentic GPT-4o-mini ---
🔄 시도 1 (gpt-4o-mini)...
   결과: '3+2*2'에서 실패. 예상: 7, 결과: 10
🔄 시도 2 (gpt-4o-mini)...
   결과: 모든 테스트 통과!
✅ 2번 시도로 해결!

def parallel_solve(problem, num_candidates=3):
    # 여러 솔루션을 병렬로 생성
    candidates = [generate_code(problem) for _ in range(num_candidates)]
 
    # 모든 후보 테스트
    for code in candidates:
        success, _ = run_and_test(code, TEST_CASES)
        if success:
            return code
 
    # 모두 실패하면 최선의 실패를 선택하고 반복
    return iterative_fix(best_candidate)

jupyter notebook week2_reasoning/project_self_correcting_coder.ipynb