From a3d64fb843415e50fb498c4ba1d784a80df296f7 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Mon, 27 Oct 2025 17:48:35 -0700 Subject: [PATCH] fix omni-moderation-latest --- tests/local_testing/test_openai_moderations_hook.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/local_testing/test_openai_moderations_hook.py b/tests/local_testing/test_openai_moderations_hook.py index 2ab8669951..c63ce342b2 100644 --- a/tests/local_testing/test_openai_moderations_hook.py +++ b/tests/local_testing/test_openai_moderations_hook.py @@ -34,7 +34,7 @@ async def test_openai_moderation_error_raising(): """ openai_mod = _ENTERPRISE_OpenAI_Moderation() - litellm.openai_moderations_model_name = "text-moderation-latest" + litellm.openai_moderations_model_name = "omni-moderation-latest" _api_key = "sk-12345" _api_key = hash_token("sk-12345") user_api_key_dict = UserAPIKeyAuth(api_key=_api_key) @@ -45,9 +45,9 @@ async def test_openai_moderation_error_raising(): llm_router = litellm.Router( model_list=[ { - "model_name": "text-moderation-latest", + "model_name": "omni-moderation-latest", "litellm_params": { - "model": "text-moderation-latest", + "model": "omni-moderation-latest", "api_key": os.environ["OPENAI_API_KEY"], }, }