使用chatgpt实现微信聊天小程序(秒回复)，github开源(附带链接) ...

媒介

我在前一段时间突发奇想，就使用java来调用chatgpt的接口，然后写了一个简单小程序，也上了热榜第一，java调用chatgpt接口，实现专属于本身的人工智能助手，事实上，这个程序毛病挺多的，最不能让人接受的一点就是返回速度非常缓慢(即使使用非常好的外网服务器)。
如今，我改进了一下程序，使用异步哀求的方式，基本可以实现秒回复。而且还基于webSocket编写了一个微信小程序来进行交互，可以直接使用微信小程序来进行体验。
如今我将所有代码都上传了github(链接在文章结尾)，各人可以clone下来，部署到服务器上，真正实现本身的聊天呆板人！！！

---

结果展示

部分截图如下

---

原理说明

在 java调用chatgpt接口，实现专属于本身的人工智能助手 我说明白java调用chatgpt的基本原理，这里的代码就是对这个代码的改进，使用异步哀求的方式来进行。

注意看官方文档，我们在哀求时可以提供一个参数stream，然后就可以实现按照流的形式进行返回，这种方式基本可以做到没有延迟就给出答案。
由于这次改进的思路主要就是将哀求改为了异步，其他的基本一样，以是就不做解释，直接给出代码了，代码上面都有注释

1.     /**
2.      * 这个方法用于测试的，可以在控制台打印输出结果
3.      *
4.      * @param chatGptRequestParameter 请求的参数
5.      * @param question                问题
6.      */
7.     public void printAnswer(ChatRequestParameter chatGptRequestParameter, String question) {
8.         asyncClient.start();
9.         // 创建一个post请求
10.         AsyncRequestBuilder asyncRequest = AsyncRequestBuilder.post(url);
12.         // 设置请求参数
13.         chatGptRequestParameter.addMessages(new ChatMessage("user", question));
15.         // 请求的参数转换为字符串
16.         String valueAsString = null;
17.         try {
18.             valueAsString = objectMapper.writeValueAsString(chatGptRequestParameter);
19.         } catch (JsonProcessingException e) {
20.             e.printStackTrace();
21.         }
23.         // 设置编码和请求参数
24.         ContentType contentType = ContentType.create("text/plain", charset);
25.         asyncRequest.setEntity(valueAsString, contentType);
26.         asyncRequest.setCharset(charset);
28.         // 设置请求头
29.         asyncRequest.setHeader(HttpHeaders.CONTENT_TYPE, "application/json");
30.         // 设置登录凭证
31.         asyncRequest.setHeader(HttpHeaders.AUTHORIZATION, "Bearer " + apiKey);
33.         // 下面就是生产者消费者模型
34.         CountDownLatch latch = new CountDownLatch(1);
35.         // 用于记录返回的答案
36.         StringBuilder sb = new StringBuilder();
37.         // 消费者
38.         AbstractCharResponseConsumer<HttpResponse> consumer = new AbstractCharResponseConsumer<HttpResponse>() {
39.             HttpResponse response;
41.             @Override
42.             protected void start(HttpResponse response, ContentType contentType) throws HttpException, IOException {
43.                 setCharset(charset);
44.                 this.response = response;
45.             }
47.             @Override
48.             protected int capacityIncrement() {
49.                 return Integer.MAX_VALUE;
50.             }
52.             @Override
53.             protected void data(CharBuffer src, boolean endOfStream) throws IOException {
54.                 // 收到一个请求就进行处理
55.                 String ss = src.toString();
56.                 // 通过data:进行分割，如果不进行此步，可能返回的答案会少一些内容
57.                 for (String s : ss.split("data:")) {
58.                     // 去除掉data:
59.                     if (s.startsWith("data:")) {
60.                         s = s.substring(5);
61.                     }
62.                     // 返回的数据可能是（DONE）
63.                     if (s.length() > 8) {
64.                         // 转换为对象
65.                         ChatResponseParameter responseParameter = objectMapper.readValue(s, ChatResponseParameter.class);
66.                         // 处理结果
67.                         for (Choice choice : responseParameter.getChoices()) {
68.                             String content = choice.getDelta().getContent();
69.                             if (content != null && !"".equals(content)) {
70.                                 // 保存结果
71.                                 sb.append(content);
72.                                 // 将结果使用webSocket传送过去
73.                                 System.out.print(content);
74.                             }
75.                         }
76.                     }
77.                 }
78.             }
80.             @Override
81.             protected HttpResponse buildResult() throws IOException {
82.                 return response;
83.             }
85.             @Override
86.             public void releaseResources() {
87.             }
88.         };
90.         // 执行请求
91.         asyncClient.execute(asyncRequest.build(), consumer, new FutureCallback<HttpResponse>() {
93.             @Override
94.             public void completed(HttpResponse response) {
95.                 latch.countDown();
96.                 chatGptRequestParameter.addMessages(new ChatMessage("assistant", sb.toString()));
97.                 System.out.println("回答结束！！！");
98.             }
100.             @Override
101.             public void failed(Exception ex) {
102.                 latch.countDown();
103.                 System.out.println("failed");
104.                 ex.printStackTrace();
105.             }
107.             @Override
108.             public void cancelled() {
109.                 latch.countDown();
110.                 System.out.println("cancelled");
111.             }
113.         });
114.         try {
115.             latch.await();
116.         } catch (InterruptedException e) {
117.             e.printStackTrace();
118.         }
119.     }

复制代码

  各人代码可以直接不看，反正终极的结果就是可以实现问了题目就返回结果。运行结果如下
  

   可以发现，输出就雷同于官方的那种结果，一个字一个字的输出
  

---

服务器端代码说明

我使用java搭建了一个简单的服务器端程序，提供最底子的用户登录校验功能，以及提供了WebSocket通信。
用户校验的代码

1. package com.ttpfx.controller;
3. import com.ttpfx.entity.User;
4. import com.ttpfx.service.UserService;
5. import com.ttpfx.utils.R;
6. import org.springframework.web.bind.annotation.RequestMapping;
7. import org.springframework.web.bind.annotation.RestController;
9. import javax.annotation.Resource;
10. import java.util.Objects;
11. import java.util.concurrent.ConcurrentHashMap;
13. /**
14. * @author ttpfx
15. * @date 2023/3/29
16. */
17. @RestController
18. @RequestMapping("/user")
19. public class UserController {
21.     @Resource
22.     private UserService userService;
24.     public static ConcurrentHashMap<String, User> loginUser = new ConcurrentHashMap<>();
26.     public static ConcurrentHashMap<String, Long> loginUserKey = new ConcurrentHashMap<>();
27.     @RequestMapping("/login")
28.     public R login(String username, String password) {
29.         if (username == null) return R.fail("必须填写用户名");
32.         User user = userService.queryByName(username);
33.         if (user == null) return R.fail("用户名不存在");
34.         String targetPassword = user.getPassword();
35.         if (targetPassword == null) return R.fail("用户密码异常");
36.         if (!targetPassword.equals(password)) return R.fail("密码错误");
38.         loginUser.put(username, user);
39.         loginUserKey.put(username, System.currentTimeMillis());
40.         return R.ok(String.valueOf(loginUserKey.get(username)));
41.     }
43.     @RequestMapping("/logout")
44.     public R logout(String username) {
45.         loginUser.remove(username);
46.         loginUserKey.remove(username);
47.         return R.ok();
48.     }
50.     @RequestMapping("/checkUserKey")
51.     public R checkUserKey(String username, Long key){
52.         if (username==null || key == null)return R.fail("用户校验异常");
53.         if (!Objects.equals(loginUserKey.get(username), key)){
54.             return R.fail("用户在其他地方登录！！！");
55.         }
56.         return R.ok();
57.     }
59.     @RequestMapping("/loginUser")
60.     public R loginUser(){
61.         return R.ok("success",loginUser.keySet());
62.     }
63. }

复制代码

基于webSocket通信的代码

1. package com.ttpfx.server;
3. import com.fasterxml.jackson.databind.ObjectMapper;
4. import com.ttpfx.entity.UserLog;
5. import com.ttpfx.model.ChatModel;
6. import com.ttpfx.service.UserLogService;
7. import com.ttpfx.service.UserService;
8. import com.ttpfx.vo.chat.ChatRequestParameter;
9. import org.springframework.stereotype.Component;
11. import javax.annotation.Resource;
12. import javax.websocket.*;
13. import javax.websocket.server.PathParam;
14. import javax.websocket.server.ServerEndpoint;
15. import java.io.IOException;
16. import java.time.LocalDateTime;
17. import java.util.concurrent.ConcurrentHashMap;
19. /**
20. * @author ttpfx
21. * @date 2023/3/28
22. */
23. @Component
24. @ServerEndpoint("/chatWebSocket/{username}")
25. public class ChatWebSocketServer {
27.     /**
28.      * 静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
29.      */
30.     private static int onlineCount = 0;
31.     /**
32.      * concurrent包的线程安全Map，用来存放每个客户端对应的MyWebSocket对象。
33.      */
34.     private static ConcurrentHashMap<String, ChatWebSocketServer> chatWebSocketMap = new ConcurrentHashMap<>();
36.     /**
37.      * 与某个客户端的连接会话，需要通过它来给客户端发送数据
38.      */
39.     private Session session;
40.     /**
41.      * 接收的username
42.      */
43.     private String username = "";
45.     private UserLog userLog;
47.     private static UserService userService;
48.     private static UserLogService userLogService;
50.     @Resource
51.     public void setUserService(UserService userService) {
52.         ChatWebSocketServer.userService = userService;
53.     }
55.     @Resource
56.     public void setUserLogService(UserLogService userLogService) {
57.         ChatWebSocketServer.userLogService = userLogService;
58.     }
60.     private ObjectMapper objectMapper = new ObjectMapper();
61.     private static ChatModel chatModel;
63.     @Resource
64.     public void setChatModel(ChatModel chatModel) {
65.         ChatWebSocketServer.chatModel = chatModel;
66.     }
68.     ChatRequestParameter chatRequestParameter = new ChatRequestParameter();
70.     /**
71.      * 建立连接
72.      * @param session 会话
73.      * @param username 连接用户名称
74.      */
75.     @OnOpen
76.     public void onOpen(Session session, @PathParam("username") String username) {
77.         this.session = session;
78.         this.username = username;
79.         this.userLog = new UserLog();
80.         // 这里的用户id不可能为null，出现null，那么就是非法请求
81.         try {
82.             this.userLog.setUserId(userService.queryByName(username).getId());
83.         } catch (Exception e) {
84.             e.printStackTrace();
85.             try {
86.                 session.close();
87.             } catch (IOException ex) {
88.                 ex.printStackTrace();
89.             }
90.         }
91.         this.userLog.setUsername(username);
92.         chatWebSocketMap.put(username, this);
93.         onlineCount++;
94.         System.out.println(username + "--open");
95.     }
97.     @OnClose
98.     public void onClose() {
99.         chatWebSocketMap.remove(username);
100.         System.out.println(username + "--close");
101.     }
103.     @OnMessage
104.     public void onMessage(String message, Session session) {
105.         System.out.println(username + "--" + message);
106.         // 记录日志
107.         this.userLog.setDateTime(LocalDateTime.now());
108.         this.userLog.setPreLogId(this.userLog.getLogId() == null ? -1 : this.userLog.getLogId());
109.         this.userLog.setLogId(null);
110.         this.userLog.setQuestion(message);
111.         long start = System.currentTimeMillis();
112.         // 这里就会返回结果
113.         String answer = chatModel.getAnswer(session, chatRequestParameter, message);
114.         long end = System.currentTimeMillis();
115.         this.userLog.setConsumeTime(end - start);
116.         this.userLog.setAnswer(answer);
117.         userLogService.save(userLog);
118.     }
120.     @OnError
121.     public void onError(Session session, Throwable error) {
122.         error.printStackTrace();
123.     }
125.     public void sendMessage(String message) throws IOException {
126.         this.session.getBasicRemote().sendText(message);
127.     }
129.     public static void sendInfo(String message, String toUserId) throws IOException {
130.         chatWebSocketMap.get(toUserId).sendMessage(message);
131.     }
132. }

复制代码

  我们只需要编写简单的前端代码，就可以实现和后端的socket通信。对于后端，我们只需要改一下apiKey和数据库配置就可以直接运行了。
  

---

微信小程序代码说明

我写了一个简单微信小程序来和后端进行通信，界面如下

各人只需要下载源代码，然将程序中的ip改为本身服务器的ip即可

---

代码链接

github的所在为 https://github.com/c-ttpfx/chatgpt-java-wx
可以直接使用 git clone https://github.com/c-ttpfx/chatgpt-java-wx.git 下载代码到本地
我在github里面说明白安装使用的基本步调，各人按照步调使用即可
总结

上面聊天小程序就是我花2天写出来的，可能会有一些bug，我本身测试的时间倒是没有怎么遇到bug，聊天和登录功能都能正常使用。
对于微信小程序，由于我不是专业搞前端的，就只东拼西凑实现了最基本的功能（登录、聊天），各人可以本身写一个，反正后端接口都提供好了嘛，也不是很难，不想写也可以迁就使用我的。
最后，也是最重要的，各人帮我的代码star一下！！！ 感谢各人了(≥▽≤)/(≥▽≤)/
更新日记

2023/5/13 14:42更新

对代码进行了重构，最新的代码已经支持代理，通过在application.yaml里面进行简单配置即可使用

1. gpt:
2.   proxy:
3.     host: 127.0.0.1
4.     port: 7890

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。