chewt/ai-assistant
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Add tracking of entire conversation

Browse source
This commit is contained in:
Hayden Johnson 2024-10-01 11:52:31 -07:00
parent 0fb11b0691
commit d074c97645
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
assistant.py
View file

@ -150,10 +150,12 @@ def chat(message, stream=True):
return result return result
def chat2(args, user_input, stream=True): def chat2(args, user_input, stream=True):
conversation += 'user: ' + user_input + '\n'
if args.reflect: if args.reflect:
result = reflection_mode(user_input, stream) result = reflection_mode(user_input, stream)
else: else:
result = chat(user_input, stream) result = chat(user_input, stream)
conversation += 'assistant: ' + result + '\n'
return result return result
def highlightify_text(full_text): def highlightify_text(full_text):