fix test_in_memory_redis_sync_e2e

BerriAI · Nov 24, 2024 · e5c7189 · e5c7189
1 parent 33a0744
commit e5c7189
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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!",
     )