Add server.removeTool('name')

Author: chipgpt

src/server/mcp.test.ts

@@ -505,6 +505,60 @@ describe('tool()', () => {
         ]);
     });
 
+    /***
+     * Test: Tool self removal
+     */
+    test("should remove tool when using tool.remove()", async () => {
+      const mcpServer = new McpServer({
+        name: "test server",
+        version: "1.0",
+      });
+  
+      // Register initial tool
+      const tool = mcpServer.tool("test", async () => ({
+        content: [
+          {
+            type: "text",
+            text: "Test response",
+          },
+        ],
+      }));
+  
+      expect(mcpServer['_registeredTools']['test']).toBeDefined();
+  
+      // Now delete the tool
+      tool.remove();
+  
+      expect(mcpServer['_registeredTools']['test']).toBeUndefined();
+    });
+  
+    /***
+     * Test: Tool server removal
+     */
+    test("should remove tool when using server.removeTool(...)", async () => {
+      const mcpServer = new McpServer({
+        name: "test server",
+        version: "1.0",
+      });
+  
+      // Register initial tool
+      mcpServer.tool("test", async () => ({
+        content: [
+          {
+            type: "text",
+            text: "Test response",
+          },
+        ],
+      }));
+  
+      expect(mcpServer['_registeredTools']['test']).toBeDefined();
+  
+      // Now delete the tool
+      mcpServer.removeTool("test");
+  
+      expect(mcpServer['_registeredTools']['test']).toBeUndefined();
+    });
+
     /***
      * Test: Tool Registration with Parameters
      */

src/server/mcp.ts

@@ -816,6 +816,17 @@ export class McpServer {
         );
     }
 
+    /**
+     * Removes a tool from the server by name.
+     * Does nothing if the tool is not registered.
+     */
+    removeTool(name: string) {
+        const tool = this._registeredTools[name];
+        if (tool) {
+        tool.update({ name: null });
+        }
+    };
+
     /**
      * Registers a zero-argument prompt `name`, which will run the given function when the client calls it.
      */