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

Warn if litellm is not giving us good validate_environment() responses

Browse source
This commit is contained in:
Paul Gauthier 2024-04-20 08:01:50 -07:00
parent 0969380ce9
commit e513174da4
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
aider/main.py
View file

@ -605,12 +605,16 @@ def main(argv=None, input=None, output=None, force_git_root=None):
# Is the model known and are all needed keys/params available? # Is the model known and are all needed keys/params available?
res = litellm.validate_environment(args.model) res = litellm.validate_environment(args.model)
missing_keys = res.get("missing_keys") missing_keys = res.get("missing_keys")
if missing_keys: keys_in_environment = res.get("keys_in_environment")
if not keys_in_environment and not missing_keys:
io.tool_error(f"Unable to check environment variables for model {args.model}")
elif missing_keys:
io.tool_error(f"To use model {args.model}, please set these environment variables:") io.tool_error(f"To use model {args.model}, please set these environment variables:")
for key in missing_keys: for key in missing_keys:
io.tool_error(f"- {key}") io.tool_error(f"- {key}")
return 1 return 1
elif not res["keys_in_environment"] and args.require_model_info: elif not keys_in_environment and args.require_model_info:
io.tool_error(models.check_model_name(args.model)) io.tool_error(models.check_model_name(args.model))
return 1 return 1