From 5ded503d2b5eab34b8b22d869fb743f8233120db Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 23 Aug 2024 14:20:15 -0700 Subject: [PATCH] fix: Only include extra_headers in send_completion if not None --- aider/sendchat.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/aider/sendchat.py b/aider/sendchat.py index e6ab1dc57..38c76fb4d 100644 --- a/aider/sendchat.py +++ b/aider/sendchat.py @@ -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