Skip to content

Commit b674ad7

Browse files
add resource registration as a toolset concern
1 parent 1969f25 commit b674ad7

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

internal/ghmcp/server.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,8 +120,8 @@ func NewMCPServer(cfg MCPServerConfig) (*server.MCPServer, error) {
120120
return nil, fmt.Errorf("failed to enable toolsets: %w", err)
121121
}
122122

123-
// Register the tools with the server
124-
tsg.RegisterTools(ghServer)
123+
// Register all mcp functionality with the server
124+
tsg.RegisterAll(ghServer)
125125

126126
if cfg.DynamicToolsets {
127127
dynamic := github.InitDynamicToolset(ghServer, tsg, cfg.Translator)

pkg/toolsets/toolsets.go

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,12 +40,13 @@ func NewServerResourceTemplate(resourceTemplate mcp.ResourceTemplate, handler se
4040
}
4141
}
4242

43-
// ServerResource represents a resource that can be registered with the MCP server.
43+
// ServerResourceTemplate represents a resource template that can be registered with the MCP server.
4444
type ServerResourceTemplate struct {
4545
resourceTemplate mcp.ResourceTemplate
4646
handler server.ResourceTemplateHandlerFunc
4747
}
4848

49+
// Toolset represents a collection of MCP functionality that can be enabled or disabled as a group.
4950
type Toolset struct {
5051
Name string
5152
Description string
@@ -220,9 +221,10 @@ func (tg *ToolsetGroup) EnableToolset(name string) error {
220221
return nil
221222
}
222223

223-
func (tg *ToolsetGroup) RegisterTools(s *server.MCPServer) {
224+
func (tg *ToolsetGroup) RegisterAll(s *server.MCPServer) {
224225
for _, toolset := range tg.Toolsets {
225226
toolset.RegisterTools(s)
227+
toolset.RegisterResourcesTemplates(s)
226228
}
227229
}
228230

0 commit comments

Comments
 (0)