perf: bill framwork

src/service/events/generateQA.ts

@@ -9,6 +9,7 @@ import { SplitDataSchema } from '@/types/mongoSchema';
 import { modelServiceToolMap } from '../utils/chat';
 import { ChatRoleEnum } from '@/constants/chat';
 import { getErrText } from '@/utils/tools';
+import { BillTypeEnum } from '@/constants/user';
 
 export async function generateQA(next = false): Promise<any> {
   if (process.env.queueTask !== '1') {
@@ -98,7 +99,7 @@ A2:
             pushSplitDataBill({
               isPay: !userOpenAiKey && result.length > 0,
               userId: dataItem.userId,
-              type: 'QA',
+              type: BillTypeEnum.QA,
               textLen: responseMessages.map((item) => item.value).join('').length,
               totalTokens
             });

src/service/events/pushBill.ts

@@ -8,7 +8,8 @@ export const pushChatBill = async ({
   userId,
   chatId,
   textLen,
-  tokens
+  tokens,
+  type
 }: {
   isPay: boolean;
   chatModel: ChatModelType;
@@ -16,6 +17,7 @@ export const pushChatBill = async ({
   chatId?: '' | string;
   textLen: number;
   tokens: number;
+  type: BillTypeEnum.chat | BillTypeEnum.openapiChat;
 }) => {
   console.log(`chat generate success. text len: ${textLen}. token len: ${tokens}. pay:${isPay}`);
   if (!isPay) return;
@@ -33,7 +35,7 @@ export const pushChatBill = async ({
       // 插入 Bill 记录
       const res = await Bill.create({
         userId,
-        type: 'chat',
+        type,
         modelName: chatModel,
         chatId: chatId ? chatId : undefined,
         textLen,
@@ -83,7 +85,7 @@ export const pushSplitDataBill = async ({
   userId: string;
   totalTokens: number;
   textLen: number;
-  type: `${BillTypeEnum}`;
+  type: BillTypeEnum.QA;
 }) => {
   console.log(
     `splitData generate success. text len: ${textLen}. token len: ${totalTokens}. pay:${isPay}`