refactor: Remove Opus and Llama model variants from legend labels

This commit is contained in:
Paul Gauthier (aider) 2024-11-20 20:07:52 -08:00
parent c2c7ee1047
commit c15ac341e2

View file

@ -8,8 +8,6 @@ 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:
return "Opus"
if "claude-3-sonnet" in model: if "claude-3-sonnet" in model:
return "Claude 3 Sonnet" return "Claude 3 Sonnet"
if "o1-preview" in model: if "o1-preview" in model: