diff --git a/aider/getinput.py b/aider/getinput.py
index 4840cbb7f..487b438d2 100644
--- a/aider/getinput.py
+++ b/aider/getinput.py
@@ -54,7 +54,9 @@ class FileContentCompleter(Completer):
 
 
 class InputOutput:
-    def __init__(self, pretty, yes, input_history_file, chat_history_file):
+    def __init__(self, pretty, yes, input_history_file, chat_history_file, input=None, output=None):
+        self.input = input
+        self.output = output
         self.pretty = pretty
         self.yes = yes
         self.input_history_file = input_history_file
@@ -123,6 +125,8 @@ class InputOutput:
                 style=style,
                 reserve_space_for_menu=4,
                 complete_style=CompleteStyle.MULTI_COLUMN,
+                input=self.input,
+                output=self.output,
             )
             if line.strip() == "{" and not multiline_input:
                 multiline_input = True
diff --git a/aider/main.py b/aider/main.py
index 048fe6ae9..0376cae9f 100644
--- a/aider/main.py
+++ b/aider/main.py
@@ -88,7 +88,7 @@ def main(args=None, input=None, output=None):
     fnames = args.files
     pretty = args.pretty
 
-    io = InputOutput(args.pretty, args.yes, args.input_history_file, args.chat_history_file)
+    io = InputOutput(args.pretty, args.yes, args.input_history_file, args.chat_history_file, input=input, output=output)
 
     coder = Coder(
         args.model, fnames, pretty, args.show_diffs, args.auto_commits, io,