Store original message as metadata

This commit is contained in:
Mark Ceter
2023-05-17 18:39:34 +00:00
parent 64ac2b2f58
commit c675f09c54

View File

@ -55,6 +55,7 @@ async function addMessages(chat_id, messages) {
role: m.is_user ? 'user' : 'assistant', role: m.is_user ? 'user' : 'assistant',
content: m.mes, content: m.mes,
date: m.send_date, date: m.send_date,
meta: JSON.stringify(m),
})); }));
const addMessagesResult = await fetch(url, { const addMessagesResult = await fetch(url, {
@ -113,13 +114,7 @@ window.chromadb_interceptGeneration = async () => {
queriedMessages.sort((a, b) => a.date - b.date); queriedMessages.sort((a, b) => a.date - b.date);
const newChat = [ const newChat = [
...queriedMessages.map(m => ({ ...queriedMessages.map(m => JSON.parse(m.meta)),
name: m.role === 'user' ? context.name1: context.name2,
is_user: m.role === 'user',
is_name: true,
send_date: m.date,
mes: m.content,
})),
...messagesToKeep, ...messagesToKeep,
]; ];