cli: Tweak code_search / list directory tool components

Author: jahooma

cli/src/components/tools/code-search.tsx

@@ -12,54 +12,61 @@ import type { ToolRenderConfig } from './types'
  */
 export const CodeSearchComponent = defineToolComponent({
   toolName: 'code_search',
-  
+
   render(toolBlock, theme, options): ToolRenderConfig | null {
     const input = toolBlock.input as any
     const pattern = input?.pattern ?? ''
     const flags = input?.flags ?? ''
-    
+    const cwd = input?.cwd ?? ''
+
     // Count results from output
     let totalResults = 0
     let fileCount = 0
-    
+
     if (toolBlock.output && typeof toolBlock.output === 'string') {
       const lines = toolBlock.output.split('\n')
       const files = new Set<string>()
-      
+
       for (const line of lines) {
         const trimmed = line.trim()
-        
+
         // File paths end with a colon and typically start with ./ or /
-        if (trimmed.endsWith(':') && (trimmed.startsWith('./') || trimmed.startsWith('/'))) {
+        if (
+          trimmed.endsWith(':') &&
+          (trimmed.startsWith('./') || trimmed.startsWith('/'))
+        ) {
           files.add(trimmed.slice(0, -1)) // Remove trailing colon
         }
         // Result lines start with a number followed by a colon
         else if (/^\d+:/.test(trimmed)) {
           totalResults++
         }
       }
-      
+
       fileCount = files.size
     }
-    
+
     // Build single-line summary
-    let summary = `"${pattern}"`
-    
-    if (flags) {
-      summary += ` ${flags}`
-    }
-    
-    summary += ` → ${totalResults} result${totalResults === 1 ? '' : 's'}`
-    
-    if (fileCount > 0) {
-      summary += ` across ${fileCount} file${fileCount === 1 ? '' : 's'}`
+    let summary = ''
+
+    summary += `${pattern}`
+
+    if (cwd) {
+      summary += ` in ${cwd}`
     }
-    
+
+    // Disable showing flags since they are noisy.
+    // if (flags) {
+    //   summary += ` ${flags}`
+    // }
+
+    summary += ` (${totalResults} result${totalResults === 1 ? '' : 's'})`
+
     // Return as content using SimpleToolCallItem
     return {
       content: (
         <SimpleToolCallItem
-          name="Code Search"
+          name="Search"
           description={summary}
           branchChar={options.branchChar}
         />

cli/src/components/tools/list-directory.tsx

@@ -12,32 +12,39 @@ import type { ToolRenderConfig } from './types'
  */
 export const ListDirectoryComponent = defineToolComponent({
   toolName: 'list_directory',
-  
+
   render(toolBlock, theme, options): ToolRenderConfig | null {
     const input = toolBlock.input as any
-    
+
     // Extract directories from input
     let directories: string[] = []
-    
+
     if (Array.isArray(input?.directories)) {
       directories = input.directories
-        .map((dir: any) => typeof dir === 'object' && dir.path ? dir.path : dir)
-        .filter((path: any) => typeof path === 'string' && path.trim().length > 0)
-    } else if (typeof input?.path === 'string' && input.path.trim().length > 0) {
+        .map((dir: any) =>
+          typeof dir === 'object' && dir.path ? dir.path : dir,
+        )
+        .filter(
+          (path: any) => typeof path === 'string' && path.trim().length > 0,
+        )
+    } else if (
+      typeof input?.path === 'string' &&
+      input.path.trim().length > 0
+    ) {
       directories = [input.path.trim()]
     }
-    
+
     if (directories.length === 0) {
       return null
     }
-    
+
     // Format directory list
     const description = directories.join(', ')
-    
+
     return {
       content: (
         <SimpleToolCallItem
-          name="List Directories"
+          name="List"
           description={description}
           branchChar={options.branchChar}
         />

cli/src/components/tools/tool-call-item.tsx

@@ -135,7 +135,7 @@ export const SimpleToolCallItem = ({
 
   return (
     <box style={{ flexDirection: 'row', alignItems: 'center', width: '100%' }}>
-      <text style={{ wrapMode: 'none' }}>
+      <text style={{ wrapMode: 'word' }}>
         <span fg={theme.foreground}>{branchChar || bulletChar}</span>
         <span fg={theme.foreground} attributes={TextAttributes.BOLD}>
           {name}