dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-02 18:54:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

fixed benchmark

Browse source
This commit is contained in:
Paul Gauthier 2023-12-05 11:21:11 -08:00
parent d92a93221c
commit bf03f43b44
4 changed files with 18 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

2
aider/coders/base_coder.py
View file

@ -480,7 +480,6 @@ class Coder:
except ExhaustedContextWindow:
exhausted = True
except openai.BadRequestError as err:
dump(err)
if "maximum context length" in str(err):
exhausted = True
else:
@ -955,7 +954,6 @@ class Coder:
def check_model_availability(io, client, main_model):
available_models = client.models.list()
dump(available_models)
model_ids = sorted(model.id for model in available_models)
if main_model.name in model_ids:
return True

17
aider/main.py
View file

@ -177,7 +177,7 @@ def main(argv=None, input=None, output=None, force_git_root=None):
model_group.add_argument(
"--openai-api-base",
metavar="OPENAI_API_BASE",
help="Specify the api_base (default: https://api.openai.com/v1)",
help="Specify the api base url",
)
model_group.add_argument(
"--openai-api-type",
@ -497,14 +497,15 @@ def main(argv=None, input=None, output=None, force_git_root=None):
)
else:
kwargs = dict()
if args.openai_api_base and "openrouter.ai" in args.openai_api_base:
kwargs["default_headers"] = {"HTTP-Referer": "http://aider.chat", "X-Title": "Aider"}
if args.openai_api_base:
kwargs["base_url"] = args.openai_api_base
if "openrouter.ai" in args.openai_api_base:
kwargs["default_headers"] = {
"HTTP-Referer": "http://aider.chat",
"X-Title": "Aider",
}
client = openai.OpenAI(
api_key=args.openai_api_key,
base_url=args.openai_api_base,
**kwargs,
)
client = openai.OpenAI(api_key=args.openai_api_key, **kwargs)
main_model = models.Model.create(args.model, client)