@@ -384,6 +384,7 @@ func (m *UI) setSessionMessages(msgs []message.Message) tea.Cmd {
384384 return tea.Batch(cmds...)
385385}
386386
387+// appendSessionMessage appends a new message to the current session in the chat
387388func (m *UI) appendSessionMessage(msg message.Message) tea.Cmd {
388389 items := chat.GetMessageItems(m.com.Styles, &msg, nil)
389390 var cmds []tea.Cmd
@@ -399,6 +400,8 @@ func (m *UI) appendSessionMessage(msg message.Message) tea.Cmd {
399400 return tea.Batch(cmds...)
400401}
401402
403+// updateSessionMessage updates an existing message in the current session in the chat
404+// INFO: currently only updates the assistant when I add tools this will get a bit more complex
402405func (m *UI) updateSessionMessage(msg message.Message) tea.Cmd {
403406 existingItem := m.chat.GetMessageItem(msg.ID)
404407 switch msg.Role {