refactor: update confirm_ask to use prompt_session or input

2025-05-30 01:04:59 +00:00 · 2024-09-10 14:06:51 -07:00 · 2024-09-10 14:06:51 -07:00 · d4e47bc7e7
commit d4e47bc7e7
parent 79a424bc10
1 changed files with 8 additions and 4 deletions
--- 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