feat: chat quote

src/pages/api/chat/saveChat.ts

@@ -6,15 +6,17 @@ import { authModel } from '@/service/utils/auth';
 import { authUser } from '@/service/utils/auth';
 import mongoose from 'mongoose';
 
+type Props = {
+  newChatId?: string;
+  chatId?: string;
+  modelId: string;
+  prompts: [ChatItemType, ChatItemType];
+};
+
 /* 聊天内容存存储 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { chatId, modelId, prompts, newChatId } = req.body as {
-      newChatId: '' | string;
-      chatId: '' | string;
-      modelId: string;
-      prompts: [ChatItemType, ChatItemType];
-    };
+    const { chatId, modelId, prompts, newChatId } = req.body as Props;
 
     if (!prompts) {
       throw new Error('缺少参数');
@@ -22,44 +24,17 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     const { userId } = await authUser({ req, authToken: true });
 
-    await connectToDatabase();
+    const nId = await saveChat({
+      chatId,
+      modelId,
+      prompts,
+      newChatId,
+      userId
+    });
 
-    const content = prompts.map((item) => ({
-      _id: new mongoose.Types.ObjectId(item._id),
-      obj: item.obj,
-      value: item.value,
-      systemPrompt: item.systemPrompt
-    }));
-
-    await authModel({ modelId, userId, authOwner: false });
-
-    // 没有 chatId, 创建一个对话
-    if (!chatId) {
-      const { _id } = await Chat.create({
-        _id: newChatId ? new mongoose.Types.ObjectId(newChatId) : undefined,
-        userId,
-        modelId,
-        content,
-        title: content[0].value.slice(0, 20),
-        latestChat: content[1].value
-      });
-      return jsonRes(res, {
-        data: _id
-      });
-    } else {
-      // 已经有记录，追加入库
-      await Chat.findByIdAndUpdate(chatId, {
-        $push: {
-          content: {
-            $each: content
-          }
-        },
-        title: content[0].value.slice(0, 20),
-        latestChat: content[1].value,
-        updateTime: new Date()
-      });
-    }
-    jsonRes(res);
+    jsonRes(res, {
+      data: nId
+    });
   } catch (err) {
     jsonRes(res, {
       code: 500,
@@ -67,3 +42,46 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     });
   }
 }
+
+export async function saveChat({
+  chatId,
+  newChatId,
+  modelId,
+  prompts,
+  userId
+}: Props & { userId: string }) {
+  await connectToDatabase();
+  await authModel({ modelId, userId, authOwner: false });
+
+  const content = prompts.map((item) => ({
+    _id: item._id ? new mongoose.Types.ObjectId(item._id) : undefined,
+    obj: item.obj,
+    value: item.value,
+    quote: item.quote
+  }));
+
+  // 没有 chatId, 创建一个对话
+  if (!chatId) {
+    const { _id } = await Chat.create({
+      _id: newChatId ? new mongoose.Types.ObjectId(newChatId) : undefined,
+      userId,
+      modelId,
+      content,
+      title: content[0].value.slice(0, 20),
+      latestChat: content[1].value
+    });
+    return _id;
+  } else {
+    // 已经有记录，追加入库
+    await Chat.findByIdAndUpdate(chatId, {
+      $push: {
+        content: {
+          $each: content
+        }
+      },
+      title: content[0].value.slice(0, 20),
+      latestChat: content[1].value,
+      updateTime: new Date()
+    });
+  }
+}