perf: ts type

package.json

@@ -26,6 +26,7 @@
     "axios": "^1.3.3",
     "crypto": "^1.0.1",
     "dayjs": "^1.11.7",
+    "delay": "^5.0.0",
     "eventsource-parser": "^0.1.0",
     "formidable": "^2.1.1",
     "framer-motion": "^9.0.6",

src/service/utils/chat/index.ts

@@ -6,20 +6,39 @@ import { OpenAiChatEnum } from '@/constants/model';
 import { chatResponse, openAiStreamResponse } from './openai';
 import type { NextApiResponse } from 'next';
 import type { PassThrough } from 'stream';
+import delay from 'delay';
 
 export type ChatCompletionType = {
   apiKey: string;
   temperature: number;
   messages: ChatItemSimpleType[];
   stream: boolean;
+  params?: any;
+};
+export type ChatCompletionResponseType = {
+  streamResponse: any;
+  responseMessages: ChatItemSimpleType[];
+  responseText: string;
+  totalTokens: number;
 };
 export type StreamResponseType = {
   stream: PassThrough;
   chatResponse: any;
   prompts: ChatItemSimpleType[];
 };
+export type StreamResponseReturnType = {
+  responseContent: string;
+  totalTokens: number;
+  finishMessages: ChatItemSimpleType[];
+};
 
-export const modelServiceToolMap = {
+export const modelServiceToolMap: Record<
+  ChatModelType,
+  {
+    chatCompletion: (data: ChatCompletionType) => Promise<ChatCompletionResponseType>;
+    streamResponse: (data: StreamResponseType) => Promise<StreamResponseReturnType>;
+  }
+> = {
   [OpenAiChatEnum.GPT35]: {
     chatCompletion: (data: ChatCompletionType) =>
       chatResponse({ model: OpenAiChatEnum.GPT35, ...data }),
@@ -142,16 +161,16 @@ export const resStreamResponse = async ({
     prompts
   });
 
-  setTimeout(() => {
-    // push system prompt
-    !stream.destroyed &&
-      systemPrompt &&
-      stream.push(`${SYSTEM_PROMPT_PREFIX}${systemPrompt.replace(/\n/g, '<br/>')}`);
+  await delay(100);
+
+  // push system prompt
+  !stream.destroyed &&
+    systemPrompt &&
+    stream.push(`${SYSTEM_PROMPT_PREFIX}${systemPrompt.replace(/\n/g, '<br/>')}`);
 
-    // close stream
-    !stream.destroyed && stream.push(null);
-    stream.destroy();
-  }, 100);
+  // close stream
+  !stream.destroyed && stream.push(null);
+  stream.destroy();
 
   return { responseContent, totalTokens, finishMessages };
 };

src/utils/chat/index.ts

@@ -2,10 +2,17 @@ import { OpenAiChatEnum } from '@/constants/model';
 import type { ChatModelType } from '@/constants/model';
 import type { ChatItemSimpleType } from '@/types/chat';
 import { countOpenAIToken, getOpenAiEncMap, adaptChatItem_openAI } from './openai';
+import { ChatCompletionRequestMessage } from 'openai';
 
 export type CountTokenType = { messages: ChatItemSimpleType[] };
 
-export const modelToolMap = {
+export const modelToolMap: Record<
+  ChatModelType,
+  {
+    countTokens: (data: CountTokenType) => number;
+    adaptChatMessages: (data: CountTokenType) => ChatCompletionRequestMessage[];
+  }
+> = {
   [OpenAiChatEnum.GPT35]: {
     countTokens: ({ messages }: CountTokenType) =>
       countOpenAIToken({ model: OpenAiChatEnum.GPT35, messages }),