phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #4854 from Manouchehri/oidc-azure-tests-3

Browse source 
OIDC azure tests 3
This commit is contained in:
Ishaan Jaff 2024-07-24 08:53:53 -07:00 • committed by GitHub
commit f325e29147
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 19 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/tests/test_embedding.py
View file

@ -197,10 +197,11 @@ def test_openai_azure_embedding():
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",
)
# @pytest.mark.skipif(
# os.environ.get("CIRCLE_OIDC_TOKEN") is None,
# reason="Cannot run without being in CircleCI Runner",
# )
@pytest.mark.skip(reason="Temporarily skipping this test.")
def test_openai_azure_embedding_with_oidc_and_cf():
# TODO: Switch to our own Azure account, currently using ai.moda's account
os.environ["AZURE_TENANT_ID"] = "17c0a27a-1246-4aa1-a3b6-d294e80e783c"

14
litellm/tests/test_secret_manager.py
View file

@ -12,6 +12,7 @@ sys.path.insert(
import pytest
from litellm import get_secret
from litellm.proxy.secret_managers.aws_secret_manager import load_aws_secret_manager
from litellm.llms.azure import get_azure_ad_token_from_oidc
@pytest.mark.skip(reason="AWS Suspended Account")
@ -76,3 +77,16 @@ def test_oidc_circleci_v2():
)
print(f"secret_val: {redact_oidc_signature(secret_val)}")
@pytest.mark.skipif(
os.environ.get("CIRCLE_OIDC_TOKEN") is None,
reason="Cannot run without being in CircleCI Runner",
)
def test_oidc_circleci_with_azure():
# TODO: Switch to our own Azure account, currently using ai.moda's account
os.environ["AZURE_TENANT_ID"] = "17c0a27a-1246-4aa1-a3b6-d294e80e783c"
os.environ["AZURE_CLIENT_ID"] = "4faf5422-b2bd-45e8-a6d7-46543a38acd0"
azure_ad_token = get_azure_ad_token_from_oidc("oidc/circleci/")
print(f"secret_val: {redact_oidc_signature(azure_ad_token)}")