From d4e47bc7e73382e4d7bebbf67c863dd060a4bfb8 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Tue, 10 Sep 2024 14:06:51 -0700
Subject: [PATCH] refactor: update confirm_ask to use prompt_session or input

---
 aider/io.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/aider/io.py b/aider/io.py
index 3bdbc3589..ba38758be 100644
--- a/aider/io.py
+++ b/aider/io.py
@@ -471,10 +471,14 @@ class InputOutput:
             self.user_input(f"{question}{res}", log_only=False)
         else:
             while True:
-                res = prompt(
-                    question,
-                    style=Style.from_dict(style),
-                )
+                if self.prompt_session:
+                    res = self.prompt_session.prompt(
+                        question,
+                        style=Style.from_dict(style),
+                    )
+                else:
+                    res = input(question)
+                
                 if not res:
                     res = "y"  # Default to Yes if no input
                     break