feat: chat quote

src/utils/chat/claude.ts

@@ -1,3 +0,0 @@
-export const ClaudeSliceTextByToken = ({ text, length }: { text: string; length: number }) => {
-  return text.slice(0, length);
-};

src/utils/file.ts

@@ -1,6 +1,6 @@
 import mammoth from 'mammoth';
 import Papa from 'papaparse';
-import { getOpenAiEncMap } from './chat/openai';
+import { getOpenAiEncMap } from './plugin/openai';
 
 /**
  * 读取 txt 文件内容

src/utils/plugin/index.ts

@@ -2,29 +2,37 @@ import { ClaudeEnum, OpenAiChatEnum } from '@/constants/model';
 import type { ChatModelType } from '@/constants/model';
 import type { ChatItemSimpleType } from '@/types/chat';
 import { countOpenAIToken, openAiSliceTextByToken } from './openai';
-import { ClaudeSliceTextByToken } from './claude';
+import { gpt_chatItemTokenSlice } from '@/pages/api/openapi/text/gptMessagesSlice';
 
 export const modelToolMap: Record<
   ChatModelType,
   {
     countTokens: (data: { messages: ChatItemSimpleType[] }) => number;
     sliceText: (data: { text: string; length: number }) => string;
+    tokenSlice: (data: {
+      messages: ChatItemSimpleType[];
+      maxToken: number;
+    }) => ChatItemSimpleType[];
   }
 > = {
   [OpenAiChatEnum.GPT35]: {
     countTokens: ({ messages }) => countOpenAIToken({ model: OpenAiChatEnum.GPT35, messages }),
-    sliceText: (data) => openAiSliceTextByToken({ model: OpenAiChatEnum.GPT35, ...data })
+    sliceText: (data) => openAiSliceTextByToken({ model: OpenAiChatEnum.GPT35, ...data }),
+    tokenSlice: (data) => gpt_chatItemTokenSlice({ model: OpenAiChatEnum.GPT35, ...data })
   },
   [OpenAiChatEnum.GPT4]: {
     countTokens: ({ messages }) => countOpenAIToken({ model: OpenAiChatEnum.GPT4, messages }),
-    sliceText: (data) => openAiSliceTextByToken({ model: OpenAiChatEnum.GPT4, ...data })
+    sliceText: (data) => openAiSliceTextByToken({ model: OpenAiChatEnum.GPT4, ...data }),
+    tokenSlice: (data) => gpt_chatItemTokenSlice({ model: OpenAiChatEnum.GPT4, ...data })
   },
   [OpenAiChatEnum.GPT432k]: {
     countTokens: ({ messages }) => countOpenAIToken({ model: OpenAiChatEnum.GPT432k, messages }),
-    sliceText: (data) => openAiSliceTextByToken({ model: OpenAiChatEnum.GPT432k, ...data })
+    sliceText: (data) => openAiSliceTextByToken({ model: OpenAiChatEnum.GPT432k, ...data }),
+    tokenSlice: (data) => gpt_chatItemTokenSlice({ model: OpenAiChatEnum.GPT432k, ...data })
   },
   [ClaudeEnum.Claude]: {
     countTokens: ({ messages }) => countOpenAIToken({ model: OpenAiChatEnum.GPT35, messages }),
-    sliceText: (data) => openAiSliceTextByToken({ model: OpenAiChatEnum.GPT35, ...data })
+    sliceText: (data) => openAiSliceTextByToken({ model: OpenAiChatEnum.GPT35, ...data }),
+    tokenSlice: (data) => gpt_chatItemTokenSlice({ model: OpenAiChatEnum.GPT35, ...data })
   }
 };

src/utils/tools.ts

@@ -126,3 +126,10 @@ export const getErrText = (err: any, def = '') => {
   msg && console.log('error =>', msg);
   return msg;
 };
+
+export const delay = (ms: number) =>
+  new Promise((resolve) => {
+    setTimeout(() => {
+      resolve('');
+    }, ms);
+  });