.NET: AG-UI support for workflow as agent

I've been trying to use AG-UI in the following setup:

Copilotkit app (Nextjs) connecting to .net AG-UI endpoint.

Now, I can successfully kick off the orchestration, but after the workflow is done, I don't see the events being sent back to the client.
Do I need to do some sort of mapping between the workflow events and the AG-UI events manually or is that handled automatically by the AG-UI server extension?