fix: fixed ROLE_COLOR for assistant

ErikBjare · Oct 30, 2023 · f5bcd2e · f5bcd2e
1 parent f26469c
commit f5bcd2e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/gptme/constants.py b/gptme/constants.py
@@ -4,16 +4,18 @@
 
 # Prompts
 ROLE_COLOR = {
-    "user": "bright_green",
-    "assistant": "bright_cyan",
+    "user": "green",
+    "assistant": "green",
     "system": "grey42",
 }
 
 # colors wrapped in \001 and \002 to inform readline about non-printable characters
 PROMPT_USER = (
     f"\001[bold {ROLE_COLOR['user']}]\002User\001[/bold {ROLE_COLOR['user']}]\002"
 )
-PROMPT_ASSISTANT = f"[bold {ROLE_COLOR['user']}]Assistant[/bold {ROLE_COLOR['user']}]"
+PROMPT_ASSISTANT = (
+    f"[bold {ROLE_COLOR['assistant']}]Assistant[/bold {ROLE_COLOR['assistant']}]"
+)
 
 # Config
 CONFIG_PATH = Path("~/.config/gptme").expanduser()