Adding color to the error message

avsthiago · Apr 13, 2023 · 526cb8b · 526cb8b
1 parent 539294d
commit 526cb8b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/kopylot/llm.py b/kopylot/llm.py
@@ -2,15 +2,16 @@
 import sys
 
 import openai
+import rich
 
 openai.api_key = os.getenv("KOPYLOT_AUTH_TOKEN", "")
 
 
 def validate_token() -> None:
     if openai.api_key == "":
-        print(
-            "Please set your OpenAI API key as an environment variable named KOPYLOT_AUTH_TOKEN",
-            file=sys.stderr,
+        rich.print(
+            "[bold red]Error:[/bold red] Please set your OpenAI API key as an environment variable named"
+            " KOPYLOT_AUTH_TOKEN"
         )
         sys.exit(1)