Trace Context • AI Agent SDK

trace_context: Context manager for trace emitter lifecycle.

trace_context

Function

</div>

This function is defined in the context_events module.

Context manager for trace emitter lifecycle.

Automatically sets the emitter on entry and resets on exit, even if an exception occurs. This is the recommended way to use custom trace emitters.

Signature

python

def trace_context(emitter: 'ContextTraceEmitter') -> Generator['ContextTraceEmitter', None, None]

Parameters

ParamField ContextTraceEmitter

The emitter to use within the context. The emitter for use within the context.

Returns

ResponseField Generator['ContextTraceEmitter', None, None]

The result of the operation.

Usage

python

sink = MyCustomSink()
    emitter = ContextTraceEmitter(sink=sink, session_id="my-session", enabled=True)

    with trace_context(emitter) as ctx:
        agent = Agent(...)
        agent.chat("Hello")  # Events go to MyCustomSink

Uses

set_context_emitter
reset_context_emitter

Source

View on GitHub

praisonaiagents/trace/context_events.py at line 122