dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-28 16:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

fix: Only include extra_headers in send_completion if not None

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-23 14:20:15 -07:00
parent 50c987e261
commit 5ded503d2b
1 changed files with 10 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

17
aider/sendchat.py
View file

@ -88,13 +88,16 @@ def send_completion(
@lazy_litellm_retry_decorator
def simple_send_with_retries(model_name, messages, extra_headers=None):
try:
_hash, response = send_completion(
model_name=model_name,
messages=messages,
functions=None,
stream=False,
extra_headers=extra_headers,
)
kwargs = {
"model_name": model_name,
"messages": messages,
"functions": None,
"stream": False,
}
if extra_headers is not None:
kwargs["extra_headers"] = extra_headers
_hash, response = send_completion(**kwargs)
return response.choices[0].message.content
except (AttributeError, litellm.exceptions.BadRequestError):
return