Skip to content

Commit 33228e1

Browse files
committed
keep scope minial
Signed-off-by: hungran <26101787+hungran@users.noreply.github.com>
1 parent aac23eb commit 33228e1

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
@@ -141,11 +141,11 @@ func NewMCPServer(cfg MCPServerConfig) (*server.MCPServer, error) {
141141
}
142142

143143
// Register all mcp functionality with the server
144-
tsg.RegisterAllWithPrefix(ghServer, cfg.ToolPrefix)
144+
tsg.RegisterAll(ghServer, cfg.ToolPrefix)
145145

146146
if cfg.DynamicToolsets {
147147
dynamic := github.InitDynamicToolset(ghServer, tsg, cfg.Translator)
148-
dynamic.RegisterToolsWithPrefix(ghServer, cfg.ToolPrefix)
148+
dynamic.RegisterTools(ghServer, cfg.ToolPrefix)
149149
}
150150

151151
return ghServer, nil

pkg/toolsets/toolsets.go

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

106-
func (t *Toolset) RegisterTools(s *server.MCPServer) {
107-
t.RegisterToolsWithPrefix(s, "")
108-
}
109-
110-
func (t *Toolset) RegisterToolsWithPrefix(s *server.MCPServer, prefix string) {
106+
func (t *Toolset) RegisterTools(s *server.MCPServer, prefix string) {
111107
if !t.Enabled {
112108
return
113109
}
@@ -276,13 +272,9 @@ func (tg *ToolsetGroup) EnableToolset(name string) error {
276272
return nil
277273
}
278274

279-
func (tg *ToolsetGroup) RegisterAll(s *server.MCPServer) {
280-
tg.RegisterAllWithPrefix(s, "")
281-
}
282-
283-
func (tg *ToolsetGroup) RegisterAllWithPrefix(s *server.MCPServer, prefix string) {
275+
func (tg *ToolsetGroup) RegisterAll(s *server.MCPServer, prefix string) {
284276
for _, toolset := range tg.Toolsets {
285-
toolset.RegisterToolsWithPrefix(s, prefix)
277+
toolset.RegisterTools(s, prefix)
286278
toolset.RegisterResourcesTemplates(s)
287279
toolset.RegisterPrompts(s)
288280
}

0 commit comments

Comments
 (0)