chore

josStorer · Apr 3, 2024 · ff6240d · ff6240d
1 parent f6614ff
commit ff6240d
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/frontend/src/pages/Chat.tsx b/frontend/src/pages/Chat.tsx
@@ -583,8 +583,9 @@ const ChatPanel: FC = observer(() => {
         }),
         signal: chatSseController?.signal,
         onmessage(e) {
+          if (finished) return;
           scrollToBottom();
-          if (!finished && e.data.trim() === '[DONE]') {
+          if (e.data.trim() === '[DONE]') {
             finish();
             return;
           }
@@ -598,7 +599,7 @@ const ChatPanel: FC = observer(() => {
           if (data.model)
             commonStore.setLastModelName(data.model);
           if (data.choices && Array.isArray(data.choices) && data.choices.length > 0) {
-            if (!finished && data.choices[0]?.finish_reason) {
+            if (data.choices[0]?.finish_reason) {
               finish();
               return;
             }

diff --git a/frontend/src/pages/Completion.tsx b/frontend/src/pages/Completion.tsx
@@ -109,8 +109,9 @@ const CompletionPanel: FC = observer(() => {
         }),
         signal: completionSseController?.signal,
         onmessage(e) {
+          if (finished) return;
           scrollToBottom();
-          if (!finished && e.data.trim() === '[DONE]') {
+          if (e.data.trim() === '[DONE]') {
             finish();
             return;
           }
@@ -124,7 +125,7 @@ const CompletionPanel: FC = observer(() => {
           if (data.model)
             commonStore.setLastModelName(data.model);
           if (data.choices && Array.isArray(data.choices) && data.choices.length > 0) {
-            if (!finished && data.choices[0]?.finish_reason) {
+            if (data.choices[0]?.finish_reason) {
               finish();
               return;
             }

diff --git a/frontend/src/pages/Composition.tsx b/frontend/src/pages/Composition.tsx
@@ -214,8 +214,9 @@ const CompositionPanel: FC = observer(() => {
         }),
         signal: compositionSseController?.signal,
         onmessage(e) {
+          if (finished) return;
           scrollToBottom();
-          if (!finished && e.data.trim() === '[DONE]') {
+          if (e.data.trim() === '[DONE]') {
             finish();
             return;
           }
@@ -229,7 +230,7 @@ const CompositionPanel: FC = observer(() => {
           if (data.model)
             commonStore.setLastModelName(data.model);
           if (data.choices && Array.isArray(data.choices) && data.choices.length > 0) {
-            if (!finished && data.choices[0]?.finish_reason) {
+            if (data.choices[0]?.finish_reason) {
               finish();
               return;
             }