new framwork

client/src/pages/api/chat/chat.ts

@@ -0,0 +1,195 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { connectToDatabase } from '@/service/mongo';
+import { authChat } from '@/service/utils/auth';
+import { modelServiceToolMap } from '@/service/utils/chat';
+import { ChatItemType } from '@/types/chat';
+import { jsonRes } from '@/service/response';
+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, QUOTE_LEN_HEADER, GUIDE_PROMPT_HEADER } from '@/constants/chat';
+import { BillTypeEnum } from '@/constants/user';
+import { sensitiveCheck } from '@/service/api/text';
+import { NEW_CHATID_HEADER } from '@/constants/chat';
+import { saveChat } from './saveChat';
+import { Types } from 'mongoose';
+
+/* 发送提示词 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  res.on('close', () => {
+    res.end();
+  });
+  res.on('error', () => {
+    console.log('error: ', 'request error');
+    res.end();
+  });
+
+  try {
+    const { chatId, prompt, modelId } = req.body as {
+      prompt: [ChatItemType, ChatItemType];
+      modelId: string;
+      chatId?: string;
+    };
+
+    if (!modelId || !prompt) {
+      throw new Error('缺少参数');
+    }
+
+    await connectToDatabase();
+    let startTime = Date.now();
+
+    const { model, showModelDetail, content, userOpenAiKey, systemAuthKey, userId } =
+      await authChat({
+        modelId,
+        chatId,
+        req
+      });
+
+    const modelConstantsData = ChatModelMap[model.chat.chatModel];
+
+    // 读取对话内容
+    const prompts = [...content, prompt[0]];
+
+    const {
+      code = 200,
+      systemPrompts = [],
+      quote = [],
+      guidePrompt = ''
+    } = await (async () => {
+      // 使用了知识库搜索
+      if (model.chat.relatedKbs.length > 0) {
+        const { code, searchPrompts, rawSearch, guidePrompt } = await appKbSearch({
+          model,
+          userId,
+          fixedQuote: content[content.length - 1]?.quote || [],
+          prompt: prompt[0],
+          similarity: ModelVectorSearchModeMap[model.chat.searchMode]?.similarity
+        });
+
+        return {
+          code,
+          quote: rawSearch,
+          systemPrompts: searchPrompts,
+          guidePrompt
+        };
+      }
+      if (model.chat.systemPrompt) {
+        return {
+          guidePrompt: model.chat.systemPrompt,
+          systemPrompts: [
+            {
+              obj: ChatRoleEnum.System,
+              value: model.chat.systemPrompt
+            }
+          ]
+        };
+      }
+      return {};
+    })();
+
+    // get conversationId. create a newId if it is null
+    const conversationId = chatId || String(new Types.ObjectId());
+    !chatId && res.setHeader(NEW_CHATID_HEADER, conversationId);
+    if (showModelDetail) {
+      guidePrompt && res.setHeader(GUIDE_PROMPT_HEADER, encodeURIComponent(guidePrompt));
+      res.setHeader(QUOTE_LEN_HEADER, quote.length);
+    }
+
+    // search result is empty
+    if (code === 201) {
+      const response = systemPrompts[0]?.value;
+      await saveChat({
+        chatId,
+        newChatId: conversationId,
+        modelId,
+        prompts: [
+          prompt[0],
+          {
+            ...prompt[1],
+            quote: [],
+            value: response
+          }
+        ],
+        userId
+      });
+      return res.end(response);
+    }
+
+    prompts.unshift(...systemPrompts);
+
+    // content check
+    await sensitiveCheck({
+      input: [...systemPrompts, prompt[0]].map((item) => item.value).join('')
+    });
+
+    // 计算温度
+    const temperature = (modelConstantsData.maxTemperature * (model.chat.temperature / 10)).toFixed(
+      2
+    );
+
+    // 发出 chat 请求
+    const { streamResponse, responseMessages } = await modelServiceToolMap[
+      model.chat.chatModel
+    ].chatCompletion({
+      apiKey: userOpenAiKey || systemAuthKey,
+      temperature: +temperature,
+      messages: prompts,
+      stream: true,
+      res,
+      chatId: conversationId
+    });
+
+    console.log('api response time:', `${(Date.now() - startTime) / 1000}s`);
+
+    if (res.closed) return res.end();
+
+    try {
+      const { totalTokens, finishMessages, responseContent } = await resStreamResponse({
+        model: model.chat.chatModel,
+        res,
+        chatResponse: streamResponse,
+        prompts: responseMessages
+      });
+
+      // save chat
+      await saveChat({
+        chatId,
+        newChatId: conversationId,
+        modelId,
+        prompts: [
+          prompt[0],
+          {
+            ...prompt[1],
+            value: responseContent,
+            quote: showModelDetail ? quote : [],
+            systemPrompt: showModelDetail ? guidePrompt : ''
+          }
+        ],
+        userId
+      });
+
+      res.end();
+
+      // 只有使用平台的 key 才计费
+      pushChatBill({
+        isPay: !userOpenAiKey,
+        chatModel: model.chat.chatModel,
+        userId,
+        chatId: conversationId,
+        textLen: finishMessages.map((item) => item.value).join('').length,
+        tokens: totalTokens,
+        type: BillTypeEnum.chat
+      });
+    } catch (error) {
+      res.end();
+      console.log('error，结束', error);
+    }
+  } catch (err: any) {
+    res.status(500);
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/chat/delChatRecordByContentId.ts

@@ -0,0 +1,44 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Chat } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { chatId, contentId } = req.query as {
+      chatId: string;
+      contentId: string;
+    };
+
+    if (!chatId || !contentId) {
+      throw new Error('缺少参数');
+    }
+
+    await connectToDatabase();
+
+    // 凭证校验
+    const { userId } = await authUser({ req, authToken: true });
+
+    const chatRecord = await Chat.findById(chatId);
+
+    if (!chatRecord) {
+      throw new Error('找不到对话');
+    }
+
+    // 删除一条数据库记录
+    await Chat.updateOne(
+      {
+        _id: chatId,
+        userId
+      },
+      { $pull: { content: { _id: contentId } } }
+    );
+
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

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

@@ -0,0 +1,39 @@
+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) {
+  try {
+    const { userId } = await authUser({ req, authToken: true });
+
+    await connectToDatabase();
+
+    const data = await Chat.find(
+      {
+        userId
+      },
+      '_id title top customTitle modelId updateTime latestChat'
+    )
+      .sort({ top: -1, updateTime: -1 })
+      .limit(20);
+
+    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, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/chat/history/getHistoryQuote.ts

@@ -0,0 +1,52 @@
+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
+    });
+  }
+}

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

@@ -0,0 +1,38 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Chat } from '@/service/mongo';
+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, customTitle, top } = req.body as Props;
+
+    const { userId } = await authUser({ req, authToken: true });
+
+    await connectToDatabase();
+
+    await Chat.findOneAndUpdate(
+      {
+        _id: chatId,
+        userId
+      },
+      {
+        ...(customTitle ? { customTitle } : {}),
+        ...(top ? { top } : { top: null })
+      }
+    );
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/chat/history/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].source': '手动修改'
+        }
+      },
+      {
+        arrayFilters: [
+          {
+            'quoteElem.id': quoteId
+          }
+        ]
+      }
+    );
+
+    jsonRes(res, {
+      data: ''
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/chat/init.ts

@@ -0,0 +1,111 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Chat, Model } from '@/service/mongo';
+import type { InitChatResponse } from '@/api/response/chat';
+import { authUser } from '@/service/utils/auth';
+import { ChatItemType } from '@/types/chat';
+import { authModel } from '@/service/utils/auth';
+import mongoose from 'mongoose';
+import { ModelStatusEnum } from '@/constants/model';
+import type { ModelSchema } from '@/types/mongoSchema';
+
+/* 初始化我的聊天框，需要身份验证 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { userId } = await authUser({ req, authToken: true });
+
+    let { modelId, chatId } = req.query as {
+      modelId: '' | string;
+      chatId: '' | string;
+    };
+
+    await connectToDatabase();
+
+    let model: ModelSchema;
+
+    // 没有 modelId 时，直接获取用户的第一个id
+    if (!modelId) {
+      const myModel = await Model.findOne({ userId });
+      if (!myModel) {
+        const { _id } = await Model.create({
+          name: '应用1',
+          userId,
+          status: ModelStatusEnum.running
+        });
+        model = (await Model.findById(_id)) as ModelSchema;
+      } else {
+        model = myModel;
+      }
+      modelId = model._id;
+    } else {
+      // 校验使用权限
+      const authRes = await authModel({
+        modelId,
+        userId,
+        authUser: false,
+        authOwner: false
+      });
+      model = authRes.model;
+    }
+
+    // 历史记录
+    let history: ChatItemType[] = [];
+
+    if (chatId) {
+      // auth chatId
+      const chat = await Chat.countDocuments({
+        _id: chatId,
+        userId
+      });
+      if (chat === 0) {
+        throw new Error('聊天框不存在');
+      }
+      // 获取 chat.content 数据
+      history = await Chat.aggregate([
+        {
+          $match: {
+            _id: new mongoose.Types.ObjectId(chatId),
+            userId: new mongoose.Types.ObjectId(userId)
+          }
+        },
+        {
+          $project: {
+            content: {
+              $slice: ['$content', -50] // 返回 content 数组的最后50个元素
+            }
+          }
+        },
+        { $unwind: '$content' },
+        {
+          $project: {
+            _id: '$content._id',
+            obj: '$content.obj',
+            value: '$content.value',
+            systemPrompt: '$content.systemPrompt',
+            quoteLen: { $size: { $ifNull: ['$content.quote', []] } }
+          }
+        }
+      ]);
+    }
+
+    jsonRes<InitChatResponse>(res, {
+      data: {
+        chatId: chatId || '',
+        modelId: modelId,
+        model: {
+          name: model.name,
+          avatar: model.avatar,
+          intro: model.share.intro,
+          canUse: model.share.isShare || String(model.userId) === userId
+        },
+        chatModel: model.chat.chatModel,
+        history
+      }
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/chat/removeHistory.ts

@@ -0,0 +1,26 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Chat } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+
+/* 获取历史记录 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { id } = req.query;
+    const { userId } = await authUser({ req, authToken: true });
+
+    await connectToDatabase();
+
+    await Chat.findOneAndRemove({
+      _id: id,
+      userId
+    });
+
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/chat/saveChat.ts

@@ -0,0 +1,101 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { ChatItemType } from '@/types/chat';
+import { connectToDatabase, Chat, Model } from '@/service/mongo';
+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 Props;
+
+    if (!prompts) {
+      throw new Error('缺少参数');
+    }
+
+    const { userId } = await authUser({ req, authToken: true });
+
+    const nId = await saveChat({
+      chatId,
+      modelId,
+      prompts,
+      newChatId,
+      userId
+    });
+
+    jsonRes(res, {
+      data: nId
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}
+
+export async function saveChat({
+  chatId,
+  newChatId,
+  modelId,
+  prompts,
+  userId
+}: Props & { userId: string }) {
+  await connectToDatabase();
+  const { model } = 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,
+    systemPrompt: item.systemPrompt,
+    quote: item.quote || []
+  }));
+
+  const [id] = await Promise.all([
+    ...(chatId // update chat
+      ? [
+          Chat.findByIdAndUpdate(chatId, {
+            $push: {
+              content: {
+                $each: content
+              }
+            },
+            title: content[0].value.slice(0, 20),
+            latestChat: content[1].value,
+            updateTime: new Date()
+          }).then(() => '')
+        ]
+      : [
+          Chat.create({
+            _id: newChatId ? new mongoose.Types.ObjectId(newChatId) : undefined,
+            userId,
+            modelId,
+            content,
+            title: content[0].value.slice(0, 20),
+            latestChat: content[1].value
+          }).then((res) => res._id)
+        ]),
+    // update model
+    ...(String(model.userId) === userId
+      ? [
+          Model.findByIdAndUpdate(modelId, {
+            updateTime: new Date()
+          })
+        ]
+      : [])
+  ]);
+
+  return {
+    id
+  };
+}

client/src/pages/api/chat/shareChat/chat.ts

@@ -0,0 +1,140 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { connectToDatabase } from '@/service/mongo';
+import { authShareChat } from '@/service/utils/auth';
+import { modelServiceToolMap } from '@/service/utils/chat';
+import { ChatItemSimpleType } from '@/types/chat';
+import { jsonRes } from '@/service/response';
+import { ChatModelMap, ModelVectorSearchModeMap } from '@/constants/model';
+import { pushChatBill, updateShareChatBill } from '@/service/events/pushBill';
+import { resStreamResponse } from '@/service/utils/chat';
+import { ChatRoleEnum } from '@/constants/chat';
+import { BillTypeEnum } from '@/constants/user';
+import { sensitiveCheck } from '@/service/api/text';
+import { appKbSearch } from '../../openapi/kb/appKbSearch';
+
+/* 发送提示词 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  res.on('error', () => {
+    console.log('error: ', 'request error');
+    res.end();
+  });
+
+  try {
+    const { shareId, password, historyId, prompts } = req.body as {
+      prompts: ChatItemSimpleType[];
+      password: string;
+      shareId: string;
+      historyId: string;
+    };
+
+    if (!historyId || !prompts) {
+      throw new Error('分享链接无效');
+    }
+
+    await connectToDatabase();
+    let startTime = Date.now();
+
+    const { model, userOpenAiKey, systemAuthKey, userId } = await authShareChat({
+      shareId,
+      password
+    });
+
+    const modelConstantsData = ChatModelMap[model.chat.chatModel];
+
+    const { code = 200, systemPrompts = [] } = await (async () => {
+      // 使用了知识库搜索
+      if (model.chat.relatedKbs.length > 0) {
+        const { code, searchPrompts } = await appKbSearch({
+          model,
+          userId,
+          fixedQuote: [],
+          prompt: prompts[prompts.length - 1],
+          similarity: ModelVectorSearchModeMap[model.chat.searchMode]?.similarity
+        });
+
+        return {
+          code,
+          systemPrompts: searchPrompts
+        };
+      }
+      if (model.chat.systemPrompt) {
+        return {
+          systemPrompts: [
+            {
+              obj: ChatRoleEnum.System,
+              value: model.chat.systemPrompt
+            }
+          ]
+        };
+      }
+      return {};
+    })();
+
+    // search result is empty
+    if (code === 201) {
+      return res.send(systemPrompts[0]?.value);
+    }
+
+    prompts.unshift(...systemPrompts);
+
+    // content check
+    await sensitiveCheck({
+      input: [...systemPrompts, prompts[prompts.length - 1]].map((item) => item.value).join('')
+    });
+
+    // 计算温度
+    const temperature = (modelConstantsData.maxTemperature * (model.chat.temperature / 10)).toFixed(
+      2
+    );
+
+    // 发出请求
+    const { streamResponse, responseMessages } = await modelServiceToolMap[
+      model.chat.chatModel
+    ].chatCompletion({
+      apiKey: userOpenAiKey || systemAuthKey,
+      temperature: +temperature,
+      messages: prompts,
+      stream: true,
+      res,
+      chatId: historyId
+    });
+
+    console.log('api response time:', `${(Date.now() - startTime) / 1000}s`);
+
+    if (res.closed) return res.end();
+
+    try {
+      const { totalTokens, finishMessages } = await resStreamResponse({
+        model: model.chat.chatModel,
+        res,
+        chatResponse: streamResponse,
+        prompts: responseMessages
+      });
+
+      res.end();
+
+      /* bill */
+      pushChatBill({
+        isPay: !userOpenAiKey,
+        chatModel: model.chat.chatModel,
+        userId,
+        textLen: finishMessages.map((item) => item.value).join('').length,
+        tokens: totalTokens,
+        type: BillTypeEnum.chat
+      });
+      updateShareChatBill({
+        shareId,
+        tokens: totalTokens
+      });
+    } catch (error) {
+      res.end();
+      console.log('error，结束', error);
+    }
+  } catch (err: any) {
+    res.status(500);
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/chat/shareChat/create.ts

@@ -0,0 +1,40 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, ShareChat } from '@/service/mongo';
+import { authModel, authUser } from '@/service/utils/auth';
+import type { ShareChatEditType } from '@/types/model';
+
+/* create a shareChat */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { modelId, name, maxContext, password } = req.body as ShareChatEditType & {
+      modelId: string;
+    };
+
+    await connectToDatabase();
+
+    const { userId } = await authUser({ req, authToken: true });
+    await authModel({
+      modelId,
+      userId,
+      authOwner: false
+    });
+
+    const { _id } = await ShareChat.create({
+      userId,
+      modelId,
+      name,
+      password,
+      maxContext
+    });
+
+    jsonRes(res, {
+      data: _id
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/chat/shareChat/delete.ts

@@ -0,0 +1,29 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, ShareChat } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+
+/* delete a shareChat by shareChatId */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { id } = req.query as {
+      id: string;
+    };
+
+    await connectToDatabase();
+
+    const { userId } = await authUser({ req, authToken: true });
+
+    await ShareChat.findOneAndRemove({
+      _id: id,
+      userId
+    });
+
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/chat/shareChat/init.ts

@@ -0,0 +1,64 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, ShareChat, User } from '@/service/mongo';
+import type { InitShareChatResponse } from '@/api/response/chat';
+import { authModel } from '@/service/utils/auth';
+import { hashPassword } from '@/service/utils/tools';
+import { HUMAN_ICON } from '@/constants/chat';
+
+/* 初始化我的聊天框，需要身份验证 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    let { shareId, password = '' } = req.query as {
+      shareId: string;
+      password: string;
+    };
+
+    if (!shareId) {
+      throw new Error('params is error');
+    }
+
+    await connectToDatabase();
+
+    // get shareChat
+    const shareChat = await ShareChat.findById(shareId);
+
+    if (!shareChat) {
+      throw new Error('分享链接已失效');
+    }
+
+    if (shareChat.password !== hashPassword(password)) {
+      return jsonRes(res, {
+        code: 501,
+        message: '密码不正确'
+      });
+    }
+
+    // 校验使用权限
+    const { model } = await authModel({
+      modelId: shareChat.modelId,
+      userId: String(shareChat.userId),
+      authOwner: false
+    });
+
+    const user = await User.findById(shareChat.userId, 'avatar');
+
+    jsonRes<InitShareChatResponse>(res, {
+      data: {
+        maxContext: shareChat.maxContext,
+        userAvatar: user?.avatar || HUMAN_ICON,
+        model: {
+          name: model.name,
+          avatar: model.avatar,
+          intro: model.share.intro
+        },
+        chatModel: model.chat.chatModel
+      }
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/chat/shareChat/list.ts

@@ -0,0 +1,43 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, ShareChat } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+import { hashPassword } from '@/service/utils/tools';
+
+/* get shareChat list by modelId */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { modelId } = req.query as {
+      modelId: string;
+    };
+
+    await connectToDatabase();
+
+    const { userId } = await authUser({ req, authToken: true });
+
+    const data = await ShareChat.find({
+      modelId,
+      userId
+    }).sort({
+      _id: -1
+    });
+
+    const blankPassword = hashPassword('');
+
+    jsonRes(res, {
+      data: data.map((item) => ({
+        _id: item._id,
+        name: item.name,
+        password: item.password === blankPassword ? '' : '1',
+        tokens: item.tokens,
+        maxContext: item.maxContext,
+        lastTime: item.lastTime
+      }))
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}