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);
+  }
+};