From d074c97645d45b9e33b45c277c2b4d241fe5b1ff Mon Sep 17 00:00:00 2001
From: Hayden Johnson <hajohn100@gmail.com>
Date: Tue, 1 Oct 2024 11:52:31 -0700
Subject: [PATCH] Add tracking of entire conversation

---
 assistant.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/assistant.py b/assistant.py
index e71c9d8..9b47ce8 100755
--- a/assistant.py
+++ b/assistant.py
@@ -150,10 +150,12 @@ def chat(message, stream=True):
     return result
 
 def chat2(args, user_input, stream=True):
+    conversation += 'user: ' + user_input + '\n'
     if args.reflect:
         result = reflection_mode(user_input, stream)
     else:
         result = chat(user_input, stream)
+    conversation += 'assistant: ' + result + '\n'
     return result
 
 def highlightify_text(full_text):