mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 19:24:27 +00:00
bc66ef9d5c
fixes the aws secret manager implementation and allows the user to set which keys they want to check thr ough it
24 lines
553 B
Python
24 lines
553 B
Python
import sys, os, uuid
|
|
import time
|
|
import traceback
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
import os
|
|
|
|
sys.path.insert(
|
|
0, os.path.abspath("../..")
|
|
) # Adds the parent directory to the system path
|
|
import pytest
|
|
from litellm import get_secret
|
|
from litellm.proxy.secret_managers.aws_secret_manager import load_aws_secret_manager
|
|
|
|
|
|
def test_aws_secret_manager():
|
|
load_aws_secret_manager(use_aws_secret_manager=True)
|
|
|
|
secret_val = get_secret("litellm_master_key")
|
|
|
|
print(f"secret_val: {secret_val}")
|
|
|
|
assert secret_val == "sk-1234"
|