diff --git a/litellm_dashboard.py b/litellm_dashboard.py index c1c11c0..e16b4cb 100644 --- a/litellm_dashboard.py +++ b/litellm_dashboard.py @@ -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:")