add mcp tool

Author: tymondesigns

config/cortex.php

@@ -211,8 +211,22 @@
         'local_http' => [
             'transport' => 'http',
             'url' => 'http://localhost:3001/sse',
-            'headers' => null,
         ],
+
+        // 'tavily' => [
+        //     'transport' => 'http',
+        //     'url' => 'https://mcp.tavily.com/mcp/?tavilyApiKey=' . env('TAVILY_API_KEY'),
+        // ],
+
+        // 'tavily' => [
+        //     'transport' => 'stdio',
+        //     'command' => 'npx',
+        //     'args' => [
+        //         '-y',
+        //         'mcp-remote',
+        //         'https://mcp.tavily.com/mcp/?tavilyApiKey=' . env('TAVILY_API_KEY'),
+        //     ],
+        // ],
     ],
 
     /*

src/LLM/Contracts/Tool.php

@@ -37,7 +37,7 @@ public function format(): array;
      *
      * @param ToolCall|array<int, mixed> $arguments
      */
-    public function invoke(ToolCall|array $arguments): mixed;
+    public function invoke(ToolCall|array $arguments = []): mixed;
 
     /**
      * Invoke the tool as a tool message.

src/Tools/ClosureTool.php

@@ -45,7 +45,7 @@ public function schema(): ObjectSchema
     /**
      * @param ToolCall|array<string, mixed> $toolCall
      */
-    public function invoke(ToolCall|array $toolCall): mixed
+    public function invoke(ToolCall|array $toolCall = []): mixed
     {
         // Get the arguments from the given tool call.
         $arguments = $this->getArguments($toolCall);

src/Tools/GoogleSerper.php

@@ -38,7 +38,7 @@ public function schema(): ObjectSchema
     /**
      * @param ToolCall|array<int, mixed> $toolCall
      */
-    public function invoke(ToolCall|array $toolCall): string
+    public function invoke(ToolCall|array $toolCall = []): string
     {
         $arguments = $this->getArguments($toolCall);
         $searchQuery = $arguments['query'];

src/Tools/McpTool.php

@@ -0,0 +1,107 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Cortex\Tools;
+
+use PhpMcp\Client\Client;
+use Cortex\Facades\McpServer;
+use Cortex\LLM\Data\ToolCall;
+use Cortex\JsonSchema\SchemaFactory;
+use Cortex\Exceptions\GenericException;
+use Cortex\JsonSchema\Types\ObjectSchema;
+use PhpMcp\Client\Model\Content\TextContent;
+use PhpMcp\Client\Contracts\ContentInterface;
+use PhpMcp\Client\Model\Definitions\ToolDefinition;
+
+class McpTool extends AbstractTool
+{
+    protected Client $client;
+
+    protected ObjectSchema $schema;
+
+    protected string $description;
+
+    public function __construct(
+        protected string $name,
+        protected ?string $mcpServer = null,
+    ) {
+        $this->client = McpServer::driver($this->mcpServer);
+        $toolDefinition = $this->getToolDefinition();
+        $this->description = $toolDefinition->description ?? '';
+        $schema = SchemaFactory::fromJson($toolDefinition->inputSchema);
+
+        if (! $schema instanceof ObjectSchema) {
+            throw new GenericException(sprintf('Schema for tool %s is not an object', $this->name));
+        }
+
+        if ($toolDefinition->description !== null) {
+            $schema->description($toolDefinition->description);
+        }
+
+        $this->schema = $schema;
+    }
+
+    public function name(): string
+    {
+        return $this->name;
+    }
+
+    public function description(): string
+    {
+        return $this->description;
+    }
+
+    public function schema(): ObjectSchema
+    {
+        return $this->schema;
+    }
+
+    /**
+     * @param ToolCall|array<string, mixed> $toolCall
+     */
+    public function invoke(ToolCall|array $toolCall = []): mixed
+    {
+        try {
+            $this->client->initialize();
+
+            // Get the arguments from the given tool call.
+            $arguments = $this->getArguments($toolCall);
+
+            // Ensure arguments are valid as per the tool's schema.
+            if ($this->schema->getPropertyKeys() !== []) {
+                $this->schema->validate($arguments);
+            }
+
+            $result = $this->client->callTool($this->name, $arguments);
+        } finally {
+            $this->client->disconnect();
+        }
+
+        // Only support text content for now.
+        $textContent = collect($result->content)
+            ->filter(fn(ContentInterface $item): bool => $item instanceof TextContent)
+            ->first();
+
+        return $textContent?->text;
+    }
+
+    protected function getToolDefinition(): ToolDefinition
+    {
+        $this->client->initialize();
+        $tools = $this->client->listTools();
+
+        $tool = collect($tools)->firstWhere('name', $this->name);
+
+        if ($tool === null) {
+            throw new GenericException(sprintf('Tool %s not found in MCP server %s', $this->name, $this->mcpServer));
+        }
+
+        return $tool;
+    }
+
+    public function __destruct()
+    {
+        $this->client->disconnect();
+    }
+}

src/Tools/SchemaTool.php

@@ -38,7 +38,7 @@ public function schema(): ObjectSchema
     /**
      * @param ToolCall|array<int, mixed> $toolCall
      */
-    public function invoke(ToolCall|array $toolCall): mixed
+    public function invoke(ToolCall|array $toolCall = []): mixed
     {
         throw new GenericException(
             'The Schema tool does not support invocation. It is only used for structured output.',

src/Tools/TavilySearch.php

@@ -38,7 +38,7 @@ public function schema(): ObjectSchema
     /**
      * @param ToolCall|array<int, mixed> $toolCall
      */
-    public function invoke(ToolCall|array $toolCall): string
+    public function invoke(ToolCall|array $toolCall = []): string
     {
         $arguments = $this->getArguments($toolCall);
         $searchQuery = $arguments['query'];

tests/Unit/Tools/McpToolTest.php

@@ -0,0 +1,53 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Cortex\Tests\Unit\Tools;
+
+use Cortex\Tools\McpTool;
+
+test('it can create a schema from an MCP tool', function (): void {
+    // $tool = new McpTool(
+    //     name: 'tavily_search',
+    //     mcpServer: 'tavily',
+    // );
+
+    $tool = new McpTool('echo');
+
+    // dump($tool->schema()->toArray());
+    // dump($tool->description());
+    $result = $tool->invoke([
+        'message' => 'foo',
+    ]);
+
+    dd($result);
+
+    // dd(json_decode($result, true));
+
+    // expect($tool->schema())->toBeInstanceOf(Schema::class);
+    // expect($tool->name())->toBe('multiply');
+    // expect($tool->description())->toBe('Multiply two numbers');
+    // expect($tool->invoke([
+    //     'a' => 2,
+    //     'b' => 2,
+    // ]))->toBe(4);
+
+    // expect($tool->format())->toBe([
+    //     'name' => 'multiply',
+    //     'description' => 'Multiply two numbers',
+    //     'parameters' => [
+    //         'type' => 'object',
+    //         'properties' => [
+    //             'a' => [
+    //                 'type' => 'integer',
+    //                 'description' => 'The first number',
+    //             ],
+    //             'b' => [
+    //                 'type' => 'integer',
+    //                 'description' => 'The second number',
+    //             ],
+    //         ],
+    //         'required' => ['a', 'b'],
+    //     ],
+    // ]);
+})->todo();