diff options
Diffstat (limited to 'packages/core/src/config')
| -rw-r--r-- | packages/core/src/config/config.test.ts | 1 | ||||
| -rw-r--r-- | packages/core/src/config/config.ts | 2 |
2 files changed, 3 insertions, 0 deletions
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); |
