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

fix make lakera ai free guardrail

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-20 14:03:22 -07:00
parent cad0352f76
commit 1a142053e5
4 changed files with 30 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -125,7 +125,7 @@ def init_guardrails_v2(all_guardrails: dict):
)
litellm.callbacks.append(_aporia_callback) # type: ignore
elif litellm_params["guardrail"] == "lakera":
from litellm.proxy.enterprise.enterprise_hooks.lakera_ai import (
from litellm.proxy.guardrails.guardrail_hooks.lakera_ai import (
lakeraAI_Moderation,
)