Merge pull request #131 from modelcontextprotocol/justin/fix-competin…

…g-completions

Fix attempted double registration of completion handlers

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@modelcontextprotocol/sdk",
-  "version": "1.3.1",
+  "version": "1.3.2",
   "description": "Model Context Protocol implementation for TypeScript",
   "license": "MIT",
   "author": "Anthropic, PBC (https://anthropic.com)",

src/server/mcp.test.ts

@@ -1283,6 +1283,69 @@ describe("prompt()", () => {
     }));
   });
 
+  test("should allow registering prompts with arguments", () => {
+    const mcpServer = new McpServer({
+      name: "test server",
+      version: "1.0",
+    });
+
+    // This should succeed
+    mcpServer.prompt(
+      "echo",
+      { message: z.string() },
+      ({ message }) => ({
+        messages: [{
+          role: "user",
+          content: {
+            type: "text",
+            text: `Please process this message: ${message}`
+          }
+        }]
+      })
+    );
+  });
+
+  test("should allow registering both resources and prompts with completion handlers", () => {
+    const mcpServer = new McpServer({
+      name: "test server",
+      version: "1.0",
+    });
+
+    // Register a resource with completion
+    mcpServer.resource(
+      "test",
+      new ResourceTemplate("test://resource/{category}", {
+        list: undefined,
+        complete: {
+          category: () => ["books", "movies", "music"],
+        },
+      }),
+      async () => ({
+        contents: [
+          {
+            uri: "test://resource/test",
+            text: "Test content",
+          },
+        ],
+      }),
+    );
+
+    // Register a prompt with completion
+    mcpServer.prompt(
+      "echo",
+      { message: completable(z.string(), () => ["hello", "world"]) },
+      ({ message }) => ({
+        messages: [{
+          role: "user",
+          content: {
+            type: "text",
+            text: `Please process this message: ${message}`
+          }
+        }]
+      })
+    );
+  });
+
   test("should throw McpError for invalid prompt name", async () => {
     const mcpServer = new McpServer({
       name: "test server",

src/server/mcp.ts

@@ -175,7 +175,13 @@ export class McpServer {
     this._toolHandlersInitialized = true;
   }
 
+  private _completionHandlerInitialized = false;
+
   private setCompletionRequestHandler() {
+    if (this._completionHandlerInitialized) {
+      return;
+    }
+
     this.server.assertCanSetRequestHandler(
       CompleteRequestSchema.shape.method.value,
     );
@@ -198,6 +204,8 @@ export class McpServer {
         }
       },
     );
+
+    this._completionHandlerInitialized = true;
   }
 
   private async handlePromptCompletion(