feat: config vector model and qa model

client/src/service/events/generateQA.ts

@@ -1,5 +1,5 @@
 import { TrainingData } from '@/service/mongo';
-import { pushSplitDataBill } from '@/service/events/pushBill';
+import { pushQABill } from '@/service/events/pushBill';
 import { pushDataToKb } from '@/pages/api/openapi/kb/pushData';
 import { TrainingModeEnum } from '@/constants/plugin';
 import { ERROR_ENUM } from '../errorCode';
@@ -60,14 +60,13 @@ export async function generateQA(): Promise<any> {
     // 请求 chatgpt 获取回答
     const response = await Promise.all(
       [data.q].map((text) => {
-        const modelTokenLimit =
-          chatModels.find((item) => item.model === data.model)?.contextMaxToken || 16000;
+        const modelTokenLimit = global.qaModel.maxToken || 16000;
         const messages: ChatCompletionRequestMessage[] = [
           {
             role: 'system',
-            content: `你是出题人.
-${data.prompt || '我会发送一段长文本'}.
-从中提取出 25 个问题和答案. 答案详细完整. 按下面格式返回: 
+            content: `你是出题人，${
+              data.prompt || '我会发送一段长文本'
+            }，请从中提取出 25 个问题和答案. 答案详细完整，并按下面格式返回: 
 Q1:
 A1:
 Q2:
@@ -88,7 +87,7 @@ A2:
         return chatAPI
           .createChatCompletion(
             {
-              model: data.model,
+              model: global.qaModel.model,
               temperature: 0.8,
               messages,
               stream: false,
@@ -106,10 +105,9 @@ A2:
             const result = formatSplitText(answer || ''); // 格式化后的QA对
             console.log(`split result length: `, result.length);
             // 计费
-            pushSplitDataBill({
+            pushQABill({
               userId: data.userId,
               totalTokens,
-              model: data.model,
               appName: 'QA 拆分'
             });
             return {
@@ -135,7 +133,6 @@ A2:
         source: data.source
       })),
       userId,
-      model: global.vectorModels[0].model,
       mode: TrainingModeEnum.index
     });
 

client/src/service/events/generateVector.ts

@@ -38,7 +38,7 @@ export async function generateVector(): Promise<any> {
       q: 1,
       a: 1,
       source: 1,
-      model: 1
+      vectorModel: 1
     });
 
     // task preemption
@@ -61,7 +61,7 @@ export async function generateVector(): Promise<any> {
 
     // 生成词向量
     const { vectors } = await getVector({
-      model: data.model,
+      model: data.vectorModel,
       input: dataItems.map((item) => item.q),
       userId
     });

client/src/service/events/pushBill.ts

@@ -76,13 +76,11 @@ export const updateShareChatBill = async ({
   }
 };
 
-export const pushSplitDataBill = async ({
+export const pushQABill = async ({
   userId,
   totalTokens,
-  model,
   appName
 }: {
-  model: string;
   userId: string;
   totalTokens: number;
   appName: string;
@@ -95,7 +93,7 @@ export const pushSplitDataBill = async ({
     await connectToDatabase();
 
     // 获取模型单价格, 都是用 gpt35 拆分
-    const unitPrice = global.chatModels.find((item) => item.model === model)?.price || 3;
+    const unitPrice = global.qaModel.price || 3;
     // 计算价格
     const total = unitPrice * totalTokens;