fix: loading status exit before response finish #14

OVINC-CN · Feb 4, 2024 · 700478a · 700478a
1 parent 3491634
commit 700478a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/components/ChatInput.vue b/src/components/ChatInput.vue
@@ -98,6 +98,7 @@ const doChat = async () => {
       .then((res) => {
         // check success
         if (!res.ok) {
+          emits('setChatLoading', false);
           Message.error(res.statusText);
           return;
         }
@@ -121,12 +122,14 @@ const doChat = async () => {
               lastResponseContent.value.content += value;
               read();
             }
+            if (chunk.done) {
+              emits('setChatLoading', false);
+            }
             emits('saveMessage');
           });
         }
         read();
-      })
-      .finally(() => emits('setChatLoading', false));
+      }, () => emits('setChatLoading', false));
 };
 
 const reGenerate = () => {