feat: 修改chat的数据结构

src/pages/api/chat/chatGpt.ts

@@ -1,24 +1,23 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { createParser, ParsedEvent, ReconnectInterval } from 'eventsource-parser';
-import { connectToDatabase, ChatWindow } from '@/service/mongo';
-import type { ModelType } from '@/types/model';
+import { connectToDatabase, Chat } from '@/service/mongo';
 import { getOpenAIApi, authChat } from '@/service/utils/chat';
 import { httpsAgent } from '@/service/utils/tools';
 import { ChatCompletionRequestMessage, ChatCompletionRequestMessageRoleEnum } from 'openai';
 import { ChatItemType } from '@/types/chat';
 import { jsonRes } from '@/service/response';
+import type { ModelSchema } from '@/types/mongoSchema';
 import { PassThrough } from 'stream';
 
 /* 发送提示词 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  const { chatId, windowId, prompt } = req.body as {
+  const { chatId, prompt } = req.body as {
     prompt: ChatItemType;
-    windowId: string;
     chatId: string;
   };
 
   try {
-    if (!windowId || !chatId || !prompt) {
+    if (!chatId || !prompt) {
       throw new Error('缺少参数');
     }
 
@@ -26,11 +25,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     const { chat, userApiKey } = await authChat(chatId);
 
-    const model: ModelType = chat.modelId;
+    const model: ModelSchema = chat.modelId;
 
     // 读取对话内容
-    const prompts: ChatItemType[] = (await ChatWindow.findById(windowId)).content;
-    prompts.push(prompt);
+    const prompts = [...chat.content, prompt];
 
     // 上下文长度过滤
     const maxContext = model.security.contextMaxLen;
@@ -49,6 +47,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
         content: item.value
       })
     );
+
     // 如果有系统提示词，自动插入
     if (model.systemPrompt) {
       formatPrompts.unshift({

src/pages/api/chat/delLastMessage.ts

@@ -1,19 +1,19 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase, ChatWindow } from '@/service/mongo';
+import { connectToDatabase, Chat } from '@/service/mongo';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { windowId } = req.query as { windowId: string };
+    const { chatId } = req.query as { chatId: string };
 
-    if (!windowId) {
+    if (!chatId) {
       throw new Error('缺少参数');
     }
 
     await connectToDatabase();
 
     // 删除最一条数据库记录, 也就是预发送的那一条
-    await ChatWindow.findByIdAndUpdate(windowId, {
+    await Chat.findByIdAndUpdate(chatId, {
       $pop: { content: 1 },
       updateTime: Date.now()
     });

src/pages/api/chat/generate.ts

@@ -2,7 +2,7 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, Model, Chat } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
-import { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
@@ -24,7 +24,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     await connectToDatabase();
 
     // 获取模型配置
-    const model: ModelType | null = await Model.findOne({
+    const model = await Model.findOne<ModelSchema>({
       _id: modelId,
       userId
     });
@@ -38,11 +38,13 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       userId,
       modelId,
       expiredTime: Date.now() + model.security.expiredTime,
-      loadAmount: model.security.maxLoadAmount
+      loadAmount: model.security.maxLoadAmount,
+      updateTime: Date.now(),
+      content: []
     });
 
     jsonRes(res, {
-      data: response._id
+      data: response._id // 即聊天框的 ID
     });
   } catch (err) {
     jsonRes(res, {

src/pages/api/chat/gpt3.ts

@@ -1,9 +1,8 @@
 // Next.js API route support: https://nextjs.org/docs/api-routes/introduction
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase, Chat } from '@/service/mongo';
-import type { ModelType } from '@/types/model';
-import { getOpenAIApi } from '@/service/utils/chat';
+import { connectToDatabase } from '@/service/mongo';
+import { getOpenAIApi, authChat } from '@/service/utils/chat';
 import { ChatItemType } from '@/types/chat';
 import { httpsAgent } from '@/service/utils/tools';
 
@@ -18,35 +17,9 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     await connectToDatabase();
 
-    // 获取 chat 数据
-    const chat = await Chat.findById(chatId)
-      .populate({
-        path: 'modelId',
-        options: {
-          strictPopulate: false
-        }
-      })
-      .populate({
-        path: 'userId',
-        options: {
-          strictPopulate: false
-        }
-      });
+    const { chat, userApiKey } = await authChat(chatId);
 
-    if (!chat || !chat.modelId || !chat.userId) {
-      throw new Error('聊天已过期');
-    }
-
-    const model: ModelType = chat.modelId;
-
-    // 获取 user 的 apiKey
-    const user = chat.userId;
-
-    const userApiKey = user.accounts?.find((item: any) => item.type === 'openai')?.value;
-
-    if (!userApiKey) {
-      throw new Error('缺少ApiKey, 无法请求');
-    }
+    const model = chat.modelId;
 
     // 获取 chatAPI
     const chatAPI = getOpenAIApi(userApiKey);

src/pages/api/chat/init.ts

@@ -1,12 +1,13 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase, Chat, ChatWindow } from '@/service/mongo';
-import type { ModelType } from '@/types/model';
+import { connectToDatabase, Chat } from '@/service/mongo';
+import type { ChatPopulate } from '@/types/mongoSchema';
+import type { InitChatResponse } from '@/api/response/chat';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { chatId, windowId } = req.query as { chatId: string; windowId?: string };
+    const { chatId } = req.query as { chatId: string };
 
     if (!chatId) {
       throw new Error('缺少参数');
@@ -15,16 +16,15 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
 
     // 获取 chat 数据
-    const chat = await Chat.findById(chatId).populate({
+    const chat = await Chat.findById<ChatPopulate>(chatId).populate({
       path: 'modelId',
       options: {
         strictPopulate: false
       }
     });
 
-    // 安全校验
-    if (!chat || chat.loadAmount === 0 || chat.expiredTime < Date.now()) {
-      throw new Error('聊天框已过期');
+    if (!chat) {
+      throw new Error('聊天框不存在');
     }
 
     if (chat.loadAmount > 0) {
@@ -38,38 +38,18 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       );
     }
 
-    const model: ModelType = chat.modelId;
-
-    /* 查找是否有记录 */
-    let history = null;
-    let responseId = windowId;
-    try {
-      history = await ChatWindow.findById(windowId);
-    } catch (error) {
-      error;
-    }
-
-    if (!history) {
-      // 没有记录，创建一个
-      const response = await ChatWindow.create({
-        chatId,
-        updateTime: Date.now(),
-        content: []
-      });
-      responseId = response._id;
-    }
-
-    jsonRes(res, {
+    const model = chat.modelId;
+    jsonRes<InitChatResponse>(res, {
+      code: 201,
       data: {
-        windowId: responseId,
-        chatSite: {
-          modelId: model._id,
-          name: model.name,
-          avatar: model.avatar,
-          secret: model.security,
-          chatModel: model.service.chatModel
-        },
-        history: history ? history.content : []
+        chatId: chat._id,
+        isExpiredTime: chat.loadAmount === 0 || chat.expiredTime <= Date.now(),
+        modelId: model._id,
+        name: model.name,
+        avatar: model.avatar,
+        secret: model.security,
+        chatModel: model.service.chatModel,
+        history: chat.content
       }
     });
   } catch (err) {

src/pages/api/chat/saveChat.ts

@@ -1,24 +1,24 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { ChatItemType } from '@/types/chat';
-import { connectToDatabase, ChatWindow } from '@/service/mongo';
+import { connectToDatabase, Chat } from '@/service/mongo';
 
 /* 聊天内容存存储 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { windowId, prompts } = req.body as {
-      windowId: string;
+    const { chatId, prompts } = req.body as {
+      chatId: string;
       prompts: ChatItemType[];
     };
 
-    if (!windowId || !prompts) {
+    if (!chatId || !prompts) {
       throw new Error('缺少参数');
     }
 
     await connectToDatabase();
 
     // 存入库
-    await ChatWindow.findByIdAndUpdate(windowId, {
+    await Chat.findByIdAndUpdate(chatId, {
       $push: {
         content: {
           $each: prompts.map((item) => ({