refactor: Simplify model color detection logic for Sonnet models

This commit is contained in:
Paul Gauthier 2024-11-20 18:31:44 -08:00 committed by Paul Gauthier (aider)
parent 35115f5707
commit 16b319174b

View file

@ -12,7 +12,7 @@ def get_model_color(model):
if model == "gpt-4o-mini":
return default
if "claude-3-sonnet" in model.lower():
if "sonnet" in model.lower():
return "orange"
if "-4o" in model: