From d62dad5575a53b5bb8058509563502ca922a4fc5 Mon Sep 17 00:00:00 2001
From: "N. Taylor Mullen" <ntaylormullen@google.com>
Date: Sun, 8 Jun 2025 16:20:43 -0700
Subject: Revert "Add batch editing capabilities to Edit Tool (#648)" (#857)

---
 packages/core/src/config/config.test.ts | 1 +
 packages/core/src/config/config.ts      | 2 ++
 2 files changed, 3 insertions(+)

(limited to 'packages/core/src/config')

diff --git a/packages/core/src/config/config.test.ts b/packages/core/src/config/config.test.ts
index 1c1996bf..52b86da8 100644
--- a/packages/core/src/config/config.test.ts
+++ b/packages/core/src/config/config.test.ts
@@ -27,6 +27,7 @@ vi.mock('../tools/grep');
 vi.mock('../tools/glob');
 vi.mock('../tools/edit');
 vi.mock('../tools/shell');
+vi.mock('../tools/write-file');
 vi.mock('../tools/web-fetch');
 vi.mock('../tools/read-many-files');
 vi.mock('../tools/memoryTool', () => ({
diff --git a/packages/core/src/config/config.ts b/packages/core/src/config/config.ts
index ff0547b7..4c8fad65 100644
--- a/packages/core/src/config/config.ts
+++ b/packages/core/src/config/config.ts
@@ -18,6 +18,7 @@ import { GrepTool } from '../tools/grep.js';
 import { GlobTool } from '../tools/glob.js';
 import { EditTool } from '../tools/edit.js';
 import { ShellTool } from '../tools/shell.js';
+import { WriteFileTool } from '../tools/write-file.js';
 import { WebFetchTool } from '../tools/web-fetch.js';
 import { ReadManyFilesTool } from '../tools/read-many-files.js';
 import { MemoryTool, setGeminiMdFilename } from '../tools/memoryTool.js';
@@ -343,6 +344,7 @@ export function createToolRegistry(config: Config): Promise<ToolRegistry> {
   registerCoreTool(GrepTool, targetDir);
   registerCoreTool(GlobTool, targetDir, config);
   registerCoreTool(EditTool, config);
+  registerCoreTool(WriteFileTool, config);
   registerCoreTool(WebFetchTool, config);
   registerCoreTool(ReadManyFilesTool, targetDir, config);
   registerCoreTool(ShellTool, config);
-- 
cgit v1.2.3