dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 17:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: Add --deepseek-beta bool arg to use DeepSeek Coder via the beta API endpoint

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-12 15:06:55 -07:00
parent 5f8df315e4
commit 485418d917
2 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
aider/args.py
View file

@ -117,6 +117,12 @@ def get_parser(default_config_files, git_root):
const=deepseek_model,
help=f"Use {deepseek_model} model for the main chat",
)
group.add_argument(
"--deepseek-beta",
action="store_true",
help="Uses DeepSeek Coder via the beta API endpoint",
default=False,
)
##########
group = parser.add_argument_group("Model Settings")

7
aider/sendchat.py
View file

@ -59,8 +59,11 @@ def send_completion(
stream=stream,
)
if model_name.startswith("deepseek/") and "DEEPSEEK_API_BASE" in os.environ:
kwargs["base_url"] = os.environ["DEEPSEEK_API_BASE"]
if model_name.startswith("deepseek/"):
if "DEEPSEEK_API_BASE" in os.environ:
kwargs["base_url"] = os.environ["DEEPSEEK_API_BASE"]
elif getattr(kwargs.get('extra_headers', {}), 'deepseek_beta', False):
kwargs["base_url"] = "https://api.deepseek.com/v1"
if functions is not None:
kwargs["functions"] = functions
if extra_headers is not None: