@@ -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
428367type 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
0 commit comments