mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 16:54:59 +00:00
feat: Add color and legend label for o1-preview models
This commit is contained in:
parent
b3e29ab20e
commit
2e00307190
1 changed files with 5 additions and 0 deletions
|
@ -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"
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue