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

test_aws_secret_manager

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-13 14:19:37 -08:00
parent dbea2eef14
commit b9b5d60a38
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/local_testing/test_secret_manager.py
View file

@ -22,15 +22,19 @@ from litellm.secret_managers.aws_secret_manager_v2 import AWSSecretsManagerV2
from litellm.secret_managers.main import get_secret from litellm.secret_managers.main import get_secret
@pytest.mark.skip(reason="AWS Suspended Account")
def test_aws_secret_manager(): def test_aws_secret_manager():
import json
AWSSecretsManagerV2.load_aws_secret_manager(use_aws_secret_manager=True) AWSSecretsManagerV2.load_aws_secret_manager(use_aws_secret_manager=True)
secret_val = get_secret("litellm_master_key") secret_val = get_secret("litellm_master_key")
print(f"secret_val: {secret_val}") print(f"secret_val: {secret_val}")
assert secret_val == "sk-1234" # cast json to dict
secret_val = json.loads(secret_val)
assert secret_val["litellm_master_key"] == "sk-1234"
def redact_oidc_signature(secret_val): def redact_oidc_signature(secret_val):