Replace hardcoded model name abbreviations with dynamic truncation

- Strip common provider prefixes (gemini/, openai/, anthropic/, mistral/, cohere/)
- Truncate names > 24 chars with ellipsis (…) instead of hardcoded abbreviations
- Works generically for any LiteLLM model, not just our Gemini config
This commit is contained in:
rin.kagamine committed 2026-06-04 17:08:51 -04:00
1 parent c92dd9cf20
commit 3d4230ccb4
1 file changed
+12 -9
+12 -9
View File
@@ -207,15 +207,18 @@ class Tools:
model_lines = []
for model_name in sorted(model_totals, key=lambda k: model_totals[k]["spend"], reverse=True):
mt = model_totals[model_name]
# Strip provider prefix and abbreviate long model names for table readability
display = model_name.replace("gemini/", "") if model_name.startswith("gemini/") else model_name
# Abbreviate known long names
display = display.replace("3.1-pro-preview-customtools", "3.1-pro-ct")
display = display.replace("3.1-flash-image-preview", "3.1-flash-img")
display = display.replace("3.1-flash-lite-preview", "3.1-flash-lite")
display = display.replace("3-flash-preview", "3-flash-prev")
# Strip common provider prefixes for readability
display = model_name
for prefix in ("gemini/", "openai/", "anthropic/", "mistral/", "cohere/"):
if display.startswith(prefix):
display = display[len(prefix):]
break
# Truncate long model names to fit table, appending ellipsis
max_name = 24
if len(display) > max_name:
display = display[:max_name - 1] + "…"
model_lines.append(
f" {display:20s} ${mt['spend']:7.2f} | "
f" {display:25s} ${mt['spend']:7.2f} | "
f"{mt['requests']:3d} ok, {mt['failed']:d} fail | "
f"{mt['total']:>10,} tokens"
)
@@ -232,7 +235,7 @@ class Tools:
lines.append(f"Cache Created: {cache_created:,} tokens")
lines.append("")
lines.append("By Model:")
lines.append(f' {"Model":20s} {"Spend":>7s} | {"Requests":>14s} | {"Tokens":>10s}')
lines.append(f' {"Model":25s} {"Spend":>7s} | {"Requests":>14s} | {"Tokens":>10s}')
lines.extend(model_lines)
lines.append("")
lines.append("By Date:")