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

correct guardrail name

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-02 17:50:23 -07:00
parent 11d1f4e430
commit 23f8d23acb
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
enterprise/enterprise_hooks/secret_detection.py
View file

@ -32,6 +32,7 @@ from litellm._logging import verbose_proxy_logger
litellm.set_verbose = True
GUARDRAIL_NAME = "hide_secrets"
_custom_plugins_path = "file://" + os.path.join(
os.path.dirname(os.path.abspath(__file__)), "secrets_plugins"
@ -466,8 +467,8 @@ class _ENTERPRISE_SecretDetection(CustomLogger):
async def should_run_check(self, user_api_key_dict: UserAPIKeyAuth) -> bool:
if user_api_key_dict.permissions is not None:
if "secret_detection" in user_api_key_dict.permissions:
if user_api_key_dict.permissions["secret_detection"] is False:
if GUARDRAIL_NAME in user_api_key_dict.permissions:
if user_api_key_dict.permissions[GUARDRAIL_NAME] is False:
return False
return True