feat: Use class defaults if not set (#47)

Author: OskarStark

src/DependencyInjection/Configuration.php

@@ -80,7 +80,7 @@ public function getConfigTreeBuilder(): TreeBuilder
                                     ->scalarNode('api_key')->isRequired()->end()
                                     ->scalarNode('index_name')->isRequired()->end()
                                     ->scalarNode('api_version')->isRequired()->end()
-                                    ->scalarNode('vector_field')->isRequired()->end()
+                                    ->scalarNode('vector_field')->end()
                                 ->end()
                             ->end()
                         ->end()
@@ -101,8 +101,8 @@ public function getConfigTreeBuilder(): TreeBuilder
                                     ->scalarNode('database')->isRequired()->end()
                                     ->scalarNode('collection')->isRequired()->end()
                                     ->scalarNode('index_name')->isRequired()->end()
-                                    ->scalarNode('vector_field')->isRequired()->end()
-                                    ->booleanNode('bulk_write')->isRequired()->end()
+                                    ->scalarNode('vector_field')->end()
+                                    ->booleanNode('bulk_write')->end()
                                 ->end()
                             ->end()
                         ->end()
@@ -115,7 +115,7 @@ public function getConfigTreeBuilder(): TreeBuilder
                                     ->arrayNode('filter')
                                         ->scalarPrototype()->end()
                                     ->end()
-                                    ->integerNode('top_k')->defaultValue(3)->end()
+                                    ->integerNode('top_k')->end()
                                 ->end()
                             ->end()
                         ->end()

src/DependencyInjection/LlmChainExtension.php

@@ -246,17 +246,22 @@ private function processStoreConfig(string $type, array $stores, ContainerBuilde
     {
         if ('azure_search' === $type) {
             foreach ($stores as $name => $store) {
+                $arguments = [
+                    '$endpointUrl' => $store['endpoint'],
+                    '$apiKey' => $store['api_key'],
+                    '$indexName' => $store['index_name'],
+                    '$apiVersion' => $store['api_version'],
+                ];
+
+                if (array_key_exists('vector_field', $store)) {
+                    $arguments['$vectorFieldName'] = $store['vector_field'];
+                }
+
                 $definition = new Definition(AzureSearchStore::class);
                 $definition
                     ->setAutowired(true)
                     ->addTag('llm_chain.store')
-                    ->setArguments([
-                        '$endpointUrl' => $store['endpoint'],
-                        '$apiKey' => $store['api_key'],
-                        '$indexName' => $store['index_name'],
-                        '$apiVersion' => $store['api_version'],
-                        '$vectorFieldName' => $store['vector_field'],
-                    ]);
+                    ->setArguments($arguments);
 
                 $container->setDefinition('llm_chain.store.'.$type.'.'.$name, $definition);
             }
@@ -276,33 +281,49 @@ private function processStoreConfig(string $type, array $stores, ContainerBuilde
 
         if ('mongodb' === $type) {
             foreach ($stores as $name => $store) {
+                $arguments = [
+                    '$databaseName' => $store['database'],
+                    '$collectionName' => $store['collection'],
+                    '$indexName' => $store['index_name'],
+                ];
+
+                if (array_key_exists('vector_field', $store)) {
+                    $arguments['$vectorFieldName'] = $store['vector_field'];
+                }
+
+                if (array_key_exists('bulk_write', $store)) {
+                    $arguments['$bulkWrite'] = $store['bulk_write'];
+                }
+
                 $definition = new Definition(MongoDBStore::class);
                 $definition
                     ->setAutowired(true)
                     ->addTag('llm_chain.store')
-                    ->setArguments([
-                        '$databaseName' => $store['database'],
-                        '$collectionName' => $store['collection'],
-                        '$indexName' => $store['index_name'],
-                        '$vectorFieldName' => $store['vector_field'],
-                        '$bulkWrite' => $store['bulk_write'],
-                    ]);
+                    ->setArguments($arguments);
 
                 $container->setDefinition('llm_chain.store.'.$type.'.'.$name, $definition);
             }
         }
 
         if ('pinecone' === $type) {
             foreach ($stores as $name => $store) {
+                $arguments = [
+                    '$namespace' => $store['namespace'],
+                ];
+
+                if (array_key_exists('filter', $store)) {
+                    $arguments['$filter'] = $store['filter'];
+                }
+
+                if (array_key_exists('top_k', $store)) {
+                    $arguments['$topK'] = $store['top_k'];
+                }
+
                 $definition = new Definition(PineconeStore::class);
                 $definition
                     ->setAutowired(true)
                     ->addTag('llm_chain.store')
-                    ->setArguments([
-                        '$namespace' => $store['namespace'],
-                        '$filter' => $store['filter'],
-                        '$topK' => $store['top_k'],
-                    ]);
+                    ->setArguments($arguments);
 
                 $container->setDefinition('llm_chain.store.'.$type.'.'.$name, $definition);
             }