DRAFT Single Agent Tool LlamaIndex Instrumentation

[shuningc]

Adding a separate workflow agent instrumentation mechanism with a wrapper function:
✅ Dual instrumentation paths for backward compatibility
✅ Non-invasive: Uses monkey-patching, no code changes needed
✅ Async-aware: Background task doesn't block user code
✅ Context propagation: Manual for workflows, automatic for callbacks
✅ Shared TelemetryHandler: Both paths use same handler instance

User code:
agent = ReActAgent(tools=[...])
handler = agent.run("Calculate 5*3+2")
↓
↓ (intercepted by wrap_agent_run)
↓

wrap_agent_run():
Line 137: handler = wrapped(*args, **kwargs)
↓
└─→ Calls original ReActAgent.run()
↓
└─→ Returns WorkflowHandler
(contains stream of events)

Line 159: await instrumentor.instrument_workflow_handler(handler, ...)
↓
└─→ Pass handler to instrumentor

Line 169: return handler # Return to user
↓
└─→ User can await it

instrument_workflow_handler(workflow_handler):
Line 43: async for event in workflow_handler.stream_events():
↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑
THIS IS THE CONSUMPTION!

       We're iterating through the event stream
       that the WorkflowHandler provides