From 51043ec80489594739c9e6fb23dfaed148ca80dc Mon Sep 17 00:00:00 2001
From: Vinnie <vinnie@Vinnies-MacBook-Pro.local>
Date: Thu, 24 Apr 2025 13:38:04 -0400
Subject: [PATCH 1/2] Add SENTRY_API_SAMPLE_RATE configuration option for
 Sentry SDK

---
 litellm/litellm_core_utils/litellm_logging.py |  6 ++++
 .../test_litellm_logging.py                   | 31 ++++++++++++++++++-
 2 files changed, 36 insertions(+), 1 deletion(-)

diff --git a/litellm/litellm_core_utils/litellm_logging.py b/litellm/litellm_core_utils/litellm_logging.py
index 77d4fd7d5d..f8b3e1ea45 100644
--- a/litellm/litellm_core_utils/litellm_logging.py
+++ b/litellm/litellm_core_utils/litellm_logging.py
@@ -2601,9 +2601,15 @@ def set_callbacks(callback_list, function_id=None):  # noqa: PLR0915
                     if "SENTRY_API_TRACE_RATE" in os.environ
                     else "1.0"
                 )
+                sentry_sample_rate = (
+                    os.environ.get("SENTRY_API_SAMPLE_RATE")
+                    if "SENTRY_API_SAMPLE_RATE" in os.environ
+                    else "1.0"
+                )
                 sentry_sdk_instance.init(
                     dsn=os.environ.get("SENTRY_DSN"),
                     traces_sample_rate=float(sentry_trace_rate),  # type: ignore
+                    sample_rate=float(sentry_sample_rate),
                 )
                 capture_exception = sentry_sdk_instance.capture_exception
                 add_breadcrumb = sentry_sdk_instance.add_breadcrumb
diff --git a/tests/litellm/litellm_core_utils/test_litellm_logging.py b/tests/litellm/litellm_core_utils/test_litellm_logging.py
index eedc15dd36..0b9b537f78 100644
--- a/tests/litellm/litellm_core_utils/test_litellm_logging.py
+++ b/tests/litellm/litellm_core_utils/test_litellm_logging.py
@@ -11,7 +11,7 @@ sys.path.insert(
 
 import time
 
-from litellm.litellm_core_utils.litellm_logging import Logging as LitellmLogging
+from litellm.litellm_core_utils.litellm_logging import Logging as LitellmLogging, set_callbacks
 
 
 @pytest.fixture
@@ -32,3 +32,32 @@ def test_get_masked_api_base(logging_obj):
     masked_api_base = logging_obj._get_masked_api_base(api_base)
     assert masked_api_base == "https://api.openai.com/v1"
     assert type(masked_api_base) == str
+
+
+def test_sentry_sample_rate():
+    existing_sample_rate = os.getenv("SENTRY_API_SAMPLE_RATE")
+    print(f"Existing SENTRY_API_SAMPLE_RATE: {existing_sample_rate}")
+    try:
+        # test with default value by removing the environment variable
+        if existing_sample_rate:
+            del os.environ["SENTRY_API_SAMPLE_RATE"]
+
+        set_callbacks(["sentry"])
+        # Check if the default sample rate is set to 1.0
+        assert os.environ.get("SENTRY_API_SAMPLE_RATE") == "1.0"
+
+        # test with custom value
+        os.environ["SENTRY_API_SAMPLE_RATE"] = "0.5"
+
+        set_callbacks(["sentry"])
+        # Check if the custom sample rate is set correctly
+        assert os.environ.get("SENTRY_API_SAMPLE_RATE") == "0.5"
+    except Exception as e:
+        print(f"Error: {e}")
+    finally:
+        # Restore the original environment variable
+        if existing_sample_rate:
+            os.environ["SENTRY_API_SAMPLE_RATE"] = existing_sample_rate
+        else:
+            if "SENTRY_API_SAMPLE_RATE" in os.environ:
+                del os.environ["SENTRY_API_SAMPLE_RATE"]
\ No newline at end of file

From b5a2fe8f648b6ac673d0dbf60b9fa48178805a15 Mon Sep 17 00:00:00 2001
From: Vinnie <vinnie@Vinnies-MacBook-Pro.local>
Date: Thu, 24 Apr 2025 14:00:10 -0400
Subject: [PATCH 2/2] removed print line

---
 tests/litellm/litellm_core_utils/test_litellm_logging.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/tests/litellm/litellm_core_utils/test_litellm_logging.py b/tests/litellm/litellm_core_utils/test_litellm_logging.py
index 0b9b537f78..767b1ff617 100644
--- a/tests/litellm/litellm_core_utils/test_litellm_logging.py
+++ b/tests/litellm/litellm_core_utils/test_litellm_logging.py
@@ -36,7 +36,6 @@ def test_get_masked_api_base(logging_obj):
 
 def test_sentry_sample_rate():
     existing_sample_rate = os.getenv("SENTRY_API_SAMPLE_RATE")
-    print(f"Existing SENTRY_API_SAMPLE_RATE: {existing_sample_rate}")
     try:
         # test with default value by removing the environment variable
         if existing_sample_rate: