[MCP] Support elicitation in create example tool (#28334)

Author: VeryEarly

tools/Mcp/src/CodegenServer.ts

@@ -2,11 +2,12 @@ import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { z } from "zod";
 import { responseSchema, toolParameterSchema, toolSchema } from "./types.js";
-import { toolServices } from "./services/toolServices.js";
-
+import { ToolsService } from "./services/toolsService.js";
 import { readFileSync } from "fs";
 import path from "path";
 import { fileURLToPath } from "url";
+import { RequestOptions } from "https";
+import { ElicitRequest, ElicitResult } from "@modelcontextprotocol/sdk/types.js";
 
 const __dirname = path.dirname(fileURLToPath(import.meta.url));
 const srcPath = path.resolve(__dirname, "..", "src");
@@ -37,6 +38,31 @@ export class CodegenServer {
         this.initPrompts();
     }
 
+    // dummy method for sending sampling request
+    public createMessage() {
+        return this._mcp.server.createMessage({
+            messages: [
+                {
+                    role: "user",
+                    content: {
+                        type: "text",
+                        text: `This is a test sampling request`,
+                    },
+                },
+            ],
+            maxTokens: 500
+        });
+    }
+
+    // server elicitation request
+    public elicitInput(
+        params: ElicitRequest["params"],
+        options?: RequestOptions
+    ): Promise<ElicitResult> {
+        //TODO: add log
+        return this._mcp.server.elicitInput(params, options);
+    }
+
     public static getInstance(): CodegenServer {
         if (!CodegenServer._instance) {
             CodegenServer._instance = new CodegenServer();
@@ -50,10 +76,11 @@ export class CodegenServer {
 
 
     initTools() {
+        const toolsService = ToolsService.getInstance().setServer(this);
         const toolSchemas = specs.tools as toolSchema[];
         for (const schema of toolSchemas) {
-            const parameter = this.createToolParameterfromSchema(schema.parameters);
-            const callBack = toolServices<{ [k: string]: z.ZodTypeAny }>(schema.callbackName, this._responses.get(schema.name));
+            const parameter = toolsService.createToolParameterfromSchema(schema.parameters);
+            const callBack = toolsService.getTools<{ [k: string]: z.ZodTypeAny }>(schema.callbackName, this._responses.get(schema.name));
             this._mcp.tool(
                 schema.name,
                 schema.description,
@@ -88,31 +115,4 @@ export class CodegenServer {
             this._responses.set(response.name, response.text);
         });
     }
-
-    createToolParameterfromSchema(schemas: toolParameterSchema[]){
-        const parameter: {[k: string]: z.ZodTypeAny} = {}; 
-        for (const schema of schemas) {
-            switch (schema.type) {
-                case "string":
-                    parameter[schema.name] = z.string().describe(schema.description);
-                    break;
-                case "number":
-                    parameter[schema.name] = z.number().describe(schema.description);
-                    break;
-                case "boolean": 
-                parameter[schema.name] = z.boolean().describe(schema.description);
-                    break;
-                case "array":
-                    parameter[schema.name] = z.array(z.string()).describe(schema.description);
-                    break;
-                // object parameter not supported yet    
-                // case "object":
-                //     parameter[schema.name] = z.object({}).describe(input.description); // Placeholder for object type
-                //     break;
-                default:
-                    throw new Error(`Unsupported parameter type: ${schema.type}`);
-            }
-        }
-        return parameter;
-    }
 }