Skip to content

Commit cf1dc07

Browse files
committed
keep scope minial
Signed-off-by: hungran <[email protected]>
1 parent f710c89 commit cf1dc07

File tree

2 files changed

+5
-13
lines changed

2 files changed

+5
-13
lines changed

internal/ghmcp/server.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -146,11 +146,11 @@ func NewMCPServer(cfg MCPServerConfig) (*server.MCPServer, error) {
146146
}
147147

148148
// Register all mcp functionality with the server
149-
tsg.RegisterAllWithPrefix(ghServer, cfg.ToolPrefix)
149+
tsg.RegisterAll(ghServer, cfg.ToolPrefix)
150150

151151
if cfg.DynamicToolsets {
152152
dynamic := github.InitDynamicToolset(ghServer, tsg, cfg.Translator)
153-
dynamic.RegisterToolsWithPrefix(ghServer, cfg.ToolPrefix)
153+
dynamic.RegisterTools(ghServer, cfg.ToolPrefix)
154154
}
155155

156156
return ghServer, nil

pkg/toolsets/toolsets.go

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -91,11 +91,7 @@ func (t *Toolset) GetAvailableTools() []server.ServerTool {
9191
return append(t.readTools, t.writeTools...)
9292
}
9393

94-
func (t *Toolset) RegisterTools(s *server.MCPServer) {
95-
t.RegisterToolsWithPrefix(s, "")
96-
}
97-
98-
func (t *Toolset) RegisterToolsWithPrefix(s *server.MCPServer, prefix string) {
94+
func (t *Toolset) RegisterTools(s *server.MCPServer, prefix string) {
9995
if !t.Enabled {
10096
return
10197
}
@@ -264,13 +260,9 @@ func (tg *ToolsetGroup) EnableToolset(name string) error {
264260
return nil
265261
}
266262

267-
func (tg *ToolsetGroup) RegisterAll(s *server.MCPServer) {
268-
tg.RegisterAllWithPrefix(s, "")
269-
}
270-
271-
func (tg *ToolsetGroup) RegisterAllWithPrefix(s *server.MCPServer, prefix string) {
263+
func (tg *ToolsetGroup) RegisterAll(s *server.MCPServer, prefix string) {
272264
for _, toolset := range tg.Toolsets {
273-
toolset.RegisterToolsWithPrefix(s, prefix)
265+
toolset.RegisterTools(s, prefix)
274266
toolset.RegisterResourcesTemplates(s)
275267
toolset.RegisterPrompts(s)
276268
}

0 commit comments

Comments
 (0)