AI 에이전트 개발 실전 #7 실전 프로젝트: 이슈 트리아지 에이전트

import requests

API = "https://api.github.com"
HEADERS = {"Authorization": f"Bearer {GITHUB_TOKEN}"}
MAX_BODY = 3000   # 4편: 도구 결과 다이어트

def list_open_issues(repo: str) -> str:
    """라벨이 없는 열린 이슈를 최신순으로 최대 10건 반환한다."""
    r = requests.get(f"{API}/repos/{repo}/issues",
                     params={"state": "open", "per_page": 30}, headers=HEADERS)
    r.raise_for_status()
    unlabeled = [i for i in r.json() if not i["labels"] and "pull_request" not in i]
    lines = [f"#{i['number']} {i['title']}" for i in unlabeled[:10]]
    return "\n".join(lines) or "라벨 없는 열린 이슈가 없습니다."

def get_issue(repo: str, number: int) -> str:
    """이슈 번호로 제목과 본문을 조회한다. 분류 전에 반드시 본문을 읽는다."""
    r = requests.get(f"{API}/repos/{repo}/issues/{number}", headers=HEADERS)
    r.raise_for_status()
    issue = r.json()
    body = (issue["body"] or "")[:MAX_BODY]
    return f"제목: {issue['title']}\n본문:\n{body}"

def add_labels(repo: str, number: int, labels: list) -> str:
    """이슈에 라벨을 붙인다. 되돌릴 수 있지만 공개 저장소에 보이는 변경이므로 승인 대상."""
    r = requests.post(f"{API}/repos/{repo}/issues/{number}/labels",
                      json={"labels": labels}, headers=HEADERS)
    r.raise_for_status()
    return f"#{number}에 라벨 {labels}를 붙였습니다."

def post_comment(repo: str, number: int, body: str) -> str:
    """이슈에 코멘트를 게시한다. 외부에 공개되는 행동이므로 반드시 승인 대상."""
    r = requests.post(f"{API}/repos/{repo}/issues/{number}/comments",
                      json={"body": body}, headers=HEADERS)
    r.raise_for_status()
    return f"#{number}에 코멘트를 게시했습니다."

NEEDS_APPROVAL = {"add_labels", "post_comment"}

def execute_tool(block) -> dict:
    if block.name in NEEDS_APPROVAL:
        print(f"\n[승인 요청] {block.name} {block.input}")
        if input("실행할까요? (y/n) ").strip().lower() != "y":
            return {
                "type": "tool_result",
                "tool_use_id": block.id,
                "content": "사용자가 승인하지 않았습니다. 제안만 정리해서 보고하세요.",
                "is_error": True,
            }
    try:
        return {"type": "tool_result", "tool_use_id": block.id,
                "content": run_tool(block.name, block.input)}
    except Exception as e:
        return {"type": "tool_result", "tool_use_id": block.id,
                "content": f"도구 실행 실패: {e}", "is_error": True}

SYSTEM = """너는 GitHub 이슈 트리아지 에이전트다.

작업 절차:
1. list_open_issues로 라벨 없는 이슈를 확인하고 처리 계획을 나열한다.
2. 이슈마다 get_issue로 본문을 읽은 뒤 분류한다. 제목만으로 분류하지 않는다.
3. 분류 기준: 오류 재현 내용이 있으면 bug, 새 기능 제안이면 enhancement,
   사용법 질문이면 question, 문서 개선이면 documentation.
4. add_labels로 라벨을 제안한다. 확신이 없으면 라벨을 붙이지 말고 이유를 보고한다.
5. question 이슈에만 답변 초안을 작성해 post_comment로 제안한다.

규칙:
- 한 번 승인이 거부된 행동은 다시 시도하지 않는다.
- 마지막에 처리한 이슈, 붙인 라벨, 보류한 이슈를 표로 정리해 보고한다.
"""

GOLDEN = [
    {"number": 101, "expected": ["bug"]},
    {"number": 95,  "expected": ["question"]},
    {"number": 88,  "expected": ["enhancement"]},
    # 과거 이슈 20건 정도면 변화를 감지하기에 충분하다
]

def evaluate(repo: str) -> float:
    correct = 0
    for case in GOLDEN:
        proposed = triage_one(repo, case["number"])   # 승인 게이트 없이 제안만 받는 모드
        if set(proposed) == set(case["expected"]):
            correct += 1
        else:
            print(f"#{case['number']}: 기대 {case['expected']} / 제안 {proposed}")
    return correct / len(GOLDEN)

print(f"분류 정확도: {evaluate('owner/repo'):.0%}")