rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Import all chats from a characterAI chat history dump

Browse Source 
Instead of only first chat found.
This commit is contained in:
maver
2023-04-18 20:10:28 +02:00
parent 57ea659540
commit 89aa307f16
1 changed files with 36 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
server.js
View File

@@ -1456,47 +1456,51 @@ app.post("/importchat", urlencodedParser, function (request, response) {
} }
const jsonData = json5.parse(data); const jsonData = json5.parse(data);
var new_chat = [];
if (jsonData.histories !== undefined) { if (jsonData.histories !== undefined) {
//console.log('/importchat confirms JSON histories are defined'); //console.log('/importchat confirms JSON histories are defined');
let i = 0; const chat = {
new_chat[i] = {}; from(history) {
new_chat[0]['user_name'] = 'You'; return [
new_chat[0]['character_name'] = ch_name; {
new_chat[0]['create_date'] = humanizedISO8601DateTime() //Date.now(); user_name: 'You',
i++; character_name: ch_name,
jsonData.histories.histories[0].msgs.forEach(function (item) { create_date: humanizedISO8601DateTime(),
new_chat[i] = {};
if (item.src.is_human == true) {
new_chat[i]['name'] = 'You';
} else {
new_chat[i]['name'] = ch_name;
}
new_chat[i]['is_user'] = item.src.is_human;
new_chat[i]['is_name'] = true;
new_chat[i]['send_date'] = humanizedISO8601DateTime() //Date.now();
new_chat[i]['mes'] = item.text;
i++;
});
const chatJsonlData = new_chat.map(JSON.stringify).join('\n');
//console.log('/importchat saving a file: '+ch_name+' - '+humanizedISO8601DateTime()+' imported.jsonl');
fs.writeFile(chatsPath + avatar_url + '/' + ch_name + ' - ' + humanizedISO8601DateTime() + ' imported.jsonl', chatJsonlData, 'utf8', function (err) { //added ch_name and replaced Date.now() with humanizedISO8601DateTime
if (err) { },
response.send(err); ...history.msgs.map(
return console.log(err); (message) => ({
//response.send(err); name: message.src.is_human ? 'You' : ch_name,
} else { is_user: message.src.is_human,
//response.redirect("/"); is_name: true,
response.send({ res: true }); send_date: humanizedISO8601DateTime(),
mes: message.text,
})
)];
} }
}
const newChats = [];
(jsonData.histories.histories ?? []).forEach((history) => {
newChats.push(chat.from(history));
}); });
const errors = [];
newChats.forEach(chat => fs.writeFile(
chatsPath + avatar_url + '/' + ch_name + ' - ' + humanizedISO8601DateTime() + ' imported.jsonl',
chat.map(JSON.stringify).join('\n'),
'utf8',
(err) => err ?? errors.push(err)
)
);
if (0 < errors.length) {
response.send('Errors occurred while writing character files. Errors: ' + JSON.stringify(errors));
}
response.send({res: true});
} else { } else {
response.send({ error: true }); response.send({ error: true });
return;
} }
}); });
} }
if (format === 'jsonl') { if (format === 'jsonl') {