tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-15 22:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(aws_secret_manager.py): fix litellm license check

Browse Source
This commit is contained in:
Krrish Dholakia
2024-07-03 22:07:48 -07:00
parent 19c982d0f9
commit a3e9bc4ae2
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/proxy/secret_managers/aws_secret_manager.py
+7 -1
View File
@@ -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."
)