perf: model framwork

src/service/utils/auth.ts

@@ -1,10 +1,33 @@
 import { Configuration, OpenAIApi } from 'openai';
-import { Chat, Model } from '../mongo';
+import type { NextApiRequest } from 'next';
+import jwt from 'jsonwebtoken';
+import { Chat, Model, OpenApi, User } from '../mongo';
 import type { ModelSchema } from '@/types/mongoSchema';
-import { authToken } from './tools';
 import { getOpenApiKey } from './openai';
 import type { ChatItemType } from '@/types/chat';
 import mongoose from 'mongoose';
+import { defaultModel } from '@/constants/model';
+import { formatPrice } from '@/utils/user';
+import { ERROR_ENUM } from '../errorCode';
+
+/* 校验 token */
+export const authToken = (token?: string): Promise<string> => {
+  return new Promise((resolve, reject) => {
+    if (!token) {
+      reject('缺少登录凭证');
+      return;
+    }
+    const key = process.env.TOKEN_KEY as string;
+
+    jwt.verify(token, key, function (err, decoded: any) {
+      if (err || !decoded?.userId) {
+        reject('凭证无效');
+        return;
+      }
+      resolve(decoded.userId);
+    });
+  });
+};
 
 export const getOpenAIApi = (apiKey: string) => {
   const configuration = new Configuration({
@@ -20,12 +43,14 @@ export const authModel = async ({
   modelId,
   userId,
   authUser = true,
-  authOwner = true
+  authOwner = true,
+  reserveDetail = false
 }: {
   modelId: string;
   userId: string;
   authUser?: boolean;
   authOwner?: boolean;
+  reserveDetail?: boolean; // focus reserve detail
 }) => {
   // 获取 model 数据
   const model = await Model.findById<ModelSchema>(modelId);
@@ -33,15 +58,21 @@ export const authModel = async ({
     return Promise.reject('模型不存在');
   }
 
-  // 使用权限校验
+  /* 
+    Access verification
+    1. authOwner=true or authUser = true ,  just owner can use
+    2. authUser = false and share, anyone can use
+  */
   if ((authOwner || (authUser && !model.share.isShare)) && userId !== String(model.userId)) {
     return Promise.reject('无权操作该模型');
   }
 
-  // detail 内容去除
-  if (!model.share.isShareDetail && userId !== String(model.userId)) {
-    model.systemPrompt = '';
-    model.temperature = 0;
+  // do not share detail info
+  if (!reserveDetail && !model.share.isShareDetail && userId !== String(model.userId)) {
+    model.chat = {
+      ...defaultModel.chat,
+      chatModel: model.chat.chatModel
+    };
   }
 
   return { model };
@@ -60,7 +91,7 @@ export const authChat = async ({
   const userId = await authToken(authorization);
 
   // 获取 model 数据
-  const { model } = await authModel({ modelId, userId, authOwner: false });
+  const { model } = await authModel({ modelId, userId, authOwner: false, reserveDetail: true });
 
   // 聊天内容
   let content: ChatItemType[] = [];
@@ -91,3 +122,41 @@ export const authChat = async ({
     model
   };
 };
+
+/* 校验 open api key */
+export const authOpenApiKey = async (req: NextApiRequest) => {
+  const { apikey: apiKey } = req.headers;
+
+  if (!apiKey) {
+    return Promise.reject(ERROR_ENUM.unAuthorization);
+  }
+
+  try {
+    const openApi = await OpenApi.findOne({ apiKey });
+    if (!openApi) {
+      return Promise.reject(ERROR_ENUM.unAuthorization);
+    }
+    const userId = String(openApi.userId);
+
+    // 余额校验
+    const user = await User.findById(userId);
+    if (!user) {
+      return Promise.reject(ERROR_ENUM.unAuthorization);
+    }
+    if (formatPrice(user.balance) <= 0) {
+      return Promise.reject(ERROR_ENUM.insufficientQuota);
+    }
+
+    // 更新使用的时间
+    await OpenApi.findByIdAndUpdate(openApi._id, {
+      lastUsedTime: new Date()
+    });
+
+    return {
+      apiKey: process.env.OPENAIKEY as string,
+      userId
+    };
+  } catch (error) {
+    return Promise.reject(error);
+  }
+};

src/service/utils/sendNote.ts

@@ -1,6 +1,5 @@
 import * as nodemailer from 'nodemailer';
 import { UserAuthTypeEnum } from '@/constants/common';
-import dayjs from 'dayjs';
 import Dysmsapi, * as dysmsapi from '@alicloud/dysmsapi20170525';
 // @ts-ignore
 import * as OpenApi from '@alicloud/openapi-client';
@@ -48,25 +47,6 @@ export const sendEmailCode = (email: string, code: string, type: `${UserAuthType
   });
 };
 
-export const sendTrainSucceed = (email: string, modelName: string) => {
-  return new Promise((resolve, reject) => {
-    const options = {
-      from: `"FastGPT" ${myEmail}`,
-      to: email,
-      subject: '模型训练完成通知',
-      html: `你的模型 ${modelName} 已于 ${dayjs().format('YYYY-MM-DD HH:mm')} 训练完成！`
-    };
-    mailTransport.sendMail(options, function (err, msg) {
-      if (err) {
-        console.log('send email  error->', err);
-        reject('邮箱异常');
-      } else {
-        resolve('');
-      }
-    });
-  });
-};
-
 export const sendPhoneCode = async (phone: string, code: string) => {
   const accessKeyId = process.env.aliAccessKeyId;
   const accessKeySecret = process.env.aliAccessKeySecret;

src/service/utils/tools.ts

@@ -1,10 +1,6 @@
-import type { NextApiRequest } from 'next';
 import crypto from 'crypto';
 import jwt from 'jsonwebtoken';
 import { ChatItemType } from '@/types/chat';
-import { OpenApi, User } from '../mongo';
-import { formatPrice } from '@/utils/user';
-import { ERROR_ENUM } from '../errorCode';
 import { countChatTokens } from '@/utils/tools';
 import { ChatCompletionRequestMessageRoleEnum, ChatCompletionRequestMessage } from 'openai';
 import { ChatModelEnum } from '@/constants/model';
@@ -46,44 +42,6 @@ export const authToken = (token?: string): Promise<string> => {
   });
 };
 
-/* 校验 open api key */
-export const authOpenApiKey = async (req: NextApiRequest) => {
-  const { apikey: apiKey } = req.headers;
-
-  if (!apiKey) {
-    return Promise.reject(ERROR_ENUM.unAuthorization);
-  }
-
-  try {
-    const openApi = await OpenApi.findOne({ apiKey });
-    if (!openApi) {
-      return Promise.reject(ERROR_ENUM.unAuthorization);
-    }
-    const userId = String(openApi.userId);
-
-    // 余额校验
-    const user = await User.findById(userId);
-    if (!user) {
-      return Promise.reject(ERROR_ENUM.unAuthorization);
-    }
-    if (formatPrice(user.balance) <= 0) {
-      return Promise.reject('Insufficient account balance');
-    }
-
-    // 更新使用的时间
-    await OpenApi.findByIdAndUpdate(openApi._id, {
-      lastUsedTime: new Date()
-    });
-
-    return {
-      apiKey: process.env.OPENAIKEY as string,
-      userId
-    };
-  } catch (error) {
-    return Promise.reject(error);
-  }
-};
-
 /* openai axios config */
 export const axiosConfig = () => ({
   httpsAgent: global.httpsAgent,