add litellm_header_name endpoint

This commit is contained in:
Ishaan Jaff 2024-07-26 21:04:31 -07:00
parent c0717133a9
commit 548adea8cf
2 changed files with 120 additions and 75 deletions

View file

@ -914,3 +914,22 @@ 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"}