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

feat(proxy_server.py): add support for blocked user lists (enterprise-only)

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-22 17:51:31 -08:00
parent 5be70b94e7
commit 028f455ad0
4 changed files with 154 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/proxy_server.py
View file

@ -1479,6 +1479,16 @@ class ProxyConfig:
llm_guard_moderation_obj = _ENTERPRISE_LLMGuard()
imported_list.append(llm_guard_moderation_obj)
elif (
isinstance(callback, str)
and callback == "blocked_user_check"
):
from litellm.proxy.enterprise.enterprise_hooks.blocked_user_list import (
_ENTERPRISE_BlockedUserList,
)
blocked_user_list = _ENTERPRISE_BlockedUserList()
imported_list.append(blocked_user_list)
else:
imported_list.append(
get_instance_fn(