changes

nmnduy · May 22, 2023 · f406b04 · f406b04
1 parent 4eb4c04
commit f406b04
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/doktor/prompt.py b/doktor/prompt.py
@@ -152,9 +152,11 @@ def get_prompt(state, # : State
                        reverse=True)
                 for msg in messages:
                     if msg.role == "user":
-                        print_green(f"You: {msg.content}" + "\n")
+                        print_green(f"You:\n")
+                        print(msg.content)
                     if msg.role == "assistant":
-                        print(f"Assistant: {msg.content}" + "\n")
+                        print_yellow(f"Assistant:\n")
+                        print(msg.content)
                 raise InputResetException()