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

refactor: Replace if/else chain with list of tuples for model selection

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-01-31 08:42:25 -08:00
parent dad3092d8d
commit 5023dfeb24
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
aider/main.py
View file

@ -749,17 +749,17 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F
models.MODEL_ALIASES[alias.strip()] = model.strip()
if not args.model:
# make a list of tuples XXX_API_KEY, "model-name" and run through it to pick
# a model based on the first key set
# ANTHROPIC -> sonnet
# Select model based on available API keys
model_key_pairs = [
("ANTHROPIC_API_KEY", "claude-3-5-sonnet-20241022"),
("DEEPSEEK_API_KEY", "deepseek-chat"),
("GEMINI_API_KEY", "flash"),
]
args.model = "gpt-4o-2024-08-06" # default
if os.environ.get("ANTHROPIC_API_KEY"):
args.model = "claude-3-5-sonnet-20241022"
elif os.environ.get("DEEPSEEK_API_KEY"):
args.model = "deepseek-chat"
elif os.environ.get("GEMINI_API_KEY"):
args.model = "flash"
for env_key, model_name in model_key_pairs:
if os.environ.get(env_key):
args.model = model_name
break
main_model = models.Model(
args.model,