[MCP Bundle] autoconfigure request and notification handlers

Author: soyuka

src/mcp-bundle/src/DependencyInjection/McpPass.php

@@ -27,13 +27,6 @@ public function process(ContainerBuilder $container): void
             return;
         }
 
-        $definition = $container->getDefinition('mcp.server.builder');
-
-        $loaderReferences = $this->findAndSortTaggedServices('mcp.loader', $container);
-        if ([] !== $loaderReferences) {
-            $definition->addMethodCall('addLoaders', $loaderReferences);
-        }
-
         $allMcpServices = [];
         $mcpTags = ['mcp.tool', 'mcp.prompt', 'mcp.resource', 'mcp.resource_template'];
 
@@ -52,6 +45,6 @@ public function process(ContainerBuilder $container): void
         }
 
         $serviceLocatorRef = ServiceLocatorTagPass::register($container, $serviceReferences);
-        $definition->addMethodCall('setContainer', [$serviceLocatorRef]);
+        $container->getDefinition('mcp.server.builder')->addMethodCall('setContainer', [$serviceLocatorRef]);
     }
 }

src/mcp-bundle/src/McpBundle.php

@@ -17,6 +17,8 @@
 use Mcp\Capability\Attribute\McpResourceTemplate;
 use Mcp\Capability\Attribute\McpTool;
 use Mcp\Capability\Registry\Loader\LoaderInterface;
+use Mcp\Server\Handler\Notification\NotificationHandlerInterface;
+use Mcp\Server\Handler\Request\RequestHandlerInterface;
 use Mcp\Server\Session\FileSessionStore;
 use Mcp\Server\Session\InMemorySessionStore;
 use Symfony\AI\McpBundle\Command\McpCommand;
@@ -62,6 +64,12 @@ public function loadExtension(array $config, ContainerConfigurator $container, C
         $builder->registerForAutoconfiguration(LoaderInterface::class)
             ->addTag('mcp.loader');
 
+        $builder->registerForAutoconfiguration(RequestHandlerInterface::class)
+            ->addTag('mcp.request_handler');
+
+        $builder->registerForAutoconfiguration(NotificationHandlerInterface::class)
+            ->addTag('mcp.notification_handler');
+
         if ($builder->getParameter('kernel.debug')) {
             $traceableRegistry = (new Definition('mcp.traceable_registry'))
                 ->setClass(TraceableRegistry::class)

src/mcp-bundle/tests/DependencyInjection/McpBundleTest.php

@@ -12,6 +12,8 @@
 namespace Symfony\AI\McpBundle\Tests\DependencyInjection;
 
 use Mcp\Capability\Registry\Loader\LoaderInterface;
+use Mcp\Server\Handler\Notification\NotificationHandlerInterface;
+use Mcp\Server\Handler\Request\RequestHandlerInterface;
 use PHPUnit\Framework\Attributes\DataProvider;
 use PHPUnit\Framework\TestCase;
 use Symfony\AI\McpBundle\McpBundle;
@@ -359,6 +361,24 @@ public function testLoaderInterfaceAutoconfiguration()
         $this->assertTrue($definition->hasTag('mcp.loader'));
     }
 
+    public function testRequestHandlerInterfaceAutoconfiguration()
+    {
+        $container = $this->buildContainer([]);
+        $autoconfigured = $container->getAutoconfiguredInstanceof();
+        $this->assertArrayHasKey(RequestHandlerInterface::class, $autoconfigured);
+        $definition = $autoconfigured[RequestHandlerInterface::class];
+        $this->assertTrue($definition->hasTag('mcp.request_handler'));
+    }
+
+    public function testNotificationHandlerInterfaceAutoconfiguration()
+    {
+        $container = $this->buildContainer([]);
+        $autoconfigured = $container->getAutoconfiguredInstanceof();
+        $this->assertArrayHasKey(NotificationHandlerInterface::class, $autoconfigured);
+        $definition = $autoconfigured[NotificationHandlerInterface::class];
+        $this->assertTrue($definition->hasTag('mcp.notification_handler'));
+    }
+
     private function buildContainer(array $configuration): ContainerBuilder
     {
         $container = new ContainerBuilder();

src/mcp-bundle/tests/DependencyInjection/McpPassTest.php

@@ -138,37 +138,4 @@ public function testHandlesPartialMcpServices()
         $this->assertInstanceOf(Reference::class, $services['tool_service']->getValues()[0]);
         $this->assertInstanceOf(Reference::class, $services['prompt_service']->getValues()[0]);
     }
-
-    public function testInjectsLoadersIntoBuilder()
-    {
-        $container = new ContainerBuilder();
-        $container->setDefinition('mcp.server.builder', new Definition());
-
-        $container->setDefinition('loader_one', (new Definition())->addTag('mcp.loader'));
-        $container->setDefinition('loader_two', (new Definition())->addTag('mcp.loader'));
-
-        $pass = new McpPass();
-        $pass->process($container);
-
-        $builderDefinition = $container->getDefinition('mcp.server.builder');
-        $methodCalls = $builderDefinition->getMethodCalls();
-
-        $addLoadersCall = null;
-        foreach ($methodCalls as $call) {
-            if ('addLoaders' === $call[0]) {
-                $addLoadersCall = $call;
-                break;
-            }
-        }
-
-        $this->assertNotNull($addLoadersCall, 'Builder should have addLoaders method call');
-
-        // Verify arguments are References
-        $args = $addLoadersCall[1];
-        $this->assertContainsOnlyInstancesOf(Reference::class, $args);
-
-        $ids = array_map(fn (Reference $ref) => (string) $ref, $args);
-        $this->assertContains('loader_one', $ids);
-        $this->assertContains('loader_two', $ids);
-    }
 }