dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 01:04:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: Add color and legend support for Gemini Pro models

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-11-21 14:02:27 -08:00
parent 9eab021a50
commit a7fc0f9d2e
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
benchmark/over_time.py
View file

@ -17,6 +17,8 @@ class ModelData:
@property @property
def color(self) -> str: def color(self) -> str:
model = self.name.lower() model = self.name.lower()
if "gemini" in model and "pro" in model:
return "magenta"
if "qwen" in model: if "qwen" in model:
return "darkblue" return "darkblue"
if "mistral" in model: if "mistral" in model:
@ -38,6 +40,8 @@ class ModelData:
@property @property
def legend_label(self) -> str: def legend_label(self) -> str:
model = self.name.lower() model = self.name.lower()
if "gemini" in model and "pro" in model:
return "Gemini Pro"
if "claude-3-sonnet" in model: if "claude-3-sonnet" in model:
return "Sonnet" return "Sonnet"
if "o1-preview" in model: if "o1-preview" in model: