diff --git a/litellm/proxy/guardrails/guardrail_hooks/promptguard/promptguard.py b/litellm/proxy/guardrails/guardrail_hooks/promptguard/promptguard.py index 4b9152bd719..c136b05c5f3 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/promptguard/promptguard.py +++ b/litellm/proxy/guardrails/guardrail_hooks/promptguard/promptguard.py @@ -10,6 +10,7 @@ import os from typing import ( TYPE_CHECKING, Any, + Dict, List, Literal, Optional, @@ -120,7 +121,7 @@ class PromptGuardGuardrail(CustomGuardrail): direction = "input" if input_type == "request" else "output" - payload: dict[str, Any] = { + payload: Dict[str, Any] = { "messages": messages, "direction": direction, }