diff --git a/litellm/types/guardrails.py b/litellm/types/guardrails.py
index 58324742f68..8eadb1e21e4 100644
--- a/litellm/types/guardrails.py
+++ b/litellm/types/guardrails.py
@@ -887,6 +887,8 @@ class ApplyGuardrailRequest(BaseModel):
     text: str
     language: Optional[str] = None
     entities: Optional[List[PiiEntityType]] = None
+    input_type: str = "request"
+    messages: Optional[List[Dict[str, Any]]] = None
 
 
 class ApplyGuardrailResponse(BaseModel):