dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-29 16:54:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: Add dynamic model selection based on available API keys

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

11
aider/main.py
View file

@ -752,13 +752,14 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F
# 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
# OPENAI -> gpt-4o
# DEEPSEEK -> deepseek
# GEMINI -> flash
# ai!
args.model = "gpt-4o-2024-08-06"
# Select model based on available API keys
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"
main_model = models.Model(
args.model,