Преглед изворни кода

feat: 优化playground样式

1808837298@qq.com пре 1 година
родитељ
комит
04a68a85dd
1 измењених фајлова са 8 додато и 3 уклоњено
  1. 8 3
      web/src/components/Playground.js

+ 8 - 3
web/src/components/Playground.js

@@ -144,7 +144,7 @@ const Playground = () => {
 
     source.addEventListener("error", (e) => {
       generateMockResponse(e.data)
-      completeMessage();
+      completeMessage('error')
     });
 
     source.addEventListener("readystatechange", (e) => {
@@ -206,12 +206,17 @@ const Playground = () => {
     });
   }, [getSystemMessage]);
 
-  const completeMessage = useCallback(() => {
+  const completeMessage = useCallback((status = 'complete') => {
+    // console.log("Complete Message: ", status)
     setMessage((prevMessage) => {
       const lastMessage = prevMessage[prevMessage.length - 1];
+      // only change the status if the last message is not complete and not error
+      if (lastMessage.status === 'complete' || lastMessage.status === 'error') {
+        return prevMessage;
+      }
       return [
         ...prevMessage.slice(0, -1),
-        { ...lastMessage, status: 'complete' }
+        { ...lastMessage, status: status }
       ];
     });
   }, [])