From aa74625f9657013a3fd2d6076f382a845559de0e Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Wed, 3 May 2023 10:27:06 +0800
Subject: [PATCH] fix: system prompt response

---
 src/api/fetch.ts | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/src/api/fetch.ts b/src/api/fetch.ts
index 3296fe71c..ab07e15d5 100644
--- a/src/api/fetch.ts
+++ b/src/api/fetch.ts
@@ -39,15 +39,16 @@ export const streamFetch = ({ url, data, onMessage, abortSignal }: StreamFetchPr
 
             return;
           }
-          const text = decoder.decode(value).replace(/<br\/>/g, '\n');
-
+          let text = decoder.decode(value).replace(/<br\/>/g, '\n');
           // check system prompt
-          if (text.startsWith(SYSTEM_PROMPT_PREFIX)) {
-            systemPrompt = text.replace(SYSTEM_PROMPT_PREFIX, '');
-          } else {
-            responseText += text;
-            onMessage(text);
+          if (text.includes(SYSTEM_PROMPT_PREFIX)) {
+            const arr = text.split(SYSTEM_PROMPT_PREFIX);
+            systemPrompt = arr.pop() || '';
+
+            text = arr.join('');
           }
+          responseText += text;
+          onMessage(text);
 
           read();
         } catch (err: any) {