diff --git a/litellm/proxy/secret_managers/aws_secret_manager.py b/litellm/proxy/secret_managers/aws_secret_manager.py index c4afaedc21..88d2029f7e 100644 --- a/litellm/proxy/secret_managers/aws_secret_manager.py +++ b/litellm/proxy/secret_managers/aws_secret_manager.py @@ -79,7 +79,13 @@ class AWSKeyManagementService_V2: raise ValueError("Missing required environment variable - AWS_REGION_NAME") ## CHECK IF LICENSE IN ENV ## - premium feature - if os.getenv("LITELLM_LICENSE", None) is None: + is_litellm_license_in_env: bool = False + + if os.getenv("LITELLM_LICENSE", None) is not None: + is_litellm_license_in_env = True + elif os.getenv("LITELLM_SECRET_AWS_KMS_LITELLM_LICENSE", None) is not None: + is_litellm_license_in_env = True + if is_litellm_license_in_env is False: raise ValueError( "AWSKeyManagementService V2 is an Enterprise Feature. Please add a valid LITELLM_LICENSE to your envionment." )