feat: quote change

src/api/chat.ts

@@ -5,7 +5,7 @@ import { RequestPaging } from '../types/index';
 import type { ShareChatSchema } from '@/types/mongoSchema';
 import type { ShareChatEditType } from '@/types/model';
 import { Obj2Query } from '@/utils/tools';
-import { Response as LastChatResultResponseType } from '@/pages/api/openapi/chat/lastChatResult';
+import { QuoteItemType } from '@/pages/api/openapi/kb/appKbSearch';
 
 /**
  * 获取初始化聊天内容
@@ -25,10 +25,20 @@ export const getChatHistory = (data: RequestPaging) =>
 export const delChatHistoryById = (id: string) => GET(`/chat/removeHistory?id=${id}`);
 
 /**
- * get latest chat result by chatId
+ * get history quotes
  */
-export const getChatResult = (chatId: string) =>
-  GET<LastChatResultResponseType>('/openapi/chat/lastChatResult', { chatId });
+export const getHistoryQuote = (params: { chatId: string; historyId: string }) =>
+  GET<(QuoteItemType & { _id: string })[]>(`/chat/getHistoryQuote`, params);
+
+/**
+ * update history quote status
+ */
+export const updateHistoryQuote = (params: {
+  chatId: string;
+  historyId: string;
+  quoteId: string;
+}) => GET(`/chat/updateHistoryQuote`, params);
+
 /**
  * 删除一句对话
  */

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 || '请求异常');
+      }
     }
-  });
+  );

src/api/plugins/kb.ts

@@ -40,10 +40,8 @@ export const getTrainingData = (kbId: string) =>
     embeddingQueue: number;
   }>(`/plugins/kb/data/getTrainingData?kbId=${kbId}`);
 
-/**
- * 获取 web 页面内容
- */
-export const getWebContent = (url: string) => POST<string>(`/model/data/fetchingUrlData`, { url });
+export const getKbDataItemById = (dataId: string) =>
+  GET(`/plugins/kb/data/getDataById`, { dataId });
 
 /**
  * 直接push数据