Skip to content

Commit be6c66f

Browse files
committed
feat: Register additional tool schemas in ToolRegistry
1 parent 6e0fdd7 commit be6c66f

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/core/tools/schemas/tool-registry.ts

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,15 @@
11
import { BaseToolSchema, generateFunctionCallSchema, generateAnthropicToolSchema } from "./base-tool-schema"
2+
import { accessMcpResourceSchema } from "./access-mcp-resource-schema"
23
import { applyDiffSchema } from "./apply-diff-schema"
4+
import { codebaseSearchSchema } from "./codebase-search-schema"
5+
import { executeCommandSchema } from "./execute-command-schema"
36
import { insertContentSchema } from "./insert-content-schema"
47
import { listCodeDefinitionNamesSchema } from "./list-code-definition-names-schema"
58
import { listFilesSchema } from "./list-files-schema"
69
import { readFileSchema } from "./read-file-schema"
710
import { searchAndReplaceSchema } from "./search-and-replace-schema"
811
import { searchFilesSchema } from "./search-files-schema"
12+
import { switchModeSchema } from "./switch-mode-schema"
913
import { writeToFileSchema } from "./write-to-file-schema"
1014
import { ToolArgs } from "../../prompts/tools/types"
1115

@@ -18,13 +22,17 @@ export class ToolRegistry {
1822

1923
private constructor() {
2024
// Register supported tools
25+
this.registerTool(accessMcpResourceSchema)
2126
this.registerTool(applyDiffSchema)
27+
this.registerTool(codebaseSearchSchema)
28+
this.registerTool(executeCommandSchema)
2229
this.registerTool(insertContentSchema)
2330
this.registerTool(listCodeDefinitionNamesSchema)
2431
this.registerTool(listFilesSchema)
2532
this.registerTool(readFileSchema)
2633
this.registerTool(searchAndReplaceSchema)
2734
this.registerTool(searchFilesSchema)
35+
this.registerTool(switchModeSchema)
2836
this.registerTool(writeToFileSchema)
2937
}
3038

0 commit comments

Comments
 (0)