implement with prompt type

Signed-off-by: Rohit Patil <[email protected]>

Author: ropatil010

pkg/api/prompts.go

@@ -0,0 +1,36 @@
+package api
+
+import (
+	internalk8s "github.com/containers/kubernetes-mcp-server/pkg/kubernetes"
+)
+
+// ServerPrompt represents a prompt that can be provided to the MCP server
+type ServerPrompt struct {
+	Name        string
+	Description string
+	Arguments   []PromptArgument
+	GetMessages func(arguments map[string]string) []PromptMessage
+}
+
+// PromptArgument defines an argument that can be passed to a prompt
+type PromptArgument struct {
+	Name        string
+	Description string
+	Required    bool
+}
+
+// PromptMessage represents a message in a prompt
+type PromptMessage struct {
+	Role    string // "user" or "assistant"
+	Content string
+}
+
+// PromptSet groups related prompts together
+type PromptSet interface {
+	// GetName returns the name of the prompt set
+	GetName() string
+	// GetDescription returns a description of what this prompt set provides
+	GetDescription() string
+	// GetPrompts returns all prompts in this set
+	GetPrompts(o internalk8s.Openshift) []ServerPrompt
+}

pkg/config/config.go

@@ -31,6 +31,7 @@ type StaticConfig struct {
 	// When true, disable tools annotated with destructiveHint=true
 	DisableDestructive bool     `toml:"disable_destructive,omitempty"`
 	Toolsets           []string `toml:"toolsets,omitempty"`
+	Promptsets         []string `toml:"promptsets,omitempty"`
 	EnabledTools       []string `toml:"enabled_tools,omitempty"`
 	DisabledTools      []string `toml:"disabled_tools,omitempty"`
 

pkg/mcp/mcp.go

@@ -17,6 +17,7 @@ import (
 	"github.com/containers/kubernetes-mcp-server/pkg/config"
 	internalk8s "github.com/containers/kubernetes-mcp-server/pkg/kubernetes"
 	"github.com/containers/kubernetes-mcp-server/pkg/output"
+	"github.com/containers/kubernetes-mcp-server/pkg/promptsets"
 	"github.com/containers/kubernetes-mcp-server/pkg/toolsets"
 	"github.com/containers/kubernetes-mcp-server/pkg/version"
 )
@@ -29,6 +30,7 @@ type Configuration struct {
 	*config.StaticConfig
 	listOutput output.Output
 	toolsets   []api.Toolset
+	promptsets []api.PromptSet
 }
 
 func (c *Configuration) Toolsets() []api.Toolset {
@@ -40,6 +42,23 @@ func (c *Configuration) Toolsets() []api.Toolset {
 	return c.toolsets
 }
 
+func (c *Configuration) Promptsets() []api.PromptSet {
+	if c.promptsets == nil {
+		// Default to core if no promptsets configured
+		promptsetNames := c.StaticConfig.Promptsets
+		if len(promptsetNames) == 0 {
+			promptsetNames = []string{"core"}
+		}
+		for _, promptset := range promptsetNames {
+			ps := promptsets.PromptSetFromString(promptset)
+			if ps != nil {
+				c.promptsets = append(c.promptsets, ps)
+			}
+		}
+	}
+	return c.promptsets
+}
+
 func (c *Configuration) ListOutput() output.Output {
 	if c.listOutput == nil {
 		c.listOutput = output.FromString(c.StaticConfig.ListOutput)
@@ -148,11 +167,37 @@ func (s *Server) reloadKubernetesClusterProvider() error {
 
 	s.server.SetTools(m3labsServerTools...)
 
+	// Register prompts
+	if err := s.registerPrompts(p); err != nil {
+		klog.Warningf("Failed to register prompts: %v", err)
+		// Don't fail the whole reload if prompts fail
+	}
+
 	// start new watch
 	s.p.WatchTargets(s.reloadKubernetesClusterProvider)
 	return nil
 }
 
+// registerPrompts loads and registers all prompts with the MCP server
+func (s *Server) registerPrompts(p internalk8s.Provider) error {
+	allPrompts := make([]api.ServerPrompt, 0)
+	for _, ps := range s.configuration.Promptsets() {
+		prompts := ps.GetPrompts(p)
+		allPrompts = append(allPrompts, prompts...)
+		klog.V(5).Infof("Loaded %d prompts from promptset '%s'", len(prompts), ps.GetName())
+	}
+
+	m3labsPrompts, err := ServerPromptToM3LabsPrompt(allPrompts)
+	if err != nil {
+		return fmt.Errorf("failed to convert prompts: %v", err)
+	}
+
+	s.server.SetPrompts(m3labsPrompts...)
+	klog.V(3).Infof("Registered %d prompts", len(m3labsPrompts))
+
+	return nil
+}
+
 func (s *Server) ServeStdio() error {
 	return server.ServeStdio(s.server)
 }

pkg/mcp/mcp_test.go

@@ -23,14 +23,20 @@ func TestWatchKubeConfig(t *testing.T) {
 		// Given
 		withTimeout, cancel := context.WithTimeout(c.ctx, 5*time.Second)
 		defer cancel()
-		var notification *mcp.JSONRPCNotification
+		var toolsNotification *mcp.JSONRPCNotification
+		var promptsNotification *mcp.JSONRPCNotification
 		c.mcpClient.OnNotification(func(n mcp.JSONRPCNotification) {
-			notification = &n
+			if n.Method == "notifications/tools/list_changed" {
+				toolsNotification = &n
+			}
+			if n.Method == "notifications/prompts/list_changed" {
+				promptsNotification = &n
+			}
 		})
 		// When
 		f, _ := os.OpenFile(filepath.Join(c.tempDir, "config"), os.O_APPEND|os.O_WRONLY, 0644)
 		_, _ = f.WriteString("\n")
-		for notification == nil {
+		for toolsNotification == nil || promptsNotification == nil {
 			select {
 			case <-withTimeout.Done():
 			default:
@@ -39,11 +45,19 @@ func TestWatchKubeConfig(t *testing.T) {
 		}
 		// Then
 		t.Run("WatchKubeConfig notifies tools change", func(t *testing.T) {
-			if notification == nil {
-				t.Fatalf("WatchKubeConfig did not notify")
+			if toolsNotification == nil {
+				t.Fatalf("WatchKubeConfig did not notify tools change")
+			}
+			if toolsNotification.Method != "notifications/tools/list_changed" {
+				t.Fatalf("WatchKubeConfig did not notify tools change, got %s", toolsNotification.Method)
+			}
+		})
+		t.Run("WatchKubeConfig notifies prompts change", func(t *testing.T) {
+			if promptsNotification == nil {
+				t.Fatalf("WatchKubeConfig did not notify prompts change")
 			}
-			if notification.Method != "notifications/tools/list_changed" {
-				t.Fatalf("WatchKubeConfig did not notify tools change, got %s", notification.Method)
+			if promptsNotification.Method != "notifications/prompts/list_changed" {
+				t.Fatalf("WatchKubeConfig did not notify prompts change, got %s", promptsNotification.Method)
 			}
 		})
 	})

pkg/mcp/modules.go

@@ -1,5 +1,6 @@
 package mcp
 
+import _ "github.com/containers/kubernetes-mcp-server/pkg/promptsets/core"
 import _ "github.com/containers/kubernetes-mcp-server/pkg/toolsets/config"
 import _ "github.com/containers/kubernetes-mcp-server/pkg/toolsets/core"
 import _ "github.com/containers/kubernetes-mcp-server/pkg/toolsets/helm"

pkg/mcp/prompts.go

@@ -0,0 +1,72 @@
+package mcp
+
+import (
+	"context"
+
+	"github.com/mark3labs/mcp-go/mcp"
+	"github.com/mark3labs/mcp-go/server"
+
+	"github.com/containers/kubernetes-mcp-server/pkg/api"
+)
+
+// ServerPromptToM3LabsPrompt converts our internal ServerPrompt to mcp-go ServerPrompt format
+func ServerPromptToM3LabsPrompt(prompts []api.ServerPrompt) ([]server.ServerPrompt, error) {
+	m3labsPrompts := make([]server.ServerPrompt, 0, len(prompts))
+
+	for _, prompt := range prompts {
+		// Convert arguments
+		arguments := make([]mcp.PromptArgument, 0, len(prompt.Arguments))
+		for _, arg := range prompt.Arguments {
+			arguments = append(arguments, mcp.PromptArgument{
+				Name:        arg.Name,
+				Description: arg.Description,
+				Required:    arg.Required,
+			})
+		}
+
+		// Create the prompt handler
+		handler := createPromptHandler(prompt)
+
+		m3labsPrompts = append(m3labsPrompts, server.ServerPrompt{
+			Prompt: mcp.Prompt{
+				Name:        prompt.Name,
+				Description: prompt.Description,
+				Arguments:   arguments,
+			},
+			Handler: handler,
+		})
+	}
+
+	return m3labsPrompts, nil
+}
+
+// createPromptHandler creates a handler function for a prompt
+func createPromptHandler(prompt api.ServerPrompt) server.PromptHandlerFunc {
+	return func(ctx context.Context, request mcp.GetPromptRequest) (*mcp.GetPromptResult, error) {
+		// Get arguments from the request (already a map[string]string)
+		arguments := request.Params.Arguments
+		if arguments == nil {
+			arguments = make(map[string]string)
+		}
+
+		// Get messages from the prompt
+		promptMessages := prompt.GetMessages(arguments)
+
+		// Convert to mcp-go format
+		messages := make([]mcp.PromptMessage, 0, len(promptMessages))
+		for _, msg := range promptMessages {
+			messages = append(messages, mcp.PromptMessage{
+				Role: mcp.Role(msg.Role),
+				Content: mcp.TextContent{
+					Type: "text",
+					Text: msg.Content,
+				},
+			})
+		}
+
+		return &mcp.GetPromptResult{
+			Description: prompt.Description,
+			Messages:    messages,
+		}, nil
+	}
+}