From 59e5d320f4190d2b11bdf083c63fef898c44e1e6 Mon Sep 17 00:00:00 2001
From: Br1an67 <932039080@qq.com>
Date: Mon, 2 Mar 2026 00:12:19 +0800
Subject: [PATCH] fix(agent-ui): add search_files tool rendering support

Map search_files to command_result renderer in workspace panel and add
inline argument display (pattern + path) in chat tool calls. Also add
search_files to file-related tools list and display name mapping.

Previously search_files fell through to the default JSON renderer,
showing raw JSON instead of formatted output.
---
 .../agent-ui/src/common/utils/tool-renderers/index.ts    | 1 +
 .../src/standalone/chat/Message/components/ToolCalls.tsx | 9 +++++++++
 2 files changed, 10 insertions(+)

diff --git a/multimodal/tarko/agent-ui/src/common/utils/tool-renderers/index.ts b/multimodal/tarko/agent-ui/src/common/utils/tool-renderers/index.ts
index c80351141d..f388cb33ac 100644
--- a/multimodal/tarko/agent-ui/src/common/utils/tool-renderers/index.ts
+++ b/multimodal/tarko/agent-ui/src/common/utils/tool-renderers/index.ts
@@ -15,6 +15,7 @@ const TOOL_TO_RENDERER_CONFIG: ToolToRendererCondition[] = [
   { toolName: 'edit_file', renderer: 'diff_result' },
   { toolName: 'run_command', renderer: 'command_result' },
   { toolName: 'run_script', renderer: 'script_result' },
+  { toolName: 'search_files', renderer: 'command_result' },
   { toolName: 'LinkReader', renderer: 'link_reader' },
   { toolName: 'Search', renderer: 'search_result' },
   { toolName: 'execute_bash', renderer: 'command_result' },
diff --git a/multimodal/tarko/agent-ui/src/standalone/chat/Message/components/ToolCalls.tsx b/multimodal/tarko/agent-ui/src/standalone/chat/Message/components/ToolCalls.tsx
index fa48b6aa19..ff66d74e7f 100644
--- a/multimodal/tarko/agent-ui/src/standalone/chat/Message/components/ToolCalls.tsx
+++ b/multimodal/tarko/agent-ui/src/standalone/chat/Message/components/ToolCalls.tsx
@@ -136,6 +136,12 @@ export const ToolCalls: React.FC<ToolCallsProps> = ({
               : '';
         case 'run_command':
           return args.command || (status === 'constructing' ? 'preparing command...' : '');
+        case 'search_files':
+          return args.pattern
+            ? `${args.pattern}${args.path ? ` in ${args.path}` : ''}`
+            : status === 'constructing'
+              ? 'preparing search...'
+              : '';
         case 'read_file':
         case 'write_file':
         case 'edit_file':
@@ -201,6 +207,8 @@ export const ToolCalls: React.FC<ToolCallsProps> = ({
         return 'Web Search';
       case 'list_directory':
         return 'List Files';
+      case 'search_files':
+        return 'Search Files';
       case 'run_command':
         return 'Run Command';
       case 'read_file':
@@ -223,6 +231,7 @@ export const ToolCalls: React.FC<ToolCallsProps> = ({
       'write_file',
       'edit_file',
       'list_directory',
+      'search_files',
       'str_replace_editor',
     ];
     return fileTools.includes(toolName);