diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index 940f10e88..a5a2adb75 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -196,6 +196,7 @@ def test_openai_azure_embedding(): except Exception as e: pytest.fail(f"Error occurred: {e}") + @pytest.mark.skipif( os.environ.get("CIRCLE_OIDC_TOKEN") is None, reason="Cannot run without being in CircleCI Runner", @@ -210,7 +211,7 @@ def test_openai_azure_embedding_with_oidc_and_cf(): model="azure/text-embedding-ada-002", input=["Hello"], azure_ad_token="oidc/circleci/", - api_base="https://gateway.ai.cloudflare.com/v1/0399b10e77ac6668c80404a5ff49eb37/litellm-test/azure-openai/eastus2-litellm", + api_base="https://eastus2-litellm.openai.azure.com/", api_version="2024-06-01", ) print(response)