diff --git a/litellm/proxy/guardrails/guardrail_hooks/block_code_execution/block_code_execution.py b/litellm/proxy/guardrails/guardrail_hooks/block_code_execution/block_code_execution.py index 99abdf2821..e76a02a6e4 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/block_code_execution/block_code_execution.py +++ b/litellm/proxy/guardrails/guardrail_hooks/block_code_execution/block_code_execution.py @@ -25,6 +25,7 @@ from fastapi import HTTPException from litellm.integrations.custom_guardrail import ( CustomGuardrail, ModifyResponseException, + log_guardrail_information, ) from litellm.types.guardrails import GuardrailEventHooks from litellm.types.proxy.guardrails.guardrail_hooks.base import GuardrailConfigModel @@ -536,6 +537,7 @@ class BlockCodeExecutionGuardrail(CustomGuardrail): detection_info={"language": language}, ) + @log_guardrail_information async def apply_guardrail( self, inputs: GenericGuardrailAPIInputs,