phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

build(litellm_server/main.py): fix keys in environment

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-26 16:54:06 -07:00
parent 89c94cf14f
commit ffa036566b
3 changed files with 8 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm_server/main.py
View file

@ -91,7 +91,8 @@ async def chat_completion(request: Request):
try:
data = await request.json()
# default to always using the "ENV" variables, only if AUTH_STRATEGY==DYNAMIC then reads headers
if os.getenv("AUTH_STRATEGY", None) == "DYNAMIC" and "authorization" in request.headers: # if users pass LLM api keys as part of header
keys_in_environment, _ = litellm.validate_environment(model=data["model"])
if (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()
if len(api_key.strip()) > 0: