fix: 账单余额问题

src/service/events/generateAbstract.ts

@@ -39,7 +39,7 @@ export async function generateAbstract(next = false): Promise<any> {
     // 获取 openapi Key
     let userApiKey, systemKey;
     try {
-      const key = await getOpenApiKey(dataItem.userId, true);
+      const key = await getOpenApiKey(dataItem.userId);
       userApiKey = key.userApiKey;
       systemKey = key.systemKey;
     } catch (error: any) {

src/service/events/generateQA.ts

@@ -38,7 +38,7 @@ export async function generateQA(next = false): Promise<any> {
     // 获取 openapi Key
     let userApiKey, systemKey;
     try {
-      const key = await getOpenApiKey(dataItem.userId, true);
+      const key = await getOpenApiKey(dataItem.userId);
       userApiKey = key.userApiKey;
       systemKey = key.systemKey;
     } catch (error: any) {
@@ -48,7 +48,7 @@ export async function generateQA(next = false): Promise<any> {
           textList: [],
           errorText: error.message
         });
-        throw new Error('账号余额不足');
+        throw new Error(error?.message);
       }
 
       throw new Error('获取 openai key 失败');

src/service/events/generateVector.ts

@@ -39,7 +39,19 @@ export async function generateVector(next = false): Promise<any> {
     };
 
     // 获取 openapi Key
-    const { userApiKey, systemKey } = await getOpenApiKey(dataItem.userId);
+    let userApiKey, systemKey;
+    try {
+      const res = await getOpenApiKey(dataItem.userId);
+      userApiKey = res.userApiKey;
+      systemKey = res.systemKey;
+    } catch (error: any) {
+      if (error?.code === 501) {
+        await redis.del(dataItem.id);
+        throw new Error(error?.message);
+      }
+
+      throw new Error('获取 openai key 失败');
+    }
 
     // 生成词向量
     const { vector } = await openaiCreateEmbedding({

src/service/events/pushBill.ts

@@ -90,7 +90,7 @@ export const pushSplitDataBill = async ({
       try {
         // 获取模型单价格, 都是用 gpt35 拆分
         const modelItem = modelList.find((item) => item.model === ChatModelNameEnum.GPT35);
-        const unitPrice = modelItem?.price || 5;
+        const unitPrice = modelItem?.price || 3;
         // 计算价格
         const price = unitPrice * tokens.length;