feat: quote change

src/pages/api/chat/chat.ts

@@ -8,7 +8,7 @@ import { ChatModelMap, ModelVectorSearchModeMap } from '@/constants/model';
 import { pushChatBill } from '@/service/events/pushBill';
 import { resStreamResponse } from '@/service/utils/chat';
 import { appKbSearch } from '../openapi/kb/appKbSearch';
-import { ChatRoleEnum } from '@/constants/chat';
+import { ChatRoleEnum, QUOTE_LEN_HEADER } from '@/constants/chat';
 import { BillTypeEnum } from '@/constants/user';
 import { sensitiveCheck } from '@/service/api/text';
 import { NEW_CHATID_HEADER } from '@/constants/chat';
@@ -85,7 +85,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     // get conversationId. create a newId if it is null
     const conversationId = chatId || String(new Types.ObjectId());
-    !chatId && res?.setHeader(NEW_CHATID_HEADER, conversationId);
+    !chatId && res.setHeader(NEW_CHATID_HEADER, conversationId);
+    res.setHeader(QUOTE_LEN_HEADER, quote.length);
 
     // search result is empty
     if (code === 201) {

src/pages/api/chat/getHistoryQuote.ts

@@ -0,0 +1,49 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Chat } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+import { Types } from 'mongoose';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { chatId, historyId } = req.query as { chatId: string; historyId: string };
+    await connectToDatabase();
+
+    const { userId } = await authUser({ req, authToken: true });
+
+    if (!chatId || !historyId) {
+      throw new Error('params is error');
+    }
+
+    const history = await Chat.aggregate([
+      {
+        $match: {
+          _id: new Types.ObjectId(chatId),
+          userId: new Types.ObjectId(userId)
+        }
+      },
+      {
+        $unwind: '$content'
+      },
+      {
+        $match: {
+          'content._id': new Types.ObjectId(historyId)
+        }
+      },
+      {
+        $project: {
+          quote: '$content.quote'
+        }
+      }
+    ]);
+
+    jsonRes(res, {
+      data: history[0]?.quote || []
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/chat/init.ts

@@ -73,7 +73,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
             _id: '$content._id',
             obj: '$content.obj',
             value: '$content.value',
-            quote: '$content.quote'
+            quoteLen: { $size: '$content.quote' }
           }
         }
       ]);

src/pages/api/chat/saveChat.ts

@@ -57,7 +57,10 @@ export async function saveChat({
     _id: item._id ? new mongoose.Types.ObjectId(item._id) : undefined,
     obj: item.obj,
     value: item.value,
-    quote: item.quote
+    quote: item.quote?.map((item) => ({
+      ...item,
+      isEdit: false
+    }))
   }));
 
   // 没有 chatId, 创建一个对话

src/pages/api/chat/updateHistoryQuote.ts

@@ -0,0 +1,51 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Chat } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+import { Types } from 'mongoose';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    let { chatId, historyId, quoteId } = req.query as {
+      chatId: string;
+      historyId: string;
+      quoteId: string;
+    };
+    await connectToDatabase();
+
+    const { userId } = await authUser({ req, authToken: true });
+
+    if (!chatId || !historyId || !quoteId) {
+      throw new Error('params is error');
+    }
+
+    await Chat.updateOne(
+      {
+        _id: new Types.ObjectId(chatId),
+        userId: new Types.ObjectId(userId),
+        'content._id': new Types.ObjectId(historyId)
+      },
+      {
+        $set: {
+          'content.$.quote.$[quoteElem].isEdit': true
+        }
+      },
+      {
+        arrayFilters: [
+          {
+            'quoteElem.id': quoteId
+          }
+        ]
+      }
+    );
+
+    jsonRes(res, {
+      data: ''
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}