monorepo packages (#344)

projects/app/src/pages/api/chat/chatTest.ts

@@ -0,0 +1,93 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { connectToDatabase } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+import { sseErrRes } from '@/service/response';
+import { sseResponseEventEnum } from '@/constants/chat';
+import { sseResponse } from '@/service/utils/tools';
+import { AppModuleItemType } from '@/types/app';
+import { dispatchModules } from '../openapi/v1/chat/completions';
+import { pushChatBill } from '@/service/common/bill/push';
+import { BillSourceEnum } from '@/constants/user';
+import { ChatItemType } from '@/types/chat';
+
+export type Props = {
+  history: ChatItemType[];
+  prompt: string;
+  modules: AppModuleItemType[];
+  variables: Record<string, any>;
+  appId: string;
+  appName: string;
+};
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  res.on('close', () => {
+    res.end();
+  });
+  res.on('error', () => {
+    console.log('error: ', 'request error');
+    res.end();
+  });
+
+  let { modules = [], history = [], prompt, variables = {}, appName, appId } = req.body as Props;
+  try {
+    if (!history || !modules || !prompt) {
+      throw new Error('Prams Error');
+    }
+    if (!Array.isArray(modules)) {
+      throw new Error('history is not array');
+    }
+
+    await connectToDatabase();
+
+    /* user auth */
+    const { userId, user } = await authUser({ req, authBalance: true });
+
+    if (!user) {
+      throw new Error('user not found');
+    }
+
+    /* start process */
+    const { responseData } = await dispatchModules({
+      res,
+      modules: modules,
+      variables,
+      user,
+      params: {
+        history,
+        userChatInput: prompt
+      },
+      stream: true,
+      detail: true
+    });
+
+    sseResponse({
+      res,
+      event: sseResponseEventEnum.answer,
+      data: '[DONE]'
+    });
+    sseResponse({
+      res,
+      event: sseResponseEventEnum.appStreamResponse,
+      data: JSON.stringify(responseData)
+    });
+    res.end();
+
+    pushChatBill({
+      appName,
+      appId,
+      userId,
+      source: BillSourceEnum.fastgpt,
+      response: responseData
+    });
+  } catch (err: any) {
+    res.status(500);
+    sseErrRes(res, err);
+    res.end();
+  }
+}
+
+export const config = {
+  api: {
+    responseLimit: '20mb'
+  }
+};

projects/app/src/pages/api/chat/delChatRecordByContentId.ts

@@ -0,0 +1,29 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, ChatItem } 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 };
+
+    await connectToDatabase();
+
+    // 凭证校验
+    const { userId } = await authUser({ req, authToken: true });
+
+    // 删除一条数据库记录
+    await ChatItem.deleteOne({
+      dataId: contentId,
+      chatId,
+      userId
+    });
+
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

projects/app/src/pages/api/chat/feedback/adminUpdate.ts

@@ -0,0 +1,40 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, ChatItem } from '@/service/mongo';
+import { AdminUpdateFeedbackParams } from '@/api/request/chat';
+import { authUser } from '@/service/utils/auth';
+
+/* 初始化我的聊天框，需要身份验证 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    await connectToDatabase();
+    const { chatItemId, kbId, dataId, content = undefined } = req.body as AdminUpdateFeedbackParams;
+
+    if (!chatItemId || !kbId || !dataId || !content) {
+      throw new Error('missing parameter');
+    }
+
+    const { userId } = await authUser({ req, authToken: true });
+
+    await ChatItem.findOneAndUpdate(
+      {
+        userId,
+        dataId: chatItemId
+      },
+      {
+        adminFeedback: {
+          kbId,
+          dataId,
+          content
+        }
+      }
+    );
+
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

projects/app/src/pages/api/chat/feedback/userUpdate.ts

@@ -0,0 +1,34 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, ChatItem } from '@/service/mongo';
+
+/* 初始化我的聊天框，需要身份验证 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    await connectToDatabase();
+    const { chatItemId, userFeedback = undefined } = req.body as {
+      chatItemId: string;
+      userFeedback?: string;
+    };
+
+    if (!chatItemId) {
+      throw new Error('chatItemId is required');
+    }
+
+    await ChatItem.findOneAndUpdate(
+      {
+        dataId: chatItemId
+      },
+      {
+        ...(userFeedback ? { userFeedback } : { $unset: { userFeedback: '' } })
+      }
+    );
+
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

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

@@ -0,0 +1,43 @@
+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 { ChatHistoryItemType } from '@/types/chat';
+import { ChatSourceEnum } from '@/constants/chat';
+
+/* 获取历史记录 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { appId } = req.body as { appId?: string };
+    const { userId } = await authUser({ req, authToken: true });
+
+    await connectToDatabase();
+
+    const data = await Chat.find(
+      {
+        userId,
+        source: ChatSourceEnum.online,
+        ...(appId && { appId })
+      },
+      'chatId title top customTitle appId updateTime'
+    )
+      .sort({ top: -1, updateTime: -1 })
+      .limit(20);
+
+    jsonRes<ChatHistoryItemType[]>(res, {
+      data: data.map((item) => ({
+        chatId: item.chatId,
+        updateTime: item.updateTime,
+        appId: item.appId,
+        customTitle: item.customTitle,
+        title: item.title,
+        top: item.top
+      }))
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

projects/app/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(
+      {
+        chatId,
+        userId
+      },
+      {
+        ...(customTitle ? { customTitle } : {}),
+        ...(top ? { top } : { top: null })
+      }
+    );
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

projects/app/src/pages/api/chat/init.ts

@@ -0,0 +1,108 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { Chat, ChatItem } from '@/service/mongo';
+import type { InitChatResponse } from '@/api/response/chat';
+import { authUser } from '@/service/utils/auth';
+import { ChatItemType } from '@/types/chat';
+import { authApp } from '@/service/utils/auth';
+import type { ChatSchema } from '@/types/mongoSchema';
+import { getSpecialModule, getChatModelNameList } from '@/components/ChatBox/utils';
+import { TaskResponseKeyEnum } from '@/constants/chat';
+
+/* 初始化我的聊天框，需要身份验证 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { userId } = await authUser({ req, authToken: true });
+
+    let { appId, chatId } = req.query as {
+      appId: '' | string;
+      chatId: '' | string;
+    };
+
+    if (!appId) {
+      return jsonRes(res, {
+        code: 501,
+        message: "You don't have an app yet"
+      });
+    }
+
+    // 校验使用权限
+    const app = (
+      await authApp({
+        appId,
+        userId,
+        authUser: false,
+        authOwner: false
+      })
+    ).app;
+
+    // get app and history
+    const { chat, history = [] }: { chat?: ChatSchema; history?: ChatItemType[] } =
+      await (async () => {
+        if (chatId) {
+          // auth chatId
+          const [chat, history] = await Promise.all([
+            Chat.findOne(
+              {
+                chatId,
+                userId,
+                appId
+              },
+              'title variables'
+            ),
+            ChatItem.find(
+              {
+                chatId,
+                userId,
+                appId
+              },
+              `dataId obj value adminFeedback userFeedback ${TaskResponseKeyEnum.responseData}`
+            )
+              .sort({ _id: -1 })
+              .limit(30)
+          ]);
+          if (!chat) {
+            throw new Error('聊天框不存在');
+          }
+          history.reverse();
+          return { app, history, chat };
+        }
+        return {};
+      })();
+
+    if (!app) {
+      throw new Error('Auth App Error');
+    }
+
+    const isOwner = String(app.userId) === userId;
+
+    jsonRes<InitChatResponse>(res, {
+      data: {
+        chatId,
+        appId,
+        app: {
+          ...getSpecialModule(app.modules),
+          chatModels: getChatModelNameList(app.modules),
+          name: app.name,
+          avatar: app.avatar,
+          intro: app.intro,
+          canUse: app.share.isShare || isOwner
+        },
+        title: chat?.title || '新对话',
+        variables: chat?.variables || {},
+        history
+      }
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}
+
+export const config = {
+  api: {
+    responseLimit: '10mb'
+  }
+};

projects/app/src/pages/api/chat/removeHistory.ts

@@ -0,0 +1,57 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Chat, ChatItem } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+import { ChatSourceEnum } from '@/constants/chat';
+
+type Props = {
+  chatId?: string;
+  appId?: string;
+};
+
+/* clear chat history */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { chatId, appId } = req.query as Props;
+    const { userId } = await authUser({ req, authToken: true });
+
+    await connectToDatabase();
+
+    if (chatId) {
+      await Promise.all([
+        Chat.findOneAndRemove({
+          chatId,
+          userId
+        }),
+        ChatItem.deleteMany({
+          userId,
+          chatId
+        })
+      ]);
+    }
+    if (appId) {
+      const chats = await Chat.find({
+        appId,
+        userId,
+        source: ChatSourceEnum.online
+      }).select('_id');
+      const chatIds = chats.map((chat) => chat._id);
+
+      await Promise.all([
+        Chat.deleteMany({
+          _id: { $in: chatIds }
+        }),
+        ChatItem.deleteMany({
+          chatId: { $in: chatIds }
+        })
+      ]);
+    }
+
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}