Swarm: Allow functions to update agent's state, including system message, before replying

Swarm: Allow functions to update agent's state, including system message, before replying #597