恢复testcontroller

Author: 黄可欣

framework/fel/java/plugins/tool-mcp-test/src/main/java/modelengine/fel/tool/mcp/test/TestController.java

@@ -6,7 +6,6 @@
 
 package modelengine.fel.tool.mcp.test;
 
-import io.modelcontextprotocol.spec.McpSchema;
 import modelengine.fel.tool.mcp.client.McpClient;
 import modelengine.fel.tool.mcp.client.McpClientFactory;
 import modelengine.fel.tool.mcp.entity.Tool;
@@ -19,7 +18,6 @@
 import modelengine.fitframework.log.Logger;
 
 import java.io.IOException;
-import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 
@@ -48,12 +46,12 @@ public TestController(McpClientFactory mcpClientFactory) {
     }
 
     /**
-     * Initializes the MCP client .
-     * This method creates an instance using the provided factory and initializes it.
+     * Initializes the MCP client by creating an instance using the provided factory and initializing it.
+     * This method sets up the connection to the MCP server and prepares it for further interactions.
      *
      * @param baseUri The base URI of the MCP server.
      * @param sseEndpoint The SSE endpoint of the MCP server.
-     * @return A map with clientId and status message.
+     * @return A string indicating that the initialization was successful.
      */
     @PostMapping(path = "/initialize")
     public String initialize(@RequestQuery(name = "baseUri") String baseUri,
@@ -102,28 +100,4 @@ public List<Tool> toolsList() {
     public Object toolsCall(@RequestQuery(name = "name") String name, @RequestBody Map<String, Object> jsonArgs) {
         return this.client.callTool(name, jsonArgs);
     }
-
-    /**
-     * Custom logging consumer for MCP client.
-     *
-     * @param notification The logging message notification from MCP server.
-     */
-    private void loggingConsumer(McpSchema.LoggingMessageNotification notification) {
-        log.info("Custom logging handler received message. [level={}, data={}]",
-                notification.level(),
-                notification.data());
-    }
-
-    /**
-     * Elicitation handler for MCP client.
-     *
-     * @param request The elicitation request from MCP server.
-     * @return The elicitation result with action and user data.
-     */
-    private McpSchema.ElicitResult elicitationHandler(McpSchema.ElicitRequest request) {
-        log.info("Elicitation request received. [message={}, schema={}]", request.message(), request.requestedSchema());
-        Map<String, Object> userData = new HashMap<>();
-        userData.put("response", "Auto-accepted by test controller");
-        return new McpSchema.ElicitResult(McpSchema.ElicitResult.Action.ACCEPT, userData);
-    }
 }