Commit f19c2d2c by archer

perf: 去掉raw content

parent 84d91f3f
......@@ -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)
......
......@@ -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',
......
......@@ -68,8 +68,6 @@ export async function generateVector(next = false): Promise<any> {
dataItem.id,
'vector',
vectorToBuffer(vector),
'rawVector',
JSON.stringify(vector),
'status',
ModelDataStatusEnum.ready
]);
......
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}`);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment