dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-04 19:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

all models to be registered with litellm

Browse source
This commit is contained in:
Krazer 2024-06-11 13:07:19 -05:00
parent d05f8e77b4
commit d91707f73e
3 changed files with 39 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
aider/main.py
View file

@ -332,6 +332,16 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F
if args.openai_organization_id:
os.environ["OPENAI_ORGANIZATION"] = args.openai_organization_id
model_def_files = []
model_def_fname = Path(".aider.models.json")
model_def_files.append(Path.home() / model_def_fname) # homedir
if git_root:
model_def_files.append(Path(git_root) / model_def_fname) # git root
if args.models:
model_def_files.append(args.models)
model_def_files = list(map(str, model_def_files))
models.register_models(model_def_files)
main_model = models.Model(args.model, weak_model=args.weak_model)
lint_cmds = parse_lint_cmds(args.lint_cmd, io)