From 965ae5d27a2f55a9214cf17cf365c90fc2f727d5 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 14 May 2025 18:29:01 -0700
Subject: [PATCH] test: fix litellm_mapped_tests

---
 litellm/constants.py                                | 3 +++
 litellm/litellm_core_utils/sensitive_data_masker.py | 4 ++--
 2 files changed, 5 insertions(+), 2 deletions(-)

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