feat: custom title and set history top

src/pages/api/chat/history/getHistory.ts

@@ -2,6 +2,7 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, Chat } from '@/service/mongo';
 import { authUser } from '@/service/utils/auth';
+import type { HistoryItemType } from '@/types/chat';
 
 /* 获取历史记录 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
@@ -14,13 +15,20 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       {
         userId
       },
-      '_id title modelId updateTime latestChat'
+      '_id title top customTitle modelId updateTime latestChat'
     )
-      .sort({ updateTime: -1 })
+      .sort({ top: -1, updateTime: -1 })
       .limit(20);
 
-    jsonRes(res, {
-      data
+    jsonRes<HistoryItemType[]>(res, {
+      data: data.map((item) => ({
+        _id: item._id,
+        updateTime: item.updateTime,
+        modelId: item.modelId,
+        title: item.customTitle || item.title,
+        latestChat: item.latestChat,
+        top: item.top
+      }))
     });
   } catch (err) {
     jsonRes(res, {

src/pages/api/chat/history/updateChatHistory.ts

@@ -1,30 +1,32 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, Chat } from '@/service/mongo';
-import { authModel } from '@/service/utils/auth';
 import { authUser } from '@/service/utils/auth';
 
+export type Props = {
+  chatId: '' | string;
+  customTitle?: string;
+  top?: boolean;
+};
+
 /* 更新聊天标题 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { chatId, modelId, newTitle } = req.body as {
-      chatId: '' | string;
-      modelId: '' | string;
-      newTitle: string;
-    };
+    const { chatId, customTitle, top } = req.body as Props;
 
     const { userId } = await authUser({ req, authToken: true });
 
     await connectToDatabase();
 
-    await authModel({ modelId, userId, authOwner: false });
-
-    await Chat.findByIdAndUpdate(
-      chatId,
+    await Chat.findOneAndUpdate(
       {
-        title: newTitle,
-        customTitle: true
-      } // 自定义标题}
+        _id: chatId,
+        userId
+      },
+      {
+        ...(customTitle ? { customTitle } : {}),
+        ...(top ? { top } : { top: null })
+      }
     );
     jsonRes(res);
   } catch (err) {

src/pages/api/chat/saveChat.ts

@@ -77,15 +77,13 @@ export async function saveChat({
     });
     return _id;
   } else {
-    // 已经有记录，追加入库
-    const chat = await Chat.findById(chatId);
     await Chat.findByIdAndUpdate(chatId, {
       $push: {
         content: {
           $each: content
         }
       },
-      ...(chat && !chat.customTitle ? { title: content[0].value.slice(0, 20) } : {}),
+      title: content[0].value.slice(0, 20),
       latestChat: content[1].value,
       updateTime: new Date()
     });