feat use OpenAI extra_headers param

This commit is contained in:
Ishaan Jaff 2024-05-16 14:38:17 -07:00
parent e9358684fb
commit 1fc9bcb184

View file

@ -552,7 +552,7 @@ def completion(
model_info = kwargs.get("model_info", None)
proxy_server_request = kwargs.get("proxy_server_request", None)
fallbacks = kwargs.get("fallbacks", None)
headers = kwargs.get("headers", None)
headers = kwargs.get("headers", None) or extra_headers
num_retries = kwargs.get("num_retries", None) ## deprecated
max_retries = kwargs.get("max_retries", None)
context_window_fallback_dict = kwargs.get("context_window_fallback_dict", None)