feat: 修改计费模式为tokens

src/pages/api/chat/chatGpt.ts

@@ -143,15 +143,15 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     !stream.destroyed && stream.push(null);
     stream.destroy();
 
-    const promptsLen = formatPrompts.reduce((sum, item) => sum + item.content.length, 0);
-    console.log(`responseLen: ${responseContent.length}`, `promptLen: ${promptsLen}`);
+    const promptsContent = formatPrompts.map((item) => item.content).join('');
+    console.log(`responseLen: ${responseContent.length}`, `promptLen: ${promptsContent.length}`);
     // 只有使用平台的 key 才计费
     !userApiKey &&
       pushBill({
         modelName: model.service.modelName,
         userId,
         chatId,
-        textLen: promptsLen + responseContent.length
+        text: promptsContent + responseContent
       });
   } catch (err: any) {
     if (step === 1) {

src/pages/api/chat/gpt3.ts

@@ -54,21 +54,20 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       }
     );
 
-    const responseMessage = response.data.choices[0]?.text || '';
+    const responseContent = response.data.choices[0]?.text || '';
 
-    const promptsLen = prompt.reduce((sum, item) => sum + item.value.length, 0);
-    console.log(`responseLen: ${responseMessage.length}`, `promptLen: ${promptsLen}`);
+    console.log(`responseLen: ${responseContent.length}`, `promptLen: ${formatPrompts.length}`);
     // 只有使用平台的 key 才计费
     !userApiKey &&
       pushBill({
         modelName: model.service.modelName,
         userId,
         chatId,
-        textLen: promptsLen + responseMessage.length
+        text: formatPrompts + responseContent
       });
 
     jsonRes(res, {
-      data: responseMessage
+      data: responseContent
     });
   } catch (err: any) {
     jsonRes(res, {

src/pages/api/model/list.ts

@@ -21,6 +21,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     // 根据 userId 获取模型信息
     const models = await Model.find({
       userId
+    }).sort({
+      _id: -1
     });
 
     jsonRes(res, {

src/pages/api/timer/initBill.ts

@@ -12,12 +12,15 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     }
     await connectToDatabase();
 
-    await Bill.updateMany(
-      {},
-      {
-        type: 'chat',
-        modelName: 'gpt-3.5-turbo'
-      }
+    const bills = await Bill.find({
+      tokenLen: { $exists: false }
+    });
+    await Promise.all(
+      bills.map((bill) =>
+        Bill.findByIdAndUpdate(bill._id, {
+          tokenLen: bill.textLen
+        })
+      )
     );
 
     jsonRes(res, {

src/pages/api/user/getPayCode.ts

@@ -5,6 +5,7 @@ import axios from 'axios';
 import { authToken } from '@/service/utils/tools';
 import { customAlphabet } from 'nanoid';
 import { connectToDatabase, Pay } from '@/service/mongo';
+import { PRICE_SCALE } from '@/utils/user';
 
 const nanoid = customAlphabet('abcdefghijklmnopqrstuvwxyz1234567890', 20);
 
@@ -35,7 +36,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     // 充值记录 + 1
     const payOrder = await Pay.create({
       userId,
-      price: amount * 100000,
+      price: amount * PRICE_SCALE,
       orderId: id
     });