From 9e06c42f72874dcdc30c77b68c04b81fc7dea216 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

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

diff --git a/.circleci/config.yml b/.circleci/config.yml
index 684e679e3e7..59616f19907 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -899,7 +899,6 @@ jobs:
         auth:
           username: ${DOCKERHUB_USERNAME}
           password: ${DOCKERHUB_PASSWORD}
-    resource_class: xlarge
     working_directory: ~/project
 
     steps:
diff --git a/litellm/constants.py b/litellm/constants.py
index a3ce465d4c9..0d8cfbe28d7 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