diff --git a/aider/io.py b/aider/io.py index 3dc0c352d..6583737ff 100644 --- a/aider/io.py +++ b/aider/io.py @@ -251,26 +251,32 @@ class InputOutput: else: self.console = Console(force_terminal=False, no_color=True) # non-pretty - def _get_style(self, style_dict=None): - if style_dict is None: - style_dict = {} - if self.pretty and self.user_input_color: + def _get_style(self): + style_dict = {} + if not self.pretty: + return Style.from_dict(style_dict) + + if self.user_input_color: style_dict.setdefault("", self.user_input_color) style_dict.update( { "pygments.literal.string": f"bold italic {self.user_input_color}", - "completion-menu": ( - f"bg:{self.completion_menu_bg_color} {self.completion_menu_color}" - ), - "completion-menu.completion.current": ( - f"bg:{self.completion_menu_current_bg_color} " - f"{self.completion_menu_current_color}" - ), } ) - return Style.from_dict(style_dict) - else: - return {} + + style_dict.update( + { + "completion-menu": ( + f"bg:{self.completion_menu_bg_color} {self.completion_menu_color}" + ), + "completion-menu.completion.current": ( + f"bg:{self.completion_menu_current_bg_color} " + f"{self.completion_menu_current_color}" + ), + } + ) + + return Style.from_dict(style_dict) def read_image(self, filename): try: