perf: save chat and del chat content;UI

src/pages/api/chat/chat.ts

@@ -2,7 +2,7 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { connectToDatabase } from '@/service/mongo';
 import { getOpenAIApi, authChat } from '@/service/utils/auth';
 import { axiosConfig, openaiChatFilter, systemPromptFilter } from '@/service/utils/tools';
-import { ChatItemType } from '@/types/chat';
+import { ChatItemSimpleType } from '@/types/chat';
 import { jsonRes } from '@/service/response';
 import { PassThrough } from 'stream';
 import { modelList, ModelVectorSearchModeMap, ModelVectorSearchModeEnum } from '@/constants/model';
@@ -28,7 +28,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
   try {
     const { chatId, prompt, modelId } = req.body as {
-      prompt: ChatItemType;
+      prompt: ChatItemSimpleType;
       modelId: string;
       chatId: '' | string;
     };
@@ -118,7 +118,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     const filterPrompts = openaiChatFilter({
       model: model.chat.chatModel,
       prompts,
-      maxTokens: modelConstantsData.contextMaxToken - 500
+      maxTokens: modelConstantsData.contextMaxToken - 300
     });
 
     // 计算温度

src/pages/api/chat/delChatRecordByContentId.ts

@@ -5,13 +5,13 @@ import { authToken } from '@/service/utils/auth';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { chatId, index } = req.query as { chatId: string; index: string };
+    const { chatId, contentId } = req.query as { chatId: string; contentId: string };
     const { authorization } = req.headers;
 
     if (!authorization) {
       throw new Error('无权操作');
     }
-    if (!chatId || !index) {
+    if (!chatId || !contentId) {
       throw new Error('缺少参数');
     }
 
@@ -26,30 +26,13 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       throw new Error('找不到对话');
     }
 
-    // 重新计算 index，跳过已经被删除的内容
-    let unDeleteIndex = +index;
-    let deletedIndex = 0;
-    for (deletedIndex = 0; deletedIndex < chatRecord.content.length; deletedIndex++) {
-      if (!chatRecord.content[deletedIndex].deleted) {
-        unDeleteIndex--;
-        if (unDeleteIndex < 0) {
-          break;
-        }
-      }
-    }
-
-    // 删除最一条数据库记录, 也就是预发送的那一条
+    // 删除一条数据库记录
     await Chat.updateOne(
       {
         _id: chatId,
         userId
       },
-      {
-        $set: {
-          [`content.${deletedIndex}.deleted`]: true,
-          updateTime: Date.now()
-        }
-      }
+      { $pull: { content: { _id: contentId } } }
     );
 
     jsonRes(res);

src/pages/api/chat/init.ts

@@ -36,20 +36,22 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
             userId: new mongoose.Types.ObjectId(userId)
           }
         },
-        { $unwind: '$content' },
-        { $match: { 'content.deleted': false } },
-        { $sort: { 'content._id': -1 } },
-        { $limit: 50 },
         {
           $project: {
-            id: '$content._id',
+            content: {
+              $slice: ['$content', -50] // 返回 content 数组的最后50个元素
+            }
+          }
+        },
+        { $unwind: '$content' },
+        {
+          $project: {
+            _id: '$content._id',
             obj: '$content.obj',
             value: '$content.value'
           }
         }
       ]);
-
-      history.reverse();
     }
 
     jsonRes<InitChatResponse>(res, {

src/pages/api/chat/saveChat.ts

@@ -4,6 +4,7 @@ import { ChatItemType } from '@/types/chat';
 import { connectToDatabase, Chat } from '@/service/mongo';
 import { authModel } from '@/service/utils/auth';
 import { authToken } from '@/service/utils/auth';
+import mongoose from 'mongoose';
 
 /* 聊天内容存存储 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
@@ -23,6 +24,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
 
     const content = prompts.map((item) => ({
+      _id: new mongoose.Types.ObjectId(item._id),
       obj: item.obj,
       value: item.value
     }));