Skip to content

Commit c60428e

Browse files
committed
simpler
Signed-off-by: hungran <[email protected]>
1 parent 609907a commit c60428e

File tree

1 file changed

+2
-15
lines changed

1 file changed

+2
-15
lines changed

pkg/toolsets/toolsets.go

Lines changed: 2 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -29,18 +29,6 @@ func NewToolsetDoesNotExistError(name string) *ToolsetDoesNotExistError {
2929
return &ToolsetDoesNotExistError{Name: name}
3030
}
3131

32-
// createToolWithPrefixedName creates a new tool with the same properties as the original but with a prefixed name
33-
func createToolWithPrefixedName(original mcp.Tool, prefix string) mcp.Tool {
34-
// Create a new tool with the prefixed name and copy all properties
35-
newTool := mcp.Tool{
36-
Name: prefix + original.Name,
37-
Description: original.Description,
38-
InputSchema: original.InputSchema,
39-
Annotations: original.Annotations,
40-
}
41-
return newTool
42-
}
43-
4432
func NewServerTool(tool mcp.Tool, handler server.ToolHandlerFunc) server.ServerTool {
4533
return server.ServerTool{Tool: tool, Handler: handler}
4634
}
@@ -96,12 +84,11 @@ func (t *Toolset) RegisterTools(s *server.MCPServer, prefix string) {
9684
return
9785
}
9886
registerToolWithPrefix := func(tool server.ServerTool) {
99-
toolToRegister := tool.Tool
10087
if prefix != "" {
10188
// Create a new tool with the prefixed name
102-
toolToRegister = createToolWithPrefixedName(tool.Tool, prefix)
89+
tool.Tool.Name = prefix + tool.Tool.Name
10390
}
104-
s.AddTool(toolToRegister, tool.Handler)
91+
s.AddTool(tool.Tool, tool.Handler)
10592
}
10693

10794
for _, tool := range t.readTools {

0 commit comments

Comments
 (0)