From ebcf4364f594bf8401ec603ea248e1a2711c03b8 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <paul@aider.chat>
Date: Thu, 6 Feb 2025 14:05:14 -0800
Subject: [PATCH] fix: Restore multiline mode after interruptions using
 decorator

---
 aider/io.py | 10 +---------
 1 file changed, 1 insertion(+), 9 deletions(-)

diff --git a/aider/io.py b/aider/io.py
index e2df60b81..ed8490e1f 100644
--- a/aider/io.py
+++ b/aider/io.py
@@ -775,15 +775,10 @@ class InputOutput:
         hist = f"{question.strip()} {res}"
         self.append_chat_history(hist, linebreak=True, blockquote=True)
 
-        # Restore original multiline mode
-        self.multiline_mode = orig_multiline
-
         return is_yes
 
+    @restore_multiline
     def prompt_ask(self, question, default="", subject=None):
-        # Temporarily disable multiline mode for prompts
-        orig_multiline = self.multiline_mode
-        self.multiline_mode = False
         self.num_user_asks += 1
 
         if subject:
@@ -812,9 +807,6 @@ class InputOutput:
         if self.yes in (True, False):
             self.tool_output(hist)
 
-        # Restore original multiline mode
-        self.multiline_mode = orig_multiline
-
         return res
 
     def _tool_message(self, message="", strip=True, color=None):