Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 9 additions & 0 deletions datadog_lambda/wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down Expand Up @@ -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))
Expand Down
Loading