[McpTool] attribute also needs to be extended

Author: bigdevlarry

src/Capability/Discovery/DocBlockParser.php

@@ -177,7 +177,7 @@ public function getReturnDescription(?DocBlock $docBlock): ?string
         }
 
         $returnTag = $returnTags[0];
-        $description = method_exists($returnTag, 'getDescription') 
+        $description = method_exists($returnTag, 'getDescription')
             ? trim((string) $returnTag->getDescription())
             : '';
 

src/Server/Builder.php

@@ -269,6 +269,7 @@ public function setDiscovery(
      *
      * @param Handler                   $handler
      * @param array<string, mixed>|null $inputSchema
+     * @param array<string, mixed>|null $outputSchema
      */
     public function addTool(
         callable|array|string $handler,
@@ -434,7 +435,7 @@ private function registerCapabilities(
                 }
 
                 $inputSchema = $data['inputSchema'] ?? $schemaGenerator->generate($reflection);
-                $outputSchema = isset($data['outputSchema']) && \is_array($data['outputSchema']) ? $data['outputSchema'] : null;
+                $outputSchema = $data['outputSchema'] ?? $schemaGenerator->generateOutputSchema($reflection);
 
                 $tool = new Tool($name, $inputSchema, $description, $data['annotations'], $outputSchema);
                 $registry->registerTool($tool, $data['handler'], true);