引言

在团队开发中，代码审查（Code Review）是保障代码质量、减少线上故障的重要手段。但人工审查耗时耗力，资深工程师时间宝贵，初级工程师又难以发现隐藏的坑。随着大语言模型（LLM）的成熟，让AI辅助甚至自动完成第一轮代码审查已经成为现实。本文将带你从0到1，用Python和OpenAI API开发一个命令行代码审查工具，只需30分钟即可拥有一个懂代码的AI助手。

核心概念

一个典型的AI代码审查工具，本质上就是一个“聪明的中间人”：读取本地代码文件，构建高质量Prompt，调用大模型接口，最后将模型返回的审查意见展示给开发者。其核心流程如下：

1. 代码输入：支持命令行指定本地源文件。
2. Prompt工程：设计合适的提示词，让模型扮演资深审查专家，关注bug、性能、风格、安全等维度。
3. 模型调用：通过HTTP请求调用GPT系列模型（或其他兼容接口）。
4. 结果解析与展示：把模型给出的建议直接打印到终端或保存为报告。

想要让审查结果有价值，Prompt设计至关重要。我们会在实战中给出一个通用且效果出色的提示词模板。

实战：从零构建审查工具

1. 准备工作

请确保本地已安装Python 3.8+，然后安装OpenAI官方库：

pip install openai

接着设置API密钥，创建环境变量（推荐使用export，Windows用set）：

export OPENAI_API_KEY="sk-你的密钥"

本文示例使用gpt-3.5-turbo模型，性价比极高，单次审查成本通常不到0.01美元。当然你也可以换成gpt-4o-mini等模型。

2. 项目结构

我们的工具只有一个Python文件，命名为review.py。整体结构如下：

• read_code() — 读取并校验文件
• build_prompt() — 构造审查提示词
• review_code() — 调用OpenAI接口
• main() — 解析命令行参数，串联流程

3. 逐步实现

3.1 读取代码文件

使用pathlib和argparse结合，让工具可以从命令行接收一个文件路径，读取其内容。

import argparse
import sys
from pathlib import Path

def read_code(filepath):
    path = Path(filepath)
    if not path.exists():
        print(f"错误：文件 {filepath} 不存在")
        sys.exit(1)
    with open(path, 'r', encoding='utf-8') as f:
        return f.read()

如果文件不存在，直接报错退出，而不是让API去处理一个空请求。

3.2 打造高质量Prompt

Prompt决定了审查的深度和方向。我们要求模型从四个维度审查：潜在bug、性能问题、代码风格与可读性、安全漏洞，并用中文给出具体建议。

def build_prompt(code: str) -> str:
    return f"""
你是一位资深代码审查专家。请审查以下代码，提供以下方面的建议：
1. 潜在的bug和错误
2. 性能问题
3. 代码风格和可读性
4. 安全性漏洞
并用中文给出具体的修改建议。代码：

{code}

"""

将代码包裹在Markdown代码块中，有助于模型识别代码边界，提升输出质量。

3.3 调用OpenAI接口

采用最新的1.x版本API风格，创建客户端，发送消息。我们设置较低的temperature（0.3）保证审查结果稳定，同时限制max_tokens防止输出过长。

from openai import OpenAI

def review_code(client: OpenAI, model: str, prompt: str) -> str:
    try:
        response = client.chat.completions.create(
            model=model,
            messages=[
                {"role": "system", "content": "你是一个严格的代码审查助手。"},
                {"role": "user", "content": prompt}
            ],
            temperature=0.3,
            max_tokens=2000
        )
        return response.choices[0].message.content
    except Exception as e:
        print(f"调用API失败：{e}")
        sys.exit(1)

3.4 命令行入口

完善参数解析，支持指定审查文件和使用--model切换模型。然后串联所有步骤。

```python
def main():
parser = argparse.ArgumentParser(description="AI代码审查工具")
parser.add_argument("file", help="待审查的代码文件路径")
parser.add_argument("--model", default="gpt-3.5-turbo",
help="使用的模型，默认为gpt-3.5-turbo")
args = parser.parse_args()

client = OpenAI()          # 从环境变量OPENAI_API_KEY读取
code = read_code(args.file)
prompt = build_prompt(code)
print("正在审查代码，请稍候...")
result = review_code(client, args.model, prompt)
print("\n