From 7026f4fdfe7c68cde372a0b861e7a0361bbcb01a Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 27 Nov 2024 15:51:47 -0800 Subject: [PATCH] test_pass_through_moderation_endpoint_factory --- tests/router_unit_tests/test_router_helper_utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/router_unit_tests/test_router_helper_utils.py b/tests/router_unit_tests/test_router_helper_utils.py index 3c51c619e..f247c33e3 100644 --- a/tests/router_unit_tests/test_router_helper_utils.py +++ b/tests/router_unit_tests/test_router_helper_utils.py @@ -1040,8 +1040,11 @@ def test_pattern_match_deployment_set_model_name( async def test_pass_through_moderation_endpoint_factory(model_list): router = Router(model_list=model_list) response = await router._pass_through_moderation_endpoint_factory( - original_function=litellm.amoderation, input="this is valid good text" + original_function=litellm.amoderation, + input="this is valid good text", + model=None, ) + assert response is not None @pytest.mark.parametrize(