dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-04 19:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: Add extra_headers parameter to simple_send_with_retries

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-23 14:17:44 -07:00
parent 9c2bd9d0a5
commit e81ddcc1a6
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
aider/sendchat.py
View file

@ -86,13 +86,14 @@ def send_completion(
@lazy_litellm_retry_decorator @lazy_litellm_retry_decorator
def simple_send_with_retries(model_name, messages): def simple_send_with_retries(model_name, messages, extra_headers=None):
try: try:
_hash, response = send_completion( _hash, response = send_completion(
model_name=model_name, model_name=model_name,
messages=messages, messages=messages,
functions=None, functions=None,
stream=False, stream=False,
extra_headers=extra_headers,
) )
return response.choices[0].message.content return response.choices[0].message.content
except (AttributeError, litellm.exceptions.BadRequestError): except (AttributeError, litellm.exceptions.BadRequestError):