From 7cde2cd77f9c39306dddd8c614769e49a507b84b Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Tue, 1 Sep 2026 22:39:17 -0700
Subject: [PATCH] test(guardrails): type the presidio sibling test helpers
 precisely

---
 .../test_litellm/proxy/guardrails/test_guardrail_registry.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/test_litellm/proxy/guardrails/test_guardrail_registry.py b/tests/test_litellm/proxy/guardrails/test_guardrail_registry.py
index b8a58f5e3da..5cbdef5f92f 100644
--- a/tests/test_litellm/proxy/guardrails/test_guardrail_registry.py
+++ b/tests/test_litellm/proxy/guardrails/test_guardrail_registry.py
@@ -1,3 +1,4 @@
+from collections.abc import Iterable
 from unittest.mock import AsyncMock, MagicMock
 
 import pytest
@@ -495,7 +496,7 @@ PRESIDIO_SIBLINGS_GID = "55555555-5555-5555-5555-555555555555"
 PRESIDIO_SIBLINGS_NAME = "presidio-siblings"
 
 
-def _presidio_db_guardrail(pii_entities_config: dict) -> Guardrail:
+def _presidio_db_guardrail(pii_entities_config: dict[str, str]) -> Guardrail:
     return Guardrail(
         guardrail_id=PRESIDIO_SIBLINGS_GID,
         guardrail_name=PRESIDIO_SIBLINGS_NAME,
@@ -512,7 +513,7 @@ def _presidio_db_guardrail(pii_entities_config: dict) -> Guardrail:
     )
 
 
-def _presidio_callbacks_in(cb_list) -> list:
+def _presidio_callbacks_in(cb_list: Iterable[object]) -> list[CustomGuardrail]:
     return [
         callback
         for callback in cb_list