Skip to content

Commit 7f97baf

Browse files
committed
remove disable_toolset_prompts flag and openshift flag
Signed-off-by: Nader Ziada <[email protected]>
1 parent c8bb280 commit 7f97baf

File tree

12 files changed

+19
-97
lines changed

12 files changed

+19
-97
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -311,7 +311,7 @@ role = "user"
311311
content = "Help me with {{resource_name}}"
312312
```
313313

314-
2. Toolset prompts implemented by toolset developers (can be disabled with disable_toolset_prompts = true)
314+
2. Toolset prompts implemented by toolset developers
315315

316316
See docs/PROMPTS.md for detailed documentation.
317317

docs/PROMPTS.md

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ Toolsets can provide built-in prompts by implementing the `GetPrompts()` method.
6868
### Implementing Toolset Prompts
6969

7070
```go
71-
func (t *MyToolset) GetPrompts(o internalk8s.Openshift) []api.ServerPrompt {
71+
func (t *MyToolset) GetPrompts() []api.ServerPrompt {
7272
return []api.ServerPrompt{
7373
{
7474
Prompt: api.Prompt{
@@ -104,18 +104,6 @@ func (t *MyToolset) GetPrompts(o internalk8s.Openshift) []api.ServerPrompt {
104104
}
105105
```
106106

107-
### Disabling Toolset Prompts
108-
109-
Add `disable_toolset_prompts = true` in your `config.toml` to use only config-defined prompts:
110-
111-
```toml
112-
disable_toolset_prompts = true
113-
114-
[[prompts]]
115-
name = "my-custom-prompt"
116-
# ... rest of config
117-
```
118-
119107
### Prompt Merging
120108

121109
When both toolset and config prompts exist:

pkg/api/toolsets.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ type Toolset interface {
4545
GetTools(o Openshift) []ServerTool
4646
// GetPrompts returns the prompts provided by this toolset.
4747
// Returns nil if the toolset doesn't provide any prompts.
48-
GetPrompts(o Openshift) []ServerPrompt
48+
GetPrompts() []ServerPrompt
4949
}
5050

5151
type ToolCallRequest interface {

pkg/config/config.go

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -81,9 +81,6 @@ type StaticConfig struct {
8181
promptsDefined bool // Internal: tracks if prompts were defined in config
8282
promptsMetadata toml.MetaData // Internal: metadata for prompts decoding
8383

84-
// When true, disable toolset-defined prompts (only use config-defined prompts)
85-
DisableToolsetPrompts bool `toml:"disable_toolset_prompts,omitempty"`
86-
8784
// Server instructions to be provided by the MCP server to the MCP client
8885
// This can be used to provide specific instructions on how the client should use the server
8986
ServerInstructions string `toml:"server_instructions,omitempty"`

pkg/mcp/mcp.go

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -171,17 +171,15 @@ func (s *Server) reloadToolsets() error {
171171
applicablePrompts := make([]api.ServerPrompt, 0)
172172
s.enabledPrompts = make([]string, 0)
173173

174-
// Load embedded toolset prompts (unless disabled)
175-
if !s.configuration.DisableToolsetPrompts {
176-
for _, toolset := range s.configuration.Toolsets() {
177-
prompts := toolset.GetPrompts(s.p)
178-
if prompts == nil {
179-
continue
180-
}
181-
for _, prompt := range prompts {
182-
applicablePrompts = append(applicablePrompts, prompt)
183-
s.enabledPrompts = append(s.enabledPrompts, prompt.Prompt.Name)
184-
}
174+
// Load embedded toolset prompts
175+
for _, toolset := range s.configuration.Toolsets() {
176+
prompts := toolset.GetPrompts()
177+
if prompts == nil {
178+
continue
179+
}
180+
for _, prompt := range prompts {
181+
applicablePrompts = append(applicablePrompts, prompt)
182+
s.enabledPrompts = append(s.enabledPrompts, prompt.Prompt.Name)
185183
}
186184
}
187185

pkg/mcp/mcp_toolset_prompts_test.go

Lines changed: 1 addition & 62 deletions
Original file line numberDiff line numberDiff line change
@@ -363,67 +363,6 @@ func (s *McpToolsetPromptsSuite) TestPromptsNotExposedWhenToolsetDisabled() {
363363
})
364364
}
365365

366-
func (s *McpToolsetPromptsSuite) TestDisableToolsetPromptsConfig() {
367-
testToolset := &mockToolsetWithPrompts{
368-
name: "test-toolset",
369-
description: "Test toolset",
370-
prompts: []api.ServerPrompt{
371-
{
372-
Prompt: api.Prompt{
373-
Name: "toolset-prompt",
374-
Description: "Should be disabled",
375-
},
376-
Handler: func(params api.PromptHandlerParams) (*api.PromptCallResult, error) {
377-
return api.NewPromptCallResult("Toolset", []api.PromptMessage{}, nil), nil
378-
},
379-
},
380-
},
381-
}
382-
383-
toolsets.Clear()
384-
toolsets.Register(testToolset)
385-
386-
// Add config prompt
387-
cfg, err := config.ReadToml([]byte(`
388-
toolsets = ["test-toolset"]
389-
disable_toolset_prompts = true
390-
391-
[[prompts]]
392-
name = "config-prompt"
393-
description = "Config prompt only"
394-
395-
[[prompts.messages]]
396-
role = "user"
397-
content = "From config"
398-
`))
399-
s.Require().NoError(err)
400-
// Preserve kubeconfig from SetupTest
401-
cfg.KubeConfig = s.Cfg.KubeConfig
402-
s.Cfg = cfg
403-
404-
s.InitMcpClient()
405-
406-
prompts, err := s.ListPrompts(s.T().Context(), mcp.ListPromptsRequest{})
407-
408-
s.Run("ListPrompts returns prompts", func() {
409-
s.NoError(err)
410-
s.Require().NotNil(prompts)
411-
})
412-
413-
s.Run("only config prompts are available", func() {
414-
s.Require().NotNil(prompts)
415-
s.Require().Len(prompts.Prompts, 1)
416-
s.Equal("config-prompt", prompts.Prompts[0].Name)
417-
})
418-
419-
s.Run("toolset prompts are not available when disabled", func() {
420-
s.Require().NotNil(prompts)
421-
for _, prompt := range prompts.Prompts {
422-
s.NotEqual("toolset-prompt", prompt.Name)
423-
}
424-
})
425-
}
426-
427366
// Mock toolset for testing
428367
type mockToolsetWithPrompts struct {
429368
name string
@@ -443,7 +382,7 @@ func (m *mockToolsetWithPrompts) GetTools(_ api.Openshift) []api.ServerTool {
443382
return nil
444383
}
445384

446-
func (m *mockToolsetWithPrompts) GetPrompts(_ api.Openshift) []api.ServerPrompt {
385+
func (m *mockToolsetWithPrompts) GetPrompts() []api.ServerPrompt {
447386
return m.prompts
448387
}
449388

pkg/toolsets/config/toolset.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ func (t *Toolset) GetTools(_ api.Openshift) []api.ServerTool {
2525
)
2626
}
2727

28-
func (t *Toolset) GetPrompts(_ api.Openshift) []api.ServerPrompt {
28+
func (t *Toolset) GetPrompts() []api.ServerPrompt {
2929
// Config toolset does not provide prompts
3030
return nil
3131
}

pkg/toolsets/core/toolset.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ func (t *Toolset) GetTools(o api.Openshift) []api.ServerTool {
2929
)
3030
}
3131

32-
func (t *Toolset) GetPrompts(_ api.Openshift) []api.ServerPrompt {
32+
func (t *Toolset) GetPrompts() []api.ServerPrompt {
3333
// Core toolset prompts will be added in Feature 3
3434
return nil
3535
}

pkg/toolsets/helm/toolset.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ func (t *Toolset) GetTools(_ api.Openshift) []api.ServerTool {
2525
)
2626
}
2727

28-
func (t *Toolset) GetPrompts(_ api.Openshift) []api.ServerPrompt {
28+
func (t *Toolset) GetPrompts() []api.ServerPrompt {
2929
// Helm toolset does not provide prompts
3030
return nil
3131
}

pkg/toolsets/kiali/toolset.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ func (t *Toolset) GetTools(_ api.Openshift) []api.ServerTool {
3232
)
3333
}
3434

35-
func (t *Toolset) GetPrompts(_ api.Openshift) []api.ServerPrompt {
35+
func (t *Toolset) GetPrompts() []api.ServerPrompt {
3636
// Kiali toolset does not provide prompts
3737
return nil
3838
}

0 commit comments

Comments
 (0)