mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-31 01:35:00 +00:00
fix: Only include extra_headers in send_completion if not None
This commit is contained in:
parent
50c987e261
commit
5ded503d2b
1 changed files with 10 additions and 7 deletions
|
@ -88,13 +88,16 @@ def send_completion(
|
||||||
@lazy_litellm_retry_decorator
|
@lazy_litellm_retry_decorator
|
||||||
def simple_send_with_retries(model_name, messages, extra_headers=None):
|
def simple_send_with_retries(model_name, messages, extra_headers=None):
|
||||||
try:
|
try:
|
||||||
_hash, response = send_completion(
|
kwargs = {
|
||||||
model_name=model_name,
|
"model_name": model_name,
|
||||||
messages=messages,
|
"messages": messages,
|
||||||
functions=None,
|
"functions": None,
|
||||||
stream=False,
|
"stream": False,
|
||||||
extra_headers=extra_headers,
|
}
|
||||||
)
|
if extra_headers is not None:
|
||||||
|
kwargs["extra_headers"] = extra_headers
|
||||||
|
|
||||||
|
_hash, response = send_completion(**kwargs)
|
||||||
return response.choices[0].message.content
|
return response.choices[0].message.content
|
||||||
except (AttributeError, litellm.exceptions.BadRequestError):
|
except (AttributeError, litellm.exceptions.BadRequestError):
|
||||||
return
|
return
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue