From e81ddcc1a6624bd68a3d419b3a18f39957bc1869 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Fri, 23 Aug 2024 14:17:44 -0700
Subject: [PATCH] feat: Add extra_headers parameter to simple_send_with_retries

---
 aider/sendchat.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/aider/sendchat.py b/aider/sendchat.py
index 1914a618b..e6ab1dc57 100644
--- a/aider/sendchat.py
+++ b/aider/sendchat.py
@@ -86,13 +86,14 @@ def send_completion(
 
 
 @lazy_litellm_retry_decorator
-def simple_send_with_retries(model_name, messages):
+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,
         )
         return response.choices[0].message.content
     except (AttributeError, litellm.exceptions.BadRequestError):