import exception replay from tracer and enable if DD_EXCEPTION_REPLAY_ENABLED=true

Author: nhulston

datadog_lambda/wrapper.py

@@ -53,6 +53,10 @@
 if llmobs_env_var:
     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:
+    from ddtrace.debugging._exception.replay import SpanExceptionHandler
+
 logger = logging.getLogger(__name__)
 
 DD_FLUSH_TO_LOG = "DD_FLUSH_TO_LOG"
@@ -224,6 +228,11 @@ def __init__(self, func):
             if llmobs_env_var:
                 LLMObs.enable()
 
+            # Enable Exception Replay
+            if exception_replay_env_var:
+                logger.debug("Enabling exception replay")
+                SpanExceptionHandler.enable()
+
             logger.debug("datadog_lambda_wrapper initialized")
         except Exception as e:
             logger.error(format_err_with_traceback(e))