diff --git a/src/pages/api/data/getDataItems.ts b/src/pages/api/data/getDataItems.ts index 3f9b4ad78..68cb7efd5 100644 --- a/src/pages/api/data/getDataItems.ts +++ b/src/pages/api/data/getDataItems.ts @@ -23,8 +23,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse) await authToken(authorization); const dataItems = await DataItem.find({ - dataId, - status: 0 + dataId }) .sort({ _id: -1 }) // 按照创建时间倒序排列 .skip((pageNum - 1) * pageSize) diff --git a/src/service/events/generateQA.ts b/src/service/events/generateQA.ts index a2385c913..b85e96b59 100644 --- a/src/service/events/generateQA.ts +++ b/src/service/events/generateQA.ts @@ -86,6 +86,7 @@ export async function generateQA(next = false): Promise { await DataItem.findByIdAndUpdate(dataItem._id, { status: dataItem.temperature >= 90 ? 0 : 1, // 需要生成 4 组内容。0,0.3,0.6,0.9 temperature: dataItem.temperature >= 90 ? dataItem.temperature : dataItem.temperature + 30, + rawResponse: content, $push: { result: { $each: splitResponse @@ -94,11 +95,17 @@ export async function generateQA(next = false): Promise { }); // 计费 !userApiKey && + splitResponse.length > 0 && pushSplitDataBill({ userId: dataItem.userId, text: systemPrompt.content + dataItem.text + content }); - console.log('生成QA成功,time:', `${(Date.now() - startTime) / 1000}s`); + console.log( + '生成QA成功,time:', + `${(Date.now() - startTime) / 1000}s`, + 'QA数量:', + splitResponse.length + ); } catch (error: any) { console.log('error: 生成QA错误', dataItem?._id); console.log('response:', error?.response); diff --git a/src/service/models/dataItem.ts b/src/service/models/dataItem.ts index 93d83c792..53454fdc8 100644 --- a/src/service/models/dataItem.ts +++ b/src/service/models/dataItem.ts @@ -24,6 +24,10 @@ const DataItemSchema = new Schema({ type: Number, required: true }, + rawResponse: { + type: [String], + default: '' + }, result: { type: [ { diff --git a/src/types/mongoSchema.d.ts b/src/types/mongoSchema.d.ts index 051f807dd..731b53b5e 100644 --- a/src/types/mongoSchema.d.ts +++ b/src/types/mongoSchema.d.ts @@ -111,6 +111,7 @@ export interface DataItemSchema { times: number; temperature: number; text: string; + rawResponse: string[]; result: { q: string; a: string;