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 = []
|
model_lines = []
|
||||||
for model_name in sorted(model_totals, key=lambda k: model_totals[k]["spend"], reverse=True):
|
for model_name in sorted(model_totals, key=lambda k: model_totals[k]["spend"], reverse=True):
|
||||||
mt = model_totals[model_name]
|
mt = model_totals[model_name]
|
||||||
# Strip provider prefix and abbreviate long model names for table readability
|
# Strip common provider prefixes for readability
|
||||||
display = model_name.replace("gemini/", "") if model_name.startswith("gemini/") else model_name
|
display = model_name
|
||||||
# Abbreviate known long names
|
for prefix in ("gemini/", "openai/", "anthropic/", "mistral/", "cohere/"):
|
||||||
display = display.replace("3.1-pro-preview-customtools", "3.1-pro-ct")
|
if display.startswith(prefix):
|
||||||
display = display.replace("3.1-flash-image-preview", "3.1-flash-img")
|
display = display[len(prefix):]
|
||||||
display = display.replace("3.1-flash-lite-preview", "3.1-flash-lite")
|
break
|
||||||
display = display.replace("3-flash-preview", "3-flash-prev")
|
# 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(
|
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['requests']:3d} ok, {mt['failed']:d} fail | "
|
||||||
f"{mt['total']:>10,} tokens"
|
f"{mt['total']:>10,} tokens"
|
||||||
)
|
)
|
||||||
@@ -232,7 +235,7 @@ class Tools:
|
|||||||
lines.append(f"Cache Created: {cache_created:,} tokens")
|
lines.append(f"Cache Created: {cache_created:,} tokens")
|
||||||
lines.append("")
|
lines.append("")
|
||||||
lines.append("By Model:")
|
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.extend(model_lines)
|
||||||
lines.append("")
|
lines.append("")
|
||||||
lines.append("By Date:")
|
lines.append("By Date:")
|
||||||
|
|||||||
Reference in new issue
Block a user