Fix Java MCP generator prompt
- Add missing import statements for handler classes - Add handler registration in test server setup
This commit is contained in:
parent
5a493169d0
commit
6ce404ebec
@ -186,6 +186,9 @@ tasks.test {
|
|||||||
```java
|
```java
|
||||||
package com.example.mcp;
|
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.McpServer;
|
||||||
import io.mcp.server.McpServerBuilder;
|
import io.mcp.server.McpServerBuilder;
|
||||||
import io.mcp.server.transport.StdioServerTransport;
|
import io.mcp.server.transport.StdioServerTransport;
|
||||||
@ -612,10 +615,15 @@ class McpServerTest {
|
|||||||
|
|
||||||
private McpServer createTestServer() {
|
private McpServer createTestServer() {
|
||||||
// Same setup as main application
|
// Same setup as main application
|
||||||
return McpServerBuilder.builder()
|
McpServer server = McpServerBuilder.builder()
|
||||||
.serverInfo("test-server", "1.0.0")
|
.serverInfo("test-server", "1.0.0")
|
||||||
.capabilities(cap -> cap.tools(true))
|
.capabilities(cap -> cap.tools(true))
|
||||||
.build();
|
.build();
|
||||||
|
|
||||||
|
// Register handlers
|
||||||
|
ToolHandlers.register(server);
|
||||||
|
|
||||||
|
return server;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user