feat: adopt universal chain rework (#5)

Author: chr-hertel

src/Resources/config/services.php

@@ -4,7 +4,7 @@
 
 namespace Symfony\Component\DependencyInjection\Loader\Configurator;
 
-use PhpLlm\LlmChain\Chat;
+use PhpLlm\LlmChain\Chain;
 use PhpLlm\LlmChain\DocumentEmbedder;
 use PhpLlm\LlmChain\OpenAI\Model\Embeddings;
 use PhpLlm\LlmChain\OpenAI\Model\Gpt;
@@ -15,7 +15,6 @@
 use PhpLlm\LlmChain\ToolBox\Registry;
 use PhpLlm\LlmChain\ToolBox\RegistryInterface;
 use PhpLlm\LlmChain\ToolBox\ToolAnalyzer;
-use PhpLlm\LlmChain\ToolChain;
 use PhpLlm\LlmChain\Store\Azure\SearchStore as AzureSearchStore;
 use PhpLlm\LlmChain\Store\ChromaDb\Store as ChromaDbStore;
 use PhpLlm\LlmChainBundle\DataCollector;
@@ -27,9 +26,9 @@
             ->autowire()
             ->autoconfigure()
 
-        // chains
-        ->set(Chat::class)
-        ->set(ToolChain::class)
+        // high level feature
+        ->set(Chain::class)
+        ->set(DocumentEmbedder::class)
 
         // runtimes
         ->set(AzureRuntime::class)
@@ -73,9 +72,6 @@
                 '$collectionName' => abstract_arg('Name of ChromaDB collection'),
             ])
 
-        // embedder
-        ->set(DocumentEmbedder::class)
-
         // tools
         ->set(Registry::class)
             ->args([

src/Resources/views/data_collector.html.twig

@@ -76,7 +76,7 @@
                                                 {% for message in call.messages %}
                                                     <li>
                                                         <strong>{{ message.role.value|title }}:</strong>
-                                                        {{ message.hasFunctionCall ? dump(message.functionCall) : message.content }}
+                                                        {{ message.hasToolCalls ? dump(message.toolCalls) : message.content }}
                                                     </li>
                                                 {% endfor %}
                                             </ol>
@@ -114,7 +114,7 @@
                 <tbody>
                     <tr>
                         <th>Arguments</th>
-                        <td>{{ call.arguments }}</td>
+                        <td>{{ dump(call.arguments) }}</td>
                     </tr>
                     <tr>
                         <th>Response</th>

src/TraceableLanguageModel.php

@@ -6,6 +6,7 @@
 
 use PhpLlm\LlmChain\LanguageModel;
 use PhpLlm\LlmChain\Message\MessageBag;
+use PhpLlm\LlmChain\Response\Response;
 
 final class TraceableLanguageModel implements LanguageModel
 {
@@ -17,7 +18,7 @@ public function __construct(
     ) {
     }
 
-    public function call(MessageBag $messages, array $options = []): array
+    public function call(MessageBag $messages, array $options = []): Response
     {
         $response = $this->llm->call($messages, $options);
 
@@ -30,6 +31,16 @@ public function call(MessageBag $messages, array $options = []): array
         return $response;
     }
 
+    public function hasToolSupport(): bool
+    {
+        return $this->llm->hasToolSupport();
+    }
+
+    public function hasStructuredOutputSupport(): bool
+    {
+        return $this->llm->hasStructuredOutputSupport();
+    }
+
     public function getName(): string
     {
         return $this->name;

src/TraceableToolRegistry.php

@@ -4,6 +4,7 @@
 
 namespace PhpLlm\LlmChainBundle;
 
+use PhpLlm\LlmChain\Response\ToolCall;
 use PhpLlm\LlmChain\ToolBox\Registry;
 use PhpLlm\LlmChain\ToolBox\RegistryInterface;
 
@@ -21,13 +22,13 @@ public function getMap(): array
         return $this->toolRegistry->getMap();
     }
 
-    public function execute(string $name, string $arguments): string
+    public function execute(ToolCall $toolCall): string
     {
-        $response = $this->toolRegistry->execute($name, $arguments);
+        $response = $this->toolRegistry->execute($toolCall);
 
         $this->calls[] = [
-            'name' => $name,
-            'arguments' => $arguments,
+            'name' => $toolCall->name,
+            'arguments' => $toolCall->arguments,
             'response' => $response,
         ];