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

Modify cost formatting in usage_report to show at least 2 decimal places, with 2 non-zero digits.

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-07-29 08:24:24 -03:00
parent effa5236c6
commit d62baf0520
1 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

13
aider/coders/base_coder.py
View file

@ -11,6 +11,7 @@ import threading
import time
import traceback
from collections import defaultdict
import math
from datetime import datetime
from json.decoder import JSONDecodeError
from pathlib import Path
@ -1263,7 +1264,17 @@ class Coder:
if self.main_model.info.get("output_cost_per_token"):
cost += completion_tokens * self.main_model.info.get("output_cost_per_token")
self.total_cost += cost
self.usage_report += f" Cost: ${cost:.6f} request, ${self.total_cost:.6f} session."
def format_cost(value):
if value == 0:
return "0.00"
magnitude = abs(value)
if magnitude >= 0.01:
return f"{value:.2f}"
else:
return f"{value:.{max(2, 2 - int(math.log10(magnitude)))}f}"
self.usage_report += f" Cost: ${format_cost(cost)} request, ${format_cost(self.total_cost)} session."
def get_multi_response_content(self, final=False):
cur = self.multi_response_content