Python: WorkflowAgent created via .as_agent() does not emit outputs when using ctx.yield_output #2813

from typing import List
from agent_framework import WorkflowBuilder, executor, WorkflowContext, ChatMessage

@executor
async def my_executor(text: str | List[ChatMessage], ctx: WorkflowContext):
    await ctx.yield_output(f"processing: {text}")


async def main():
    workflow = (
        WorkflowBuilder()
        .set_start_executor(my_executor)
        .build()
    )

    print("Running workflow...")
    result = await workflow.run("hello from workflow")
    print("Workflow result:", result)

    print("\nRunning as agent...")
    agent = workflow.as_agent("wf-agent")
    result = await agent.run("hello from agent")
    print("WorkflowAgent result:", result)

    print("END")


if __name__ == "__main__":
    import asyncio
    asyncio.run(main())


# Expected:
# Running workflow...
# Workflow result: [ExecutorInvokedEvent(executor_id=my_executor, data=None), WorkflowOutputEvent(data=processing: hello from workflow, source_executor_id=my_executor), ExecutorCompletedEvent(executor_id=my_executor, data=None)]

# Running as agent...
# WorkflowAgent result: [ExecutorInvokedEvent(executor_id=my_executor, data=None), WorkflowOutputEvent(data=processing: hello from workflow, source_executor_id=my_executor), ExecutorCompletedEvent(executor_id=my_executor, data=None)]
# END


# Actual:
# Running workflow...
# Workflow result: [ExecutorInvokedEvent(executor_id=my_executor, data=None), WorkflowOutputEvent(data=processing: hello from workflow, source_executor_id=my_executor), ExecutorCompletedEvent(executor_id=my_executor, data=None)]

# Running as agent...

# END