89 lines
2.3 KiB
TypeScript
89 lines
2.3 KiB
TypeScript
import type { NextApiRequest, NextApiResponse } from 'next';
|
||
import { jsonRes } from '@/service/response';
|
||
import { Chat, Model, Training, connectToDatabase } from '@/service/mongo';
|
||
import { authToken } from '@/service/utils/tools';
|
||
import { getUserApiOpenai } from '@/service/utils/openai';
|
||
import { TrainingStatusEnum } from '@/constants/model';
|
||
import { TrainingItemType } from '@/types/training';
|
||
import { httpsAgent } from '@/service/utils/tools';
|
||
import { PgClient } from '@/service/pg';
|
||
|
||
/* 获取我的模型 */
|
||
export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
|
||
try {
|
||
const { modelId } = req.query as { modelId: string };
|
||
const { authorization } = req.headers;
|
||
|
||
if (!authorization) {
|
||
throw new Error('无权操作');
|
||
}
|
||
|
||
if (!modelId) {
|
||
throw new Error('参数错误');
|
||
}
|
||
|
||
// 凭证校验
|
||
const userId = await authToken(authorization);
|
||
|
||
// 验证是否是该用户的 model
|
||
const model = await Model.findOne({
|
||
_id: modelId,
|
||
userId
|
||
});
|
||
|
||
if (!model) {
|
||
throw new Error('无权操作该模型');
|
||
}
|
||
|
||
await connectToDatabase();
|
||
|
||
// 删除 pg 中所有该模型的数据
|
||
await PgClient.delete('modelData', {
|
||
where: [['user_id', userId], 'AND', ['model_id', modelId]]
|
||
});
|
||
|
||
// 删除对应的聊天
|
||
await Chat.deleteMany({
|
||
modelId
|
||
});
|
||
|
||
// 查看是否正在训练
|
||
const training: TrainingItemType | null = await Training.findOne({
|
||
modelId,
|
||
status: TrainingStatusEnum.pending
|
||
});
|
||
|
||
// 如果正在训练，需要删除openai上的相关信息
|
||
if (training) {
|
||
const { openai } = await getUserApiOpenai(userId);
|
||
// 获取训练记录
|
||
const tuneRecord = await openai.retrieveFineTune(training.tuneId, {
|
||
httpsAgent: httpsAgent(false)
|
||
});
|
||
|
||
// 删除训练文件
|
||
openai.deleteFile(tuneRecord.data.training_files[0].id, { httpsAgent: httpsAgent(false) });
|
||
// 取消训练
|
||
openai.cancelFineTune(training.tuneId, { httpsAgent: httpsAgent(false) });
|
||
}
|
||
|
||
// 删除对应训练记录
|
||
await Training.deleteMany({
|
||
modelId
|
||
});
|
||
|
||
// 删除模型
|
||
await Model.deleteOne({
|
||
_id: modelId,
|
||
userId
|
||
});
|
||
|
||
jsonRes(res);
|
||
} catch (err) {
|
||
jsonRes(res, {
|
||
code: 500,
|
||
error: err
|
||
});
|
||
}
|
||
}
|