scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-22 00:31:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 51

test(proxy): type the moderation override stub in hook detection tests

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-17 01:59:04 +00:00
parent b1255a6f2c
commit fc77914df3
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/test_litellm/proxy/test_proxy_logging_hook_detection.py
View file

@ -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"})