Formatting script for NobodyExistsOnTheInternet/toxicqa

This commit is contained in:
James S 2024-05-25 04:36:17 +00:00
parent a572dc62f1
commit 59264b9421

22
data/procToxicQA.js Normal file
View File

@ -0,0 +1,22 @@
const fs = require('node:fs');
var lineReader = require('readline').createInterface({
input: fs.createReadStream('toxicQA.json')
});
var outstream = fs.createWriteStream('toxicQAfinal.json');
fs.unlinkSync('toxicQAfinal.json');
lineReader.on('line', function (line) {
const dialogue = JSON.parse(line)["conversations"];
const newdialogue = [];
for (const dialogueLine of dialogue) {
newdialogue.push({
role: dialogueLine["from"] === "human" ? "user" : "assistant",
content: dialogueLine["value"]
});
}
outstream.write(JSON.stringify(newdialogue) + '\n');
});
lineReader.on('close', function () {
console.log('all done, son');
});