From 843836ea87770c5f5a726375f7e5b118672007da Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Mon, 12 Aug 2024 15:08:04 -0700 Subject: [PATCH] feat: Add --deepseek-beta bool arg and set model name and DEEPSEEK_API_BASE when used --- aider/main.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/aider/main.py b/aider/main.py index de50dbc70..72308f7a1 100644 --- a/aider/main.py +++ b/aider/main.py @@ -461,6 +461,10 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F if os.environ.get("ANTHROPIC_API_KEY"): args.model = "claude-3-5-sonnet-20240620" + if args.deepseek_beta: + args.model = "deepseek-coder" + os.environ["DEEPSEEK_API_BASE"] = "https://api.deepseek.com/beta" + main_model = models.Model(args.model, weak_model=args.weak_model) lint_cmds = parse_lint_cmds(args.lint_cmd, io)