fix: Update Gemini Pro legend label to Gemini 1.5 Pro

This commit is contained in:
Paul Gauthier 2024-11-21 14:19:03 -08:00 committed by Paul Gauthier (aider)
parent 2412c81d92
commit 781a40df52

View file

@ -41,7 +41,7 @@ class ModelData:
def legend_label(self) -> str:
model = self.name.lower()
if "gemini" in model and "pro" in model:
return "Gemini Pro"
return "Gemini 1.5 Pro"
if "claude-3-sonnet" in model:
return "Sonnet"
if "o1-preview" in model: