diff --git a/litellm/integrations/custom_guardrail.py b/litellm/integrations/custom_guardrail.py
index cf75fb38b75..5859e7f3693 100644
--- a/litellm/integrations/custom_guardrail.py
+++ b/litellm/integrations/custom_guardrail.py
@@ -1129,7 +1129,7 @@ def log_guardrail_information(func):
             "async_post_call_streaming_hook",
         ):
             return GuardrailEventHooks.post_call
-        elif func_name == "apply_guardrail" and kwargs:
+        elif func_name == "apply_guardrail" and kwargs and "input_type" in kwargs:
             input_type = kwargs.get("input_type")
             if input_type == "request":
                 return GuardrailEventHooks.pre_call
diff --git a/litellm/types/utils.py b/litellm/types/utils.py
index e33e2335525..1f54a3e63a8 100644
--- a/litellm/types/utils.py
+++ b/litellm/types/utils.py
@@ -2698,7 +2698,7 @@ GuardrailStatus = Literal["success", "guardrail_intervened", "guardrail_failed_t
 class StandardLoggingGuardrailInformation(TypedDict, total=False):
     guardrail_name: Optional[str]
     guardrail_provider: Optional[str]
-    guardrail_mode: Optional[Union[GuardrailEventHooks, List[GuardrailEventHooks], GuardrailMode]]
+    guardrail_mode: Optional[Union[GuardrailEventHooks, List[GuardrailEventHooks], GuardrailMode, str]]
     guardrail_request: Optional[Union[str, dict]]
     guardrail_response: Optional[Union[dict, str, List[dict]]]
     guardrail_status: GuardrailStatus