From 41fda47587a497bcfb23756cf6fbfb8b35c96f65 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 24 Jul 2024 08:04:27 -0700 Subject: [PATCH] test(test_embedding.py): fix base url --- litellm/tests/test_embedding.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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)