From 901cbf8adc63848c7c5cd3bbd2ea8d4aab743f3f Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 23 Jul 2025 13:04:28 -0700 Subject: [PATCH] test_router_content_policy_fallbacks --- tests/local_testing/test_router_fallbacks.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/local_testing/test_router_fallbacks.py b/tests/local_testing/test_router_fallbacks.py index ced2c4dd9e0..b17825ab22a 100644 --- a/tests/local_testing/test_router_fallbacks.py +++ b/tests/local_testing/test_router_fallbacks.py @@ -1139,9 +1139,9 @@ async def test_router_content_policy_fallbacks( router = Router( model_list=[ { - "model_name": "claude-2.1", + "model_name": "claude-3-5-sonnet-latest", "litellm_params": { - "model": "claude-2.1", + "model": "anthropic/claude-3-5-sonnet-latest", "api_key": "", "mock_response": mock_response, }, @@ -1173,14 +1173,14 @@ async def test_router_content_policy_fallbacks( { "model_name": "my-context-window-model", "litellm_params": { - "model": "claude-2.1", + "model": "anthropic/claude-3-5-sonnet-latest", "api_key": "", "mock_response": Exception("Should not have called this."), }, }, ], content_policy_fallbacks=( - [{"claude-2.1": ["my-fallback-model"]}] + [{"claude-3-5-sonnet-latest": ["my-fallback-model"]}] if fallback_type == "model-specific" else None ), @@ -1191,12 +1191,12 @@ async def test_router_content_policy_fallbacks( if sync_mode is True: response = router.completion( - model="claude-2.1", + model="claude-3-5-sonnet-latest", messages=[{"role": "user", "content": "Hey, how's it going?"}], ) else: response = await router.acompletion( - model="claude-2.1", + model="claude-3-5-sonnet-latest", messages=[{"role": "user", "content": "Hey, how's it going?"}], )