feat: update for 0.16 - gemini & fault tolerant toolbox (#67)

Author: chr-hertel

README.md

@@ -24,7 +24,7 @@ llm_chain:
                 name: 'GPT'
 ```
 
-### Advanced Example with Anthropic, Azure and multiple chains
+### Advanced Example with Anthropic, Azure, Google and multiple chains
 ```yaml
 # config/packages/llm_chain.yaml
 llm_chain:
@@ -38,6 +38,8 @@ llm_chain:
                 deployment: '%env(AZURE_OPENAI_GPT)%'
                 api_key: '%env(AZURE_OPENAI_KEY)%'
                 api_version: '%env(AZURE_GPT_VERSION)%'
+        google:
+            api_key: '%env(GOOGLE_API_KEY)%'
     chain:
         rag:
             platform: 'llm_chain.platform.azure.gpt_deployment'
@@ -54,6 +56,7 @@ llm_chain:
                 name: 'Claude'
             tools: # If undefined, all tools are injected into the chain, use "tools: false" to disable tools.
                 - 'PhpLlm\LlmChain\Chain\ToolBox\Tool\Wikipedia'
+            fault_tolerant_toolbox: false # Disables fault tolerant toolbox, default is true
     store:
         # also azure_search, mongodb and pinecone are supported as store type
         chroma_db:
@@ -112,6 +115,8 @@ services:
     PhpLlm\LlmChain\Chain\ToolBox\Tool\SerpApi:
         $apiKey: '%env(SERP_API_KEY)%'
     PhpLlm\LlmChain\Chain\ToolBox\Tool\SimilaritySearch: ~
+    PhpLlm\LlmChain\Chain\ToolBox\Tool\Tavily:
+      $apiKey: '%env(TAVILY_API_KEY)%'
     PhpLlm\LlmChain\Chain\ToolBox\Tool\Wikipedia: ~
     PhpLlm\LlmChain\Chain\ToolBox\Tool\YouTubeTranscriber: ~
 ```

composer.json

@@ -11,14 +11,14 @@
     ],
     "require": {
         "php": ">=8.2",
-        "php-llm/llm-chain": "^0.15",
+        "php-llm/llm-chain": "^0.16",
         "symfony/config": "^6.4 || ^7.0",
         "symfony/dependency-injection": "^6.4 || ^7.0",
         "symfony/framework-bundle": "^6.4 || ^7.0",
         "symfony/string": "^6.4 || ^7.0"
     },
     "require-dev": {
-        "php-cs-fixer/shim": "^3.68",
+        "php-cs-fixer/shim": "^3.69",
         "phpstan/phpstan": "^2.1",
         "phpunit/phpunit": "^11.5"
     },

src/DependencyInjection/Configuration.php

@@ -38,6 +38,11 @@ public function getConfigTreeBuilder(): TreeBuilder
                                 ->end()
                             ->end()
                         ->end()
+                        ->arrayNode('google')
+                            ->children()
+                                ->scalarNode('api_key')->isRequired()->end()
+                            ->end()
+                        ->end()
                         ->arrayNode('openai')
                             ->children()
                                 ->scalarNode('api_key')->isRequired()->end()
@@ -93,6 +98,7 @@ public function getConfigTreeBuilder(): TreeBuilder
                                     ->end()
                                 ->end()
                             ->end()
+                            ->booleanNode('fault_tolerant_toolbox')->defaultTrue()->end()
                         ->end()
                     ->end()
                 ->end()

src/DependencyInjection/LlmChainExtension.php

@@ -9,6 +9,8 @@
 use PhpLlm\LlmChain\Bridge\Azure\OpenAI\PlatformFactory as AzureOpenAIPlatformFactory;
 use PhpLlm\LlmChain\Bridge\Azure\Store\SearchStore as AzureSearchStore;
 use PhpLlm\LlmChain\Bridge\ChromaDB\Store as ChromaDBStore;
+use PhpLlm\LlmChain\Bridge\Google\Gemini;
+use PhpLlm\LlmChain\Bridge\Google\PlatformFactory as GooglePlatformFactory;
 use PhpLlm\LlmChain\Bridge\Meta\Llama;
 use PhpLlm\LlmChain\Bridge\MongoDB\Store as MongoDBStore;
 use PhpLlm\LlmChain\Bridge\OpenAI\Embeddings;
@@ -23,6 +25,7 @@
 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\Chain\ToolBox\FaultTolerantToolBox;
 use PhpLlm\LlmChain\ChainInterface;
 use PhpLlm\LlmChain\Embedder;
 use PhpLlm\LlmChain\Model\EmbeddingsModel;
@@ -132,16 +135,22 @@ public function load(array $configs, ContainerBuilder $container): void
      */
     private function processPlatformConfig(string $type, array $platform, ContainerBuilder $container): void
     {
-        if ('openai' === $type) {
-            $platformId = 'llm_chain.platform.openai';
+        if ('anthropic' === $type) {
+            $platformId = 'llm_chain.platform.anthropic';
             $definition = (new Definition(Platform::class))
-                ->setFactory(OpenAIPlatformFactory::class.'::create')
+                ->setFactory(AnthropicPlatformFactory::class.'::create')
                 ->setAutowired(true)
                 ->setLazy(true)
                 ->addTag('proxy', ['interface' => PlatformInterface::class])
-                ->setArguments(['$apiKey' => $platform['api_key']])
+                ->setArguments([
+                    '$apiKey' => $platform['api_key'],
+                ])
                 ->addTag('llm_chain.platform');
 
+            if (isset($platform['version'])) {
+                $definition->replaceArgument('$version', $platform['version']);
+            }
+
             $container->setDefinition($platformId, $definition);
 
             return;
@@ -169,21 +178,30 @@ private function processPlatformConfig(string $type, array $platform, ContainerB
             return;
         }
 
-        if ('anthropic' === $type) {
-            $platformId = 'llm_chain.platform.anthropic';
+        if ('google' === $type) {
+            $platformId = 'llm_chain.platform.google';
             $definition = (new Definition(Platform::class))
-                ->setFactory(AnthropicPlatformFactory::class.'::create')
+                ->setFactory(GooglePlatformFactory::class.'::create')
                 ->setAutowired(true)
                 ->setLazy(true)
                 ->addTag('proxy', ['interface' => PlatformInterface::class])
-                ->setArguments([
-                    '$apiKey' => $platform['api_key'],
-                ])
+                ->setArguments(['$apiKey' => $platform['api_key']])
                 ->addTag('llm_chain.platform');
 
-            if (isset($platform['version'])) {
-                $definition->replaceArgument('$version', $platform['version']);
-            }
+            $container->setDefinition($platformId, $definition);
+
+            return;
+        }
+
+        if ('openai' === $type) {
+            $platformId = 'llm_chain.platform.openai';
+            $definition = (new Definition(Platform::class))
+                ->setFactory(OpenAIPlatformFactory::class.'::create')
+                ->setAutowired(true)
+                ->setLazy(true)
+                ->addTag('proxy', ['interface' => PlatformInterface::class])
+                ->setArguments(['$apiKey' => $platform['api_key']])
+                ->addTag('llm_chain.platform');
 
             $container->setDefinition($platformId, $definition);
 
@@ -205,6 +223,7 @@ private function processChainConfig(string $name, array $config, ContainerBuilde
             'gpt' => GPT::class,
             'claude' => Claude::class,
             'llama' => Llama::class,
+            'gemini' => Gemini::class,
             default => throw new \InvalidArgumentException(sprintf('Model "%s" is not supported.', $modelName)),
         };
         $llmDefinition = new Definition($llmClass);
@@ -235,6 +254,14 @@ private function processChainConfig(string $name, array $config, ContainerBuilde
                     ->replaceArgument('$tools', $tools);
                 $container->setDefinition('llm_chain.toolbox.'.$name, $toolboxDefinition);
 
+                if ($config['fault_tolerant_toolbox']) {
+                    $faultTolerantToolboxDefinition = (new Definition('llm_chain.fault_tolerant_toolbox.'.$name))
+                        ->setClass(FaultTolerantToolBox::class)
+                        ->setAutowired(true)
+                        ->setDecoratedService('llm_chain.toolbox.'.$name);
+                    $container->setDefinition('llm_chain.fault_tolerant_toolbox.'.$name, $faultTolerantToolboxDefinition);
+                }
+
                 if ($container->getParameter('kernel.debug')) {
                     $traceableToolboxDefinition = (new Definition('llm_chain.traceable_toolbox.'.$name))
                         ->setClass(TraceableToolBox::class)