From a3e9bc4ae25964cabd60d38b00a303298c55b45b Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 3 Jul 2024 22:07:48 -0700 Subject: [PATCH] fix(aws_secret_manager.py): fix litellm license check --- litellm/proxy/secret_managers/aws_secret_manager.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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." )