Add server.removeTool('name')

Author: chipgpt

src/server/mcp.test.ts

@@ -528,6 +528,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

@@ -955,6 +955,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.
    */