chat box

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

@@ -0,0 +1,77 @@
+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 { type ChatCompletionRequestMessage } from 'openai';
+import { AppModuleItemType } from '@/types/app';
+import { dispatchModules } from '../openapi/v1/chat/completions2';
+
+export type MessageItemType = ChatCompletionRequestMessage & { _id?: string };
+export type Props = {
+  history: MessageItemType[];
+  prompt: string;
+  modules: AppModuleItemType[];
+  variable: Record<string, any>;
+};
+export type ChatResponseType = {
+  newChatId: string;
+  quoteLen?: number;
+};
+
+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, variable = {} } = 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 } = await authUser({ req });
+
+    /* start process */
+    const { responseData } = await dispatchModules({
+      res,
+      modules: modules,
+      variable,
+      params: {
+        history,
+        userChatInput: prompt
+      },
+      stream: true
+    });
+
+    sseResponse({
+      res,
+      event: sseResponseEventEnum.answer,
+      data: '[DONE]'
+    });
+    sseResponse({
+      res,
+      event: sseResponseEventEnum.appStreamResponse,
+      data: JSON.stringify(responseData)
+    });
+    res.end();
+
+    // bill
+  } catch (err: any) {
+    res.status(500);
+    sseErrRes(res, err);
+    res.end();
+  }
+}

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

@@ -3,33 +3,36 @@ import { jsonRes } from '@/service/response';
 import { connectToDatabase, ShareChat } from '@/service/mongo';
 import { authApp, authUser } from '@/service/utils/auth';
 import type { ShareChatEditType } from '@/types/app';
+import { customAlphabet } from 'nanoid';
+const nanoid = customAlphabet('abcdefghijklmnopqrstuvwxyz1234567890', 24);
 
 /* create a shareChat */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { modelId, name, maxContext, password } = req.body as ShareChatEditType & {
-      modelId: string;
+    const { appId, name, maxContext } = req.body as ShareChatEditType & {
+      appId: string;
     };
 
     await connectToDatabase();
 
     const { userId } = await authUser({ req, authToken: true });
     await authApp({
-      appId: modelId,
+      appId,
       userId,
       authOwner: false
     });
 
-    const { _id } = await ShareChat.create({
+    const shareId = nanoid();
+    await ShareChat.create({
+      shareId,
       userId,
-      modelId,
+      appId,
       name,
-      password,
       maxContext
     });
 
     jsonRes(res, {
-      data: _id
+      data: shareId
     });
   } catch (err) {
     jsonRes(res, {

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

@@ -5,13 +5,14 @@ import type { InitShareChatResponse } from '@/api/response/chat';
 import { authApp } from '@/service/utils/auth';
 import { hashPassword } from '@/service/utils/tools';
 import { HUMAN_ICON } from '@/constants/chat';
+import { FlowModuleTypeEnum } from '@/constants/flow';
+import { SystemInputEnum } from '@/constants/app';
 
 /* 初始化我的聊天框，需要身份验证 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    let { shareId, password = '' } = req.query as {
+    let { shareId } = req.query as {
       shareId: string;
-      password: string;
     };
 
     if (!shareId) {
@@ -21,22 +22,18 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
 
     // get shareChat
-    const shareChat = await ShareChat.findById(shareId);
+    const shareChat = await ShareChat.findOne({ shareId });
 
     if (!shareChat) {
-      throw new Error('分享链接已失效');
-    }
-
-    if (shareChat.password !== hashPassword(password)) {
       return jsonRes(res, {
         code: 501,
-        message: '密码不正确'
+        error: '分享链接已失效'
       });
     }
 
     // 校验使用权限
     const { app } = await authApp({
-      appId: shareChat.modelId,
+      appId: shareChat.appId,
       userId: String(shareChat.userId),
       authOwner: false
     });
@@ -45,15 +42,22 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     jsonRes<InitShareChatResponse>(res, {
       data: {
-        appId: shareChat.modelId,
-        maxContext: shareChat.maxContext,
         userAvatar: user?.avatar || HUMAN_ICON,
-        model: {
+        maxContext:
+          app.modules
+            ?.find((item) => item.flowType === FlowModuleTypeEnum.historyNode)
+            ?.inputs?.find((item) => item.key === 'maxContext')?.value || 0,
+        app: {
+          variableModules: app.modules
+            .find((item) => item.flowType === FlowModuleTypeEnum.userGuide)
+            ?.inputs?.find((item) => item.key === SystemInputEnum.variables)?.value,
+          welcomeText: app.modules
+            .find((item) => item.flowType === FlowModuleTypeEnum.userGuide)
+            ?.inputs?.find((item) => item.key === SystemInputEnum.welcomeText)?.value,
           name: app.name,
           avatar: app.avatar,
           intro: app.intro
-        },
-        chatModel: app.chat.chatModel
+        }
       }
     });
   } catch (err) {

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

@@ -4,11 +4,11 @@ import { connectToDatabase, ShareChat } from '@/service/mongo';
 import { authUser } from '@/service/utils/auth';
 import { hashPassword } from '@/service/utils/tools';
 
-/* get shareChat list by modelId */
+/* get shareChat list by appId */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { modelId } = req.query as {
-      modelId: string;
+    const { appId } = req.query as {
+      appId: string;
     };
 
     await connectToDatabase();
@@ -16,19 +16,17 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     const { userId } = await authUser({ req, authToken: true });
 
     const data = await ShareChat.find({
-      modelId,
+      appId,
       userId
     }).sort({
       _id: -1
     });
 
-    const blankPassword = hashPassword('');
-
     jsonRes(res, {
       data: data.map((item) => ({
         _id: item._id,
+        shareId: item.shareId,
         name: item.name,
-        password: item.password === blankPassword ? '' : '1',
         tokens: item.tokens,
         maxContext: item.maxContext,
         lastTime: item.lastTime