phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

build(litellm_server/main.py): make /chat/completions azure compatible

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-28 13:25:22 -07:00
parent 438b285451
commit 9eb53acd8e
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm_server/main.py
View file

@ -106,7 +106,7 @@ async def chat_completion(request: Request, model: Optional[str] = None):
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
api_key = request.headers.get("authorization")
api_key = api_key.replace("Bearer", "").strip()
api_key = api_key.split(" ")[1]
if len(api_key) > 0:
api_key = api_key
data["api_key"] = api_key