dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-02 02:34:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

refactor: Move format_tokens function to utils.py

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-26 15:46:26 -07:00
parent 93e37b60d0
commit bd9e6d7f58
2 changed files with 10 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

11
aider/coders/base_coder.py
View file

@ -22,7 +22,8 @@ import git
from rich.console import Console, Text
from rich.markdown import Markdown
from aider import __version__, models, prompts, urls, utils
from aider import __version__, models, prompts, urls
from aider.utils import format_tokens, utils
from aider.commands import Commands
from aider.history import ChatSummary
from aider.io import ConfirmGroup, InputOutput
@ -1488,14 +1489,6 @@ class Coder:
self.message_tokens_received += completion_tokens
def format_tokens(count):
if count < 1000:
return f"{count}"
elif count < 10000:
return f"{count / 1000:.1f}k"
else:
return f"{round(count / 1000)}k"
tokens_report = f"Tokens: {format_tokens(self.message_tokens_sent)} sent"
if cache_write_tokens: