feat: lafClaude

src/pages/api/chat/chat.ts

@@ -42,11 +42,12 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
     let startTime = Date.now();
 
-    const { model, showModelDetail, content, userApiKey, systemApiKey, userId } = await authChat({
-      modelId,
-      chatId,
-      authorization
-    });
+    const { model, showModelDetail, content, userOpenAiKey, systemAuthKey, userId } =
+      await authChat({
+        modelId,
+        chatId,
+        authorization
+      });
 
     const modelConstantsData = ChatModelMap[model.chat.chatModel];
 
@@ -56,8 +57,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     // 使用了知识库搜索
     if (model.chat.useKb) {
       const { code, searchPrompt } = await searchKb({
-        userApiKey,
-        systemApiKey,
+        userOpenAiKey,
         prompts,
         similarity: ModelVectorSearchModeMap[model.chat.searchMode]?.similarity,
         model,
@@ -86,10 +86,12 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     // 发出请求
     const { streamResponse } = await modelServiceToolMap[model.chat.chatModel].chatCompletion({
-      apiKey: userApiKey || systemApiKey,
+      apiKey: userOpenAiKey || systemAuthKey,
       temperature: +temperature,
       messages: prompts,
-      stream: true
+      stream: true,
+      res,
+      chatId
     });
 
     console.log('api response time:', `${(Date.now() - startTime) / 1000}s`);
@@ -108,7 +110,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     // 只有使用平台的 key 才计费
     pushChatBill({
-      isPay: !userApiKey,
+      isPay: !userOpenAiKey,
       chatModel: model.chat.chatModel,
       userId,
       chatId,

src/pages/api/chat/saveChat.ts

@@ -9,7 +9,8 @@ import mongoose from 'mongoose';
 /* 聊天内容存存储 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { chatId, modelId, prompts } = req.body as {
+    const { chatId, modelId, prompts, newChatId } = req.body as {
+      newChatId: '' | string;
       chatId: '' | string;
       modelId: string;
       prompts: ChatItemType[];
@@ -35,6 +36,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     // 没有 chatId, 创建一个对话
     if (!chatId) {
       const { _id } = await Chat.create({
+        _id: newChatId ? new mongoose.Types.ObjectId(newChatId) : undefined,
         userId,
         modelId,
         content,

src/pages/api/openapi/chat/chat.ts

@@ -65,7 +65,6 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       const similarity = ModelVectorSearchModeMap[model.chat.searchMode]?.similarity || 0.22;
 
       const { code, searchPrompt } = await searchKb({
-        systemApiKey: apiKey,
         prompts,
         similarity,
         model,

src/pages/api/openapi/chat/lafGpt.ts

@@ -116,7 +116,6 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     // 获取向量匹配到的提示词
     const { searchPrompt } = await searchKb({
-      systemApiKey: apiKey,
       similarity: ModelVectorSearchModeMap[model.chat.searchMode]?.similarity,
       prompts,
       model,

src/pages/chat/index.tsx

@@ -206,7 +206,7 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
       };
 
       // 流请求，获取数据
-      const { responseText, systemPrompt } = await streamFetch({
+      let { responseText, systemPrompt, newChatId } = await streamFetch({
         url: '/api/chat/chat',
         data: {
           prompt,
@@ -234,10 +234,10 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
         return;
       }
 
-      let newChatId = '';
       // save chat record
       try {
         newChatId = await postSaveChat({
+          newChatId, // 如果有newChatId，会自动以这个Id创建对话框
           modelId,
           chatId,
           prompts: [