mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 16:54:59 +00:00
feat: Add dynamic model selection based on available API keys
This commit is contained in:
parent
ab77c032de
commit
dad3092d8d
1 changed files with 6 additions and 5 deletions
|
@ -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
|
# make a list of tuples XXX_API_KEY, "model-name" and run through it to pick
|
||||||
# a model based on the first key set
|
# a model based on the first key set
|
||||||
# ANTHROPIC -> sonnet
|
# ANTHROPIC -> sonnet
|
||||||
# OPENAI -> gpt-4o
|
# Select model based on available API keys
|
||||||
# DEEPSEEK -> deepseek
|
args.model = "gpt-4o-2024-08-06" # default
|
||||||
# GEMINI -> flash
|
|
||||||
# ai!
|
|
||||||
args.model = "gpt-4o-2024-08-06"
|
|
||||||
if os.environ.get("ANTHROPIC_API_KEY"):
|
if os.environ.get("ANTHROPIC_API_KEY"):
|
||||||
args.model = "claude-3-5-sonnet-20241022"
|
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(
|
main_model = models.Model(
|
||||||
args.model,
|
args.model,
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue