dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 17:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refactor: conditionally initialize and use PromptSession based on pretty flag

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-09-10 13:49:22 -07:00
parent ed866d33e4
commit 79a424bc10
1 changed files with 27 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

8
aider/io.py
View file

@ -219,6 +219,8 @@ class InputOutput:
current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
self.append_chat_history(f"\n# aider chat started at {current_time}\n\n")
self.prompt_session = None
if self.pretty:
# Initialize PromptSession
session_kwargs = {
"input": self.input,
@ -339,6 +341,7 @@ class InputOutput:
show = ". "
try:
if self.prompt_session:
line = self.prompt_session.prompt(
show,
completer=completer_instance,
@ -347,6 +350,8 @@ class InputOutput:
style=style,
key_bindings=kb,
)
else:
line = input(show)
except UnicodeEncodeError as err:
self.tool_error(str(err))
return ""
@ -519,7 +524,10 @@ class InputOutput:
elif self.yes is False:
res = "no"
else:
if self.prompt_session:
res = self.prompt_session.prompt(question + " ", default=default, style=style)
else:
res = input(question + " ")
hist = f"{question.strip()} {res.strip()}"
self.append_chat_history(hist, linebreak=True, blockquote=True)