diff --git a/tests/local_testing/test_router.py b/tests/local_testing/test_router.py index 4be7e78f97c..3a07238eb36 100644 --- a/tests/local_testing/test_router.py +++ b/tests/local_testing/test_router.py @@ -334,10 +334,8 @@ async def test_router_retries(sync_mode): { "model_name": "gpt-3.5-turbo", "litellm_params": { - "model": "azure/gpt-4o-new-test", - "api_key": os.getenv("AZURE_API_KEY"), - "api_base": os.getenv("AZURE_API_BASE"), - "api_version": os.getenv("AZURE_API_VERSION"), + "model": "gpt-4.1-nano", + "api_key": os.getenv("OPENAI_API_KEY"), }, }, ]