style: Format code with linter and improve readability

This commit is contained in:
Paul Gauthier (aider) 2025-03-11 11:35:31 -07:00
parent 5608db0892
commit c38cdef220

View file

@ -211,11 +211,13 @@ 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"]):
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:
@ -223,7 +225,7 @@ class Coder:
else:
formatted_budget = f"{budget/1000:.1f}K"
output += f", {formatted_budget} think tokens"
lines.append(output)
if self.edit_format == "architect":