Add annotations

[rolandtannous]

Is your feature request related to a problem? Please describe.
Current memory is hard to follow as messages are not labeled adequately to distinguish human from agent messages.

Describe the solution you'd like
label messages as User or Agent

Describe alternatives you've considered
Post processing result

Additional context
Add any other context or screenshots about the feature request here.