Better rendering for search_files args

Signed-off-by: David Gageot <david.gageot@docker.com>

Author: dgageot

internal/tui/components/tool/builtins.go

@@ -0,0 +1,31 @@
+package tool
+
+import (
+	"encoding/json"
+	"strings"
+
+	"github.com/docker/cagent/pkg/tools"
+	"github.com/docker/cagent/pkg/tools/builtin"
+)
+
+func render_search_files(toolCall tools.ToolCall) string {
+	var args builtin.SearchFilesArgs
+	if err := json.Unmarshal([]byte(toolCall.Function.Arguments), &args); err != nil {
+		return ""
+	}
+
+	// Search pattern
+	output := args.Pattern
+
+	// Optional search path
+	if path := args.Path; path != "" && path != "." {
+		output += " in " + path
+	}
+
+	// Optional exclude patterns
+	if exclude := args.ExcludePatterns; len(exclude) > 0 {
+		output += " excluding [" + strings.Join(exclude, ", ") + "]"
+	}
+
+	return output
+}

internal/tui/components/tool/tool.go

@@ -85,30 +85,31 @@ func (mv *toolModel) View() string {
 	return mv.Render(mv.width)
 }
 
-// MessageView specific methods
-
 // Render renders the message view content
 func (mv *toolModel) Render(width int) string {
 	msg := mv.message
 
-	// Add spinner for pending and running tools
-	var spinnerText string
-	if msg.ToolStatus == types.ToolStatusPending || msg.ToolStatus == types.ToolStatusRunning {
-		spinnerText = " " + mv.spinner.View()
-	}
-
 	// Ask the tool what's its display name
 	team := mv.app.Team()
 	agent := team.Agent(msg.Sender)
 	displayName := agent.ToolDisplayName(context.TODO(), msg.ToolCall.Function.Name)
-	content := fmt.Sprintf("%s %s%s", icon(msg.ToolStatus), styles.HighlightStyle.Render(displayName), spinnerText)
+	content := fmt.Sprintf("%s %s", icon(msg.ToolStatus), styles.HighlightStyle.Render(displayName))
 
 	if msg.ToolCall.Function.Arguments != "" {
-		lines := wrapLines(msg.ToolCall.Function.Arguments, mv.width-2)
-		argsViewport := viewport.New(viewport.WithWidth(mv.width), viewport.WithHeight(len(lines)))
-		argsViewport.SetContent(styles.MutedStyle.Render(strings.Join(lines, "\n")))
-		argsViewport.GotoBottom()
-		content += "\n" + argsViewport.View()
+		if msg.ToolCall.Function.Name == "search_files" {
+			content += " " + render_search_files(msg.ToolCall)
+		} else {
+			lines := wrapLines(msg.ToolCall.Function.Arguments, mv.width-2)
+			argsViewport := viewport.New(viewport.WithWidth(mv.width), viewport.WithHeight(len(lines)))
+			argsViewport.SetContent(styles.MutedStyle.Render(strings.Join(lines, "\n")))
+			argsViewport.GotoBottom()
+			content += "\n" + argsViewport.View()
+		}
+	}
+
+	// Add spinner for pending and running tools
+	if msg.ToolStatus == types.ToolStatusPending || msg.ToolStatus == types.ToolStatusRunning {
+		content += " " + mv.spinner.View()
 	}
 
 	// Add tool result content if available (for completed tools with content)