From c675f09c549fbad8d75ea1da7370670a91bae744 Mon Sep 17 00:00:00 2001
From: Mark Ceter <133643956+maceter@users.noreply.github.com>
Date: Wed, 17 May 2023 18:39:34 +0000
Subject: [PATCH] Store original message as metadata

---
 public/scripts/extensions/infinity-context/index.js | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

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