phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +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 4665487558
commit 9d809e8404
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,
)