diff --git a/litellm/proxy/guardrails/guardrail_hooks/microsoft_purview/purview_dlp.py b/litellm/proxy/guardrails/guardrail_hooks/microsoft_purview/purview_dlp.py index 05bf48417a6..8d7aa9d3d74 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/microsoft_purview/purview_dlp.py +++ b/litellm/proxy/guardrails/guardrail_hooks/microsoft_purview/purview_dlp.py @@ -21,6 +21,7 @@ from litellm.integrations.custom_guardrail import ( log_guardrail_information, ) from litellm.types.guardrails import GuardrailEventHooks +from litellm.types.utils import GuardrailStatus from .base import PurviewGuardrailBase @@ -113,7 +114,7 @@ class MicrosoftPurviewDLPGuardrail(PurviewGuardrailBase, CustomGuardrail): The processContent response dict. """ start_time = datetime.now() - status = "success" + status: GuardrailStatus = "success" response: Dict[str, Any] = {} try: