diff --git a/litellm/proxy/guardrails/guardrail_hooks/model_armor/model_armor.py b/litellm/proxy/guardrails/guardrail_hooks/model_armor/model_armor.py index de581857b5f..3f197c26d3d 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/model_armor/model_armor.py +++ b/litellm/proxy/guardrails/guardrail_hooks/model_armor/model_armor.py @@ -842,6 +842,7 @@ class ModelArmorGuardrail(CustomGuardrail, VertexBase): return response + @log_guardrail_information async def apply_guardrail( self, inputs: GenericGuardrailAPIInputs,