phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add enabled_roles to Guardrails configuration, Update Lakera guardrail moderation hook

Browse source
This commit is contained in:
Vinnie Giarrusso 2024-07-16 01:52:08 -07:00
parent a99cb5deeb
commit 6ff863ee00
5 changed files with 199 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/guardrails/init_guardrails.py
View file

@ -24,7 +24,7 @@ def initialize_guardrails(
"""
one item looks like this:
{'prompt_injection': {'callbacks': ['lakera_prompt_injection', 'prompt_injection_api_2'], 'default_on': True}}
{'prompt_injection': {'callbacks': ['lakera_prompt_injection', 'prompt_injection_api_2'], 'default_on': True, 'enabled_roles': ['user']}}
"""
for k, v in item.items():
guardrail_item = GuardrailItem(**v, guardrail_name=k)