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:
1 parent
c92dd9cf20
commit
3d4230ccb4
1 file changed
+12
-9
+12
-9
@@ -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:")
|
||||
|
||||
Reference in new issue
Block a user