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

feat(lakera_ai.py): control running prompt injection between pre-call and in_parallel

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-22 20:04:42 -07:00
parent a32a7af215
commit 99a5436ed5
6 changed files with 211 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -38,6 +38,8 @@ def initialize_guardrails(
verbose_proxy_logger.debug(guardrail.guardrail_name)
verbose_proxy_logger.debug(guardrail.default_on)
callback_specific_params.update(guardrail.callback_args)
if guardrail.default_on is True:
# add these to litellm callbacks if they don't exist
for callback in guardrail.callbacks:
@ -46,7 +48,7 @@ def initialize_guardrails(
if guardrail.logging_only is True:
if callback == "presidio":
callback_specific_params["logging_only"] = True
callback_specific_params["logging_only"] = True # type: ignore
default_on_callbacks_list = list(default_on_callbacks)
if len(default_on_callbacks_list) > 0: