From b8b22c984441bc23dc2ec86835e9c46bd15df3af Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaan@berri.ai>
Date: Thu, 23 Apr 2026 18:40:44 -0700
Subject: [PATCH] fix(types/guardrails): add input_type and messages to
 ApplyGuardrailRequest

---
 litellm/types/guardrails.py | 2 ++
 1 file changed, 2 insertions(+)

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):