mirror of
https://github.com/labring/FastGPT.git
synced 2025-07-22 20:37:48 +00:00

Co-authored-by: Mufei <327958099@qq.com> Co-authored-by: heheer <71265218+newfish-cmyk@users.noreply.github.com>
42 lines
1.0 KiB
TypeScript
42 lines
1.0 KiB
TypeScript
import { DatasetTrainingSchemaType } from '@fastgpt/global/core/dataset/type';
|
|
import { addLog } from '../../../common/system/log';
|
|
import { getErrText } from '@fastgpt/global/common/error/utils';
|
|
import { MongoDatasetTraining } from './schema';
|
|
|
|
export const checkInvalidChunkAndLock = async ({
|
|
err,
|
|
errText,
|
|
data
|
|
}: {
|
|
err: any;
|
|
errText: string;
|
|
data: DatasetTrainingSchemaType;
|
|
}) => {
|
|
if (err?.response) {
|
|
addLog.error(`openai error: ${errText}`, {
|
|
status: err.response?.status,
|
|
statusText: err.response?.statusText,
|
|
data: err.response?.data
|
|
});
|
|
} else {
|
|
addLog.error(getErrText(err, errText), err);
|
|
}
|
|
|
|
if (
|
|
err?.message === 'invalid message format' ||
|
|
err?.type === 'invalid_request_error' ||
|
|
err?.code === 500
|
|
) {
|
|
addLog.info('Lock training data');
|
|
console.log(err);
|
|
|
|
try {
|
|
await MongoDatasetTraining.findByIdAndUpdate(data._id, {
|
|
lockTime: new Date('2998/5/5')
|
|
});
|
|
} catch (error) {}
|
|
return true;
|
|
}
|
|
return false;
|
|
};
|