result = await agent.run("Research three AI agent runtimes.", session_id="user_1")

events = await agent.get_telemetry_events_after(
    cursor=None,
    session_id=result["session_id"],
    run_id=result["run_id"],
)

for event in events:
    print(event.event_type, event.model_dump())

exact_trace = await agent.get_trace(result["trace_id"])
latest_session_trace = await agent.get_latest_trace(result["session_id"])
run_correlated_trace = await agent.get_trace(run_id=result["run_id"])
normalized_trace = await agent.get_trace(result["trace_id"], normalize=True)

session_id = "user_1"
run_id = "run_ui_1"
cursor = await agent.get_telemetry_stream_cursor(session_id=session_id)

task = asyncio.create_task(
    agent.run("Use tools if needed.", session_id=session_id, run_id=run_id)
)

async for event in agent.stream_telemetry_after(
    cursor=cursor,
    session_id=session_id,
    run_id=run_id,
):
    print(event.event_type)
    if event.event_type == "final_answer":
        break

await task