toolkit

Author: tymondesigns

src/Support/helpers.php

@@ -19,6 +19,8 @@
  * Helper function to create a chat prompt builder.
  *
  * @param MessageCollection|array<int, \Cortex\LLM\Contracts\Message>|string|null $messages
+ *
+ * @return ($messages is null ? \Cortex\Prompts\Prompt : ($messages is string ? \Cortex\Prompts\Builders\TextPromptBuilder : \Cortex\Prompts\Builders\ChatPromptBuilder))
  */
 function prompt(MessageCollection|array|string|null $messages): Prompt|PromptBuilder
 {

src/Tools/McpTool.php

@@ -24,17 +24,22 @@ class McpTool extends AbstractTool
 
     public function __construct(
         protected string $name,
-        protected ?string $mcpServer = null,
+        protected ?string $server = null,
+        protected ?ToolDefinition $toolDefinition = null,
     ) {
-        $this->client = McpServer::driver($this->mcpServer);
-        $toolDefinition = $this->getToolDefinition();
-        $this->description = $toolDefinition->description ?? '';
-        $schema = SchemaFactory::fromJson($toolDefinition->inputSchema);
+        $this->client = McpServer::driver($this->server);
+
+        // If a tool definition is provided then we don't need to connect to the MCP server yet.
+        $this->toolDefinition ??= $this->getToolDefinition();
+
+        $schema = SchemaFactory::from($this->toolDefinition->inputSchema);
 
         if (! $schema instanceof ObjectSchema) {
             throw new GenericException(sprintf('Schema for tool %s is not an object', $this->name));
         }
 
+        $this->description = $this->toolDefinition->description ?? '';
+
         if ($toolDefinition->description !== null) {
             $schema->description($toolDefinition->description);
         }
@@ -68,7 +73,7 @@ public function invoke(ToolCall|array $toolCall = []): mixed
             // Get the arguments from the given tool call.
             $arguments = $this->getArguments($toolCall);
 
-            // Ensure arguments are valid as per the tool's schema.
+            // Ensure arguments are valid as per the tool's schema (if it has properties).
             if ($this->schema->getPropertyKeys() !== []) {
                 $this->schema->validate($arguments);
             }
@@ -94,7 +99,7 @@ protected function getToolDefinition(): ToolDefinition
         $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));
+            throw new GenericException(sprintf('Tool [%s] not found in MCP server [%s]', $this->name, $this->server));
         }
 
         return $tool;

src/Tools/ToolKits/McpToolKit.php

@@ -0,0 +1,48 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Cortex\Tools\ToolKits;
+
+use Cortex\Tools\McpTool;
+use PhpMcp\Client\Client;
+use Cortex\Facades\McpServer;
+use PhpMcp\Client\Model\Definitions\ToolDefinition;
+
+class McpToolKit
+{
+    protected Client $client;
+
+    public function __construct(
+        protected ?string $server = null,
+    ) {
+        $this->client = McpServer::driver($this->server);
+    }
+
+    /**
+     * @return array<array-key, \Cortex\LLM\Contracts\Tool>
+     */
+    public function getTools(): array
+    {
+        try {
+            $this->client->initialize();
+            $tools = $this->client->listTools();
+
+            return collect($tools)
+                ->map(fn(ToolDefinition $toolDefinition): McpTool => new McpTool(
+                    $toolDefinition->name,
+                    $this->server,
+                    $toolDefinition,
+                ))
+                ->all();
+
+        } finally {
+            $this->client->disconnect();
+        }
+    }
+
+    public function __destruct()
+    {
+        $this->client->disconnect();
+    }
+}

tests/Unit/Experimental/PlaygroundTest.php

@@ -5,18 +5,11 @@
 use Cortex\Cortex;
 use Cortex\Pipeline;
 use Cortex\Facades\LLM;
-use Cortex\Facades\Image;
-use Cortex\Attributes\Tool;
-use Cortex\Chat\GenericChat;
 use Cortex\Facades\ModelInfo;
-use Cortex\Memory\ChatMemory;
-use Cortex\Facades\Embeddings;
 use Cortex\Events\ChatModelEnd;
-use Cortex\Facades\VectorStore;
 use Cortex\Tasks\Enums\TaskType;
 use Cortex\Events\ChatModelStart;
 use Cortex\JsonSchema\SchemaFactory;
-use Cortex\Memory\ChatSummaryMemory;
 use Illuminate\Support\Facades\Event;
 use Cortex\JsonSchema\Types\StringSchema;
 use Cortex\LLM\Data\Messages\UserMessage;
@@ -100,14 +93,6 @@
     dd('done');
 })->skip();
 
-test('image playground', function (): void {
-    $generator = Image::driver();
-
-    $result = $generator->invoke('A beautiful landscape with a river and mountains');
-
-    dd($result);
-})->skip();
-
 test('piping tasks with structured output', function (): void {
     Event::listen(ChatModelStart::class, function (ChatModelStart $event): void {
         dump($event->parameters);
@@ -249,58 +234,6 @@
     dd($result);
 })->skip();
 
-test('chat', function (): void {
-    $llm = LLM::provider('openai')->withModel('gpt-4o-mini');
-
-    $add = #[Tool(name: 'add', description: 'Add two numbers')]
-    function (int $a, int $b): int {
-        return $a + $b;
-    };
-
-    $chat = new GenericChat($llm, new ChatMemory(), tools: [$add]);
-
-    $result = $chat->sendMessage(new UserMessage('Hello, my name is Sean, how are you?'));
-    dump('Result 1: ' . $result->text());
-
-    $result = $chat->sendMessage(new UserMessage('What is 2 + 2?'));
-    dump('Result 2: ' . $result->text());
-
-    $result = $chat->sendMessage(new UserMessage('What is my name?'));
-    dump('Result 3: ' . $result->text());
-
-    dd($chat->getMessages());
-})->skip();
-
-test('chat summary memory', function (): void {
-    $summaryLLM = LLM::provider('ollama')->withModel('llama3.2');
-    $chatLLM = LLM::provider('ollama')->withModel('llama3.1');
-
-    $chat = new GenericChat($chatLLM, new ChatSummaryMemory($summaryLLM));
-
-    $result = $chat->sendMessage(new UserMessage('Hello, my name is Sean, how are you?'));
-    dump('Result 1: ' . $result->text());
-
-    $result = $chat->sendMessage(new UserMessage('My favourite food is pizza'));
-    dump('Result 2: ' . $result->text());
-
-    $result = $chat->sendMessage(new UserMessage('What is my favourite food?'));
-    dump('Result 3: ' . $result->text());
-})->skip();
-
-test('qdrant', function (): void {
-    $result = Embeddings::driver('ollama')->invoke([
-        "The State of the Union Address (sometimes abbreviated to SOTU) is an annual message delivered by the president of the United States to a joint session of the United States Congress near the beginning of most calendar years on the current condition of the nation.[3][4] The State of the Union Address generally includes reports on the nation's budget, economy, news, agenda, progress, achievements and the president's priorities and legislative proposals",
-        "The address fulfills the requirement in Article II, Section 3, Clause 1 of the U.S. Constitution for the president to periodically \"give to the Congress Information of the State of the Union, and recommend to their Consideration such Measures as he shall judge necessary and expedient.\" During most of the country's first century, the president primarily submitted only a written report to Congress. After 1913, Woodrow Wilson, the 28th U.S. president, began the regular practice of delivering the address to Congress in person as a way to rally support for the president's agenda, while also submitting a more detailed report.[3] With the advent of radio and television, the address is now broadcast live in all United States time zones on many networks.",
-        'The speech is generally held in January or February, and an invitation to the president is extended to use the chamber of the House by the speaker of the House. Starting in 1981, Ronald Reagan, the 40th U.S. president, began the practice of newly inaugurated presidents delivering an address to Congress in the first year of their term but not designating that speech an official "State of the Union".',
-    ]);
-    $qdrant = VectorStore::driver('qdrant');
-    $qdrant->ensureCollectionExists('test', $result->size);
-
-    // foreach ($result->embeddings as $embedding) {
-    //     $qdrant->add('test', $embedding);
-    // }
-})->skip();
-
 test('parallel group 1', function (): void {
     $dogJoke = task('tell_a_dog_joke', TaskType::Structured)
         ->system('You are a comedian.')

tests/Unit/Tools/McpToolTest.php

@@ -5,11 +5,12 @@
 namespace Cortex\Tests\Unit\Tools;
 
 use Cortex\Tools\McpTool;
+use Cortex\Tools\ToolKits\McpToolKit;
 
 test('it can create a schema from an MCP tool', function (): void {
     // $tool = new McpTool(
     //     name: 'tavily_search',
-    //     mcpServer: 'tavily',
+    //     server: 'tavily',
     // );
 
     $tool = new McpTool('echo');
@@ -51,3 +52,15 @@
     //     ],
     // ]);
 })->todo();
+
+test('it can create an MCP tool kit', function (): void {
+    $kit = new McpToolKit();
+
+    $tools = collect($kit->getTools());
+
+    dd($tools->map(fn(McpTool $tool): array => [
+        'name' => $tool->name(),
+        // 'description' => $tool->description(),
+        'schema' => $tool->schema()->toArray(),
+    ])->all());
+})->todo();