Add exception_replay_enabled.

Author: purple4reina

datadog_lambda/config.py

@@ -60,6 +60,7 @@ class Config:
     capture_payload_max_depth = _get_env("DD_CAPTURE_PAYLOAD_MAX_DEPTH", 10, int)
     profiling_enabled = _get_env("DD_PROFILING_ENABLED", "false", as_bool)
     llmobs_enabled = _get_env("DD_LLMOBS_ENABLED", "false", as_bool)
+    exception_replay_enabled = _get_env("DD_EXCEPTION_REPLAY_ENABLED", "false", as_bool)
 
     @property
     def fips_mode_enabled(self):

datadog_lambda/wrapper.py

@@ -51,10 +51,7 @@
 if config.llmobs_enabled:
     from ddtrace.llmobs import LLMObs
 
-exception_replay_env_var = os.environ.get(
-    "DD_EXCEPTION_REPLAY_ENABLED", "false"
-).lower() in ("true", "1")
-if exception_replay_env_var:
+if config.exception_replay_enabled:
     from ddtrace.debugging._exception.replay import SpanExceptionHandler
 
 logger = logging.getLogger(__name__)
@@ -206,7 +203,7 @@ def __init__(self, func):
                 LLMObs.enable()
 
             # Enable Exception Replay
-            if exception_replay_env_var:
+            if config.exception_replay_enabled:
                 logger.debug("Enabling exception replay")
                 SpanExceptionHandler.enable()
 

tests/test_config.py

@@ -155,6 +155,15 @@ def set_env(key, value):
     ("DD_LLMOBS_ENABLED", "llmobs_enabled", "1", True),  # CHANGED
     ("DD_LLMOBS_ENABLED", "llmobs_enabled", "0", False),
     ("DD_LLMOBS_ENABLED", "llmobs_enabled", "purple", False),
+    ("DD_EXCEPTION_REPLAY_ENABLED", "exception_replay_enabled", None, False),
+    ("DD_EXCEPTION_REPLAY_ENABLED", "exception_replay_enabled", "", False),
+    ("DD_EXCEPTION_REPLAY_ENABLED", "exception_replay_enabled", "true", True),
+    ("DD_EXCEPTION_REPLAY_ENABLED", "exception_replay_enabled", "TRUE", True),
+    ("DD_EXCEPTION_REPLAY_ENABLED", "exception_replay_enabled", "false", False),
+    ("DD_EXCEPTION_REPLAY_ENABLED", "exception_replay_enabled", "FALSE", False),
+    ("DD_EXCEPTION_REPLAY_ENABLED", "exception_replay_enabled", "1", True),  # CHANGED
+    ("DD_EXCEPTION_REPLAY_ENABLED", "exception_replay_enabled", "0", False),
+    ("DD_EXCEPTION_REPLAY_ENABLED", "exception_replay_enabled", "purple", False),
 )