From 2e00307190d0470167818533ef1bbbefe9e43a12 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Wed, 20 Nov 2024 20:03:49 -0800
Subject: [PATCH] feat: Add color and legend label for o1-preview models

---
 benchmark/over_time.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/benchmark/over_time.py b/benchmark/over_time.py
index 3955c563a..3d0bb9ae4 100644
--- a/benchmark/over_time.py
+++ b/benchmark/over_time.py
@@ -26,6 +26,8 @@ def get_legend_label(model):
         return "DeepSeek"
     if "mistral" in model:
         return "Mistral"
+    if "o1-preview" in model:
+        return "o1-preview"
     return model
 
 
@@ -50,6 +52,9 @@ def get_model_color(model):
     if "sonnet" in model.lower():
         return "orange"
 
+    if "o1-preview" in model.lower():
+        return "magenta"
+
     if "-4o" in model:
         return "purple"