ChatGPT Idea插件开发实战：从零构建你的第一个AI辅助编程工具

1次阅读

没有评论

共计 2366 个字符，预计需要花费 6 分钟才能阅读完成。

在日常开发中，我们经常遇到一些重复性工作，比如：

编写样板代码（如 Getter/Setter、DTO 转换）
查找并修复常见语法错误
为复杂逻辑编写测试用例
查阅 API 文档和示例代码

这些工作不仅耗时，还容易让人分心。ChatGPT 这类 AI 工具能够理解自然语言，可以帮我们快速生成代码片段、解释错误信息，甚至提供优化建议。如果能将这些能力直接集成到 IDE 中，开发效率将大幅提升。

当前主流的 AI 编程辅助方案主要有三种：

GitHub Copilot：专为编程优化，但闭源且需要订阅
本地模型（如 CodeLlama）：数据隐私性好，但对硬件要求高
OpenAI API：响应快、能力强，按使用量计费

我们选择 ChatGPT API 是因为：

成熟的对话能力适合交互式编程场景
支持代码理解（能处理带上下文的问题）
提供稳定的 SDK 和清晰的计费模式

使用 Gradle 模板初始化项目（需安装 IntelliJ Plugin 开发工具包）：

// build.gradle.kts
plugins {id("java")
    id("org.jetbrains.intellij") version "1.15.0"
}

intellij {version.set("2023.2")  // 匹配你的 IDE 版本
    plugins.set(listOf("java"))
}

添加依赖并封装基础请求逻辑：

public class ChatGPTClient {
    private final OpenAiService service;

    public ChatGPTClient(String apiKey) {this.service = new OpenAiService(apiKey, Duration.ofSeconds(30));
    }

    public String getCodeSuggestion(String prompt) {CompletionRequest request = CompletionRequest.builder()
            .model("gpt-3.5-turbo")
            .prompt("作为编程助手，请生成 Kotlin 代码：" + prompt)
            .temperature(0.3)  // 降低随机性
            .build();

        return service.createCompletion(request)
            .getChoices()
            .get(0)
            .getText();}
}

实现右键菜单触发 AI 建议（关键部分）：

class CodeAction : AnAction() {override fun actionPerformed(e: AnActionEvent) {val editor = e.getData(CommonDataKeys.EDITOR) ?: return
        val document = editor.document

        // 获取选中文本作为提示词
        val selectionModel = editor.selectionModel
        val prompt = selectionModel.selectedText ?: document.text

        // 异步调用避免阻塞 UI
        ApplicationManager.getApplication().executeOnPooledThread {val suggestion = ChatGPTClient(API_KEY).getCodeSuggestion(prompt)

            // 回到 UI 线程更新文档
            ApplicationManager.getApplication().invokeLater {document.insertString(selectionModel.selectionStart, suggestion)
            }
        }
    }
}

请求压缩：移除代码中的空行和注释再发送
上下文缓存：对相同文件保留最近的 3 条对话历史
预加载：在 IDE 启动时建立 API 连接

// 使用指数退避重试
fun callWithRetry(prompt: String, retries: Int = 3): String {
    try {return chatGPTClient.getCodeSuggestion(prompt)
    } catch (e: OpenAIException) {if (e.statusCode == 429 && retries > 0) {Thread.sleep(1000 * (4 - retries))  // 1s, 2s, 3s
            return callWithRetry(prompt, retries - 1)
        }
        throw e
    }
}