Trae AI 是一款智能编程助手,在 IntelliJ IDEA 中通过插件形式提供代码补全、生成、解释及单元测试等功能,支持 Java、Python、Go 等多种语言。
安装与基础配置
安装插件
打开 IDEA 设置界面(File → Settings → Plugins),进入市场搜索 "Trae AI" 或 "Trae Code AI"。点击 Install 安装后重启 IDE 即可生效。
模型配置
首次使用需指定推理源:
- 云端模型:推荐新手选择 Cloud Model,需在官网注册获取 API Key 填入。
- 本地模型:若部署了私有服务,选择 Local Model,输入地址如
http://localhost:8080/v1并调整 temperature 等参数。
验证安装是否成功可观察菜单栏是否新增 "Trae AI" 选项,以及编辑器右键菜单和工具栏图标是否正常显示。
核心功能与调用方式
三种调用路径
- 右键菜单:选中代码块,右键选择 "Trae AI" 子菜单,执行生成、解释或优化操作。
- 快捷键:在 Keymap 设置中为常用功能绑定快捷指令,例如将 Generate Code 设为 Alt+A。
- 命令面板:按 Ctrl+Shift+A 唤起面板,输入 "Trae AI:" 快速定位功能。
功能概览
| 功能 | 说明 | 适用场景 |
|---|---|---|
| 代码补全 | 基于上下文提供单行或多行建议 | 循环、条件判断、方法调用 |
| 代码生成 | 根据自然语言描述生成完整逻辑 | CRUD 方法、工具函数、接口实现 |
| 代码解释 | 逐行解析功能与依赖关系 | 理解复杂逻辑、快速上手项目 |
| 注释生成 | 自动生成函数级或行间注释 | 提高可读性、补充 API 文档 |
| 单元测试 | 为片段生成测试用例 | 快速编写 JUnit、pytest 测试 |
| 智能修复 | 检测语法错误并提供修复方案 | 空指针、越界、资源未关闭等问题 |
实战示例(Java + Spring Boot)
示例 1:生成 CRUD 方法
场景:在 UserService 接口中补充根据 ID 查询用户的方法。
直接在需要的位置右键选择 "Generate Code",输入提示词:根据用户 ID 查询用户信息,返回 User 对象,如果用户不存在返回 null。
生成的代码通常如下:
public User getUserById(Long id) {
return userRepository.findById(id).orElse(null);
}
示例 2:代码解释与优化
场景:重构一段嵌套过深的业务逻辑。
先选中代码块,右键选择 "Explain Code" 理解现有逻辑,随后选择 "Optimize Code" 查看结构建议。


