73 lines
1.8 KiB
TypeScript
73 lines
1.8 KiB
TypeScript
import type { NextApiRequest, NextApiResponse } from 'next';
|
|
import { jsonRes } from '@/service/response';
|
|
import { connectToDatabase, Model } from '@/service/mongo';
|
|
import { authToken } from '@/service/utils/tools';
|
|
import { ModelDataSchema } from '@/types/mongoSchema';
|
|
import { generateVector } from '@/service/events/generateVector';
|
|
import { connectRedis } from '@/service/redis';
|
|
import { VecModelDataPrefix, ModelDataStatusEnum } from '@/constants/redis';
|
|
|
|
export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
|
|
try {
|
|
const { modelId, data } = req.body as {
|
|
modelId: string;
|
|
data: { text: ModelDataSchema['text']; q: ModelDataSchema['q'] }[];
|
|
};
|
|
const { authorization } = req.headers;
|
|
|
|
if (!authorization) {
|
|
throw new Error('无权操作');
|
|
}
|
|
|
|
if (!modelId || !Array.isArray(data)) {
|
|
throw new Error('缺少参数');
|
|
}
|
|
|
|
// 凭证校验
|
|
const userId = await authToken(authorization);
|
|
|
|
await connectToDatabase();
|
|
const redis = await connectRedis();
|
|
|
|
// 验证是否是该用户的 model
|
|
const model = await Model.findOne({
|
|
_id: modelId,
|
|
userId
|
|
});
|
|
|
|
if (!model) {
|
|
throw new Error('无权操作该模型');
|
|
}
|
|
|
|
const insertRes = await Promise.allSettled(
|
|
data.map((item) => {
|
|
return redis.sendCommand([
|
|
'HMSET',
|
|
`${VecModelDataPrefix}:${item.q.id}`,
|
|
'userId',
|
|
userId,
|
|
'modelId',
|
|
modelId,
|
|
'q',
|
|
item.q.text,
|
|
'text',
|
|
item.text,
|
|
'status',
|
|
ModelDataStatusEnum.waiting
|
|
]);
|
|
})
|
|
);
|
|
|
|
generateVector();
|
|
|
|
jsonRes(res, {
|
|
data: insertRes.filter((item) => item.status === 'rejected').length
|
|
});
|
|
} catch (err) {
|
|
jsonRes(res, {
|
|
code: 500,
|
|
error: err
|
|
});
|
|
}
|
|
}
|