perf: 去掉raw content

src/pages/api/model/data/exportModelData.ts

@@ -32,7 +32,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       VecModelDataIdx,
       `@modelId:{${modelId}} @userId:{${userId}}`,
       {
-        RETURN: ['q', 'text', 'rawVector'],
+        RETURN: ['q', 'text'],
         LIMIT: {
           from: 0,
           size: 10000
@@ -40,21 +40,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       }
     );
 
-    const data = searchRes.documents
-      .filter((item) => {
-        if (!item?.value?.rawVector) return false;
-        try {
-          JSON.parse(item.value.rawVector as string);
-          return true;
-        } catch (error) {
-          return false;
-        }
-      })
-      .map((item: any) => ({
-        prompt: item.value.q,
-        completion: item.value.text,
-        vector: JSON.parse(item.value.rawVector)
-      }));
+    const data = searchRes.documents.map((item: any) => ({
+      prompt: item.value.q,
+      completion: item.value.text
+    }));
 
     jsonRes(res, {
       data: JSON.stringify(data)

src/pages/api/model/data/pushModelDataJson.ts

@@ -53,9 +53,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
           userId,
           'modelId',
           String(modelId),
-          ...(vector
-            ? ['vector', vectorToBuffer(formatVector(vector)), 'rawVector', JSON.stringify(vector)]
-            : []),
+          ...(vector ? ['vector', vectorToBuffer(formatVector(vector))] : []),
           'q',
           item.prompt,
           'text',

src/service/events/generateVector.ts

@@ -68,8 +68,6 @@ export async function generateVector(next = false): Promise<any> {
       dataItem.id,
       'vector',
       vectorToBuffer(vector),
-      'rawVector',
-      JSON.stringify(vector),
       'status',
       ModelDataStatusEnum.ready
     ]);

src/service/events/pushBill.ts

@@ -1,7 +1,6 @@
 import { connectToDatabase, Bill, User } from '../mongo';
 import { modelList, ChatModelNameEnum } from '@/constants/model';
 import { encode } from 'gpt-token-utils';
-import { formatPrice } from '@/utils/user';
 import { BillTypeEnum } from '@/constants/user';
 import type { DataType } from '@/types/data';
 
@@ -22,7 +21,7 @@ export const pushChatBill = async ({
 
   try {
     // 计算 token 数量
-    const tokens = encode(text).length;
+    const tokens = Math.floor(encode(text).length * 0.7);
 
     console.log(`chat generate success. text len: ${text.length}. token len: ${tokens}`);