dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-31 09:44:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

feat: Add extra_headers parameter to send_with_retries function

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-01 17:27:31 -03:00
parent 83b72c5a5d
commit 38b411a6cb
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
aider/sendchat.py
View file

@ -57,7 +57,7 @@ def lazy_litellm_retry_decorator(func):
@lazy_litellm_retry_decorator
def send_with_retries(model_name, messages, functions, stream, temperature=0):
def send_with_retries(model_name, messages, functions, stream, temperature=0, extra_headers=None):
from aider.llm import litellm
kwargs = dict(
@ -68,6 +68,8 @@ def send_with_retries(model_name, messages, functions, stream, temperature=0):
)
if functions is not None:
kwargs["functions"] = functions
if extra_headers is not None:
kwargs["extra_headers"] = extra_headers
key = json.dumps(kwargs, sort_keys=True).encode()