up 4.0.2

Author: PlexPt

src/main/java/com/plexpt/chatgpt/exception/ChatException.java

@@ -1,13 +1,18 @@
 package com.plexpt.chatgpt.exception;
 
 /**
- * 异常
+ * Custom exception class for chat-related errors
  *
  * @author plexpt
  */
 public class ChatException extends RuntimeException {
 
 
+    /**
+     * Constructs a new ChatException with the specified detail message.
+     *
+     * @param message the detail message (which is saved for later retrieval by the getMessage() method)
+     */
     public ChatException(String msg) {
         super(msg);
     }

src/main/java/com/plexpt/chatgpt/listener/AbstractStreamListener.java

@@ -19,45 +19,59 @@
 import okhttp3.sse.EventSourceListener;
 
 /**
- * sse
+ * EventSource listener for chat-related events.
  *
  * @author plexpt
  */
 @Slf4j
 public abstract class AbstractStreamListener extends EventSourceListener {
 
-    protected String last = "";
+    protected String lastMessage = "";
+
+
+    /**
+     * Called when all new message are received.
+     *
+     * @param message the new message
+     */
     @Setter
     @Getter
     protected Consumer<String> onComplate = s -> {
 
     };
 
-
     /**
+     * Called when a new message is received.
      * 收到消息 单个字
+     *
+     * @param message the new message
      */
-    public abstract void onMsg(String msg);
-
+    public abstract void onMsg(String message);
 
     /**
-     * 出错了
+     * Called when an error occurs.
+     * 出错时调用
+     *
+     * @param throwable the throwable that caused the error
+     * @param response  the response associated with the error, if any
      */
-    public abstract void onError(Throwable t, String response);
+    public abstract void onError(Throwable throwable, String response);
 
     @Override
     public void onOpen(EventSource eventSource, Response response) {
+        // do nothing
     }
 
     @Override
     public void onClosed(EventSource eventSource) {
+        // do nothing
     }
 
     @Override
     public void onEvent(EventSource eventSource, String id, String type, String data) {
         if (data.equals("[DONE]")) {
-            log.info("回答完成：{}", last);
-            onComplate.accept(last);
+            log.info("Chat session completed: {}", lastMessage);
+            onComplate.accept(lastMessage);
             return;
         }
 
@@ -71,7 +85,7 @@ public void onEvent(EventSource eventSource, String id, String type, String data
         String text = delta.getContent();
 
         if (text != null) {
-            last += text;
+            lastMessage += text;
 
             onMsg(text);
 
@@ -82,28 +96,37 @@ public void onEvent(EventSource eventSource, String id, String type, String data
 
     @SneakyThrows
     @Override
-    public void onFailure(EventSource eventSource, Throwable t, Response response) {
+    public void onFailure(EventSource eventSource, Throwable throwable, Response response) {
 
         try {
-            log.error("stream连接异常:{}", t);
+            log.error("Stream connection error: {}", throwable);
 
-            String res = "";
+            String responseText = "";
 
             if (Objects.nonNull(response)) {
-                res = response.body().string();
+                responseText = response.body().string();
             }
 
-            log.error("response：{}", res);
+            log.error("response：{}", responseText);
 
-            String seq = "Your access was terminated due to violation of our policies";
+            String forbiddenText = "Your access was terminated due to violation of our policies";
 
-            if (StrUtil.contains(res, seq)) {
+            if (StrUtil.contains(responseText, forbiddenText)) {
+                log.error("Chat session has been terminated due to policy violation");
                 log.error("检测到号被封了");
             }
 
-            onError(t, res);
+            String overloadedText = "That model is currently overloaded with other requests.";
+
+            if (StrUtil.contains(responseText, overloadedText)) {
+                log.error("检测到官方超载了，赶紧优化你的代码，做重试吧");
+            }
+
+            this.onError(throwable, responseText);
 
         } catch (Exception e) {
+            log.warn("onFailure error:{}", e);
+            // do nothing
 
         } finally {
             eventSource.cancel();

src/main/java/com/plexpt/chatgpt/listener/ConsoleStreamListener.java

@@ -4,6 +4,7 @@
 
 /**
  * 控制台测试
+ * Console Stream Test Listener
  *
  * @author plexpt
  */
@@ -12,12 +13,12 @@ public class ConsoleStreamListener extends AbstractStreamListener {
 
 
     @Override
-    public void onMsg(String msg) {
-        System.out.print(msg);
+    public void onMsg(String message) {
+        System.out.print(message);
     }
 
     @Override
-    public void onError(Throwable t, String response) {
+    public void onError(Throwable throwable, String response) {
 
     }
 

src/main/java/com/plexpt/chatgpt/listener/SseStreamListener.java

@@ -20,12 +20,12 @@ public class SseStreamListener extends AbstractStreamListener {
 
 
     @Override
-    public void onMsg(String msg) {
-        SseHelper.send(sseEmitter, msg);
+    public void onMsg(String message) {
+        SseHelper.send(sseEmitter, message);
     }
 
     @Override
-    public void onError(Throwable t, String response) {
+    public void onError(Throwable throwable, String response) {
         SseHelper.complete(sseEmitter);
     }