For each reply by AI, use reply to the message to differentiate which message AI responds to.

Background & Feedback from users:

End users type many messages in sequence and AI generates response for each message seperately and sometimes it may cause confusion. If we use reply to message feature for each AI response, it may avoid that confusion.