coderabit fix

Author: JBUinfo

client/elicitation_test.go

@@ -144,6 +144,23 @@ func TestClient_Initialize_WithElicitationHandler(t *testing.T) {
 t.Fatalf("expected initialize method, got %s", request.Method)
 }
 
+// Verify that elicitation capability is included in the request
+paramsBytes, err := json.Marshal(request.Params)
+if err != nil {
+t.Fatalf("failed to marshal params: %v", err)
+}
+
+var initParams struct {
+Capabilities mcp.ClientCapabilities `json:"capabilities"`
+}
+if err := json.Unmarshal(paramsBytes, &initParams); err != nil {
+t.Fatalf("failed to unmarshal params: %v", err)
+}
+
+if initParams.Capabilities.Elicitation == nil {
+t.Error("expected elicitation capability to be declared")
+}
+
 // Return successful initialization response
 result := mcp.InitializeResult{
 ProtocolVersion: mcp.LATEST_PROTOCOL_VERSION,

examples/elicitation/main.go

@@ -58,7 +58,11 @@ func demoElicitationHandler(s *server.MCPServer) server.ToolHandlerFunc {
 return nil, fmt.Errorf("unexpected response format")
 }
 
-projectName := data["projectName"].(string)
+projectName, ok := data["projectName"].(string)
+if !ok || projectName == "" {
+return nil, fmt.Errorf("invalid or missing 'projectName' in elicitation response")
+}
+
 framework := "none"
 if fw, ok := data["framework"].(string); ok {
 framework = fw
@@ -123,8 +127,7 @@ func main() {
 mcp.WithString("data", mcp.Required(), mcp.Description("Data to process")),
 ),
 func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
-data := request.GetArguments()["data"].(string)
-
+data := request.GetString("data", "")
 // Only request elicitation if data seems sensitive
 if len(data) > 100 {
 elicitationRequest := mcp.ElicitationRequest{
@@ -160,7 +163,11 @@ func main() {
 }, nil
 }
 
-responseData := result.Response.Value.(map[string]interface{})
+responseData, ok := result.Response.Value.(map[string]interface{})
+if !ok {
+responseData = make(map[string]interface{})
+}
+
 if proceed, ok := responseData["proceed"].(bool); !ok || !proceed {
 reason := "No reason provided"
 if r, ok := responseData["reason"].(string); ok && r != "" {