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

fix test secret manager

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-13 13:21:20 -08:00
parent 5d8f7758d1
commit 46e3b13dbb
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/local_testing/test_secret_manager.py
View file

@ -18,13 +18,13 @@ import pytest
from litellm.llms.AzureOpenAI.azure import get_azure_ad_token_from_oidc from litellm.llms.AzureOpenAI.azure import get_azure_ad_token_from_oidc
from litellm.llms.bedrock.chat import BedrockConverseLLM, BedrockLLM from litellm.llms.bedrock.chat import BedrockConverseLLM, BedrockLLM
from litellm.secret_managers.aws_secret_manager import load_aws_secret_manager 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") @pytest.mark.skip(reason="AWS Suspended Account")
def test_aws_secret_manager(): def test_aws_secret_manager():
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")