AI로 코드 리뷰 자동화하기 — GitHub Actions + LLM 연동 실전 가이드

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11

PR 생성/업데이트
    ↓
GitHub Actions 트리거
    ↓
변경된 파일 diff 추출 (GitHub API)
    ↓
파일별로 Claude API 호출 (병렬 처리)
    ↓
리뷰 결과 집계
    ↓
PR에 인라인 댓글 + 요약 댓글 게시

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

name: AI Code Review

on:
  pull_request:
    types: [opened, synchronize, reopened]
    # 리뷰하지 않을 브랜치 제외
    branches-ignore:
      - 'dependabot/**'
      - 'renovate/**'

# 동일 PR에 새 커밋이 오면 이전 실행 취소
concurrency:
  group: ai-review-${{ github.event.pull_request.number }}
  cancel-in-progress: true

jobs:
  ai-review:
    runs-on: ubuntu-latest
    # 봇이 만든 PR은 리뷰 건너뜀
    if: github.event.pull_request.user.type != 'Bot'

    permissions:
      contents: read
      pull-requests: write

    env:
      ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
      GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

    steps:
      - name: Checkout code
        uses: actions/checkout@v4
        with:
          fetch-depth: 0

      - name: Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '3.12'

      - name: Install dependencies
        run: |
          pip install anthropic PyGithub python-dotenv

      - name: Run AI Code Review
        run: python .github/scripts/ai_review.py
        env:
          PR_NUMBER: ${{ github.event.pull_request.number }}
          REPO_NAME: ${{ github.repository }}
          BASE_SHA: ${{ github.event.pull_request.base.sha }}
          HEAD_SHA: ${{ github.event.pull_request.head.sha }}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

import os
import anthropic
from github import Github
from typing import Optional

# 환경 변수
ANTHROPIC_API_KEY = os.environ["ANTHROPIC_API_KEY"]
GITHUB_TOKEN = os.environ["GITHUB_TOKEN"]
REPO_NAME = os.environ["REPO_NAME"]
PR_NUMBER = int(os.environ["PR_NUMBER"])
BASE_SHA = os.environ["BASE_SHA"]
HEAD_SHA = os.environ["HEAD_SHA"]

# 리뷰하지 않을 파일 패턴
SKIP_PATTERNS = [
    ".lock", ".sum", "package-lock.json",
    ".min.js", ".min.css", ".svg", ".png",
    "migrations/", "__pycache__/", ".pyc"
]

# 리뷰 대상 확장자
REVIEW_EXTENSIONS = {
    ".py", ".ts", ".tsx", ".js", ".jsx",
    ".go", ".java", ".rs", ".cpp", ".c",
    ".rb", ".php", ".swift", ".kt"
}

def should_skip(filename: str) -> bool:
    """리뷰를 건너뛸 파일인지 판단"""
    if not any(filename.endswith(ext) for ext in REVIEW_EXTENSIONS):
        return True
    return any(pattern in filename for pattern in SKIP_PATTERNS)

def get_diff_for_file(repo, base_sha: str, head_sha: str, filename: str) -> Optional[str]:
    """특정 파일의 diff를 가져옴"""
    comparison = repo.compare(base_sha, head_sha)
    for file in comparison.files:
        if file.filename == filename:
            return file.patch
    return None

def review_code_with_claude(filename: str, diff: str) -> dict:
    """Claude API로 코드 리뷰 수행"""
    client = anthropic.Anthropic(api_key=ANTHROPIC_API_KEY)

    system_prompt = """당신은 15년 경력의 시니어 소프트웨어 엔지니어입니다.
코드 diff를 보고 다음을 검토합니다:
1. 버그 및 잠재적 오류
2. 보안 취약점 (SQL injection, XSS, 인증 우회 등)
3. 성능 이슈 (N+1 쿼리, 불필요한 루프, 메모리 누수)
4. 엣지 케이스 미처리
5. 코드 스타일 및 가독성 (심각한 문제만)

응답은 반드시 아래 JSON 형식으로만 하세요:
{
  "issues": [
    {
      "severity": "critical|warning|info",
      "line": 줄 번호 (diff 기준, 없으면 null),
      "description": "문제 설명 (한국어)",
      "suggestion": "개선 방법 (한국어)"
    }
  ],
  "summary": "전체 리뷰 요약 (한국어, 2~3문장)",
  "score": 1~10
}

이슈가 없으면 issues는 빈 배열로, score는 9~10으로 응답하세요.
사소한 스타일 이슈는 severity info로만 표시하고 최대 3개까지만 포함하세요."""

    user_prompt = f"""파일: {filename}

변경 사항 (diff):

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

def post_inline_comment(pr, commit_sha: str, filename: str, line: int, body: str):
    """특정 줄에 인라인 리뷰 댓글 게시"""
    try:
        pr.create_review_comment(
            body=body,
            commit_id=commit_sha,
            path=filename,
            line=line,
            side="RIGHT"  # 변경 후 코드 기준
        )
    except Exception as e:
        print(f"인라인 댓글 실패 ({filename}:{line}): {e}")
        # 인라인 댓글 실패해도 요약 댓글은 게시

# main()에서 critical 이슈는 인라인 댓글로 추가
for issue in review.get("issues", []):
    if issue["severity"] == "critical" and issue.get("line"):
        inline_body = f"🔴 **{issue['description']}**\n\n{issue['suggestion']}"
        post_inline_comment(
            pr,
            HEAD_SHA,
            file.filename,
            issue["line"],
            inline_body
        )

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12

response = client.messages.create(
    model="claude-haiku-3-5",
    max_tokens=1500,
    system=[
        {
            "type": "text",
            "text": system_prompt,
            "cache_control": {"type": "ephemeral"}  # 캐싱 활성화
        }
    ],
    messages=[{"role": "user", "content": user_prompt}]
)

1
2
3
4
5
6

MAX_DIFF_CHARS = 3000  # 파일당 diff 최대 길이

def truncate_diff(diff: str, max_chars: int = MAX_DIFF_CHARS) -> str:
    if len(diff) <= max_chars:
        return diff
    return diff[:max_chars] + f"\n... (이하 {len(diff) - max_chars}자 생략)"

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12

LANGUAGE_HINTS = {
    ".py": "Django/FastAPI 패턴, PEP 8, 타입 힌트 확인",
    ".ts": "TypeScript strict 모드, React 훅 규칙, 비동기 처리 확인",
    ".go": "goroutine 누수, 에러 핸들링, defer 사용 확인",
    ".sql": "인덱스 사용 여부, N+1 쿼리, SQL injection 확인",
}

def get_language_hint(filename: str) -> str:
    for ext, hint in LANGUAGE_HINTS.items():
        if filename.endswith(ext):
            return f"\n추가 검토 사항: {hint}"
    return ""

1
2
3
4

# PR에 'skip-ai-review' 레이블이 있으면 건너뜀
def should_skip_pr(pr) -> bool:
    labels = [label.name for label in pr.labels]
    return "skip-ai-review" in labels

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

// .github/scripts/ai-review.ts
import Anthropic from "@anthropic-ai/sdk";
import { Octokit } from "@octokit/rest";

const client = new Anthropic({ apiKey: process.env.ANTHROPIC_API_KEY });
const octokit = new Octokit({ auth: process.env.GITHUB_TOKEN });

const [owner, repo] = process.env.REPO_NAME!.split("/");
const prNumber = parseInt(process.env.PR_NUMBER!);

async function reviewFile(filename: string, patch: string): Promise<ReviewResult> {
  const response = await client.messages.create({
    model: "claude-haiku-3-5",
    max_tokens: 1500,
    system: SYSTEM_PROMPT,
    messages: [
      {
        role: "user",
        content: `파일: ${filename}\n\n\`\`\`\n${patch.slice(0, 3000)}\n\`\`\``,
      },
    ],
  });

  return JSON.parse(response.content[0].type === "text" ? response.content[0].text : "{}");
}

async function main() {
  const { data: files } = await octokit.pulls.listFiles({
    owner, repo, pull_number: prNumber,
  });

  const reviews = await Promise.all(
    files
      .filter((f) => f.patch && !shouldSkip(f.filename))
      .slice(0, 10)
      .map((f) => reviewFile(f.filename, f.patch!))
  );

  await postSummaryComment(octokit, owner, repo, prNumber, reviews);
}

main().catch(console.error);

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

{
  "scripts": {
    "review": "ts-node .github/scripts/ai-review.ts"
  },
  "dependencies": {
    "@anthropic-ai/sdk": "^0.30.0",
    "@octokit/rest": "^20.0.0"
  },
  "devDependencies": {
    "ts-node": "^10.9.0",
    "typescript": "^5.0.0"
  }
}

1
2
3
4
5
6
7
8
9

# 포크 PR 안전하게 처리하기
on:
  pull_request_target:
    types: [opened, synchronize]

jobs:
  ai-review:
    # 외부 기여자 PR은 승인 후 실행
    environment: ${{ github.event.pull_request.head.repo.full_name != github.repository && 'external' || 'internal' }}

1
2
3
4
5
6
7

# 월별 비용 한도 초과 시 리뷰 건너뜀 (간단한 방어 로직)
import datetime

def is_within_monthly_budget(max_reviews_per_day: int = 50) -> bool:
    # 실제로는 DB나 외부 카운터로 추적
    # 여기서는 예시만 제시
    return True