check cluster health

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

Author: ropatil010

README.md

@@ -243,6 +243,11 @@ In case multi-cluster support is enabled (default) and you have access to multip
 - **events_list** - List all the Kubernetes events in the current cluster from all namespaces
   - `namespace` (`string`) - Optional Namespace to retrieve the events from. If not provided, will list events from all namespaces
 
+- **cluster_health_check** - Perform comprehensive health check on Kubernetes/OpenShift cluster and report issues. Examines cluster operators (OpenShift), nodes, deployments, pods, persistent volumes, and events to identify problems affecting cluster stability or workload availability.
+  - `check_events` (`boolean`) - Include recent warning events in the health check (may increase execution time)
+  - `output_format` (`string`) - Output format for results: 'text' (human-readable) or 'json' (machine-readable)
+  - `verbose` (`boolean`) - Enable detailed output with additional context and resource-level details
+
 - **namespaces_list** - List all the Kubernetes namespaces in the current cluster
 
 - **projects_list** - List all the OpenShift projects in the current cluster

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

@@ -9,12 +9,14 @@ import (
 
 	"github.com/modelcontextprotocol/go-sdk/mcp"
 	authenticationapiv1 "k8s.io/api/authentication/v1"
+	"k8s.io/klog/v2"
 	"k8s.io/utils/ptr"
 
 	"github.com/containers/kubernetes-mcp-server/pkg/api"
 	"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"
 )
@@ -27,6 +29,7 @@ type Configuration struct {
 	*config.StaticConfig
 	listOutput output.Output
 	toolsets   []api.Toolset
+	promptsets []api.PromptSet
 }
 
 func (c *Configuration) Toolsets() []api.Toolset {
@@ -38,6 +41,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)
@@ -77,7 +97,7 @@ func NewServer(configuration Configuration) (*Server, error) {
 			},
 			&mcp.ServerOptions{
 				HasResources: false,
-				HasPrompts:   false,
+				HasPrompts:   true,
 				HasTools:     true,
 			}),
 	}
@@ -165,11 +185,42 @@ func (s *Server) reloadKubernetesClusterProvider() error {
 		s.server.AddTool(goSdkTool, goSdkToolHandler)
 	}
 
+	// 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())
+	}
+
+	goSdkPrompts, goSdkHandlers, err := ServerPromptToGoSdkPrompt(s, allPrompts)
+	if err != nil {
+		return fmt.Errorf("failed to convert prompts: %v", err)
+	}
+
+	// Register each prompt with its handler
+	for name, prompt := range goSdkPrompts {
+		handler := goSdkHandlers[name]
+		s.server.AddPrompt(prompt, handler)
+	}
+
+	klog.V(3).Infof("Registered %d prompts", len(goSdkPrompts))
+
+	return nil
+}
+
 func (s *Server) ServeStdio(ctx context.Context) error {
 	return s.server.Run(ctx, &mcp.LoggingTransport{Transport: &mcp.StdioTransport{}, Writer: os.Stderr})
 }

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,75 @@
+package mcp
+
+import (
+	"context"
+
+	"github.com/modelcontextprotocol/go-sdk/mcp"
+
+	"github.com/containers/kubernetes-mcp-server/pkg/api"
+)
+
+// ServerPromptToGoSdkPrompt converts our internal ServerPrompt to go-sdk Prompt format
+func ServerPromptToGoSdkPrompt(s *Server, prompts []api.ServerPrompt) (map[string]*mcp.Prompt, map[string]mcp.PromptHandler, error) {
+	goSdkPrompts := make(map[string]*mcp.Prompt)
+	goSdkHandlers := make(map[string]mcp.PromptHandler)
+
+	for _, prompt := range prompts {
+		// Convert arguments to PromptArgument pointers
+		var arguments []*mcp.PromptArgument
+		for _, arg := range prompt.Arguments {
+			arguments = append(arguments, &mcp.PromptArgument{
+				Name:        arg.Name,
+				Description: arg.Description,
+				Required:    arg.Required,
+			})
+		}
+
+		goSdkPrompt := &mcp.Prompt{
+			Name:        prompt.Name,
+			Description: prompt.Description,
+			Arguments:   arguments,
+		}
+
+		// Create the prompt handler
+		handler := createPromptHandler(s, prompt)
+
+		goSdkPrompts[prompt.Name] = goSdkPrompt
+		goSdkHandlers[prompt.Name] = handler
+	}
+
+	return goSdkPrompts, goSdkHandlers, nil
+}
+
+// createPromptHandler creates a handler function for a prompt
+func createPromptHandler(s *Server, prompt api.ServerPrompt) mcp.PromptHandler {
+	return func(ctx context.Context, request *mcp.GetPromptRequest) (*mcp.GetPromptResult, error) {
+		// Get arguments from the request
+		params := request.GetParams()
+		arguments := make(map[string]string)
+		if params != nil {
+			// Cast to concrete type to access Arguments field
+			if getPromptParams, ok := params.(*mcp.GetPromptParams); ok && getPromptParams.Arguments != nil {
+				arguments = getPromptParams.Arguments
+			}
+		}
+
+		// Get messages from the prompt
+		promptMessages := prompt.GetMessages(arguments)
+
+		// Convert to mcp-go format - need to use pointers
+		messages := make([]*mcp.PromptMessage, 0, len(promptMessages))
+		for _, msg := range promptMessages {
+			messages = append(messages, &mcp.PromptMessage{
+				Role: mcp.Role(msg.Role),
+				Content: &mcp.TextContent{
+					Text: msg.Content,
+				},
+			})
+		}
+
+		return &mcp.GetPromptResult{
+			Description: prompt.Description,
+			Messages:    messages,
+		}, nil
+	}
+}