fix(core): Improve shell tool reliability and test portability (#1036)

This commit is contained in:
Sijie Wang 2025-06-15 02:19:19 -07:00 committed by GitHub
parent 2b799cbbf0
commit 7352cb403c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 10 additions and 10 deletions

View File

@ -174,8 +174,8 @@ export class Config {
setGeminiMdFilename(params.contextFileName);
}
this.toolRegistry = createToolRegistry(this);
this.geminiClient = new GeminiClient(this);
this.toolRegistry = createToolRegistry(this);
if (this.telemetrySettings.enabled) {
initializeTelemetry(this);

View File

@ -7,6 +7,7 @@
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
import { getCoreSystemPrompt } from './prompts.js'; // Adjust import path
import * as process from 'node:process';
import * as os from 'node:os';
import { isGitRepository } from '../utils/gitUtils.js';
// Mock tool names if they are dynamically generated or complex
@ -24,6 +25,9 @@ vi.mock('../tools/shell', () => ({
vi.mock('../tools/write-file', () => ({
WriteFileTool: { Name: 'write_file' },
}));
vi.mock('node:os', () => ({
platform: vi.fn(),
}));
vi.mock('../utils/gitUtils', () => ({
isGitRepository: vi.fn(),
}));
@ -35,6 +39,7 @@ describe('Core System Prompt (prompts.ts)', () => {
beforeEach(() => {
// Store original value before each test
originalSandboxEnv = process.env.SANDBOX;
vi.mocked(os.platform).mockReturnValue('darwin');
});
afterEach(() => {

View File

@ -170,17 +170,16 @@ export class ShellTool extends BaseTool<ShellToolParams, ToolResult> {
}
const isWindows = os.platform() === 'win32';
const tempFileName = `shell_pgrep_${crypto
.randomBytes(6)
.toString('hex')}.tmp`;
const tempFilePath = path.join(os.tmpdir(), tempFileName);
// pgrep is not available on Windows, so we can't get background PIDs
const command = isWindows
? params.command
: (() => {
// wrap command to append subprocess pids (via pgrep) to temporary file
const tempFileName = `shell_pgrep_${crypto
.randomBytes(6)
.toString('hex')}.tmp`;
const tempFilePath = path.join(os.tmpdir(), tempFileName);
let command = params.command.trim();
if (!command.endsWith('&')) command += ';';
return `{ ${command} }; __code=$?; pgrep -g 0 >${tempFilePath} 2>&1; exit $__code;`;
@ -293,10 +292,6 @@ export class ShellTool extends BaseTool<ShellToolParams, ToolResult> {
// parse pids (pgrep output) from temporary file and remove it
const backgroundPIDs: number[] = [];
if (os.platform() !== 'win32') {
const tempFileName = `shell_pgrep_${crypto
.randomBytes(6)
.toString('hex')}.tmp`;
const tempFilePath = path.join(os.tmpdir(), tempFileName);
if (fs.existsSync(tempFilePath)) {
const pgrepLines = fs
.readFileSync(tempFilePath, 'utf8')