Fix issue with last line not being printed

This commit is contained in:
Hayden Johnson 2024-09-25 13:19:03 -07:00
parent fb57930c69
commit c0a9377849

View file

@ -132,7 +132,7 @@ def chat(message, stream=True):
if not stream:
result = completion['message']['content']
if stream:
print()
print(large_text, flush=True)
history.append({"role": 'assistant', 'content': result})
return result