From fc77914df3cd59bf79bfc0cca8e163bb48c38ede Mon Sep 17 00:00:00 2001 From: yucheng Date: Thu, 17 Sep 2026 01:59:04 +0000 Subject: [PATCH] test(proxy): type the moderation override stub in hook detection tests Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../proxy/test_proxy_logging_hook_detection.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/test_litellm/proxy/test_proxy_logging_hook_detection.py b/tests/test_litellm/proxy/test_proxy_logging_hook_detection.py index 34d1488a4e5..58ee8ff656c 100644 --- a/tests/test_litellm/proxy/test_proxy_logging_hook_detection.py +++ b/tests/test_litellm/proxy/test_proxy_logging_hook_detection.py @@ -8,6 +8,7 @@ from litellm.integrations.custom_logger import CustomLogger from litellm.proxy._types import UserAPIKeyAuth from litellm.proxy.utils import ProxyLogging from litellm.types.guardrails import GuardrailEventHooks +from litellm.types.utils import CallTypesLiteral def test_has_post_call_response_headers_callbacks_ignores_empty_callbacks( @@ -609,7 +610,12 @@ class _RejectsInModeration(CustomLogger): super().__init__() self.moderated: list[str] = [] - async def async_moderation_hook(self, data, user_api_key_dict, call_type): + async def async_moderation_hook( + self, + data: dict, + user_api_key_dict: UserAPIKeyAuth, + call_type: CallTypesLiteral, + ) -> None: self.moderated.append(call_type) raise HTTPException(status_code=400, detail={"error": "rejected"})