Day 11 — 다단계 추론 패턴

prompt = """다음 코드의 버그를 찾아주세요.

코드:
def calculate_discount(price, discount_pct):
    return price - price * discount_pct

calculate_discount(100, 20)  # 기대값: 80, 실제값: -1900

단계별로 분석해주세요:
1단계) 함수가 의도한 동작 파악
2단계) 실제 계산 과정 추적
3단계) 버그 원인 특정
4단계) 수정 코드 제시
"""

import anthropic

client = anthropic.Anthropic()

def run(prompt: str) -> str:
    msg = client.messages.create(
        model="claude-sonnet-4-6",
        max_tokens=2048,
        messages=[{"role": "user", "content": prompt}]
    )
    return msg.content[0].text

# 요구사항 문서 → 개발 태스크 분해 파이프라인
requirements = """
사용자가 CSV 파일을 업로드하면 자동으로 데이터를 정제하고
이상치를 탐지한 뒤, 요약 리포트를 이메일로 발송하는 시스템
"""

# Step 1: 핵심 컴포넌트 추출
components = run(f"""
다음 요구사항에서 독립적인 기능 컴포넌트를 목록으로 추출해주세요.
요구사항: {requirements}
형식: 번호 매긴 목록, 각 항목은 한 문장
""")

# Step 2: 각 컴포넌트를 개발 태스크로 변환
tasks = run(f"""
다음 컴포넌트 목록을 Jira 태스크 형식으로 변환해주세요.
각 태스크: 제목, 수락 기준 2개, 예상 공수(시간)

컴포넌트 목록:
{components}
""")

print(tasks)

draft = run("""
Python으로 이진 탐색 함수를 작성해주세요.
정렬된 리스트에서 target 값을 찾아 인덱스를 반환합니다.
없으면 -1을 반환합니다.
""")

final = run(f"""
다음 코드를 검토하고 개선해주세요.

[초안 코드]
{draft}

검토 기준:
1. 엣지 케이스 처리 (빈 리스트, 단일 원소 등)
2. 시간/공간 복잡도
3. Pythonic한 스타일
4. 독스트링 완성도

개선된 버전을 제시해주세요.
""")

print(final)

import anthropic

client = anthropic.Anthropic()

def multi_step_code_review(code: str, language: str = "Python") -> dict:
    """3단계 코드 리뷰 파이프라인"""

    def run(prompt: str) -> str:
        msg = client.messages.create(
            model="claude-sonnet-4-6",
            max_tokens=2048,
            messages=[{"role": "user", "content": prompt}]
        )
        return msg.content[0].text

    # Step 1: 코드 이해
    summary = run(f"""
다음 {language} 코드가 무엇을 하는지 3문장으로 요약해주세요.
기술적 세부사항보다 비즈니스 목적 중심으로 설명해주세요.

```{language.lower()}
{code}
```
""")

    # Step 2: 잠재적 문제 탐지
    issues = run(f"""
다음 코드의 잠재적 문제를 찾아주세요.

코드 요약: {summary}

```{language.lower()}
{code}
```

아래 카테고리별로 분류해주세요:
- [버그] 실제 오류 가능성
- [보안] 취약점
- [성능] 비효율
- [가독성] 유지보수 어려움

문제가 없는 카테고리는 "없음"으로 표기하세요.
""")

    # Step 3: 개선 우선순위 결정
    priority = run(f"""
다음 코드 리뷰 결과를 바탕으로 수정 우선순위를 결정해주세요.

발견된 이슈:
{issues}

출력 형식:
1. [즉시 수정] ...
2. [다음 PR에서 수정] ...
3. [기술 부채로 등록] ...
""")

    return {
        "summary": summary,
        "issues": issues,
        "priority": priority
    }


# 사용 예시
code = """
def get_user(user_id):
    query = f"SELECT * FROM users WHERE id = {user_id}"
    result = db.execute(query)
    return result[0]
"""

review = multi_step_code_review(code)
for step, content in review.items():
    print(f"=== {step.upper()} ===")
    print(content)
    print()