From e5c718992216a7da4a9395b904cf830c44c5418c Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 23 Nov 2024 16:48:36 -0800 Subject: [PATCH] fix test_in_memory_redis_sync_e2e --- tests/local_testing/test_router_provider_budgets.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_router_provider_budgets.py b/tests/local_testing/test_router_provider_budgets.py index 415a93f4d85a..3f67dd0ccf2f 100644 --- a/tests/local_testing/test_router_provider_budgets.py +++ b/tests/local_testing/test_router_provider_budgets.py @@ -291,7 +291,7 @@ async def test_in_memory_redis_sync_e2e(): router = Router( model_list=[ { - "model_name": "gpt-3.5-turbo", + "model_name": "gpt-3.5-turbo-very-new", "litellm_params": { "model": "openai/gpt-3.5-turbo", }, @@ -317,7 +317,7 @@ async def test_in_memory_redis_sync_e2e(): # Make a completion call to trigger spend tracking response = await router.acompletion( - model="gpt-3.5-turbo", + model="gpt-3.5-turbo-very-new", messages=[{"role": "user", "content": "Hello"}], mock_response="Hello there!", )