feat(ag-ui): Tool call returns and thinking parts

Add support for the new tool call returns and thinking parts to the
AG UI adapter. This eliminates the need to build a message snapshot
while sill providing the necessary information for the UI.

Author: stevenh