add support for ripgrep search (#30)

Author: homanp

package-lock.json

@@ -34,6 +34,7 @@
     "ink-markdown": "^1.0.4",
     "openai": "^5.10.1",
     "react": "^17.0.2",
+    "ripgrep-node": "^1.0.0",
     "tiktoken": "^1.0.21"
   },
   "devDependencies": {

package.json

@@ -1,6 +1,12 @@
 import { GrokClient, GrokMessage, GrokToolCall } from "../grok/client";
 import { GROK_TOOLS } from "../grok/tools";
-import { TextEditorTool, BashTool, TodoTool, ConfirmationTool } from "../tools";
+import {
+  TextEditorTool,
+  BashTool,
+  TodoTool,
+  ConfirmationTool,
+  SearchTool,
+} from "../tools";
 import { ToolResult } from "../types";
 import { EventEmitter } from "events";
 import { createTokenCounter, TokenCounter } from "../utils/token-counter";
@@ -31,6 +37,7 @@ export class GrokAgent extends EventEmitter {
   private bash: BashTool;
   private todoTool: TodoTool;
   private confirmationTool: ConfirmationTool;
+  private search: SearchTool;
   private chatHistory: ChatEntry[] = [];
   private messages: GrokMessage[] = [];
   private tokenCounter: TokenCounter;
@@ -43,6 +50,7 @@ export class GrokAgent extends EventEmitter {
     this.bash = new BashTool();
     this.todoTool = new TodoTool();
     this.confirmationTool = new ConfirmationTool();
+    this.search = new SearchTool();
     this.tokenCounter = createTokenCounter("grok-4-latest");
 
     // Load custom instructions
@@ -61,6 +69,7 @@ You have access to these tools:
 - create_file: Create new files with content (ONLY use this for files that don't exist yet)
 - str_replace_editor: Replace text in existing files (ALWAYS use this to edit or update existing files)
 - bash: Execute bash commands (use for searching, file discovery, navigation, and system operations)
+- search: Unified search tool for finding text content or files (similar to Cursor's search functionality)
 - create_todo_list: Create a visual todo list for planning and tracking tasks
 - update_todo_list: Update existing todos in your todo list
 
@@ -74,9 +83,9 @@ IMPORTANT TOOL USAGE RULES:
 - Use create_file ONLY when creating entirely new files that don't exist
 
 SEARCHING AND EXPLORATION:
-- Use bash with commands like 'find', 'grep', 'rg' (ripgrep), 'ls', etc. for searching files and content
-- Examples: 'find . -name "*.js"', 'grep -r "function" src/', 'rg "import.*react"'
-- Use bash for directory navigation, file discovery, and content searching
+- Use search for fast, powerful text search across files or finding files by name (unified search tool)
+- Examples: search for text content like "import.*react", search for files like "component.tsx"
+- Use bash with commands like 'find', 'grep', 'rg', 'ls' for complex file operations and navigation
 - view_file is best for reading specific files you already know exist
 
 When a user asks you to edit, update, modify, or change an existing file:
@@ -550,6 +559,19 @@ Current working directory: ${process.cwd()}`,
         case "update_todo_list":
           return await this.todoTool.updateTodoList(args.updates);
 
+        case "search":
+          return await this.search.search(args.query, {
+            searchType: args.search_type,
+            includePattern: args.include_pattern,
+            excludePattern: args.exclude_pattern,
+            caseSensitive: args.case_sensitive,
+            wholeWord: args.whole_word,
+            regex: args.regex,
+            maxResults: args.max_results,
+            fileTypes: args.file_types,
+            includeHidden: args.include_hidden,
+          });
+
         default:
           return {
             success: false,

src/agent/grok-agent.ts

@@ -62,15 +62,17 @@ export const GROK_TOOLS: GrokTool[] = [
           },
           old_str: {
             type: "string",
-            description: "Text to replace (must match exactly, or will use fuzzy matching for multi-line strings)",
+            description:
+              "Text to replace (must match exactly, or will use fuzzy matching for multi-line strings)",
           },
           new_str: {
             type: "string",
             description: "Text to replace with",
           },
           replace_all: {
             type: "boolean",
-            description: "Replace all occurrences (default: false, only replaces first occurrence)",
+            description:
+              "Replace all occurrences (default: false, only replaces first occurrence)",
           },
         },
         required: ["path", "old_str", "new_str"],
@@ -94,6 +96,66 @@ export const GROK_TOOLS: GrokTool[] = [
       },
     },
   },
+  {
+    type: "function",
+    function: {
+      name: "search",
+      description:
+        "Unified search tool for finding text content or files (similar to Cursor's search)",
+      parameters: {
+        type: "object",
+        properties: {
+          query: {
+            type: "string",
+            description: "Text to search for or file name/path pattern",
+          },
+          search_type: {
+            type: "string",
+            enum: ["text", "files", "both"],
+            description:
+              "Type of search: 'text' for content search, 'files' for file names, 'both' for both (default: 'both')",
+          },
+          include_pattern: {
+            type: "string",
+            description:
+              "Glob pattern for files to include (e.g. '*.ts', '*.js')",
+          },
+          exclude_pattern: {
+            type: "string",
+            description:
+              "Glob pattern for files to exclude (e.g. '*.log', 'node_modules')",
+          },
+          case_sensitive: {
+            type: "boolean",
+            description:
+              "Whether search should be case sensitive (default: false)",
+          },
+          whole_word: {
+            type: "boolean",
+            description: "Whether to match whole words only (default: false)",
+          },
+          regex: {
+            type: "boolean",
+            description: "Whether query is a regex pattern (default: false)",
+          },
+          max_results: {
+            type: "number",
+            description: "Maximum number of results to return (default: 50)",
+          },
+          file_types: {
+            type: "array",
+            items: { type: "string" },
+            description: "File types to search (e.g. ['js', 'ts', 'py'])",
+          },
+          include_hidden: {
+            type: "boolean",
+            description: "Whether to include hidden files (default: false)",
+          },
+        },
+        required: ["query"],
+      },
+    },
+  },
   {
     type: "function",
     function: {

src/grok/tools.ts

@@ -1,4 +1,5 @@
-export { TextEditorTool } from "./text-editor";
 export { BashTool } from "./bash";
+export { TextEditorTool } from "./text-editor";
 export { TodoTool } from "./todo-tool";
 export { ConfirmationTool } from "./confirmation-tool";
+export { SearchTool } from "./search";