diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index 5f6ef2e211..82ae1646b6 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -162,18 +162,16 @@ def test_router_azure_acompletion(): "api_version": os.getenv("AZURE_API_VERSION"), "api_base": os.getenv("AZURE_API_BASE") }, - "tpm": 240000, "rpm": 1800 }, { "model_name": "gpt-3.5-turbo", # openai model name "litellm_params": { # params for litellm completion/embedding call - "model": "azure/chatgpt-v-2", - "api_key": old_api_key, + "model": "azure/gpt-turbo", + "api_key": os.getenv("AZURE_FRANCE_API_KEY"), "api_version": os.getenv("AZURE_API_VERSION"), - "api_base": os.getenv("AZURE_API_BASE") + "api_base": "https://openai-france-1234.openai.azure.com" }, - "tpm": 240000, "rpm": 1800 } ]