38 lines
1.0 KiB
TypeScript
38 lines
1.0 KiB
TypeScript
// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
|
|
import type { NextApiRequest, NextApiResponse } from 'next';
|
|
import { jsonRes } from '@/service/response';
|
|
import { authUser } from '@/service/utils/auth';
|
|
import { connectToDatabase, TrainingData } from '@/service/mongo';
|
|
import { TrainingTypeEnum } from '@/constants/plugin';
|
|
|
|
export default async function handler(req: NextApiRequest, res: NextApiResponse) {
|
|
try {
|
|
await authUser({ req, authRoot: true });
|
|
|
|
await connectToDatabase();
|
|
|
|
// split queue data
|
|
const result = await TrainingData.aggregate([
|
|
{
|
|
$group: {
|
|
_id: '$mode',
|
|
count: { $sum: 1 }
|
|
}
|
|
}
|
|
]);
|
|
|
|
jsonRes(res, {
|
|
data: {
|
|
qaListLen: result.find((item) => item._id === TrainingTypeEnum.qa)?.count || 0,
|
|
vectorListLen: result.find((item) => item._id === TrainingTypeEnum.index)?.count || 0
|
|
}
|
|
});
|
|
} catch (error) {
|
|
console.log(error);
|
|
jsonRes(res, {
|
|
code: 500,
|
|
error
|
|
});
|
|
}
|
|
}
|