feat: quote change

src/api/fetch.ts

@@ -1,4 +1,4 @@
-import { NEW_CHATID_HEADER } from '@/constants/chat';
+import { NEW_CHATID_HEADER, QUOTE_LEN_HEADER } from '@/constants/chat';
 
 interface StreamFetchProps {
   url: string;
@@ -7,52 +7,57 @@ interface StreamFetchProps {
   abortSignal: AbortController;
 }
 export const streamFetch = ({ url, data, onMessage, abortSignal }: StreamFetchProps) =>
-  new Promise<{ responseText: string; newChatId: string }>(async (resolve, reject) => {
-    try {
-      const res = await fetch(url, {
-        method: 'POST',
-        headers: {
-          'Content-Type': 'application/json'
-        },
-        body: JSON.stringify(data),
-        signal: abortSignal.signal
-      });
-      const reader = res.body?.getReader();
-      if (!reader) return;
+  new Promise<{ responseText: string; newChatId: string; quoteLen: number }>(
+    async (resolve, reject) => {
+      try {
+        const res = await fetch(url, {
+          method: 'POST',
+          headers: {
+            'Content-Type': 'application/json'
+          },
+          body: JSON.stringify(data),
+          signal: abortSignal.signal
+        });
+        const reader = res.body?.getReader();
+        if (!reader) return;
 
-      const decoder = new TextDecoder();
+        const decoder = new TextDecoder();
 
-      const newChatId = decodeURIComponent(res.headers.get(NEW_CHATID_HEADER) || '');
+        const newChatId = decodeURIComponent(res.headers.get(NEW_CHATID_HEADER) || '');
+        const quoteLen = res.headers.get(QUOTE_LEN_HEADER)
+          ? Number(res.headers.get(QUOTE_LEN_HEADER))
+          : 0;
 
-      let responseText = '';
+        let responseText = '';
 
-      const read = async () => {
-        try {
-          const { done, value } = await reader?.read();
-          if (done) {
-            if (res.status === 200) {
-              resolve({ responseText, newChatId });
-            } else {
-              const parseError = JSON.parse(responseText);
-              reject(parseError?.message || '请求异常');
+        const read = async () => {
+          try {
+            const { done, value } = await reader?.read();
+            if (done) {
+              if (res.status === 200) {
+                resolve({ responseText, newChatId, quoteLen });
+              } else {
+                const parseError = JSON.parse(responseText);
+                reject(parseError?.message || '请求异常');
+              }
+
+              return;
             }
-
-            return;
+            const text = decoder.decode(value);
+            responseText += text;
+            onMessage(text);
+            read();
+          } catch (err: any) {
+            if (err?.message === 'The user aborted a request.') {
+              return resolve({ responseText, newChatId, quoteLen: 0 });
+            }
+            reject(typeof err === 'string' ? err : err?.message || '请求异常');
           }
-          const text = decoder.decode(value);
-          responseText += text;
-          onMessage(text);
-          read();
-        } catch (err: any) {
-          if (err?.message === 'The user aborted a request.') {
-            return resolve({ responseText, newChatId });
-          }
-          reject(typeof err === 'string' ? err : err?.message || '请求异常');
-        }
-      };
-      read();
-    } catch (err: any) {
-      console.log(err, '====');
-      reject(typeof err === 'string' ? err : err?.message || '请求异常');
+        };
+        read();
+      } catch (err: any) {
+        console.log(err, '====');
+        reject(typeof err === 'string' ? err : err?.message || '请求异常');
+      }
     }
-  });
+  );