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

feat: Add percentage column to model token usage summary

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-12-13 12:33:14 -08:00
parent 2d5f613984
commit 66e5e9c1ce
1 changed files with 10 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

16
scripts/my_models.py
View file

@ -30,14 +30,18 @@ with open(analytics_path) as f:
except json.JSONDecodeError:
continue
# Calculate total for percentages
total_tokens = sum(model_stats.values())
# Print results
print("\nModel Token Usage Summary:")
print("-" * 60)
print(f"{'Model Name':<40} {'Total Tokens':>15}")
print("-" * 60)
print("-" * 80)
print(f"{'Model Name':<40} {'Total Tokens':>15} {'Percent':>10}")
print("-" * 80)
for model, tokens in sorted(model_stats.items(), key=lambda x: x[1], reverse=True):
print(f"{model:<40} {tokens:>15,}")
percentage = (tokens / total_tokens) * 100 if total_tokens > 0 else 0
print(f"{model:<40} {tokens:>15,} {percentage:>9.1f}%")
print("-" * 60)
print(f"{'TOTAL':<40} {sum(model_stats.values()):>15,}")
print("-" * 80)
print(f"{'TOTAL':<40} {total_tokens:>15,} {100:>9.1f}%")