feat: refresh max token

client/src/pages/api/app/data/totalUsage.ts

@@ -2,7 +2,6 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, Bill } from '@/service/mongo';
 import { authUser } from '@/service/utils/auth';
-import type { ChatHistoryItemType } from '@/types/chat';
 import { Types } from 'mongoose';
 
 /* get one app chat history content number. */
@@ -28,14 +27,14 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
             month: { $month: '$time' },
             day: { $dayOfMonth: '$time' }
           },
-          tokenLen: { $sum: '$tokenLen' } // 对tokenLen的值求和
+          total: { $sum: '$total' }
         }
       },
       {
         $project: {
           _id: 0,
           date: { $dateFromParts: { year: '$_id.year', month: '$_id.month', day: '$_id.day' } },
-          tokenLen: 1
+          total: 1
         }
       },
       { $sort: { date: 1 } }

client/src/pages/api/openapi/v1/chat/completions.ts

@@ -15,7 +15,7 @@ import { Types } from 'mongoose';
 import { moduleFetch } from '@/service/api/request';
 import { AppModuleItemType, RunningModuleItemType } from '@/types/app';
 import { FlowInputItemTypeEnum } from '@/constants/flow';
-import { finishTaskBill, createTaskBill } from '@/service/events/pushBill';
+import { finishTaskBill, createTaskBill, delTaskBill } from '@/service/events/pushBill';
 import { BillSourceEnum } from '@/constants/user';
 
 export type MessageItemType = ChatCompletionRequestMessage & { _id?: string };
@@ -56,6 +56,8 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
     variables = {}
   } = req.body as Props;
 
+  let billId = '';
+
   try {
     if (!messages) {
       throw new Error('Prams Error');
@@ -108,11 +110,11 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
       res.setHeader('newHistoryId', String(newHistoryId));
     }
 
-    const billId = await createTaskBill({
+    billId = await createTaskBill({
       userId,
       appName: app.name,
       appId,
-      source: BillSourceEnum.fastgpt
+      source: authType === 'apikey' ? BillSourceEnum.api : BillSourceEnum.fastgpt
     });
 
     /* start process */
@@ -125,7 +127,7 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
         userChatInput: prompt.value
       },
       stream,
-      billId: ''
+      billId
     });
 
     // save chat
@@ -184,6 +186,8 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
       billId
     });
   } catch (err: any) {
+    delTaskBill(billId);
+
     if (stream) {
       res.status(500);
       sseErrRes(res, err);