dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-03 19:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

added --assistant-output-color

Browse source
This commit is contained in:
Paul Gauthier 2023-06-19 15:59:47 -07:00
parent 824ec054c6
commit d76bf0b8b0
2 changed files with 16 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
aider/coder.py
View file

@ -51,6 +51,7 @@ class Coder:
verbose=False, verbose=False,
openai_api_key=None, openai_api_key=None,
openai_api_base=None, openai_api_base=None,
assistant_output_color="blue",
): ):
if not openai_api_key: if not openai_api_key:
raise MissingAPIKeyError("No OpenAI API key provided.") raise MissingAPIKeyError("No OpenAI API key provided.")
@ -69,6 +70,7 @@ class Coder:
self.auto_commits = auto_commits self.auto_commits = auto_commits
self.dirty_commits = dirty_commits self.dirty_commits = dirty_commits
self.assistant_output_color = assistant_output_color
self.dry_run = dry_run self.dry_run = dry_run
self.pretty = pretty self.pretty = pretty
@ -493,7 +495,9 @@ class Coder:
show_resp = self.update_files_gpt35(self.resp, mode="diff") show_resp = self.update_files_gpt35(self.resp, mode="diff")
except ValueError: except ValueError:
pass 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) live.update(md)
else: else:
sys.stdout.write(text) sys.stdout.write(text)

13
aider/main.py
View file

@ -4,7 +4,7 @@ import sys
import configargparse import configargparse
import git import git
from aider import models, __version__ from aider import __version__, models
from aider.coder import Coder from aider.coder import Coder
from aider.io import InputOutput from aider.io import InputOutput
@ -38,7 +38,10 @@ def main(args=None, input=None, output=None):
) )
parser.add_argument( 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( parser.add_argument(
@ -117,6 +120,11 @@ def main(args=None, input=None, output=None):
default="red", default="red",
help="Set the color for tool error messages (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( parser.add_argument(
"--apply", "--apply",
metavar="FILE", metavar="FILE",
@ -232,6 +240,7 @@ def main(args=None, input=None, output=None):
verbose=args.verbose, verbose=args.verbose,
openai_api_key=args.openai_api_key, openai_api_key=args.openai_api_key,
openai_api_base=args.openai_api_base, openai_api_base=args.openai_api_base,
assistant_output_color=args.assistant_output_color,
) )
if args.dirty_commits: if args.dirty_commits: