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

fix(proxy_server.py): add license protection for 'allowed_ip' address feature

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-08 16:04:44 -07:00
parent f982e93d24
commit 0ecf94d32e
2 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/proxy/_new_secret_config.yaml
View file

@ -19,3 +19,4 @@ model_list:
general_settings: general_settings:
alerting: ["slack"] alerting: ["slack"]
alerting_threshold: 10 alerting_threshold: 10
allowed_ips: ["192.168.1.1"]

6
litellm/proxy/proxy_server.py
View file

@ -1690,6 +1690,12 @@ class ProxyConfig:
ui_access_mode = general_settings.get( ui_access_mode = general_settings.get(
"ui_access_mode", "all" "ui_access_mode", "all"
) # can be either ["admin_only" or "all"] ) # can be either ["admin_only" or "all"]
### ALLOWED IP ###
allowed_ips = general_settings.get("allowed_ips", None)
if allowed_ips is not None and premium_user is False:
raise ValueError(
"allowed_ips is an Enterprise Feature. Please add a valid LITELLM_LICENSE to your envionment."
)
## BUDGET RESCHEDULER ## ## BUDGET RESCHEDULER ##
proxy_budget_rescheduler_min_time = general_settings.get( proxy_budget_rescheduler_min_time = general_settings.get(
"proxy_budget_rescheduler_min_time", proxy_budget_rescheduler_min_time "proxy_budget_rescheduler_min_time", proxy_budget_rescheduler_min_time