diff --git a/aider/io.py b/aider/io.py index f2b89828a..4c8f812e6 100644 --- a/aider/io.py +++ b/aider/io.py @@ -13,6 +13,7 @@ from pathlib import Path def restore_multiline(func): """Decorator to restore multiline mode after function execution""" + @functools.wraps(func) def wrapper(self, *args, **kwargs): orig_multiline = self.multiline_mode @@ -23,8 +24,10 @@ def restore_multiline(func): raise finally: self.multiline_mode = orig_multiline + return wrapper + from prompt_toolkit.completion import Completer, Completion, ThreadedCompleter from prompt_toolkit.cursor_shapes import ModalCursorShapeConfig from prompt_toolkit.enums import EditingMode