Clean up DEEPSEEK_API_BASE

This commit is contained in:
Paul Gauthier 2024-08-12 15:35:32 -07:00
parent 27f92b26c6
commit 2a1fb7d150
3 changed files with 0 additions and 16 deletions

View file

@ -117,12 +117,6 @@ 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")

View file

@ -461,10 +461,6 @@ 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/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)

View file

@ -1,6 +1,5 @@
import hashlib
import json
import os
import backoff
@ -59,11 +58,6 @@ def send_completion(
stream=stream,
)
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: