diff --git a/litellm/constants.py b/litellm/constants.py index 1ee2e9517f3..be1c05da6a4 100644 --- a/litellm/constants.py +++ b/litellm/constants.py @@ -6,6 +6,9 @@ DEFAULT_BATCH_SIZE = int(os.getenv("DEFAULT_BATCH_SIZE", 512)) DEFAULT_FLUSH_INTERVAL_SECONDS = int(os.getenv("DEFAULT_FLUSH_INTERVAL_SECONDS", 5)) DEFAULT_MAX_RETRIES = int(os.getenv("DEFAULT_MAX_RETRIES", 2)) DEFAULT_MAX_RECURSE_DEPTH = int(os.getenv("DEFAULT_MAX_RECURSE_DEPTH", 100)) +DEFAULT_MAX_RECURSE_DEPTH_SENSITIVE_DATA_MASKER = int( + os.getenv("DEFAULT_MAX_RECURSE_DEPTH_SENSITIVE_DATA_MASKER", 10) +) DEFAULT_FAILURE_THRESHOLD_PERCENT = float( os.getenv("DEFAULT_FAILURE_THRESHOLD_PERCENT", 0.5) ) # default cooldown a deployment if 50% of requests fail in a given minute diff --git a/litellm/litellm_core_utils/sensitive_data_masker.py b/litellm/litellm_core_utils/sensitive_data_masker.py index 23b9ec32fc7..900239602df 100644 --- a/litellm/litellm_core_utils/sensitive_data_masker.py +++ b/litellm/litellm_core_utils/sensitive_data_masker.py @@ -1,6 +1,6 @@ from typing import Any, Dict, Optional, Set -from litellm.constants import DEFAULT_MAX_RECURSE_DEPTH +from litellm.constants import DEFAULT_MAX_RECURSE_DEPTH_SENSITIVE_DATA_MASKER class SensitiveDataMasker: @@ -44,7 +44,7 @@ class SensitiveDataMasker: self, data: Dict[str, Any], depth: int = 0, - max_depth: int = DEFAULT_MAX_RECURSE_DEPTH, + max_depth: int = DEFAULT_MAX_RECURSE_DEPTH_SENSITIVE_DATA_MASKER, ) -> Dict[str, Any]: if depth >= max_depth: return data