doc gpt V0.2

src/service/utils/chat.ts

@@ -0,0 +1,45 @@
+import { Configuration, OpenAIApi } from 'openai';
+import { Chat } from '../mongo';
+
+export const getOpenAIApi = (apiKey: string) => {
+  const configuration = new Configuration({
+    apiKey
+  });
+
+  return new OpenAIApi(configuration, undefined);
+};
+
+export const authChat = async (chatId: string) => {
+  // 获取 chat 数据
+  const chat = await Chat.findById(chatId)
+    .populate({
+      path: 'modelId',
+      options: {
+        strictPopulate: false
+      }
+    })
+    .populate({
+      path: 'userId',
+      options: {
+        strictPopulate: false
+      }
+    });
+
+  if (!chat || !chat.modelId || !chat.userId) {
+    return Promise.reject('聊天已过期');
+  }
+
+  // 获取 user 的 apiKey
+  const user = chat.userId;
+
+  const userApiKey = user.accounts?.find((item: any) => item.type === 'openai')?.value;
+
+  if (!userApiKey) {
+    return Promise.reject('该用户缺少ApiKey, 无法请求');
+  }
+
+  return {
+    userApiKey,
+    chat
+  };
+};

src/service/utils/sendEmail.ts

@@ -0,0 +1,63 @@
+import * as nodemailer from 'nodemailer';
+import { EmailTypeEnum } from '@/constants/common';
+import dayjs from 'dayjs';
+
+const myEmail = process.env.MY_MAIL;
+let mailTransport = nodemailer.createTransport({
+  // host: 'smtp.qq.email',
+  service: 'qq',
+  secure: true, //安全方式发送,建议都加上
+  auth: {
+    user: myEmail,
+    pass: process.env.MAILE_CODE
+  }
+});
+
+const emailMap: { [key: string]: any } = {
+  [EmailTypeEnum.register]: {
+    subject: '注册 DocGPT 账号',
+    html: (code: string) => `<div>您正在注册 DocGPT 账号，验证码为：${code}</div>`
+  },
+  [EmailTypeEnum.findPassword]: {
+    subject: '修改 DocGPT 密码',
+    html: (code: string) => `<div>您正在修改 DocGPT 账号密码，验证码为：${code}</div>`
+  }
+};
+
+export const sendCode = (email: string, code: string, type: `${EmailTypeEnum}`) => {
+  return new Promise((resolve, reject) => {
+    const options = {
+      from: `"DocGPT" ${myEmail}`,
+      to: email,
+      subject: emailMap[type]?.subject,
+      html: emailMap[type]?.html(code)
+    };
+    mailTransport.sendMail(options, function (err, msg) {
+      if (err) {
+        console.log(err);
+        reject('邮箱异常');
+      } else {
+        resolve('');
+      }
+    });
+  });
+};
+
+export const sendTrainSucceed = (email: string, modelName: string) => {
+  return new Promise((resolve, reject) => {
+    const options = {
+      from: `"DocGPT" ${myEmail}`,
+      to: email,
+      subject: '模型训练完成通知',
+      html: `你的模型 ${modelName} 已于 ${dayjs().format('YYYY-MM-DD HH:mm')} 训练完成！`
+    };
+    mailTransport.sendMail(options, function (err, msg) {
+      if (err) {
+        console.log(err);
+        reject('邮箱异常');
+      } else {
+        resolve('');
+      }
+    });
+  });
+};

src/service/utils/tools.ts

@@ -0,0 +1,63 @@
+import crypto from 'crypto';
+import jwt from 'jsonwebtoken';
+import { User } from '../models/user';
+import tunnel from 'tunnel';
+
+/* 密码加密 */
+export const hashPassword = (psw: string) => {
+  return crypto.createHash('sha256').update(psw).digest('hex');
+};
+
+/* 生成 token */
+export const generateToken = (userId: string) => {
+  const key = process.env.TOKEN_KEY as string;
+  const token = jwt.sign(
+    {
+      userId,
+      exp: Math.floor(Date.now() / 1000) + 60 * 60 * 24 * 7
+    },
+    key
+  );
+  return token;
+};
+
+/* 校验 token */
+export const authToken = (token: string): Promise<string> => {
+  return new Promise((resolve, reject) => {
+    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);
+    });
+  });
+};
+
+/* 获取用户的 openai APIkey */
+export const getUserOpenaiKey = async (userId: string) => {
+  const user = await User.findById(userId);
+
+  const userApiKey = user?.accounts?.find((item: any) => item.type === 'openai')?.value;
+  if (!userApiKey) {
+    return Promise.reject('缺少ApiKey, 无法请求');
+  }
+
+  return Promise.resolve(userApiKey);
+};
+
+/* 代理 */
+export const openaiProxy: any =
+  process.env.AXIOS_PROXY_PORT && process.env.AXIOS_PROXY_HOST
+    ? {
+        httpsAgent: tunnel.httpsOverHttp({
+          proxy: {
+            host: process.env.AXIOS_PROXY_HOST,
+            port: +process.env.AXIOS_PROXY_PORT
+          }
+        }),
+        proxy: false
+      }
+    : undefined;