fix(server): check if text is already in train data

This commit is contained in:
GramingFoxTeam
2022-11-18 12:58:02 +03:00
parent d332043b1a
commit edba5f7f3a

View File

@@ -6,9 +6,12 @@ export default function addTrainData(eventData) {
join(global.__dirname, global.config.fasttext.trainFile),
'utf-8'
);
const data = file.split('\n');
const { label, text } = eventData;
if (file.includes(text)) return;
const data = file.split('\n');
const labelIndex = data.findIndex((data) => data.startsWith(label));
data.splice(labelIndex === -1 ? 0 : labelIndex, 0, `${label} ${text}`);