Added possibility to set custom ServerCapabilities.

Author: ineersa

examples/stdio-explicit-registration/server.php

@@ -14,6 +14,7 @@
 chdir(__DIR__);
 
 use Mcp\Example\StdioExplicitRegistration\SimpleHandlers;
+use Mcp\Schema\ServerCapabilities;
 use Mcp\Server;
 use Mcp\Server\Transport\StdioTransport;
 
@@ -27,6 +28,17 @@
     ->addResource([SimpleHandlers::class, 'getAppVersion'], 'app://version', 'application_version', mimeType: 'text/plain')
     ->addPrompt([SimpleHandlers::class, 'greetingPrompt'], 'personalized_greeting')
     ->addResourceTemplate([SimpleHandlers::class, 'getItemDetails'], 'item://{itemId}/details', 'get_item_details', mimeType: 'application/json')
+    ->setServerCapabilities(new ServerCapabilities(
+        tools: true,
+        toolsListChanged: false,
+        resources: true,
+        resourcesSubscribe: false,
+        resourcesListChanged: false,
+        prompts: true,
+        promptsListChanged: false,
+        logging: false,
+        completions: false,
+    ))
     ->build();
 
 $transport = new StdioTransport(logger: logger());

src/Capability/Registry.php

@@ -63,6 +63,8 @@ final class Registry implements ReferenceProviderInterface, ReferenceRegistryInt
      */
     private array $resourceTemplates = [];
 
+    private ServerCapabilities $serverCapabilities;
+
     public function __construct(
         private readonly ?EventDispatcherInterface $eventDispatcher = null,
         private readonly LoggerInterface $logger = new NullLogger(),
@@ -74,18 +76,11 @@ public function getCapabilities(): ServerCapabilities
         if (!$this->hasElements()) {
             $this->logger->info('No capabilities registered on server.');
         }
+        if (!isset($this->serverCapabilities)) {
+            $this->setServerCapabilities(null);
+        }
 
-        return new ServerCapabilities(
-            tools: [] !== $this->tools,
-            toolsListChanged: $this->eventDispatcher instanceof EventDispatcherInterface,
-            resources: [] !== $this->resources || [] !== $this->resourceTemplates,
-            resourcesSubscribe: false,
-            resourcesListChanged: $this->eventDispatcher instanceof EventDispatcherInterface,
-            prompts: [] !== $this->prompts,
-            promptsListChanged: $this->eventDispatcher instanceof EventDispatcherInterface,
-            logging: false,
-            completions: true,
-        );
+        return $this->serverCapabilities;
     }
 
     public function registerTool(Tool $tool, callable|array|string $handler, bool $isManual = false): void
@@ -453,4 +448,23 @@ private function paginateResults(array $items, int $limit, ?string $cursor = nul
 
         return array_values(\array_slice($items, $offset, $limit));
     }
+
+    public function setServerCapabilities(?ServerCapabilities $serverCapabilities): void
+    {
+        if ($serverCapabilities) {
+            $this->serverCapabilities = $serverCapabilities;
+        } else {
+            $this->serverCapabilities = new ServerCapabilities(
+                tools: [] !== $this->tools,
+                toolsListChanged: $this->eventDispatcher instanceof EventDispatcherInterface,
+                resources: [] !== $this->resources || [] !== $this->resourceTemplates,
+                resourcesSubscribe: false,
+                resourcesListChanged: $this->eventDispatcher instanceof EventDispatcherInterface,
+                prompts: [] !== $this->prompts,
+                promptsListChanged: $this->eventDispatcher instanceof EventDispatcherInterface,
+                logging: false,
+                completions: true,
+            );
+        }
+    }
 }

src/Server/Builder.php

@@ -35,6 +35,7 @@
 use Mcp\Schema\PromptArgument;
 use Mcp\Schema\Resource;
 use Mcp\Schema\ResourceTemplate;
+use Mcp\Schema\ServerCapabilities;
 use Mcp\Schema\Tool;
 use Mcp\Schema\ToolAnnotations;
 use Mcp\Server;
@@ -128,6 +129,8 @@ final class Builder
      */
     private array $discoveryExcludeDirs = [];
 
+    private ?ServerCapabilities $serverCapabilities = null;
+
     /**
      * Sets the server's identity. Required.
      */
@@ -237,6 +240,13 @@ public function setDiscovery(
         return $this;
     }
 
+    public function setServerCapabilities(ServerCapabilities $serverCapabilities): self
+    {
+        $this->serverCapabilities = $serverCapabilities;
+
+        return $this;
+    }
+
     /**
      * Manually registers a tool handler.
      */
@@ -318,6 +328,7 @@ public function build(): Server
         $promptGetter = $this->promptGetter ??= new PromptGetter($registry, $referenceHandler, $logger);
 
         $this->registerCapabilities($registry, $logger);
+        $registry->setServerCapabilities($this->serverCapabilities);
 
         if (null !== $this->discoveryBasePath) {
             $discovery = new Discoverer($registry, $logger);

tests/Unit/Capability/Registry/RegistryTest.php

@@ -80,6 +80,42 @@ public function testGetCapabilitiesWhenPopulated(): void
         $this->assertFalse($capabilities->logging);
     }
 
+    public function testSetCustomCapabilities(): void
+    {
+        $serverCapabilities = new ServerCapabilities(
+            tools: false,
+            toolsListChanged: true,
+            resources: false,
+            resourcesSubscribe: false,
+            resourcesListChanged: false,
+            prompts: false,
+            promptsListChanged: false,
+            logging: true,
+            completions: true,
+        );
+        $tool = $this->createValidTool('test_tool');
+        $resource = $this->createValidResource('test://resource');
+        $prompt = $this->createValidPrompt('test_prompt');
+        $template = $this->createValidResourceTemplate('test://{id}');
+
+        $this->registry->registerTool($tool, fn () => 'result');
+        $this->registry->registerResource($resource, fn () => 'content');
+        $this->registry->registerPrompt($prompt, fn () => []);
+        $this->registry->registerResourceTemplate($template, fn () => 'template');
+
+        $this->registry->setServerCapabilities($serverCapabilities);
+
+        $capabilities = $this->registry->getCapabilities();
+
+        $this->assertFalse($capabilities->tools);
+        $this->assertFalse($capabilities->resources);
+        $this->assertFalse($capabilities->prompts);
+        $this->assertTrue($capabilities->completions);
+        $this->assertFalse($capabilities->resourcesSubscribe);
+        $this->assertTrue($capabilities->logging);
+        $this->assertTrue($capabilities->toolsListChanged);
+    }
+
     public function testRegisterToolWithManualFlag(): void
     {
         $tool = $this->createValidTool('test_tool');