如何在 SpringBoot 项目中接入 ChartGPT

大家好，我是公子骏。最近体验了火爆全网的 ChartGPT，深刻体会了其强大的能力，这让我们程序猿对AI的未来突然有了广大的畅想空间。
我也在网上看到不少大牛通过 ChartGPT 来获取收益，就寻思着能否自己接入 ChartGPT 的 API 来做些什么事情。
然后到网上找了不少 Java 接入 ChatGPT 的文章案例，成功实现了接入，而本文就是我整理了这些案例后，使用 Forest 框架重写了一遍接入方法。
一、前期工作

• 注册 OpenAI 的账号
  
• 获取 OpenAI 的 API KEY
  

如何完成这两步网上的资料很多，我这里就不再赘述了。
需要注意的是，API KEY 在创建好之后一定要保存好，否则退出创建页面后就可能找不到了。
二. 添加依赖

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

1. <dependency>
2.     <groupId>com.dtflys.forest</groupId>
3.     <artifactId>forest-spring-boot-starter</artifactId>
4.     <version>1.5.30</version>
5. </dependency>
8. <dependency>
9.     <groupId>com.alibaba</groupId>
10.     <artifactId>fastjson</artifactId>
11.     <version>1.2.83</version>
12. </dependency>

复制代码

 

1. 三. 配置

复制代码

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

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

复制代码

 

1. 四. 定义 ChartGPT 的接入接口

复制代码

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

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

复制代码

 

1. 五. 调用接口

复制代码

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

1. @Resource
2. private ChartGPT chartGPT;
4. public void run() {
5.     String response = chartGPT.send("你好");
6.     System.out.println(response);
7. }

复制代码

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

1. {"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}}

复制代码

 

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

复制代码

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

我们添加两个数据类
在 ChartGPT 返回的 JSON 字符串后，Forest 框架会自动将其反序列化成这些数据类
添加的第一个数据类 GPTResponse: 用于接受 ChartGPT 响应结果最外层 JSON 结构的数据类

1. @Data
2. public class GPTResponse {
4.     private String id;
6.     private String object;
8.     private String created;
10.     private String model;
12.     private List<GPTChoice> choices;
13. }

复制代码

 

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

复制代码

1. @Data
2. public class GPTChoice {
4.     private String text;
6.     private Integer index;
7. }

复制代码

 

1. 改一下接口的返回类型

复制代码

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

复制代码

 

1. 改一下调用部分的代码

复制代码

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

复制代码

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

复制代码

1. 你好！很高兴见到你！<br>

复制代码

示例代码

如果以上步骤都完成了，祝贺你！你是一名优秀的程序猿~
完不成也没关系，我在开源仓库里也提供了示例工程代码，方便大家参考
https://gitee.com/dromara/forest/tree/master/forest-examples/example-chartgpt
或者关注公众号 程序猿公子骏，回复消息 chartgpt，即可获得示例工程源码
 
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！