refresh-gpt-chat v0.0.3

1.新增打字机效果

pom.xml

@@ -10,7 +10,7 @@
     </parent>
     <groupId>com.yyandywt99</groupId>
     <artifactId>refresh-gpt-chat</artifactId>
-    <version>0.0.2-SNAPSHOT</version>
+    <version>0.0.1-SNAPSHOT</version>
     <name>refresh-gpt-chat</name>
     <description>refresh-gpt-chat</description>
     <properties>

src/main/java/com/refresh/gptChat/controller/chatController.java

@@ -317,23 +317,39 @@ private void outPutChat(HttpServletResponse response, Response resp, Object conv
         try {
             JSONObject jsonObject = new JSONObject(com.alibaba.fastjson2.JSON.toJSONString(conversation));
             boolean isStream = jsonObject.optBoolean("stream", false);
+            int one_messageByte;
+            int sleep_time;
             if (isStream) {
+                if (!jsonObject.optString("model", "gpt-3.5-turbo").contains("gpt-4")) {
+                    one_messageByte = 2048;
+                    sleep_time = 0;
+                } else {
+                    one_messageByte = 128;
+                    sleep_time = 25;
+                }
                 response.setContentType("text/event-stream; charset=UTF-8");
             } else {
+                one_messageByte = 8192;
+                sleep_time = 0;
                 response.setContentType("application/json; charset=utf-8");
             }
             OutputStream out = new BufferedOutputStream(response.getOutputStream());
             InputStream in = new BufferedInputStream(resp.body().byteStream());
             // 一次拿多少数据 迭代循环
-            byte[] buffer = new byte[8192];
+            byte[] buffer = new byte[one_messageByte];
             int bytesRead;
             while ((bytesRead = in.read(buffer)) != -1) {
                 out.write(buffer, 0, bytesRead);
                 out.flush();
+                try {
+                    if(sleep_time > 0){
+                        Thread.sleep(sleep_time);
+                    }
+                } catch (InterruptedException e) {
+                    throw new RuntimeException(e);
+                }
             }
-        } catch (IOException e) {
-            throw new RuntimeException(e);
-        } catch (JSONException e) {
+        } catch (IOException | JSONException e) {
             throw new RuntimeException(e);
         }
     }

src/main/java/com/refresh/gptChat/refreshGptChatApplication.java

@@ -35,7 +35,7 @@ public static void main(String[] args) {
     @PostConstruct
     public void initialize() {
         log.info("------------------------------------------------------");
-        log.info("----------原神refresh-gpt-chat启动成功------------");
+        log.info("----------原神refresh-gpt-chat v0.0.3启动成功------------");
         log.info("URL地址：http://0.0.0.0:" + serverPort + contextPath +"");
         log.info("------------------------------------------------------");
     }