diff --git a/src/pages/api/data/getDataItems.ts b/src/pages/api/data/getDataItems.ts index 68cb7efd5..83668de03 100644 --- a/src/pages/api/data/getDataItems.ts +++ b/src/pages/api/data/getDataItems.ts @@ -35,8 +35,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse) pageSize, data: dataItems, total: await DataItem.countDocuments({ - dataId, - status: 0 + dataId }) } }); diff --git a/src/pages/api/timer/updateTraining.ts b/src/pages/api/timer/updateTraining.ts index 4d4fe6d83..2886b1307 100644 --- a/src/pages/api/timer/updateTraining.ts +++ b/src/pages/api/timer/updateTraining.ts @@ -9,6 +9,7 @@ import { getUserOpenaiKey } from '@/service/utils/tools'; import { OpenAiTuneStatusEnum } from '@/service/constants/training'; import { sendTrainSucceed } from '@/service/utils/sendEmail'; import { httpsAgent } from '@/service/utils/tools'; +import { ModelPopulate } from '@/types/mongoSchema'; export default async function handler(req: NextApiRequest, res: NextApiResponse) { if (req.headers.auth !== 'archer') { @@ -31,7 +32,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse) // 删除训练文件 openai.deleteFile(data.training_files[0].id, { httpsAgent }); - const model = await Model.findById(item.modelId).populate({ + const model = await Model.findById(item.modelId).populate({ path: 'userId', options: { strictPopulate: false diff --git a/src/pages/api/user/register.ts b/src/pages/api/user/register.ts index f2b153cc0..38554c287 100644 --- a/src/pages/api/user/register.ts +++ b/src/pages/api/user/register.ts @@ -46,6 +46,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse< // 根据 id 获取用户信息 const user = await User.findById(response._id); + if (!user) { + throw new Error('获取用户信息异常'); + } + jsonRes(res, { data: { token: generateToken(user._id), diff --git a/src/pages/api/user/updatePasswordByCode.ts b/src/pages/api/user/updatePasswordByCode.ts index 45fe2f915..ad3a2c308 100644 --- a/src/pages/api/user/updatePasswordByCode.ts +++ b/src/pages/api/user/updatePasswordByCode.ts @@ -44,6 +44,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse< email }); + if (!user) { + throw new Error('获取用户信息异常'); + } + jsonRes(res, { data: { token: generateToken(user._id),