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

feat: Add thinking token budget display in model details

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-11 11:35:23 -07:00
parent 98b3722a02
commit 5608db0892
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
aider/coders/base_coder.py
View file

@ -211,6 +211,19 @@ class Coder:
output += ", prompt cache"
if main_model.info.get("supports_assistant_prefill"):
output += ", infinite output"
# Check for thinking token budget
if (main_model.extra_params and
"thinking" in main_model.extra_params and
"budget_tokens" in main_model.extra_params["thinking"]):
budget = main_model.extra_params["thinking"]["budget_tokens"]
# Format as xx.yK for thousands, xx.yM for millions
if budget >= 1000000:
formatted_budget = f"{budget/1000000:.1f}M"
else:
formatted_budget = f"{budget/1000:.1f}K"
output += f", {formatted_budget} think tokens"
lines.append(output)
if self.edit_format == "architect":