Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 8 additions & 2 deletions extensions/cli/src/tools/index.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ import { listFilesTool } from "./listFiles.js";
import { multiEditTool } from "./multiEdit.js";
import { readFileTool } from "./readFile.js";
import { runTerminalCommandTool } from "./runTerminalCommand.js";
import { searchCodeTool } from "./searchCode.js";
import { checkIfRipgrepIsInstalled, searchCodeTool } from "./searchCode.js";
import { statusTool } from "./status.js";
import {
type Tool,
Expand All @@ -45,18 +45,24 @@ const BASE_BUILTIN_TOOLS: Tool[] = [
readFileTool,
writeFileTool,
listFilesTool,
searchCodeTool,
runTerminalCommandTool,
fetchTool,
writeChecklistTool,
];

const BUILTIN_SEARCH_TOOLS: Tool[] = [searchCodeTool];

// Get all builtin tools including dynamic ones, with capability-based filtering
export async function getAllAvailableTools(
isHeadless: boolean,
): Promise<Tool[]> {
const tools = [...BASE_BUILTIN_TOOLS];

const isRipgrepInstalled = await checkIfRipgrepIsInstalled();
if (isRipgrepInstalled) {
tools.push(...BUILTIN_SEARCH_TOOLS);
}

// If model is capable, exclude editTool in favor of multiEditTool
const modelState = await serviceContainer.get<ModelServiceState>(
SERVICE_NAMES.MODEL,
Expand Down
9 changes: 9 additions & 0 deletions extensions/cli/src/tools/searchCode.ts
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,15 @@ const execPromise = util.promisify(child_process.exec);
// Default maximum number of results to display
const DEFAULT_MAX_RESULTS = 100;

export async function checkIfRipgrepIsInstalled(): Promise<boolean> {
try {
await execPromise("rg --version");
return true;
} catch {
return false;
}
}

export const searchCodeTool: Tool = {
name: "Search",
displayName: "Search",
Expand Down
Loading