From a8cbd81c614663a63e90a5bdb59f3ac831eccbb4 Mon Sep 17 00:00:00 2001
From: Kalvin C <kalvinnchau@users.noreply.github.com>
Date: Wed, 9 Apr 2025 11:39:35 -0700
Subject: [PATCH] feat: support NO_COLOR when PrettyPrinter is used (#2098)

---
 crates/goose-cli/src/session/output.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/crates/goose-cli/src/session/output.rs b/crates/goose-cli/src/session/output.rs
index a86e062f..9d330268 100644
--- a/crates/goose-cli/src/session/output.rs
+++ b/crates/goose-cli/src/session/output.rs
@@ -358,10 +358,17 @@ fn print_tool_header(call: &ToolCall) {
     println!("{}", tool_header);
 }
 
+// Respect NO_COLOR, as https://crates.io/crates/console already does
+pub fn env_no_color() -> bool {
+    // if NO_COLOR is defined at all disable colors
+    std::env::var_os("NO_COLOR").is_none()
+}
+
 fn print_markdown(content: &str, theme: Theme) {
     bat::PrettyPrinter::new()
         .input(bat::Input::from_bytes(content.as_bytes()))
         .theme(theme.as_str())
+        .colored_output(env_no_color())
         .language("Markdown")
         .wrapping_mode(WrappingMode::NoWrapping(true))
         .print()