Trae AI 是字节跳动推出的 AI 编程助手,在 IntelliJ IDEA 中通过插件形式提供智能代码补全、代码生成、代码解释、单元测试生成等功能,支持 Java、Python、Go、JavaScript 等多种语言。
一、安装与配置
1. 安装插件
在 IDEA 中打开插件市场(File → Settings → Plugins),搜索'Trae AI'或'Trae Code AI',点击 Install 安装,重启 IDEA 生效。
2. 配置模型
首次使用需配置 AI 模型:
- 打开 File → Settings → Tools → Trae AI
- 选择 Cloud Model(推荐新手),输入 API Key(在 Trae AI 官网注册获取)
- 选择模型版本(如 trae-7b-code、trae-13b-code,参数更大的模型效果更好)
- 本地模型配置:选择 Local Model,输入本地模型服务地址(如
http://localhost:8080/v1),配置 temperature 等参数
3. 验证安装成功
重启后检查:
- 菜单栏出现'Trae AI'选项
- 编辑器右键菜单包含'Trae AI: ...'相关功能
- 工具栏显示 Trae AI 图标(带闪电的机器人)
二、核心功能与使用方式
1. 三种调用方式
- 右键菜单调用:在编辑器中右键代码 → 选择'Trae AI'子菜单 → 使用 Generate Code、Explain Code、Optimize Code、Generate Tests 等功能
- 快捷键调用:Settings → Keymap → Trae AI,为常用功能绑定快捷键(如 Generate Code 绑定 Alt+A)
- 命令面板调用:Ctrl+Shift+A 打开命令面板,输入'Trae AI:'选择功能
2. 主要功能说明
| 功能 | 说明 | 适用场景 |
|---|---|---|
| 代码补全 | 基于上下文提供单行/多行代码建议 | 编写循环、条件判断、方法调用等 |
| 代码生成 | 根据自然语言描述生成完整代码 | 生成 CRUD 方法、工具函数、接口实现 |
| 代码解释 | 逐行解析代码功能、依赖关系 | 理解复杂逻辑、快速上手项目 |
| 注释生成 | 自动生成函数级或行间注释 | 提高代码可读性、生成 API 文档 |
| 单元测试生成 | 为代码片段生成单元测试用例 | 快速编写 JUnit、pytest 测试 |
| 智能修复 | 检测语法错误、逻辑 bug 并提供修复方案 | 修复空指针、数组越界、资源未关闭等问题 |
三、实战示例(Java + Spring Boot)
示例 1:生成 CRUD 方法
场景:为 UserService 生成根据 ID 查询用户的方法
操作步骤:
- 在 UserService 接口中定位到需要生成方法的位置
- 右键 → Trae AI → Generate Code
- 输入提示词:
根据用户 ID 查询用户信息,返回 User 对象,如果用户不存在返回 null - 点击生成,查看并确认生成的代码

