perf: scroll to bottom in thread

2 files changed, 6 insertions, 5 deletions

diff --git a/packages/tui/internal/components/chat/messages.go b/packages/tui/internal/components/chat/messages.go
index 9b6920adf..7d8b1b3a9 100644
--- a/packages/tui/internal/components/chat/messages.go
+++ b/packages/tui/internal/components/chat/messages.go
@@ -102,9 +102,6 @@ func (m *messagesComponent) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		m.loading = false
 		m.tail = m.viewport.AtBottom()
 		m.viewport = msg.viewport
-		if m.tail {
-			m.viewport.GotoBottom()
-		}
 		if m.dirty {
 			cmds = append(cmds, m.renderView())
 		}
@@ -138,6 +135,7 @@ func (m *messagesComponent) renderView() tea.Cmd {
 	m.rendering = true
 
 	viewport := m.viewport
+	tail := m.tail
 
 	return func() tea.Msg {
 		measure := util.Measure("messages.renderView")
@@ -402,6 +400,9 @@ func (m *messagesComponent) renderView() tea.Cmd {
 		content := "\n" + strings.Join(blocks, "\n\n")
 		viewport.SetHeight(m.height - lipgloss.Height(m.header))
 		viewport.SetContent(content)
+		if tail {
+			viewport.GotoBottom()
+		}
 
 		return renderCompleteMsg{
 			viewport:  viewport,
diff --git a/packages/tui/internal/tui/tui.go b/packages/tui/internal/tui/tui.go
index d09f3d343..c6c8fa13d 100644
--- a/packages/tui/internal/tui/tui.go
+++ b/packages/tui/internal/tui/tui.go
@@ -528,17 +528,17 @@ func (a appModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 }
 
 func (a appModel) View() string {
+	measure := util.Measure("app.View")
+	defer measure()
 	t := theme.CurrentTheme()
 
 	var mainLayout string
 
-	measure := util.Measure("app.View")
 	if a.app.Session.ID == "" {
 		mainLayout = a.home()
 	} else {
 		mainLayout = a.chat()
 	}
-	measure()
 	mainLayout = styles.NewStyle().
 		Background(t.Background()).
 		Padding(0, 2).