fix: consolidate tool config and make it work again (#56)

Author: chr-hertel

README.md

@@ -51,7 +51,7 @@ llm_chain:
             platform: 'llm_chain.platform.anthropic'
             model:
                 name: 'Claude'
-            tools: # If undefined, all tools are injected into the chain, use "[]" or "false" to have no tools.
+            tools: # If undefined, all tools are injected into the chain, use "tools: false" to disable tools.
                 - 'PhpLlm\LlmChain\Chain\ToolBox\Tool\Wikipedia'
     store:
         # also azure_search, mongodb and pinecone are supported as store type
@@ -129,6 +129,25 @@ final class CompanyName
 }
 ```
 
+The chain configuration by default will inject all known tools into the chain.
+
+To disable this behavior, set the `tools` option to `false`:
+```yaml
+llm_chain:
+    chain:
+        my_chain:
+            tools: false
+```
+
+To inject only specific tools, list them in the configuration:
+```yaml
+llm_chain:
+    chain:
+        my_chain:
+            tools:
+                - 'PhpLlm\LlmChain\Chain\ToolBox\Tool\SimilaritySearch'
+```
+
 ### Profiler
 
 The profiler panel provides insights into the chain's execution:

src/DependencyInjection/Configuration.php

@@ -65,11 +65,25 @@ public function getConfigTreeBuilder(): TreeBuilder
                             ->end()
                             ->booleanNode('structured_output')->defaultTrue()->end()
                             ->arrayNode('tools')
+                                ->addDefaultsIfNotSet()
+                                ->treatFalseLike(['enabled' => false])
+                                ->treatTrueLike(['enabled' => true])
+                                ->treatNullLike(['enabled' => true])
                                 ->beforeNormalization()
-                                    ->ifTrue(fn ($v) => false === $v)
-                                    ->then(fn () => [])
+                                    ->ifArray()
+                                    ->then(function (array $v) {
+                                        return [
+                                            'enabled' => $v['enabled'] ?? true,
+                                            'services' => $v['services'] ?? $v,
+                                        ];
+                                    })
+                                ->end()
+                                ->children()
+                                    ->booleanNode('enabled')->defaultTrue()->end()
+                                    ->arrayNode('services')
+                                        ->scalarPrototype()->end()
+                                    ->end()
                                 ->end()
-                                ->scalarPrototype()->end()
                             ->end()
                         ->end()
                     ->end()

src/DependencyInjection/LlmChainExtension.php

@@ -21,6 +21,7 @@
 use PhpLlm\LlmChain\Chain\OutputProcessor;
 use PhpLlm\LlmChain\Chain\StructuredOutput\ChainProcessor as StructureOutputProcessor;
 use PhpLlm\LlmChain\Chain\ToolBox\Attribute\AsTool;
+use PhpLlm\LlmChain\Chain\ToolBox\ChainProcessor as ToolProcessor;
 use PhpLlm\LlmChain\ChainInterface;
 use PhpLlm\LlmChain\Embedder;
 use PhpLlm\LlmChain\Model\EmbeddingsModel;
@@ -224,27 +225,33 @@ private function processChainConfig(string $name, array $config, ContainerBuilde
         $outputProcessor = [];
 
         // TOOL & PROCESSOR
-        if (0 !== count($config['tools'])) {
-            $tools = array_map(static fn (string $tool) => new Reference($tool), $config['tools']);
-            $toolboxDefinition = (new ChildDefinition('llm_chain.toolbox.abstract'))
-                ->replaceArgument('$tools', $tools);
-            $container->setDefinition('llm_chain.toolbox.'.$name, $toolboxDefinition);
-
-            if ($container->getParameter('kernel.debug')) {
-                $traceableToolboxDefinition = (new Definition('llm_chain.traceable_toolbox.'.$name))
-                    ->setClass(TraceableToolBox::class)
-                    ->setAutowired(true)
-                    ->setDecoratedService('llm_chain.toolbox.'.$name)
-                    ->addTag('llm_chain.traceable_toolbox');
-                $container->setDefinition('llm_chain.traceable_toolbox.'.$name, $traceableToolboxDefinition);
-            }
+        if ($config['tools']['enabled']) {
+            // Create specific tool box and process if tools are explicitly defined
+            if (0 !== count($config['tools']['services'])) {
+                $tools = array_map(static fn (string $tool) => new Reference($tool), $config['tools']['services']);
+                $toolboxDefinition = (new ChildDefinition('llm_chain.toolbox.abstract'))
+                    ->replaceArgument('$tools', $tools);
+                $container->setDefinition('llm_chain.toolbox.'.$name, $toolboxDefinition);
+
+                if ($container->getParameter('kernel.debug')) {
+                    $traceableToolboxDefinition = (new Definition('llm_chain.traceable_toolbox.'.$name))
+                        ->setClass(TraceableToolBox::class)
+                        ->setAutowired(true)
+                        ->setDecoratedService('llm_chain.toolbox.'.$name)
+                        ->addTag('llm_chain.traceable_toolbox');
+                    $container->setDefinition('llm_chain.traceable_toolbox.'.$name, $traceableToolboxDefinition);
+                }
 
-            $toolProcessorDefinition = (new ChildDefinition('llm_chain.tool.chain_processor.abstract'))
-                ->replaceArgument('$toolBox', new Reference('llm_chain.toolbox.'.$name));
-            $container->setDefinition('llm_chain.tool.chain_processor.'.$name, $toolProcessorDefinition);
+                $toolProcessorDefinition = (new ChildDefinition('llm_chain.tool.chain_processor.abstract'))
+                    ->replaceArgument('$toolBox', new Reference('llm_chain.toolbox.'.$name));
+                $container->setDefinition('llm_chain.tool.chain_processor.'.$name, $toolProcessorDefinition);
 
-            $inputProcessor[] = new Reference('llm_chain.tool.chain_processor.'.$name);
-            $outputProcessor[] = new Reference('llm_chain.tool.chain_processor.'.$name);
+                $inputProcessor[] = new Reference('llm_chain.tool.chain_processor.'.$name);
+                $outputProcessor[] = new Reference('llm_chain.tool.chain_processor.'.$name);
+            } else {
+                $inputProcessor[] = new Reference(ToolProcessor::class);
+                $outputProcessor[] = new Reference(ToolProcessor::class);
+            }
         }
 
         // STRUCTURED OUTPUT