Registry interface

src/Message/Message.php

@@ -22,11 +22,6 @@ public static function forSystem(string $content): self
         return new self($content, Role::System);
     }
 
-    public function isSystem(): bool
-    {
-        return Role::System === $this->role;
-    }
-
     /**
      * @param array{name: string, arguments: string} $functionCall
      */
@@ -44,4 +39,14 @@ public static function ofFunctionCall(string $name, string $content): self
     {
         return new self($content, Role::FunctionCall, null, $name);
     }
+
+    public function isSystem(): bool
+    {
+        return Role::System === $this->role;
+    }
+
+    public function hasFunctionCall(): bool
+    {
+        return null !== $this->functionCall && 0 !== count($this->functionCall);
+    }
 }

src/ToolBox/Registry.php

@@ -7,18 +7,9 @@
 use Psr\Log\LoggerInterface;
 
 /**
- * @phpstan-import-type ParameterDefinition from ParameterAnalyzer
- *
- * @phpstan-type ToolDefinition = array{
- *     type: 'function',
- *     function: array{
- *         name: string,
- *         description: string,
- *         parameters?: ParameterDefinition
- *     }
- * }
+ * @phpstan-import-type ToolDefinition from RegistryInterface
  */
-final class Registry
+final class Registry implements RegistryInterface
 {
     /**
      * @var list<object>
@@ -41,9 +32,6 @@ public function __construct(
         $this->tools = $tools instanceof \Traversable ? iterator_to_array($tools) : $tools;
     }
 
-    /**
-     * @return list<ToolDefinition>
-     */
     public function getMap(): array
     {
         if (isset($this->map)) {

src/ToolBox/RegistryInterface.php

@@ -0,0 +1,27 @@
+<?php
+
+declare(strict_types=1);
+
+namespace PhpLlm\LlmChain\ToolBox;
+
+/**
+ * @phpstan-import-type ParameterDefinition from ParameterAnalyzer
+ *
+ * @phpstan-type ToolDefinition = array{
+ *      type: 'function',
+ *      function: array{
+ *          name: string,
+ *          description: string,
+ *          parameters?: ParameterDefinition
+ *      }
+ *  }
+ */
+interface RegistryInterface
+{
+    /**
+     * @return list<ToolDefinition>
+     */
+    public function getMap(): array;
+
+    public function execute(string $name, string $arguments): string;
+}

src/ToolChain.php

@@ -6,13 +6,13 @@
 
 use PhpLlm\LlmChain\Message\Message;
 use PhpLlm\LlmChain\Message\MessageBag;
-use PhpLlm\LlmChain\ToolBox\Registry;
+use PhpLlm\LlmChain\ToolBox\RegistryInterface;
 
 final class ToolChain
 {
     public function __construct(
         private LanguageModel $llm,
-        private Registry $toolRegistry,
+        private RegistryInterface $toolRegistry,
     ) {
     }