From 6ce404ebecf60e88831f37be30b99eb4ce50fc13 Mon Sep 17 00:00:00 2001
From: Troy Taylor <troy@troystaylor.com>
Date: Thu, 16 Oct 2025 14:54:02 -0400
Subject: [PATCH] Fix Java MCP generator prompt

- Add missing import statements for handler classes
- Add handler registration in test server setup
---
 prompts/java-mcp-server-generator.prompt.md | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/prompts/java-mcp-server-generator.prompt.md b/prompts/java-mcp-server-generator.prompt.md
index 7d8a2b2..0d2b68d 100644
--- a/prompts/java-mcp-server-generator.prompt.md
+++ b/prompts/java-mcp-server-generator.prompt.md
@@ -186,6 +186,9 @@ tasks.test {
 ```java
 package com.example.mcp;
 
+import com.example.mcp.tools.ToolHandlers;
+import com.example.mcp.resources.ResourceHandlers;
+import com.example.mcp.prompts.PromptHandlers;
 import io.mcp.server.McpServer;
 import io.mcp.server.McpServerBuilder;
 import io.mcp.server.transport.StdioServerTransport;
@@ -612,10 +615,15 @@ class McpServerTest {
     
     private McpServer createTestServer() {
         // Same setup as main application
-        return McpServerBuilder.builder()
+        McpServer server = McpServerBuilder.builder()
             .serverInfo("test-server", "1.0.0")
             .capabilities(cap -> cap.tools(true))
             .build();
+        
+        // Register handlers
+        ToolHandlers.register(server);
+        
+        return server;
     }
     
     @Test