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%}")