mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-06 16:24:46 +00:00
test(test_embedding.py): fix base url
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user