feat: 拆分测试环境

src/service/events/generateAbstract.ts

@@ -7,8 +7,6 @@ import { ChatModelNameEnum } from '@/constants/model';
 import { pushSplitDataBill } from '@/service/events/pushBill';
 
 export async function generateAbstract(next = false): Promise<any> {
-  if (process.env.NODE_ENV === 'development') return;
-
   if (global.generatingAbstract && !next) return;
   global.generatingAbstract = true;
 

src/service/events/pushBill.ts

@@ -119,3 +119,60 @@ export const pushSplitDataBill = async ({
     console.log(error);
   }
 };
+
+export const pushGenerateVectorBill = async ({
+  isPay,
+  userId,
+  text,
+  type
+}: {
+  isPay: boolean;
+  userId: string;
+  text: string;
+  type: DataType;
+}) => {
+  await connectToDatabase();
+
+  let billId;
+
+  try {
+    // 计算 token 数量
+    const tokens = encode(text);
+
+    console.log('text len: ', text.length);
+    console.log('token len:', tokens.length);
+
+    if (isPay) {
+      try {
+        // 获取模型单价格, 都是用 gpt35 拆分
+        const modelItem = modelList.find((item) => item.model === ChatModelNameEnum.GPT35);
+        const unitPrice = modelItem?.price || 5;
+        // 计算价格
+        const price = unitPrice * tokens.length;
+
+        console.log(`splitData bill, price: ${formatPrice(price)}元`);
+
+        // 插入 Bill 记录
+        const res = await Bill.create({
+          userId,
+          type,
+          modelName: ChatModelNameEnum.GPT35,
+          textLen: text.length,
+          tokenLen: tokens.length,
+          price
+        });
+        billId = res._id;
+
+        // 账号扣费
+        await User.findByIdAndUpdate(userId, {
+          $inc: { balance: -price }
+        });
+      } catch (error) {
+        console.log('创建账单失败:', error);
+        billId && Bill.findByIdAndDelete(billId);
+      }
+    }
+  } catch (error) {
+    console.log(error);
+  }
+};

src/service/mongo.ts

@@ -17,7 +17,7 @@ export async function connectToDatabase(): Promise<void> {
     mongoose.set('strictQuery', true);
     global.mongodb = await mongoose.connect(process.env.MONGODB_URI as string, {
       bufferCommands: true,
-      dbName: 'doc_gpt',
+      dbName: process.env.NODE_ENV === 'development' ? 'doc_gpt_test' : 'doc_gpt',
       maxPoolSize: 5,
       minPoolSize: 1,
       maxConnecting: 5

src/service/redis.ts

@@ -30,7 +30,7 @@ export const connectRedis = async () => {
     await global.redisClient.connect();
 
     // 1 - 测试库，0 - 正式
-    await global.redisClient.select(process.env.NODE_ENV === 'development' ? 0 : 0);
+    await global.redisClient.SELECT(0);
 
     return global.redisClient;
   } catch (error) {