From 80af02a1c3c6adfc0d9a4d12be3f351922f56a0f Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Fri, 17 Oct 2025 09:39:32 -0700 Subject: [PATCH] test fix: tests.test_litellm.proxy.guardrails.guardrail_hooks.guardrails_ai.test_guardrails_ai::test_guardrails_ai_process_input --- .../guardrails/guardrail_hooks/guardrails_ai/guardrails_ai.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/litellm/proxy/guardrails/guardrail_hooks/guardrails_ai/guardrails_ai.py b/litellm/proxy/guardrails/guardrail_hooks/guardrails_ai/guardrails_ai.py index 84cbef05266..2d04517de2b 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/guardrails_ai/guardrails_ai.py +++ b/litellm/proxy/guardrails/guardrail_hooks/guardrails_ai/guardrails_ai.py @@ -167,6 +167,10 @@ class GuardrailsAI(CustomGuardrail): set_last_user_message, ) + # Only process completion-related call types + if call_type not in ["completion", "acompletion"]: + return data + if "messages" not in data: # invalid request return data