Fix compilation

Author: tiginamaria

kotlin-sdk-server/src/commonMain/kotlin/io/modelcontextprotocol/kotlin/sdk/server/KtorServer.kt

@@ -8,6 +8,7 @@ import io.ktor.server.response.respond
 import io.ktor.server.routing.Routing
 import io.ktor.server.routing.RoutingContext
 import io.ktor.server.routing.post
+import io.ktor.server.routing.route
 import io.ktor.server.routing.routing
 import io.ktor.server.sse.SSE
 import io.ktor.server.sse.ServerSSESession
@@ -60,7 +61,7 @@ public fun Routing.mcp(block: ServerSSESession.() -> Server) {
 
 @Suppress("FunctionName")
 @Deprecated("Use mcp() instead", ReplaceWith("mcp(block)"), DeprecationLevel.ERROR)
-public fun Application.MCP(block: () -> Server) {
+public fun Application.MCP(block: ServerSSESession.() -> Server) {
     mcp(block)
 }
 

kotlin-sdk-server/src/commonMain/kotlin/io/modelcontextprotocol/kotlin/sdk/server/Server.kt

@@ -152,8 +152,8 @@ public open class Server(
      * Registers a callback to be invoked when the server has completed initialization.
      */
     @Deprecated(
-        "Will be removed with Protocol inheritance. Use onConnect instead.",
-        ReplaceWith("onConnect"),
+        "Initialization moved to ServerSession, use ServerSession.onInitialized instead.",
+        ReplaceWith("ServerSession.onInitialized"),
         DeprecationLevel.WARNING
     )
     public fun onInitialized(block: () -> Unit) {

kotlin-sdk-test/src/commonTest/kotlin/io/modelcontextprotocol/kotlin/sdk/integration/SseIntegrationTest.kt

@@ -28,11 +28,11 @@ import kotlinx.coroutines.withTimeout
 import kotlin.test.Test
 import kotlin.test.assertTrue
 import io.ktor.client.engine.cio.CIO as ClientCIO
-import io.ktor.client.plugins.sse.SSE as ClientSSE
 import io.ktor.server.cio.CIO as ServerCIO
 import io.ktor.server.sse.SSE as ServerSSE
 
 private const val URL = "127.0.0.1"
+private const val PORT = 0
 
 class SseIntegrationTest {
     @Test
@@ -171,7 +171,7 @@ class SseIntegrationTest {
             )
         }
 
-        val ktorServer = embeddedServer(ServerCIO, host = URL, port = 0) {
+        val ktorServer = embeddedServer(ServerCIO, host = URL, port = PORT) {
             install(ServerSSE)
             routing {
                 mcp { server }