mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 08:44:59 +00:00
feat: Add labels for Claude 3 Opus, Sonnet, and O1 Preview models
This commit is contained in:
parent
dd3bfaee01
commit
72c46ccec6
1 changed files with 4 additions and 2 deletions
|
@ -9,9 +9,11 @@ from aider.dump import dump # noqa: 401
|
||||||
def get_legend_label(model):
|
def get_legend_label(model):
|
||||||
model = model.lower()
|
model = model.lower()
|
||||||
if "claude-3-opus" in model:
|
if "claude-3-opus" in model:
|
||||||
return "Opus"
|
return "Claude 3 Opus"
|
||||||
if "claude-3-sonnet" in model:
|
if "claude-3-sonnet" in model:
|
||||||
return "Sonnet"
|
return "Claude 3 Sonnet"
|
||||||
|
if "o1-preview" in model:
|
||||||
|
return "O1 Preview"
|
||||||
if "gpt-3.5" in model:
|
if "gpt-3.5" in model:
|
||||||
return "GPT-3.5 Turbo"
|
return "GPT-3.5 Turbo"
|
||||||
if "gpt-4-" in model and "-4o" not in model:
|
if "gpt-4-" in model and "-4o" not in model:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue