wip: refactoring tui

2 files changed, 4 insertions, 5 deletions

diff --git a/internal/tui/components/chat/message.go b/internal/tui/components/chat/message.go
index c23644638..679041d8d 100644
--- a/internal/tui/components/chat/message.go
+++ b/internal/tui/components/chat/message.go
@@ -76,7 +76,7 @@ func renderMessage(msg string, isUser bool, isFocused bool, width int, info ...s
 	return rendered
 }
 
-func renderUserMessage(msg client.SessionMessage, isFocused bool, width int, position int) uiMessage {
+func renderUserMessage(msg client.MessageInfo, isFocused bool, width int) uiMessage {
 	// var styledAttachments []string
 	t := theme.CurrentTheme()
 	baseStyle := styles.BaseStyle()
@@ -118,8 +118,8 @@ func renderUserMessage(msg client.SessionMessage, isFocused bool, width int, pos
 		}
 
 		switch part.(type) {
-		case client.SessionMessagePartText:
-			textPart := part.(client.SessionMessagePartText)
+		case client.MessagePartText:
+			textPart := part.(client.MessagePartText)
 			content = renderMessage(textPart.Text, true, isFocused, width, info...)
 		}
 	}
diff --git a/internal/tui/components/chat/messages.go b/internal/tui/components/chat/messages.go
index 652f831d3..5fe355071 100644
--- a/internal/tui/components/chat/messages.go
+++ b/internal/tui/components/chat/messages.go
@@ -185,12 +185,12 @@ func formatTimeDifference(unixTime1, unixTime2 int64) string {
 
 func (m *messagesCmp) renderView() {
 	m.uiMessages = make([]uiMessage, 0)
-	pos := 0
 	baseStyle := styles.BaseStyle()
 
 	if m.width == 0 {
 		return
 	}
+
 	for _, msg := range m.app.Messages {
 		switch msg.Role {
 		case client.User:
@@ -202,7 +202,6 @@ func (m *messagesCmp) renderView() {
 				msg,
 				msg.Id == m.currentMsgID,
 				m.width,
-				pos,
 			)
 			m.uiMessages = append(m.uiMessages, userMsg)
 			m.cachedContent[msg.Id] = cacheItem{