perf: 文件结构

6c4026cc · archer · caf31faf · 6c4026cc · 6c4026cc · 6c4026cc
Commit 6c4026cc authored Apr 03, 2023 by archer
10 changed files
--- a/src/pages/api/chat/chatGpt.ts
+++ b/src/pages/api/chat/chatGpt.ts
@@ -2,7 +2,7 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { createParser, ParsedEvent, ReconnectInterval } from 'eventsource-parser';
 import { connectToDatabase } from '@/service/mongo';
 import { getOpenAIApi, authChat } from '@/service/utils/chat';
-import { httpsAgent } from '@/service/utils/tools';
+import { httpsAgent, openaiChatFilter } from '@/service/utils/tools';
 import { ChatCompletionRequestMessage, ChatCompletionRequestMessageRoleEnum } from 'openai';
 import { ChatItemType } from '@/types/chat';
 import { jsonRes } from '@/service/response';
@@ -10,7 +10,6 @@ import type { ModelSchema } from '@/types/mongoSchema';
 import { PassThrough } from 'stream';
 import { modelList } from '@/constants/model';
 import { pushChatBill } from '@/service/events/pushBill';
-import { openaiChatFilter } from '@/service/utils/tools';
 /* 发送提示词 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {

--- a/src/pages/api/model/del.ts
+++ b/src/pages/api/model/del.ts
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { Chat, Model, Training, connectToDatabase } from '@/service/mongo';
-import { authToken, getUserApiOpenai } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/tools';
+import { getUserApiOpenai } from '@/service/utils/openai';
 import { TrainingStatusEnum } from '@/constants/model';
 import { TrainingItemType } from '@/types/training';
 import { httpsAgent } from '@/service/utils/tools';

--- a/src/pages/api/model/train/putTrainStatus.ts
+++ b/src/pages/api/model/train/putTrainStatus.ts
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, Model, Training } from '@/service/mongo';
-import { getOpenAIApi } from '@/service/utils/chat';
+import { authToken } from '@/service/utils/tools';
-import { authToken, getUserApiOpenai } from '@/service/utils/tools';
+import { getUserApiOpenai } from '@/service/utils/openai';
 import type { ModelSchema } from '@/types/mongoSchema';
 import { TrainingItemType } from '@/types/training';
 import { ModelStatusEnum, TrainingStatusEnum } from '@/constants/model';

--- a/src/pages/api/model/train/train.ts
+++ b/src/pages/api/model/train/train.ts
@@ -3,7 +3,8 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, Model, Training } from '@/service/mongo';
 import formidable from 'formidable';
-import { authToken, getUserApiOpenai } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/tools';
+import { getUserApiOpenai } from '@/service/utils/openai';
 import { join } from 'path';
 import fs from 'fs';
 import type { ModelSchema } from '@/types/mongoSchema';

--- a/src/pages/api/timer/updateTraining.ts
+++ b/src/pages/api/timer/updateTraining.ts
@@ -4,8 +4,7 @@ import { jsonRes } from '@/service/response';
 import { connectToDatabase, Training, Model } from '@/service/mongo';
 import type { TrainingItemType } from '@/types/training';
 import { TrainingStatusEnum, ModelStatusEnum } from '@/constants/model';
-import { getOpenAIApi } from '@/service/utils/chat';
+import { getUserApiOpenai } from '@/service/utils/openai';
-import { getUserApiOpenai } from '@/service/utils/tools';
 import { OpenAiTuneStatusEnum } from '@/service/constants/training';
 import { sendTrainSucceed } from '@/service/utils/sendEmail';
 import { httpsAgent } from '@/service/utils/tools';

--- a/src/service/events/generateAbstract.ts
+++ b/src/service/events/generateAbstract.ts
 import { DataItem } from '@/service/mongo';
 import { getOpenAIApi } from '@/service/utils/chat';
-import { httpsAgent, getOpenApiKey } from '@/service/utils/tools';
+import { httpsAgent } from '@/service/utils/tools';
+import { getOpenApiKey } from '../utils/openai';
 import type { ChatCompletionRequestMessage } from 'openai';
 import { DataItemSchema } from '@/types/mongoSchema';
 import { ChatModelNameEnum } from '@/constants/model';

--- a/src/service/events/generateQA.ts
+++ b/src/service/events/generateQA.ts
 import { SplitData } from '@/service/mongo';
 import { getOpenAIApi } from '@/service/utils/chat';
-import { httpsAgent, getOpenApiKey } from '@/service/utils/tools';
+import { httpsAgent } from '@/service/utils/tools';
+import { getOpenApiKey } from '../utils/openai';
 import type { ChatCompletionRequestMessage } from 'openai';
 import { ChatModelNameEnum } from '@/constants/model';
 import { pushSplitDataBill } from '@/service/events/pushBill';

--- a/src/service/utils/chat.ts
+++ b/src/service/utils/chat.ts
 import { Configuration, OpenAIApi } from 'openai';
 import { Chat } from '../mongo';
 import type { ChatPopulate } from '@/types/mongoSchema';
-import { authToken, getOpenApiKey } from './tools';
+import { authToken } from './tools';
+import { getOpenApiKey } from './openai';
 export const getOpenAIApi = (apiKey: string) => {
  const configuration = new Configuration({

--- a/src/service/utils/openai.ts
+++ b/src/service/utils/openai.ts
+import axios from 'axios';
+import { getOpenAIApi } from '@/service/utils/chat';
+import { httpsAgent } from './tools';
+import { User } from '../models/user';
+import { formatPrice } from '@/utils/user';
+/* 判断 apikey 是否还有余额 */
+export const checkKeyGrant = async (apiKey: string) => {
+  const grant = await axios.get('https://api.openai.com/dashboard/billing/credit_grants', {
+    headers: {
+      Authorization: `Bearer ${apiKey}`
+    },
+    httpsAgent
+  });
+  if (grant.data?.total_available <= 0.2) {
+    return false;
+  }
+  return true;
+};
+/* 获取用户 api 的 openai 信息 */
+export const getUserApiOpenai = 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, 无法请求');
+  }
+  // 余额校验
+  const hasGrant = await checkKeyGrant(userApiKey);
+  if (!hasGrant) {
+    return Promise.reject({
+      code: 501,
+      message: 'API 余额不足'
+    });
+  }
+  return {
+    user,
+    openai: getOpenAIApi(userApiKey),
+    apiKey: userApiKey
+  };
+};
+/* 获取 open api key，如果用户没有自己的key，就用平台的，用平台记得加账单 */
+export const getOpenApiKey = async (userId: string, checkGrant = false) => {
+  const user = await User.findById(userId);
+  if (!user) {
+    return Promise.reject('找不到用户');
+  }
+  const userApiKey = user?.accounts?.find((item: any) => item.type === 'openai')?.value;
+  // 有自己的key
+  if (userApiKey) {
+    // api 余额校验
+    if (checkGrant) {
+      const hasGrant = await checkKeyGrant(userApiKey);
+      if (!hasGrant) {
+        return Promise.reject({
+          code: 501,
+          message: 'API 余额不足'
+        });
+      }
+    }
+    return {
+      user,
+      userApiKey,
+      systemKey: ''
+    };
+  }
+  // 平台账号余额校验
+  if (formatPrice(user.balance) <= 0) {
+    return Promise.reject({
+      code: 501,
+      message: '账号余额不足'
+    });
+  }
+  return {
+    user,
+    userApiKey: '',
+    systemKey: process.env.OPENAIKEY as string
+  };
+};
--- a/src/service/utils/tools.ts
+++ b/src/service/utils/tools.ts
 import crypto from 'crypto';
 import jwt from 'jsonwebtoken';
-import { User } from '../models/user';
 import tunnel from 'tunnel';
-import { formatPrice } from '@/utils/user';
 import { ChatItemType } from '@/types/chat';
 import { encode } from 'gpt-token-utils';
-import { getOpenAIApi } from '@/service/utils/chat';
-import axios from 'axios';
 /* 密码加密 */
 export const hashPassword = (psw: string) => {
@@ -56,90 +52,6 @@ export const httpsAgent =
      })
    : undefined;
-/* 判断 apikey 是否还有余额 */
-export const checkKeyGrant = async (apiKey: string) => {
-  const grant = await axios.get('https://api.openai.com/dashboard/billing/credit_grants', {
-    headers: {
-      Authorization: `Bearer ${apiKey}`
-    },
-    httpsAgent
-  });
-  if (grant.data?.total_available <= 0.2) {
-    return false;
-  }
-  return true;
-};
-/* 获取用户 api 的 openai 信息 */
-export const getUserApiOpenai = 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, 无法请求');
-  }
-  // 余额校验
-  const hasGrant = await checkKeyGrant(userApiKey);
-  if (!hasGrant) {
-    return Promise.reject({
-      code: 501,
-      message: 'API 余额不足'
-    });
-  }
-  return {
-    user,
-    openai: getOpenAIApi(userApiKey),
-    apiKey: userApiKey
-  };
-};
-/* 获取 open api key，如果用户没有自己的key，就用平台的，用平台记得加账单 */
-export const getOpenApiKey = async (userId: string, checkGrant = false) => {
-  const user = await User.findById(userId);
-  if (!user) {
-    return Promise.reject('找不到用户');
-  }
-  const userApiKey = user?.accounts?.find((item: any) => item.type === 'openai')?.value;
-  // 有自己的key
-  if (userApiKey) {
-    // api 余额校验
-    if (checkGrant) {
-      const hasGrant = await checkKeyGrant(userApiKey);
-      if (!hasGrant) {
-        return Promise.reject({
-          code: 501,
-          message: 'API 余额不足'
-        });
-      }
-    }
-    return {
-      user,
-      userApiKey,
-      systemKey: ''
-    };
-  }
-  // 平台账号余额校验
-  if (formatPrice(user.balance) <= 0) {
-    return Promise.reject({
-      code: 501,
-      message: '账号余额不足'
-    });
-  }
-  return {
-    user,
-    userApiKey: '',
-    systemKey: process.env.OPENAIKEY as string
-  };
-};
 /* tokens 截断 */
 export const openaiChatFilter = (prompts: ChatItemType[], maxTokens: number) => {
  let res: ChatItemType[] = [];