From 72c46ccec6cb1a48b9774c7e5c5410b379ed1fb1 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Wed, 20 Nov 2024 20:06:04 -0800
Subject: [PATCH] feat: Add labels for Claude 3 Opus, Sonnet, and O1 Preview
 models

---
 benchmark/over_time.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/benchmark/over_time.py b/benchmark/over_time.py
index 8eef15cfa..91ed11d89 100644
--- a/benchmark/over_time.py
+++ b/benchmark/over_time.py
@@ -9,9 +9,11 @@ from aider.dump import dump  # noqa: 401
 def get_legend_label(model):
     model = model.lower()
     if "claude-3-opus" in model:
-        return "Opus"
+        return "Claude 3 Opus"
     if "claude-3-sonnet" in model:
-        return "Sonnet"
+        return "Claude 3 Sonnet"
+    if "o1-preview" in model:
+        return "O1 Preview"
     if "gpt-3.5" in model:
         return "GPT-3.5 Turbo"
     if "gpt-4-" in model and "-4o" not in model: