dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-25 14:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

print -> tool_output

Browse source
This commit is contained in:
Paul Gauthier 2023-06-23 10:12:19 -07:00
parent 803423cc6a
commit f9599f87a2
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
aider/commands.py
View file

@ -120,8 +120,8 @@ class Commands:
tokens = len(self.tokenizer.encode(quoted)) tokens = len(self.tokenizer.encode(quoted))
res.append((tokens, f"{relative_fname}", "use /drop to drop from chat")) res.append((tokens, f"{relative_fname}", "use /drop to drop from chat"))
print("Approximate context window usage, in tokens:") self.io.tool_output("Approximate context window usage, in tokens:")
print() self.io.tool_output()
width = 8 width = 8
@ -134,18 +134,18 @@ class Commands:
for tk, msg, tip in res: for tk, msg, tip in res:
total += tk total += tk
msg = msg.ljust(col_width) msg = msg.ljust(col_width)
print(f"{fmt(tk)} {msg} {tip}") self.io.tool_output(f"{fmt(tk)} {msg} {tip}")
print("=" * width) self.io.tool_output("=" * width)
print(f"{fmt(total)} tokens total") self.io.tool_output(f"{fmt(total)} tokens total")
limit = self.coder.main_model.max_context_tokens limit = self.coder.main_model.max_context_tokens
remaining = limit - total remaining = limit - total
if remaining > 0: if remaining > 0:
print(f"{fmt(remaining)} tokens remaining in context window") self.io.tool_output(f"{fmt(remaining)} tokens remaining in context window")
else: else:
print(f"{fmt(remaining)} tokens remaining, window exhausted!") self.io.tool_output(f"{fmt(remaining)} tokens remaining, window exhausted!")
print(f"{fmt(limit)} tokens max context window size") self.io.tool_output(f"{fmt(limit)} tokens max context window size")
def cmd_undo(self, args): def cmd_undo(self, args):
"Undo the last git commit if it was done by aider" "Undo the last git commit if it was done by aider"
@ -302,7 +302,7 @@ class Commands:
except Exception as e: except Exception as e:
self.io.tool_error(f"Error running command: {e}") self.io.tool_error(f"Error running command: {e}")
print(combined_output) self.io.tool_output(combined_output)
if self.io.confirm_ask("Add the output to the chat?", default="y"): if self.io.confirm_ask("Add the output to the chat?", default="y"):
for line in combined_output.splitlines(): for line in combined_output.splitlines():