dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-02 18:54:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

feat: store partial input when interrupting prompt

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-10-25 13:17:54 -07:00
parent 1261a335f4
commit 88c54c918c
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
aider/io.py
View file

@ -180,6 +180,7 @@ class InputOutput:
chat_history_file=None, chat_history_file=None,
input=None, input=None,
output=None, output=None,
interrupted_partial_input=None,
user_input_color="blue", user_input_color="blue",
tool_output_color=None, tool_output_color=None,
tool_error_color="red", tool_error_color="red",
@ -197,6 +198,7 @@ class InputOutput:
fancy_input=True, fancy_input=True,
): ):
self.never_prompts = set() self.never_prompts = set()
self.interrupted_partial_input = interrupted_partial_input
self.editingmode = editingmode self.editingmode = editingmode
no_color = os.environ.get("NO_COLOR") no_color = os.environ.get("NO_COLOR")
if no_color is not None and no_color != "": if no_color is not None and no_color != "":
@ -349,6 +351,8 @@ class InputOutput:
def interrupt_input(self): def interrupt_input(self):
if self.prompt_session and self.prompt_session.app: if self.prompt_session and self.prompt_session.app:
# Store any partial input before interrupting
self.interrupted_partial_input = self.prompt_session.app.current_buffer.text
self.prompt_session.app.exit() self.prompt_session.app.exit()
print("interrupting") print("interrupting")