diff --git a/litellm/llms/a2a/chat/guardrail_translation/handler.py b/litellm/llms/a2a/chat/guardrail_translation/handler.py
index 8728b9db7a1..7b132c30e00 100644
--- a/litellm/llms/a2a/chat/guardrail_translation/handler.py
+++ b/litellm/llms/a2a/chat/guardrail_translation/handler.py
@@ -444,7 +444,7 @@ class A2AGuardrailHandler(BaseTranslation):
         if depth >= max_depth:
             return ()
 
-        def _scan(part_idx: int, part: dict[str, Any]) -> tuple[tuple[str, tuple[str, ...], int, str], ...]:
+        def _scan(part_idx: int, part: _A2ATextPart) -> tuple[tuple[str, tuple[str, ...], int, str], ...]:
             kind: Final = part.get("kind")
             if kind == "text":
                 text: Final = part.get("text", "")