feat(client): add a convenience method to set the elicitation handler

Author: ptitjes

api/kotlin-sdk.api

@@ -2872,6 +2872,7 @@ public class io/modelcontextprotocol/kotlin/sdk/client/Client : io/modelcontextp
 	public final fun removeRoot (Ljava/lang/String;)Z
 	public final fun removeRoots (Ljava/util/List;)I
 	public final fun sendRootsListChanged (Lkotlin/coroutines/Continuation;)Ljava/lang/Object;
+	public final fun setElicitationHandler (Lkotlin/jvm/functions/Function1;)V
 	public final fun setLoggingLevel (Lio/modelcontextprotocol/kotlin/sdk/LoggingLevel;Lio/modelcontextprotocol/kotlin/sdk/shared/RequestOptions;Lkotlin/coroutines/Continuation;)Ljava/lang/Object;
 	public static synthetic fun setLoggingLevel$default (Lio/modelcontextprotocol/kotlin/sdk/client/Client;Lio/modelcontextprotocol/kotlin/sdk/LoggingLevel;Lio/modelcontextprotocol/kotlin/sdk/shared/RequestOptions;Lkotlin/coroutines/Continuation;ILjava/lang/Object;)Ljava/lang/Object;
 	public final fun subscribeResource (Lio/modelcontextprotocol/kotlin/sdk/SubscribeRequest;Lio/modelcontextprotocol/kotlin/sdk/shared/RequestOptions;Lkotlin/coroutines/Continuation;)Ljava/lang/Object;

src/commonMain/kotlin/io/modelcontextprotocol/kotlin/sdk/client/Client.kt

@@ -8,6 +8,8 @@ import io.modelcontextprotocol.kotlin.sdk.ClientCapabilities
 import io.modelcontextprotocol.kotlin.sdk.CompatibilityCallToolResult
 import io.modelcontextprotocol.kotlin.sdk.CompleteRequest
 import io.modelcontextprotocol.kotlin.sdk.CompleteResult
+import io.modelcontextprotocol.kotlin.sdk.CreateElicitationRequest
+import io.modelcontextprotocol.kotlin.sdk.CreateElicitationResult
 import io.modelcontextprotocol.kotlin.sdk.EmptyRequestResult
 import io.modelcontextprotocol.kotlin.sdk.GetPromptRequest
 import io.modelcontextprotocol.kotlin.sdk.GetPromptResult
@@ -579,6 +581,24 @@ public open class Client(
         notification(RootsListChangedNotification())
     }
 
+    /**
+     * Sets the elicitation handler.
+     *
+     * @param handler The elicitation handler.
+     * @throws IllegalStateException if the client does not support elicitation.
+     */
+    public fun setElicitationHandler(handler: (CreateElicitationRequest) -> CreateElicitationResult) {
+        if (capabilities.elicitation == null) {
+            logger.error { "Failed to set elicitation handler: Client does not support elicitation" }
+            throw IllegalStateException("Client does not support elicitation.")
+        }
+        logger.info { "Setting the elicitation handler" }
+
+        setRequestHandler<CreateElicitationRequest>(Method.Defined.ElicitationCreate) { request, _ ->
+            handler(request)
+        }
+    }
+
     // --- Internal Handlers ---
 
     private suspend fun handleListRoots(): ListRootsResult {

src/jvmTest/kotlin/client/ClientTest.kt

@@ -869,7 +869,7 @@ class ClientTest {
             put("name", "octocat")
         }
 
-        client.setRequestHandler<CreateElicitationRequest>(Method.Defined.ElicitationCreate) { request, _ ->
+        client.setElicitationHandler { request ->
             assertEquals(elicitationMessage, request.message)
             assertEquals(requestedSchema, request.requestedSchema)