diff --git a/packages/tui/internal/components/chat/message.go b/packages/tui/internal/components/chat/message.go
index 3fa013e3..d862e528 100644
--- a/packages/tui/internal/components/chat/message.go
+++ b/packages/tui/internal/components/chat/message.go
@@ -308,10 +308,15 @@ func renderToolInvocation(
 		if m, ok := value.(map[string]any); ok {
 			toolArgsMap = m
 
-			firstKey := ""
+			keys := make([]string, 0, len(toolArgsMap))
 			for key := range toolArgsMap {
-				firstKey = key
-				break
+				keys = append(keys, key)
+			}
+			slices.Sort(keys)
+
+			firstKey := ""
+			if len(keys) > 0 {
+				firstKey = keys[0]
 			}
 
 			toolArgs = renderArgs(&toolArgsMap, firstKey)