feat: Add color and legend label for o1-preview models

This commit is contained in:
Paul Gauthier (aider) 2024-11-20 20:03:49 -08:00
parent b3e29ab20e
commit 2e00307190

View file

@ -26,6 +26,8 @@ def get_legend_label(model):
return "DeepSeek" return "DeepSeek"
if "mistral" in model: if "mistral" in model:
return "Mistral" return "Mistral"
if "o1-preview" in model:
return "o1-preview"
return model return model
@ -50,6 +52,9 @@ def get_model_color(model):
if "sonnet" in model.lower(): if "sonnet" in model.lower():
return "orange" return "orange"
if "o1-preview" in model.lower():
return "magenta"
if "-4o" in model: if "-4o" in model:
return "purple" return "purple"