diff --git a/tests/local_testing/test_router_fallbacks.py b/tests/local_testing/test_router_fallbacks.py index 54633742684..2159c77f0d3 100644 --- a/tests/local_testing/test_router_fallbacks.py +++ b/tests/local_testing/test_router_fallbacks.py @@ -250,10 +250,8 @@ def test_sync_fallbacks_embeddings(): { # list of model deployments "model_name": "good-azure-embedding-model", # openai model name "litellm_params": { # params for litellm completion/embedding call - "model": "azure/azure-embedding-model", - "api_key": os.getenv("AZURE_API_KEY"), - "api_version": os.getenv("AZURE_API_VERSION"), - "api_base": os.getenv("AZURE_API_BASE"), + "model": "text-embedding-ada-002", + "api_key": os.getenv("OPENAI_API_KEY"), }, "tpm": 240000, "rpm": 1800, @@ -302,10 +300,8 @@ async def test_async_fallbacks_embeddings(): { # list of model deployments "model_name": "good-azure-embedding-model", # openai model name "litellm_params": { # params for litellm completion/embedding call - "model": "azure/azure-embedding-model", - "api_key": os.getenv("AZURE_API_KEY"), - "api_version": os.getenv("AZURE_API_VERSION"), - "api_base": os.getenv("AZURE_API_BASE"), + "model": "text-embedding-ada-002", + "api_key": os.getenv("OPENAI_API_KEY"), }, "tpm": 240000, "rpm": 1800,