phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test_pass_through_moderation_endpoint_factory

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-27 15:51:47 -08:00
parent 48227c1330
commit 7026f4fdfe
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/router_unit_tests/test_router_helper_utils.py
View file

@ -1040,8 +1040,11 @@ def test_pattern_match_deployment_set_model_name(
async def test_pass_through_moderation_endpoint_factory(model_list): async def test_pass_through_moderation_endpoint_factory(model_list):
router = Router(model_list=model_list) router = Router(model_list=model_list)
response = await router._pass_through_moderation_endpoint_factory( 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( @pytest.mark.parametrize(