feat: add store config with ChromaDB

Author: chr-hertel

src/DependencyInjection/Configuration.php

@@ -47,6 +47,15 @@ public function getConfigTreeBuilder(): TreeBuilder
                         ->end()
                     ->end()
                 ->end()
+                ->arrayNode('stores')
+                    ->normalizeKeys(false)
+                    ->useAttributeAsKey('name')
+                    ->arrayPrototype()
+                        ->children()
+                            ->scalarNode('engine')->isRequired()->end()
+                            ->scalarNode('collection_name')->end()
+                        ->end()
+                    ->end()
             ->end()
         ;
 

src/DependencyInjection/LlmChainExtension.php

@@ -11,6 +11,9 @@
 use PhpLlm\LlmChain\OpenAI\Runtime;
 use PhpLlm\LlmChain\OpenAI\Runtime\Azure as AzureRuntime;
 use PhpLlm\LlmChain\OpenAI\Runtime\OpenAI as OpenAIRuntime;
+use PhpLlm\LlmChain\Store\ChromaDb\Store as ChromaDbStore;
+use PhpLlm\LlmChain\Store\StoreInterface;
+use PhpLlm\LlmChain\Store\VectorStoreInterface;
 use PhpLlm\LlmChain\ToolBox\AsTool;
 use Symfony\Component\Config\FileLocator;
 use Symfony\Component\DependencyInjection\ChildDefinition;
@@ -50,6 +53,14 @@ public function load(array $configs, ContainerBuilder $container): void
             $container->setAlias(EmbeddingModel::class, 'llm_chain.embeddings.'.$name);
         }
 
+        foreach ($config['stores'] as $name => $store) {
+            $this->processStoreConfig($name, $store, $container);
+        }
+        if (1 === count($config['stores'])) {
+            $container->setAlias(VectorStoreInterface::class, 'llm_chain.store.'.$name);
+            $container->setAlias(StoreInterface::class, 'llm_chain.store.'.$name);
+        }
+
         $container->registerAttributeForAutoconfiguration(AsTool::class, static function (ChildDefinition $definition, AsTool $attribute): void {
             $definition->addTag('llm_chain.tool', [
                 'name' => $attribute->name,
@@ -102,4 +113,14 @@ private function processEmbeddingsConfig(string $name, mixed $embeddings, Contai
 
         $container->setDefinition('llm_chain.embeddings.'.$name, $definition);
     }
+
+    private function processStoreConfig(string $name, mixed $stores, ContainerBuilder $container): void
+    {
+        if ('chroma-db' === $stores['engine']) {
+            $definition = new ChildDefinition(ChromaDbStore::class);
+            $definition->replaceArgument('$collectionName', $stores['collection_name']);
+
+            $container->setDefinition('llm_chain.store.'.$name, $definition);
+        }
+    }
 }

src/Resources/config/services.php

@@ -5,16 +5,18 @@
 namespace Symfony\Component\DependencyInjection\Loader\Configurator;
 
 use PhpLlm\LlmChain\Chat;
+use PhpLlm\LlmChain\DocumentEmbedder;
 use PhpLlm\LlmChain\OpenAI\Model\Embeddings;
 use PhpLlm\LlmChain\OpenAI\Model\Gpt;
 use PhpLlm\LlmChain\OpenAI\Runtime;
+use PhpLlm\LlmChain\OpenAI\Runtime\Azure as AzureRuntime;
+use PhpLlm\LlmChain\OpenAI\Runtime\OpenAI as OpenAIRuntime;
 use PhpLlm\LlmChain\RetrievalChain;
 use PhpLlm\LlmChain\ToolBox\ParameterAnalyzer;
 use PhpLlm\LlmChain\ToolBox\Registry;
 use PhpLlm\LlmChain\ToolBox\ToolAnalyzer;
 use PhpLlm\LlmChain\ToolChain;
-use PhpLlm\LlmChain\OpenAI\Runtime\Azure as AzureRuntime;
-use PhpLlm\LlmChain\OpenAI\Runtime\OpenAI as OpenAIRuntime;
+use PhpLlm\LlmChain\Store\ChromaDb\Store as ChromaDbStore;
 
 return static function (ContainerConfigurator $container) {
     $container->services()
@@ -54,6 +56,16 @@
                 '$runtime' => service(Runtime::class),
             ])
 
+        // stores
+        ->set(ChromaDbStore::class)
+            ->abstract()
+            ->args([
+                '$collectionName' => abstract_arg('Name of ChromaDB collection'),
+            ])
+
+        // embedder
+        ->set(DocumentEmbedder::class)
+
         // tools
         ->set(Registry::class)
             ->args([