From b4b54d17967d0211352cc7d62668de97129f42d1 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Fri, 31 Jan 2025 08:46:16 -0800 Subject: [PATCH] feat: Add support for OpenRouter and GPT-4o model selection with improved model detection --- aider/main.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/aider/main.py b/aider/main.py index 8e63e3139..3c64b3576 100644 --- a/aider/main.py +++ b/aider/main.py @@ -751,14 +751,19 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F if not args.model: # Select model based on available API keys model_key_pairs = [ - ("ANTHROPIC_API_KEY", "claude-3-5-sonnet-20241022"), - ("DEEPSEEK_API_KEY", "deepseek-chat"), + ("ANTHROPIC_API_KEY", "sonnet"), + ("DEEPSEEK_API_KEY", "deepseek"), + ("OPENROUTER_API_KEY", "openrouter/anthropic/claude-3.5-sonnet"), + ("OPENAI_API_KEY", "gpt-4o"), ("GEMINI_API_KEY", "flash"), ] args.model = "gpt-4o-2024-08-06" # default for env_key, model_name in model_key_pairs: if os.environ.get(env_key): args.model = model_name + io.tool_warning( + f"Found {env_key} so using {model_name} since no --model was specified." + ) break main_model = models.Model(