Skip to content

Commit ed6ac0f

Browse files
yogesh-chauhanHarness
authored andcommitted
fix: [IDP-6590]: fix IDP toolset registration and add logging (#244)
* 8923c8 fix: [IDP-6590]: fix IDP toolset registration and add logging
1 parent ea6809a commit ed6ac0f

File tree

2 files changed

+17
-5
lines changed

2 files changed

+17
-5
lines changed

pkg/modules/idp.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,15 +39,15 @@ func (m *IDPModule) Name() string {
3939
// Toolsets returns the names of toolsets provided by this module
4040
func (m *IDPModule) Toolsets() []string {
4141
return []string{
42-
"Internal Developer Portal",
42+
"idp",
4343
}
4444
}
4545

4646
// RegisterToolsets registers all toolsets in the IDP module
4747
func (m *IDPModule) RegisterToolsets() error {
4848
for _, t := range m.Toolsets() {
4949
switch t {
50-
case "Internal Developer Portal":
50+
case "idp":
5151
if err := RegisterInternalDeveloperPortal(m.config, m.tsg); err != nil {
5252
return err
5353
}

pkg/toolsets/toolsets.go

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ package toolsets
22

33
import (
44
"fmt"
5+
"log/slog"
56

67
"github.com/mark3labs/mcp-go/mcp"
78
"github.com/mark3labs/mcp-go/server"
@@ -131,6 +132,7 @@ func (tg *ToolsetGroup) EnableToolsets(names []string) error {
131132
}
132133
return nil
133134
}
135+
134136
for _, name := range names {
135137
if name == "all" {
136138
tg.everythingOn = true
@@ -149,8 +151,8 @@ func (tg *ToolsetGroup) EnableToolsets(names []string) error {
149151
return err
150152
}
151153
}
152-
return nil
153154
}
155+
154156
return nil
155157
}
156158

@@ -160,8 +162,18 @@ func (tg *ToolsetGroup) EnableToolset(name string) error {
160162
if !exists {
161163
return fmt.Errorf("toolset %s does not exist", name)
162164
}
163-
toolset.Enabled = true
164-
tg.Toolsets[name] = toolset
165+
166+
// Only log if not already enabled
167+
if !toolset.Enabled {
168+
toolset.Enabled = true
169+
tg.Toolsets[name] = toolset
170+
171+
// Log each tool in this toolset
172+
for _, tool := range toolset.GetAvailableTools() {
173+
slog.Info("Tool enabled", "toolset", name, "tool", tool.Tool.Name)
174+
}
175+
}
176+
165177
return nil
166178
}
167179

0 commit comments

Comments
 (0)