Add tracking of entire conversation

2024-10-01 11:52:31 -07:00 · 2024-10-01 11:52:31 -07:00 · d074c97645
parent 0fb11b0691
commit d074c97645
1 changed files with 2 additions and 0 deletions
--- 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):