From 046b9d8f66dac95e105fb482c00fc49b1a41d79a Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 3 Jul 2024 18:58:36 -0700 Subject: [PATCH] fix lakera ai testing --- litellm/proxy/guardrails/guardrail_helpers.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/guardrails/guardrail_helpers.py b/litellm/proxy/guardrails/guardrail_helpers.py index 39c9a983119..8a25abf3a9d 100644 --- a/litellm/proxy/guardrails/guardrail_helpers.py +++ b/litellm/proxy/guardrails/guardrail_helpers.py @@ -43,4 +43,7 @@ async def should_proceed_based_on_metadata(data: dict, guardrail_name: str) -> b if guardrail_name in requested_callback_names: return True - return False + # Do no proceeed if - "metadata": { "guardrails": { "lakera_prompt_injection": false } } + return False + + return True