From a7fc0f9d2e512d47d530066884a0cb1bc7ab7083 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Thu, 21 Nov 2024 14:02:27 -0800
Subject: [PATCH] feat: Add color and legend support for Gemini Pro models

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

diff --git a/benchmark/over_time.py b/benchmark/over_time.py
index 411f3dda0..e4de7c152 100644
--- a/benchmark/over_time.py
+++ b/benchmark/over_time.py
@@ -17,6 +17,8 @@ class ModelData:
     @property
     def color(self) -> str:
         model = self.name.lower()
+        if "gemini" in model and "pro" in model:
+            return "magenta"
         if "qwen" in model:
             return "darkblue"
         if "mistral" in model:
@@ -38,6 +40,8 @@ class ModelData:
     @property
     def legend_label(self) -> str:
         model = self.name.lower()
+        if "gemini" in model and "pro" in model:
+            return "Gemini Pro"
         if "claude-3-sonnet" in model:
             return "Sonnet"
         if "o1-preview" in model: