Skip to content

Commit b88254f

Browse files
committed
feat: Add time to pareto plot
1 parent ed636ef commit b88254f

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

agents_mcp_usage/multi_mcp/eval_multi_mcp/dashboard_config.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@
5858
"x_axis_options": {
5959
"cost": {"column": "total_cost", "label": "Cost"},
6060
"tokens": {"column": "total_response_tokens", "label": "Tokens"},
61+
"duration": {"column": "Duration", "label": "Duration"},
6162
},
6263
"color_axis": "Duration", # Column to use for the color scale
6364
},

agents_mcp_usage/multi_mcp/eval_multi_mcp/merbench_ui.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -419,6 +419,7 @@ def create_pareto_frontier_plot(
419419
y_axis=(primary_metric_name, "mean"),
420420
total_cost=("total_cost", "mean"),
421421
total_response_tokens=("total_response_tokens", "mean"),
422+
Duration=("Duration", "mean"),
422423
color_axis=(plot_config["color_axis"], "mean"),
423424
)
424425
.reset_index()
@@ -428,7 +429,12 @@ def create_pareto_frontier_plot(
428429
x_data = model_metrics[x_axis_config["column"]]
429430
x_title = x_axis_config["label"]
430431
hover_label = x_axis_config["label"]
431-
hover_format = ":.4f" if x_axis_mode == "cost" else ":.0f"
432+
if x_axis_mode == "cost":
433+
hover_format = ":.4f"
434+
elif x_axis_mode == "duration":
435+
hover_format = ":.2f"
436+
else:
437+
hover_format = ":.0f"
432438

433439
fig.add_trace(
434440
go.Scatter(

0 commit comments

Comments
 (0)