dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 09:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Accept colors for various prompts

Browse source
This commit is contained in:
Paul Gauthier 2023-05-27 15:14:56 -07:00
parent 03cc679a5b
commit f6d07d9849
1 changed files with 14 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

19
aider/io.py
View file

@ -82,9 +82,11 @@ class InputOutput:
chat_history_file=None, chat_history_file=None,
input=None, input=None,
output=None, output=None,
tool_color="green", user_color="green",
tool_color=None,
tool_error_color="red", tool_error_color="red",
): ):
self.user_color = user_color
self.tool_color = tool_color self.tool_color = tool_color
self.tool_error_color = tool_error_color self.tool_error_color = tool_error_color
self.input = input self.input = input
@ -120,7 +122,10 @@ class InputOutput:
inp = "" inp = ""
multiline_input = False multiline_input = False
style = Style.from_dict({"": self.tool_color}) if self.user_color:
style = Style.from_dict({"": self.user_color})
else:
style = None
while True: while True:
completer_instance = FileContentCompleter( completer_instance = FileContentCompleter(
@ -132,12 +137,14 @@ class InputOutput:
session_kwargs = { session_kwargs = {
"message": show, "message": show,
"completer": completer_instance, "completer": completer_instance,
"style": style,
"reserve_space_for_menu": 4, "reserve_space_for_menu": 4,
"complete_style": CompleteStyle.MULTI_COLUMN, "complete_style": CompleteStyle.MULTI_COLUMN,
"input": self.input, "input": self.input,
"output": self.output, "output": self.output,
} }
if style:
session_kwargs["style"] = style
if self.input_history_file is not None: if self.input_history_file is not None:
session_kwargs["history"] = FileHistory(self.input_history_file) session_kwargs["history"] = FileHistory(self.input_history_file)
@ -207,7 +214,8 @@ class InputOutput:
self.append_chat_history(hist, linebreak=True, blockquote=True) self.append_chat_history(hist, linebreak=True, blockquote=True)
message = Text(message) message = Text(message)
self.console.print(message, style=self.tool_error_color) style = dict(style=self.tool_error_color) if self.tool_error_color else dict()
self.console.print(message, **style)
def tool(self, *messages, log_only=False): def tool(self, *messages, log_only=False):
if messages: if messages:
@ -217,7 +225,8 @@ class InputOutput:
if not log_only: if not log_only:
messages = list(map(Text, messages)) messages = list(map(Text, messages))
self.console.print(*messages) style = dict(style=self.tool_color) if self.tool_color else dict()
self.console.print(*messages, **style)
def append_chat_history(self, text, linebreak=False, blockquote=False): def append_chat_history(self, text, linebreak=False, blockquote=False):
if blockquote: if blockquote: