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

feat - add lakera ai prompt injection detection

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-23 15:16:46 -07:00
parent 35146d8628
commit ec41226ee8
2 changed files with 129 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/proxy/proxy_server.py
View file

@ -2325,6 +2325,18 @@ class ProxyConfig:
_ENTERPRISE_OpenAI_Moderation()
)
imported_list.append(openai_moderations_object)
elif (
isinstance(callback, str)
and callback == "lakera_prompt_injection"
):
from enterprise.enterprise_hooks.lakera_ai import (
_ENTERPRISE_lakeraAI_Moderation,
)
lakera_moderations_object = (
_ENTERPRISE_lakeraAI_Moderation()
)
imported_list.append(lakera_moderations_object)
elif (
isinstance(callback, str)
and callback == "google_text_moderation"