Merge pull request #177 from mkouba/fix-list-notifications-stdio

stdio: fix list changed notifications

Author: mkouba

transports/stdio/integration-tests/src/main/java/io/quarkiverse/mcp/server/stdio/it/ServerFeatures.java

@@ -10,15 +10,26 @@
 import io.quarkiverse.mcp.server.Resource;
 import io.quarkiverse.mcp.server.TextContent;
 import io.quarkiverse.mcp.server.Tool;
+import io.quarkiverse.mcp.server.ToolManager;
+import io.quarkiverse.mcp.server.ToolResponse;
 import io.quarkus.logging.Log;
 
 public class ServerFeatures {
 
     @Inject
     CodeService codeService;
 
+    @Inject
+    ToolManager toolManager;
+
     @Tool
     TextContent toLowerCase(String value) {
+        // Just register another tool
+        toolManager.newTool("testTool")
+                .setDescription("This tool does nothing.")
+                .setHandler(
+                        ta -> ToolResponse.success("OK"))
+                .register();
         return new TextContent(value.toLowerCase());
     }
 

transports/stdio/integration-tests/src/test/java/io/quarkiverse/mcp/server/stdio/it/ServerFeaturesIT.java

@@ -149,7 +149,7 @@ public void testTool() {
             assertEquals("string", valueProperty.getString("type"));
         });
 
-        assertToolCall(
+        assertToolCallAndNotification(
                 "loop", "toLowerCase", new JsonObject()
                         .put("value", "LooP"));
     }
@@ -226,13 +226,17 @@ private void assertTool(JsonObject tool, String name, String description, Consum
         }
     }
 
-    private void assertToolCall(String expectedText, String name, JsonObject arguments) {
+    private void assertToolCallAndNotification(String expectedText, String name, JsonObject arguments) {
         JsonObject toolGetMessage = newMessage("tools/call")
                 .put("params", new JsonObject()
                         .put("name", name)
                         .put("arguments", arguments));
         sendMessage(toolGetMessage);
 
+        // Since we're using stdio we need to await the notification first
+        JsonObject notification = awaitResponse();
+        assertEquals("notifications/tools/list_changed", notification.getString("method"));
+
         JsonObject toolGetResponse = awaitResponse();
 
         JsonObject toolGetResult = assertResponseMessage(toolGetMessage, toolGetResponse);

transports/stdio/runtime/src/main/java/io/quarkiverse/mcp/server/stdio/runtime/StdioMcpMessageHandler.java

@@ -65,6 +65,7 @@ public void initialize(PrintStream stdout, McpRuntimeConfig config) {
             String connectionId = Base64.getUrlEncoder().encodeToString(UUID.randomUUID().toString().getBytes());
             StdioMcpConnection connection = new StdioMcpConnection(connectionId, config.clientLogging().defaultLevel(),
                     trafficLogger, config.autoPingInterval(), stdout, vertx);
+            connectionManager.add(connection);
             InputStream in = System.in;
             executor.submit(new Runnable() {