大家好，我是公子骏。最近体验了火爆全网的 ChartGPT，深刻体会了其强大的能力，这让我们程序猿对AI的未来突然有了广大的畅想空间。

我也在网上看到不少大牛通过 ChartGPT 来获取收益，就寻思着能否自己接入 ChartGPT 的 API 来做些什么事情。

(资料图)

然后到网上找了不少 Java 接入 ChatGPT 的文章案例，成功实现了接入，而本文就是我整理了这些案例后，使用 Forest 框架重写了一遍接入方法。

注册 OpenAI 的账号

获取 OpenAI 的 API KEY

如何完成这两步网上的资料很多，我这里就不再赘述了。

需要注意的是，API KEY 在创建好之后一定要保存好，否则退出创建页面后就可能找不到了。

除了基本的 SpringBoot 依赖外，还需添加以下 Maven 依赖

    com.dtflys.forest    forest-spring-boot-starter    1.5.30    com.alibaba    fastjson    1.2.83

三. 配置

在application.yml文件中添加下列配置

forest:  connect-timeout: 60000      # HTTP请求连接超时时间  read-timeout: 60000         # HTTP请求读取超时时间  variables:                  # 自定义变量:    apiKey: YOUR_API_KEY      # 你的 OpenAI 的 API KEY    model: text-davinci-003   # ChartGPT 的模型    maxTokens: 50             # 最大 Token 数    temperature: 0.5          # 该值越大每次返回的结果越随机，即相似度越小

四. 定义 ChartGPT 的接入接口

创建一个interface类，名字就叫 ChartGPT （名字可以随意起）

public interface ChartGPT {    // 只要输入一个参数: 提示词    // 该方法就会发送 HTTP 请求到 OpenAI 的接口服务    // 并将响应结果以字符串形式返回回来    @Post(            url = "https://api.openai.com/v1/engines/${model}/completions",            contentType = "application/json",            headers = "Authorization: Bearer ${apiKey}",            data = "{\"prompt\": \"${prompt}\", \"max_tokens\": ${maxTokens}, \"temperature\": ${temperature}}"    )    String send(@Var("prompt") String prompt);}

五. 调用接口

在需要调用的地方注入ChartGPT接口实例，并可调用获取 ChartGPT 的结果

@Resourceprivate ChartGPT chartGPT;public void run() {    String response = chartGPT.send("你好");    System.out.println(response);}

执行程序后，会在控制台上打印出 ChartGPT 的响应结果

{"id":"cmpl-6ze62yNNtH5ZHNkaPjWUZORoEtojK","object":"text_completion","created":1680150158,"model":"text-davinci-003","choices":[{"text":"\n\n你好！","index":0,"logprobs":null,"finish_reason":"stop"}],"usage":{"prompt_tokens":4,"completion_tokens":9,"total_tokens":13}}

到这一步已经算初步成功了！

但你发现我们接受到的是一个JSON字符串，并不能直接处理它，需要进一步的加工。

我们添加两个数据类

在 ChartGPT 返回的 JSON 字符串后，Forest 框架会自动将其反序列化成这些数据类

添加的第一个数据类 GPTResponse: 用于接受 ChartGPT 响应结果最外层 JSON 结构的数据类

@Datapublic class GPTResponse {    private String id;    private String object;    private String created;    private String model;    private List choices;}

添加的二个数据类 GPTChoice: 用于接受 ChartGPT 响应结果 JSON 结构里层中文本内容部分的数据类

@Datapublic class GPTChoice {    private String text;    private Integer index;}

改一下接口的返回类型

public interface ChartGPT {    // Forest 会自动将响应结果反序列化成 GPTResponse 类型并返回出来    @Post(            url = "https://api.openai.com/v1/engines/${model}/completions",            contentType = "application/json",            headers = "Authorization: Bearer ${apiKey}",            data = "{\"prompt\": \"${prompt}\", \"max_tokens\": ${maxTokens}, \"temperature\": ${temperature}}"    )    GPTResponse send(@Var("prompt") String prompt);}

改一下调用部分的代码

GPTResponse response = chartGPT.send("你好");System.out.println(response.getChoices().get(0).getText());

执行一下，然后会在控制台上看到类似的输出内容

你好！很高兴见到你！

如果以上步骤都完成了，祝贺你！你是一名优秀的程序猿~

完不成也没关系，我在开源仓库里也提供了示例工程代码，方便大家参考

https://gitee.com/dromara/forest/tree/master/forest-examples/example-chartgpt

或者关注公众号程序猿公子骏，回复消息chartgpt，即可获得示例工程源码