02. Reflection Agents

def generator(topic: str, previous_draft: str = None, critique: str = None) -> str:
    """
    초안을 생성하거나 비평을 기반으로 수정합니다.
    """
    if previous_draft and critique:
        prompt = f"""주제: {topic}
이전 초안:
{previous_draft}
 
비평/피드백:
{critique}
 
위의 비평을 반영하여 초안을 다시 작성하세요.
"""
    else:
        prompt = f"주제: {topic}\n주제에 대한 짧은 에세이 초안을 작성하세요."
 
    response = client.chat.completions.create(
        model="gpt-4o-mini",
        messages=[{"role": "user", "content": prompt}]
    )
    return response.choices[0].message.content

def reflector(draft: str) -> str:
    """
    초안을 읽고 건설적인 비평을 제공합니다.
    """
    prompt = f"""다음 초안을 읽고 건설적인 비평을 제공하세요.
논리, 명확성, 스타일을 평가하세요. 개선이 필요한 영역을 최대 3개까지 지적하세요.
칭찬보다는 수정이 필요한 부분에 집중하세요.
 
초안:
{draft}
"""
    response = client.chat.completions.create(
        model="gpt-4o-mini",
        messages=[{"role": "user", "content": prompt}]
    )
    return response.choices[0].message.content

# Step 1: 첫 번째 초안
print("✍️ 초안 V1 작성 중...")
draft_v1 = generator(topic)
 
# Step 2: 리플렉션
print("🤔 성찰 중...")
critique_v1 = reflector(draft_v1)
 
# Step 3: 개선
print("✍️ 초안 V2 작성 (개선)...")
draft_v2 = generator(topic, previous_draft=draft_v1, critique=critique_v1)

class ReflexionAgent:
    def __init__(self, max_iterations: int = 3):
        self.max_iterations = max_iterations
 
    def run(self, topic: str):
        current_draft = generator(topic)
        print(f"📝 초기 초안:\n{current_draft}\n")
 
        for i in range(self.max_iterations):
            print(f"--- 반복 {i+1} ---")
 
            # 1. 비평
            critique = reflector(current_draft)
            print(f"🔍 비평:\n{critique}\n")
 
            # 2. 중지 조건 확인
            if "개선 필요 없음" in critique.lower():
                print("✨ Reflector가 만족합니다. 중지.")
                break
 
            # 3. 개선
            current_draft = generator(
                topic,
                previous_draft=current_draft,
                critique=critique
            )
            print(f"✍️ 개선된 초안:\n{current_draft}\n")
 
        return current_draft

def code_generator(prompt: str, error_msg: str = None, previous_code: str = None) -> str:
    if error_msg:
        full_prompt = f"""다음 코드에서 에러가 발생했습니다.
코드:
{previous_code}
 
에러:
{error_msg}
 
코드를 수정하세요. 코드만 반환하세요.
"""
    else:
        full_prompt = f"다음을 수행하는 Python 코드를 작성하세요: {prompt}. 코드만 반환하세요."
 
    response = client.chat.completions.create(
        model="gpt-4o-mini",
        messages=[{"role": "user", "content": full_prompt}]
    )
    return clean_code(response.choices[0].message.content)
 
def execute_code(code: str):
    """실행하고 (성공 여부, 출력/에러) 반환"""
    try:
        exec_globals = {}
        exec(code, exec_globals)
        return True, "성공"
    except Exception as e:
        return False, str(e)
 
# 자가 수정 루프
code = code_generator(problem)
success, output = execute_code(code)
 
if not success:
    print("⚠️ 에러 감지! 자가 수정 중...")
    fixed_code = code_generator(problem, error_msg=output, previous_code=code)

def scored_reflector(draft: str) -> tuple[str, int]:
    """(비평, 1-5 점수) 반환"""
    prompt = f"""초안을 1-5점으로 평가하고 비평을 제공하세요.
형식: 점수: X\n비평: ...
 
초안:
{draft}
"""
    response = client.chat.completions.create(...)
    # 응답에서 점수와 비평 파싱
    return critique, score
 
# 루프에서:
critique, score = scored_reflector(current_draft)
if score >= 4:
    print("✅ 품질 임계값 도달!")
    break