From 835af21bcf2a34065398880e936f75cb4786efdb Mon Sep 17 00:00:00 2001 From: David Gageot Date: Fri, 26 Dec 2025 12:52:19 +0100 Subject: [PATCH] fix the margins for handoffs and transfer tasks Signed-off-by: David Gageot --- pkg/tui/components/message/message.go | 2 +- pkg/tui/components/tool/handoff/handoff.go | 2 +- pkg/tui/components/tool/transfertask/transfertask.go | 2 +- pkg/tui/styles/styles.go | 3 +-- 4 files changed, 4 insertions(+), 5 deletions(-) diff --git a/pkg/tui/components/message/message.go b/pkg/tui/components/message/message.go index 6a59f64af..002cf959c 100644 --- a/pkg/tui/components/message/message.go +++ b/pkg/tui/components/message/message.go @@ -145,7 +145,7 @@ func (mv *messageModel) senderPrefix(sender string) string { if sender == "" { return "" } - return styles.AgentBadgeStyle.Render(sender) + "\n\n" + return styles.AgentBadgeStyle.MarginLeft(2).Render(sender) + "\n\n" } // Height calculates the height needed for this message view diff --git a/pkg/tui/components/tool/handoff/handoff.go b/pkg/tui/components/tool/handoff/handoff.go index ceaead511..73063d77c 100644 --- a/pkg/tui/components/tool/handoff/handoff.go +++ b/pkg/tui/components/tool/handoff/handoff.go @@ -22,5 +22,5 @@ func render(msg *types.Message, _ spinner.Spinner, _, _ int) string { return "" } - return styles.AgentBadgeStyle.Render(msg.Sender) + " ─► " + styles.AgentBadgeStyle.Render(params.Agent) + return styles.AgentBadgeStyle.MarginLeft(2).Render(msg.Sender) + " ─► " + styles.AgentBadgeStyle.Render(params.Agent) } diff --git a/pkg/tui/components/tool/transfertask/transfertask.go b/pkg/tui/components/tool/transfertask/transfertask.go index 65b616088..309c9da47 100644 --- a/pkg/tui/components/tool/transfertask/transfertask.go +++ b/pkg/tui/components/tool/transfertask/transfertask.go @@ -22,7 +22,7 @@ func render(msg *types.Message, _ spinner.Spinner, _, _ int) string { return "" } - return styles.AgentBadgeStyle.Render(msg.Sender) + + return styles.AgentBadgeStyle.MarginLeft(2).Render(msg.Sender) + " calls " + styles.AgentBadgeStyle.Render(params.Agent) + "\n\n" + diff --git a/pkg/tui/styles/styles.go b/pkg/tui/styles/styles.go index 4ef284934..2754710c3 100644 --- a/pkg/tui/styles/styles.go +++ b/pkg/tui/styles/styles.go @@ -471,8 +471,7 @@ var ( AgentBadgeStyle = BaseStyle. Foreground(AgentBadgeFg). Background(AgentBadgeBg). - Padding(0, 1). - MarginLeft(2) + Padding(0, 1) ) // Deprecated styles (kept for backward compatibility)