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):