一.问题背景
- 旨在批量为工程代码添加注释,探讨能否借助 AI 自动识别并完成。
- 鉴于利用豆包仅能为单个文件生成注释,考虑采用 LLM 结合 Python 脚本实现自动化操作。
- ai生成的名字真吸引人啊!
二.实现demo
1.实现思路
- 读取java工程所在下面文件夹里的java文件
- 组成prompt利用大模型为代码添加注释
- 将加载注释后的回写至文件
2.前期准备
- 创建文件夹spring-test,同时创建hello-world.java文件
- 编写测试demo
public class HelloWorld {
public static void main(String[] args) {
System.out.println("你好世界");
}
}
- 选择大模型并申请 API 密钥,可选择 ChatGPT 或热门的 DeepSeek。
3.编码
import os
import requests
import json
class DeepSeekLLM:
def __init__(self, api_key):
self.api_key = api_key
def _call(self, prompt, stop=None):
url = "https://api.deepseek.com/chat/completions"
headers =