dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-31 09:44:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

feat: add model name aliases with canonical mapping

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-11-25 20:51:37 -08:00
parent 2ff3a23606
commit 445f9fa7df
1 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

14
aider/models.py
View file

@ -61,6 +61,17 @@ claude-3-5-sonnet-20241022
ANTHROPIC_MODELS = [ln.strip() for ln in ANTHROPIC_MODELS.splitlines() if ln.strip()]
# Mapping of model aliases to their canonical names
MODEL_ALIASES = {
"gpt-4": "openai/gpt-4",
"gpt-4-turbo": "gpt-4-turbo-preview",
"gpt-3.5": "gpt-3.5-turbo",
"claude-3": "claude-3-opus-20240229",
"claude-3-opus": "claude-3-opus-20240229",
"claude-3-sonnet": "claude-3-sonnet-20240229",
"claude-3-haiku": "claude-3-haiku-20240307",
}
@dataclass
class ModelSettings:
@ -819,7 +830,8 @@ model_info_manager = ModelInfoManager()
class Model(ModelSettings):
def __init__(self, model, weak_model=None, editor_model=None, editor_edit_format=None):
self.name = model
# Map any alias to its canonical name
self.name = MODEL_ALIASES.get(model, model)
self.max_chat_history_tokens = 1024
self.weak_model = None
self.editor_model = None