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

Merge pull request #3803 from BerriAI/litellm_add_lakera_ai

Browse source 
[Feat] Add Lakera AI Prompt Injection Detection
This commit is contained in:
Ishaan Jaff 2024-05-23 16:01:24 -07:00 • committed by GitHub
commit 580a342fdf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 310 additions and 6 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"