summaryrefslogtreecommitdiff
path: root/packages/server/src
diff options
context:
space:
mode:
authorTaylor Mullen <[email protected]>2025-05-18 23:13:57 -0700
committerN. Taylor Mullen <[email protected]>2025-05-18 23:19:15 -0700
commit3d74a7061e2e64d7d8f3850e913acf31b493bcf9 (patch)
tree27314f5526383186b71a139e1c0fa4c5a396fc80 /packages/server/src
parentcd1dc7ec5918f3e3867ba1101b6228c815b7bc31 (diff)
fix(server): Use console.debug in GrepTool for less verbose logging
- Replaces `console.warn` and `console.error` calls with `console.debug` in `packages/server/src/tools/grep.ts`. This change reduces noise for the user, as `warn` and `error` messages are displayed directly, while `debug` messages are not. - Adds a comprehensive test suite for the GrepTool (`packages/server/src/tools/grep.test.ts`) to ensure its functionality remains robust after these changes and to cover various usage scenarios. - Improves error message consistency in `GrepTool`'s parameter validation and execution. Fixes https://b.corp.google.com/issues/418648813
Diffstat (limited to 'packages/server/src')
-rw-r--r--packages/server/src/tools/grep.test.ts257
-rw-r--r--packages/server/src/tools/grep.ts34
2 files changed, 274 insertions, 17 deletions
diff --git a/packages/server/src/tools/grep.test.ts b/packages/server/src/tools/grep.test.ts
new file mode 100644
index 00000000..59eb75a4
--- /dev/null
+++ b/packages/server/src/tools/grep.test.ts
@@ -0,0 +1,257 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
+import { GrepTool, GrepToolParams } from './grep.js';
+import path from 'path';
+import fs from 'fs/promises';
+import os from 'os';
+
+// Mock the child_process module to control grep/git grep behavior
+vi.mock('child_process', () => ({
+ spawn: vi.fn(() => ({
+ on: (event: string, cb: (...args: unknown[]) => void) => {
+ if (event === 'error' || event === 'close') {
+ // Simulate command not found or error for git grep and system grep
+ // to force fallback to JS implementation.
+ setTimeout(() => cb(1), 0); // cb(1) for error/close
+ }
+ },
+ stdout: { on: vi.fn() },
+ stderr: { on: vi.fn() },
+ })),
+}));
+
+describe('GrepTool', () => {
+ let tempRootDir: string;
+ let grepTool: GrepTool;
+ const abortSignal = new AbortController().signal;
+
+ beforeEach(async () => {
+ tempRootDir = await fs.mkdtemp(path.join(os.tmpdir(), 'grep-tool-root-'));
+ grepTool = new GrepTool(tempRootDir);
+
+ // Create some test files and directories
+ await fs.writeFile(
+ path.join(tempRootDir, 'fileA.txt'),
+ 'hello world\nsecond line with world',
+ );
+ await fs.writeFile(
+ path.join(tempRootDir, 'fileB.js'),
+ 'const foo = "bar";\nfunction baz() { return "hello"; }',
+ );
+ await fs.mkdir(path.join(tempRootDir, 'sub'));
+ await fs.writeFile(
+ path.join(tempRootDir, 'sub', 'fileC.txt'),
+ 'another world in sub dir',
+ );
+ await fs.writeFile(
+ path.join(tempRootDir, 'sub', 'fileD.md'),
+ '# Markdown file\nThis is a test.',
+ );
+ });
+
+ afterEach(async () => {
+ await fs.rm(tempRootDir, { recursive: true, force: true });
+ });
+
+ describe('validateToolParams', () => {
+ it('should return null for valid params (pattern only)', () => {
+ const params: GrepToolParams = { pattern: 'hello' };
+ expect(grepTool.validateToolParams(params)).toBeNull();
+ });
+
+ it('should return null for valid params (pattern and path)', () => {
+ const params: GrepToolParams = { pattern: 'hello', path: '.' };
+ expect(grepTool.validateToolParams(params)).toBeNull();
+ });
+
+ it('should return null for valid params (pattern, path, and include)', () => {
+ const params: GrepToolParams = {
+ pattern: 'hello',
+ path: '.',
+ include: '*.txt',
+ };
+ expect(grepTool.validateToolParams(params)).toBeNull();
+ });
+
+ it('should return error if pattern is missing', () => {
+ const params = { path: '.' } as unknown as GrepToolParams;
+ expect(grepTool.validateToolParams(params)).toContain(
+ 'Parameters failed schema validation',
+ );
+ });
+
+ it('should return error for invalid regex pattern', () => {
+ const params: GrepToolParams = { pattern: '[[' };
+ expect(grepTool.validateToolParams(params)).toContain(
+ 'Invalid regular expression pattern',
+ );
+ });
+
+ it('should return error if path does not exist', () => {
+ const params: GrepToolParams = { pattern: 'hello', path: 'nonexistent' };
+ // Check for the core error message, as the full path might vary
+ expect(grepTool.validateToolParams(params)).toContain(
+ 'Failed to access path stats for',
+ );
+ expect(grepTool.validateToolParams(params)).toContain('nonexistent');
+ });
+
+ it('should return error if path is a file, not a directory', async () => {
+ const filePath = path.join(tempRootDir, 'fileA.txt');
+ const params: GrepToolParams = { pattern: 'hello', path: filePath };
+ expect(grepTool.validateToolParams(params)).toContain(
+ `Path is not a directory: ${filePath}`,
+ );
+ });
+ });
+
+ describe('execute', () => {
+ it('should find matches for a simple pattern in all files', async () => {
+ const params: GrepToolParams = { pattern: 'world' };
+ const result = await grepTool.execute(params, abortSignal);
+ expect(result.llmContent).toContain(
+ 'Found 3 match(es) for pattern "world" in path "."',
+ );
+ expect(result.llmContent).toContain('File: fileA.txt');
+ expect(result.llmContent).toContain('L1: hello world');
+ expect(result.llmContent).toContain('L2: second line with world');
+ expect(result.llmContent).toContain('File: sub/fileC.txt');
+ expect(result.llmContent).toContain('L1: another world in sub dir');
+ expect(result.returnDisplay).toBe('Found 3 matche(s)');
+ });
+
+ it('should find matches in a specific path', async () => {
+ const params: GrepToolParams = { pattern: 'world', path: 'sub' };
+ const result = await grepTool.execute(params, abortSignal);
+ expect(result.llmContent).toContain(
+ 'Found 1 match(es) for pattern "world" in path "sub"',
+ );
+ expect(result.llmContent).toContain('File: fileC.txt'); // Path relative to 'sub'
+ expect(result.llmContent).toContain('L1: another world in sub dir');
+ expect(result.returnDisplay).toBe('Found 1 matche(s)');
+ });
+
+ it('should find matches with an include glob', async () => {
+ const params: GrepToolParams = { pattern: 'hello', include: '*.js' };
+ const result = await grepTool.execute(params, abortSignal);
+ expect(result.llmContent).toContain(
+ 'Found 1 match(es) for pattern "hello" in path "." (filter: "*.js")',
+ );
+ expect(result.llmContent).toContain('File: fileB.js');
+ expect(result.llmContent).toContain(
+ 'L2: function baz() { return "hello"; }',
+ );
+ expect(result.returnDisplay).toBe('Found 1 matche(s)');
+ });
+
+ it('should find matches with an include glob and path', async () => {
+ await fs.writeFile(
+ path.join(tempRootDir, 'sub', 'another.js'),
+ 'const greeting = "hello";',
+ );
+ const params: GrepToolParams = {
+ pattern: 'hello',
+ path: 'sub',
+ include: '*.js',
+ };
+ const result = await grepTool.execute(params, abortSignal);
+ expect(result.llmContent).toContain(
+ 'Found 1 match(es) for pattern "hello" in path "sub" (filter: "*.js")',
+ );
+ expect(result.llmContent).toContain('File: another.js');
+ expect(result.llmContent).toContain('L1: const greeting = "hello";');
+ expect(result.returnDisplay).toBe('Found 1 matche(s)');
+ });
+
+ it('should return "No matches found" when pattern does not exist', async () => {
+ const params: GrepToolParams = { pattern: 'nonexistentpattern' };
+ const result = await grepTool.execute(params, abortSignal);
+ expect(result.llmContent).toContain(
+ 'No matches found for pattern "nonexistentpattern" in path "."',
+ );
+ expect(result.returnDisplay).toBe('No matches found');
+ });
+
+ it('should handle regex special characters correctly', async () => {
+ const params: GrepToolParams = { pattern: 'foo.*bar' }; // Matches 'const foo = "bar";'
+ const result = await grepTool.execute(params, abortSignal);
+ expect(result.llmContent).toContain(
+ 'Found 1 match(es) for pattern "foo.*bar" in path "."',
+ );
+ expect(result.llmContent).toContain('File: fileB.js');
+ expect(result.llmContent).toContain('L1: const foo = "bar";');
+ });
+
+ it('should be case-insensitive by default (JS fallback)', async () => {
+ const params: GrepToolParams = { pattern: 'HELLO' };
+ const result = await grepTool.execute(params, abortSignal);
+ expect(result.llmContent).toContain(
+ 'Found 2 match(es) for pattern "HELLO" in path "."',
+ );
+ expect(result.llmContent).toContain('File: fileA.txt');
+ expect(result.llmContent).toContain('L1: hello world');
+ expect(result.llmContent).toContain('File: fileB.js');
+ expect(result.llmContent).toContain(
+ 'L2: function baz() { return "hello"; }',
+ );
+ });
+
+ it('should return an error if params are invalid', async () => {
+ const params = { path: '.' } as unknown as GrepToolParams; // Invalid: pattern missing
+ const result = await grepTool.execute(params, abortSignal);
+ expect(result.llmContent).toContain(
+ 'Error: Invalid parameters provided. Reason: Parameters failed schema validation',
+ );
+ expect(result.returnDisplay).toContain(
+ 'Model provided invalid parameters. Error: Parameters failed schema validation',
+ );
+ });
+ });
+
+ describe('getDescription', () => {
+ it('should generate correct description with pattern only', () => {
+ const params: GrepToolParams = { pattern: 'testPattern' };
+ expect(grepTool.getDescription(params)).toBe("'testPattern'");
+ });
+
+ it('should generate correct description with pattern and include', () => {
+ const params: GrepToolParams = {
+ pattern: 'testPattern',
+ include: '*.ts',
+ };
+ expect(grepTool.getDescription(params)).toBe("'testPattern' in *.ts");
+ });
+
+ it('should generate correct description with pattern and path', () => {
+ const params: GrepToolParams = {
+ pattern: 'testPattern',
+ path: 'src/app',
+ };
+ // The path will be relative to the tempRootDir, so we check for containment.
+ expect(grepTool.getDescription(params)).toContain("'testPattern' within");
+ expect(grepTool.getDescription(params)).toContain('src/app');
+ });
+
+ it('should generate correct description with pattern, include, and path', () => {
+ const params: GrepToolParams = {
+ pattern: 'testPattern',
+ include: '*.ts',
+ path: 'src/app',
+ };
+ expect(grepTool.getDescription(params)).toContain(
+ "'testPattern' in *.ts within",
+ );
+ expect(grepTool.getDescription(params)).toContain('src/app');
+ });
+
+ it('should use ./ for root path in description', () => {
+ const params: GrepToolParams = { pattern: 'testPattern', path: '.' };
+ expect(grepTool.getDescription(params)).toBe("'testPattern' within ./");
+ });
+ });
+});
diff --git a/packages/server/src/tools/grep.ts b/packages/server/src/tools/grep.ts
index 54391832..acdf0bc8 100644
--- a/packages/server/src/tools/grep.ts
+++ b/packages/server/src/tools/grep.ts
@@ -147,13 +147,13 @@ export class GrepTool extends BaseTool<GrepToolParams, ToolResult> {
try {
new RegExp(params.pattern);
} catch (error) {
- return `Invalid regular expression pattern provided: ${params.pattern}. Error: ${error instanceof Error ? error.message : String(error)}`;
+ return `Invalid regular expression pattern provided: ${params.pattern}. Error: ${getErrorMessage(error)}`;
}
try {
this.resolveAndValidatePath(params.path);
} catch (error) {
- return error instanceof Error ? error.message : String(error);
+ return getErrorMessage(error);
}
return null; // Parameters are valid
@@ -172,12 +172,9 @@ export class GrepTool extends BaseTool<GrepToolParams, ToolResult> {
): Promise<ToolResult> {
const validationError = this.validateToolParams(params);
if (validationError) {
- console.error(
- `GrepLogic Parameter Validation Failed: ${validationError}`,
- );
return {
llmContent: `Error: Invalid parameters provided. Reason: ${validationError}`,
- returnDisplay: `Error: Failed to execute tool.`,
+ returnDisplay: `Model provided invalid parameters. Error: ${validationError}`,
};
}
@@ -231,8 +228,7 @@ export class GrepTool extends BaseTool<GrepToolParams, ToolResult> {
};
} catch (error) {
console.error(`Error during GrepLogic execution: ${error}`);
- const errorMessage =
- error instanceof Error ? error.message : String(error);
+ const errorMessage = getErrorMessage(error);
return {
llmContent: `Error during grep search operation: ${errorMessage}`,
returnDisplay: `Error: ${errorMessage}`,
@@ -286,7 +282,7 @@ export class GrepTool extends BaseTool<GrepToolParams, ToolResult> {
return false;
} catch (error: unknown) {
if (!isNodeError(error) || error.code !== 'ENOENT') {
- console.error(
+ console.debug(
`Error checking for .git in ${currentPath}: ${error}`,
);
return false;
@@ -299,7 +295,7 @@ export class GrepTool extends BaseTool<GrepToolParams, ToolResult> {
currentPath = path.dirname(currentPath);
}
} catch (error: unknown) {
- console.error(
+ console.debug(
`Error traversing directory structure upwards from ${dirPath}: ${getErrorMessage(error)}`,
);
}
@@ -366,9 +362,13 @@ export class GrepTool extends BaseTool<GrepToolParams, ToolResult> {
description += ` in ${params.include}`;
}
if (params.path) {
- const searchDir = params.path || this.rootDirectory;
- const relativePath = makeRelative(searchDir, this.rootDirectory);
- description += ` within ${shortenPath(relativePath || './')}`;
+ const resolvedPath = path.resolve(this.rootDirectory, params.path);
+ if (resolvedPath === this.rootDirectory || params.path === '.') {
+ description += ` within ./`;
+ } else {
+ const relativePath = makeRelative(resolvedPath, this.rootDirectory);
+ description += ` within ${shortenPath(relativePath)}`;
+ }
}
return description;
}
@@ -433,7 +433,7 @@ export class GrepTool extends BaseTool<GrepToolParams, ToolResult> {
});
return this.parseGrepOutput(output, absolutePath);
} catch (gitError: unknown) {
- console.warn(
+ console.debug(
`GrepLogic: git grep failed: ${getErrorMessage(gitError)}. Falling back...`,
);
}
@@ -496,14 +496,14 @@ export class GrepTool extends BaseTool<GrepToolParams, ToolResult> {
});
return this.parseGrepOutput(output, absolutePath);
} catch (grepError: unknown) {
- console.warn(
+ console.debug(
`GrepLogic: System grep failed: ${getErrorMessage(grepError)}. Falling back...`,
);
}
}
// --- Strategy 3: Pure JavaScript Fallback ---
- console.warn(
+ console.debug(
'GrepLogic: Falling back to JavaScript grep implementation.',
);
strategyUsed = 'javascript fallback';
@@ -548,7 +548,7 @@ export class GrepTool extends BaseTool<GrepToolParams, ToolResult> {
} catch (readError: unknown) {
// Ignore errors like permission denied or file gone during read
if (!isNodeError(readError) || readError.code !== 'ENOENT') {
- console.warn(
+ console.debug(
`GrepLogic: Could not read/process ${fileAbsolutePath}: ${getErrorMessage(readError)}`,
);
}