check litellm header in login on ui

litellm/proxy/management_endpoints/key_management_endpoints.py

@@ -914,22 +914,3 @@ async def delete_verification_token(tokens: List, user_id: Optional[str] = None)
         verbose_proxy_logger.debug(traceback.format_exc())
         raise e
     return deleted_tokens
-
-
-@router.get(
-    "/litellm_header_name",
-    include_in_schema=False,
-)
-async def get_litellm_header_name():
-    """
-    Used by LiteLLM Admin UI
-
-    returns the header name that should be used for the Authorization header on requests to litellm
-    """
-    from litellm.proxy.proxy_server import general_settings
-
-    if "litellm_key_header_name" in general_settings:
-        return {"litellm_key_header_name": general_settings["litellm_key_header_name"]}
-    else:
-        # default value
-        return {"litellm_key_header_name": "Authorization"}

litellm/proxy/proxy_server.py

@@ -7773,7 +7773,7 @@ async def fallback_login(request: Request):
     "/login", include_in_schema=False
 )  # hidden since this is a helper for UI sso login
 async def login(request: Request):
-    global premium_user
+    global premium_user, general_settings
     try:
         import multipart
     except ImportError:
@@ -7875,6 +7875,9 @@ async def login(request: Request):
                 "user_role": user_role,  # this is the path without sso - we can assume only admins will use this
                 "login_method": "username_password",
                 "premium_user": premium_user,
+                "auth_header_name": general_settings.get(
+                    "litellm_key_header_name", "Authorization"
+                ),
             },
             master_key,
             algorithm="HS256",
@@ -7939,6 +7942,9 @@ async def login(request: Request):
                     "user_role": user_role,
                     "login_method": "username_password",
                     "premium_user": premium_user,
+                    "auth_header_name": general_settings.get(
+                        "litellm_key_header_name", "Authorization"
+                    ),
                 },
                 master_key,
                 algorithm="HS256",
@@ -7987,7 +7993,7 @@ async def onboarding(invite_link: str):
     - Get user from db
     - Pass in user_email if set
     """
-    global prisma_client, master_key
+    global prisma_client, master_key, general_settings
     if master_key is None:
         raise ProxyException(
             message="Master Key not set for Proxy. Please set Master Key to use Admin UI. Set `LITELLM_MASTER_KEY` in .env or set general_settings:master_key in config.yaml.  https://docs.litellm.ai/docs/proxy/virtual_keys. If set, use `--detailed_debug` to debug issue.",
@@ -8074,6 +8080,9 @@ async def onboarding(invite_link: str):
             "user_role": user_obj.user_role,
             "login_method": "username_password",
             "premium_user": premium_user,
+            "auth_header_name": general_settings.get(
+                "litellm_key_header_name", "Authorization"
+            ),
         },
         master_key,
         algorithm="HS256",
@@ -8491,6 +8500,9 @@ async def auth_callback(request: Request):
             "user_role": user_role,
             "login_method": "sso",
             "premium_user": premium_user,
+            "auth_header_name": general_settings.get(
+                "litellm_key_header_name", "Authorization"
+            ),
         },
         master_key,
         algorithm="HS256",