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

fix(proxy_server.py): allow jwt public key ttl to be passed as os.environ

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-09 07:23:13 -07:00
parent ef4c85522c
commit 0c2ae7a90e
2 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/_new_secret_config.yaml
View file

@ -42,4 +42,4 @@ general_settings:
alerting: ["slack"] alerting: ["slack"]
litellm_jwtauth: litellm_jwtauth:
admin_jwt_scope: "litellm_proxy_admin" admin_jwt_scope: "litellm_proxy_admin"
public_key_ttl: 600 public_key_ttl: os.environ/LITELLM_PUBLIC_KEY_TTL

3
litellm/proxy/proxy_server.py
View file

@ -3000,6 +3000,9 @@ async def startup_event():
## JWT AUTH ## ## JWT AUTH ##
if general_settings.get("litellm_jwtauth", None) is not None: if general_settings.get("litellm_jwtauth", None) is not None:
for k, v in general_settings["litellm_jwtauth"].items():
if isinstance(v, str) and v.startswith("os.environ/"):
general_settings["litellm_jwtauth"][k] = litellm.get_secret(v)
litellm_jwtauth = LiteLLM_JWTAuth(**general_settings["litellm_jwtauth"]) litellm_jwtauth = LiteLLM_JWTAuth(**general_settings["litellm_jwtauth"])
else: else:
litellm_jwtauth = LiteLLM_JWTAuth() litellm_jwtauth = LiteLLM_JWTAuth()