From dc3cdf3ed88fbcdae8620458d82c65441eef30fd Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 7 Aug 2024 08:15:05 -0700 Subject: [PATCH] fix use extra headers for open router --- litellm/main.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/litellm/main.py b/litellm/main.py index 1209306c8b..789863ecb0 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -1856,17 +1856,18 @@ def completion( ) openrouter_site_url = get_secret("OR_SITE_URL") or "https://litellm.ai" - openrouter_app_name = get_secret("OR_APP_NAME") or "liteLLM" - headers = ( - headers - or litellm.headers - or { - "HTTP-Referer": openrouter_site_url, - "X-Title": openrouter_app_name, - } - ) + openrouter_headers = { + "HTTP-Referer": openrouter_site_url, + "X-Title": openrouter_app_name, + } + + _headers = headers or litellm.headers + if _headers: + openrouter_headers.update(_headers) + + headers = openrouter_headers ## Load Config config = openrouter.OpenrouterConfig.get_config()