From 16b319174b44a81ce4907e22eb471e5d15f29cfa Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Wed, 20 Nov 2024 18:31:44 -0800
Subject: [PATCH] refactor: Simplify model color detection logic for Sonnet
 models

---
 benchmark/over_time.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/benchmark/over_time.py b/benchmark/over_time.py
index 455a07ccb..98d206794 100644
--- a/benchmark/over_time.py
+++ b/benchmark/over_time.py
@@ -12,7 +12,7 @@ def get_model_color(model):
     if model == "gpt-4o-mini":
         return default
 
-    if "claude-3-sonnet" in model.lower():
+    if "sonnet" in model.lower():
         return "orange"
 
     if "-4o" in model: