From e3556494b3cfe3a8ecfebd7c41788346f31eafb8 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 28 Oct 2023 14:02:05 -0700 Subject: [PATCH] build(litellm_server/main.py): support azure api key header --- litellm_server/main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm_server/main.py b/litellm_server/main.py index a43b8f17593..c040b743389 100644 --- a/litellm_server/main.py +++ b/litellm_server/main.py @@ -104,7 +104,8 @@ async def chat_completion(request: Request, model: Optional[str] = None): ## CHECK KEYS ## # default to always using the "ENV" variables, only if AUTH_STRATEGY==DYNAMIC then reads headers env_validation = litellm.validate_environment(model=data["model"]) - if (env_validation['keys_in_environment'] is False or os.getenv("AUTH_STRATEGY", None) == "DYNAMIC") and "authorization" in request.headers: # if users pass LLM api keys as part of header + print(f"request headers: {request.headers}") + if (env_validation['keys_in_environment'] is False or os.getenv("AUTH_STRATEGY", None) == "DYNAMIC") and ("authorization" in request.headers or "api-key" in request.headers): # if users pass LLM api keys as part of header api_key = request.headers.get("authorization") print(f"api_key in headers: {api_key}") api_key = api_key.split(" ")[1]