NO_COLOR support (#165)

Forethought-Technologies · Nov 27, 2023 · 6c756ee · 6c756ee
1 parent 238577f
commit 6c756ee
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/autochain/utils.py b/autochain/utils.py
@@ -7,8 +7,11 @@
 
 
 def print_with_color(text: str, color: str):
-    print(color + text)
-    print(Style.RESET_ALL)
+    if os.getenv("NO_COLOR"):
+        print(text)
+    else:
+        print(color + text)
+        print(Style.RESET_ALL)
 
 
 def get_from_dict_or_env(