new framwork

client/src/api/fetch.ts

@@ -0,0 +1,67 @@
+import { GUIDE_PROMPT_HEADER, NEW_CHATID_HEADER, QUOTE_LEN_HEADER } from '@/constants/chat';
+
+interface StreamFetchProps {
+  url: string;
+  data: any;
+  onMessage: (text: string) => void;
+  abortSignal: AbortController;
+}
+export const streamFetch = ({ url, data, onMessage, abortSignal }: StreamFetchProps) =>
+  new Promise<{
+    responseText: string;
+    newChatId: string;
+    systemPrompt: 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 newChatId = decodeURIComponent(res.headers.get(NEW_CHATID_HEADER) || '');
+      const systemPrompt = decodeURIComponent(res.headers.get(GUIDE_PROMPT_HEADER) || '').trim();
+      const quoteLen = res.headers.get(QUOTE_LEN_HEADER)
+        ? Number(res.headers.get(QUOTE_LEN_HEADER))
+        : 0;
+
+      let responseText = '';
+
+      const read = async () => {
+        try {
+          const { done, value } = await reader?.read();
+          if (done) {
+            if (res.status === 200) {
+              resolve({ responseText, newChatId, quoteLen, systemPrompt });
+            } else {
+              const parseError = JSON.parse(responseText);
+              reject(parseError?.message || '请求异常');
+            }
+
+            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, systemPrompt });
+          }
+          reject(typeof err === 'string' ? err : err?.message || '请求异常');
+        }
+      };
+      read();
+    } catch (err: any) {
+      console.log(err, '====');
+      reject(typeof err === 'string' ? err : err?.message || '请求异常');
+    }
+  });