From d76bf0b8b0684f5cdac93cd5325e0236f7ee244f Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Mon, 19 Jun 2023 15:59:47 -0700
Subject: [PATCH] added --assistant-output-color

---
 aider/coder.py |  6 +++++-
 aider/main.py  | 13 +++++++++++--
 2 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/aider/coder.py b/aider/coder.py
index 22434d18b..072f3cf86 100755
--- a/aider/coder.py
+++ b/aider/coder.py
@@ -51,6 +51,7 @@ class Coder:
         verbose=False,
         openai_api_key=None,
         openai_api_base=None,
+        assistant_output_color="blue",
     ):
         if not openai_api_key:
             raise MissingAPIKeyError("No OpenAI API key provided.")
@@ -69,6 +70,7 @@ class Coder:
 
         self.auto_commits = auto_commits
         self.dirty_commits = dirty_commits
+        self.assistant_output_color = assistant_output_color
 
         self.dry_run = dry_run
         self.pretty = pretty
@@ -493,7 +495,9 @@ class Coder:
                             show_resp = self.update_files_gpt35(self.resp, mode="diff")
                         except ValueError:
                             pass
-                    md = Markdown(show_resp, style="blue", code_theme="default")
+                    md = Markdown(
+                        show_resp, style=self.assistant_output_color, code_theme="default"
+                    )
                     live.update(md)
                 else:
                     sys.stdout.write(text)
diff --git a/aider/main.py b/aider/main.py
index 76f31af51..8d2ade72c 100644
--- a/aider/main.py
+++ b/aider/main.py
@@ -4,7 +4,7 @@ import sys
 import configargparse
 import git
 
-from aider import models, __version__
+from aider import __version__, models
 from aider.coder import Coder
 from aider.io import InputOutput
 
@@ -38,7 +38,10 @@ def main(args=None, input=None, output=None):
     )
 
     parser.add_argument(
-        "--version", action="version", version=f"%(prog)s {__version__}", help="Show the version number and exit"
+        "--version",
+        action="version",
+        version=f"%(prog)s {__version__}",
+        help="Show the version number and exit",
     )
 
     parser.add_argument(
@@ -117,6 +120,11 @@ def main(args=None, input=None, output=None):
         default="red",
         help="Set the color for tool error messages (default: red)",
     )
+    parser.add_argument(
+        "--assistant-output-color",
+        default="blue",
+        help="Set the color for assistant output (default: blue)",
+    )
     parser.add_argument(
         "--apply",
         metavar="FILE",
@@ -232,6 +240,7 @@ def main(args=None, input=None, output=None):
         verbose=args.verbose,
         openai_api_key=args.openai_api_key,
         openai_api_base=args.openai_api_base,
+        assistant_output_color=args.assistant_output_color,
     )
 
     if args.dirty_commits: