Add server.removeTool('name')

Author: chipgpt

src/server/mcp.test.ts

@@ -1961,6 +1961,60 @@ describe.each(zodTestMatrix)('$zodVersionLabel', (entry: ZodMatrixEntry) => {
             // Clean up spies
             warnSpy.mockRestore();
         });
+
+        /***
+         * 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();
+        });
     });
 
     describe('resource()', () => {

src/server/mcp.ts

@@ -1061,6 +1061,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.
      * @deprecated Use `registerPrompt` instead.