ref(aibundle): improvement on tests

Author: Guikingone

src/ai-bundle/composer.json

@@ -27,6 +27,7 @@
         "symfony/string": "^7.3|^8.0"
     },
     "require-dev": {
+        "codewithkyrian/chromadb-php": "^0.2.1 || ^0.3 || ^0.4",
         "google/auth": "^1.47",
         "mongodb/mongodb": "^1.21 || ^2.0",
         "phpstan/phpstan": "^2.1",

src/ai-bundle/src/AiBundle.php

@@ -960,19 +960,23 @@ private function processStoreConfig(string $type, array $stores, ContainerBuilde
 
         if ('cache' === $type) {
             foreach ($stores as $name => $store) {
+                $distanceCalculatorDefinition = new Definition(DistanceCalculator::class);
+                $distanceCalculatorDefinition->setLazy(true);
+
+                $container->setDefinition('ai.store.distance_calculator.'.$name, $distanceCalculatorDefinition);
+
                 $arguments = [
                     new Reference($store['service']),
-                    new Definition(DistanceCalculator::class),
+                    new Reference('ai.store.distance_calculator.'.$name),
                     $store['cache_key'] ?? $name,
                 ];
 
                 if (\array_key_exists('strategy', $store) && null !== $store['strategy']) {
-                    if (!$container->hasDefinition('ai.store.distance_calculator.'.$name)) {
-                        $distanceCalculatorDefinition = new Definition(DistanceCalculator::class);
-                        $distanceCalculatorDefinition->setArgument(0, DistanceStrategy::from($store['strategy']));
+                    $distanceCalculatorDefinition = new Definition(DistanceCalculator::class);
+                    $distanceCalculatorDefinition->setLazy(true);
+                    $distanceCalculatorDefinition->setArgument(0, DistanceStrategy::from($store['strategy']));
 
-                        $container->setDefinition('ai.store.distance_calculator.'.$name, $distanceCalculatorDefinition);
-                    }
+                    $container->setDefinition('ai.store.distance_calculator.'.$name, $distanceCalculatorDefinition);
 
                     $arguments[1] = new Reference('ai.store.distance_calculator.'.$name);
                 }
@@ -1016,8 +1020,7 @@ private function processStoreConfig(string $type, array $stores, ContainerBuilde
                     $httpClient = new Definition(HttpClientInterface::class);
                     $httpClient
                         ->setFactory([HttpClient::class, 'createForBaseUri'])
-                        ->setArguments([$store['dsn']])
-                    ;
+                        ->setArguments([$store['dsn']]);
                 }
 
                 $definition = new Definition(ClickHouseStore::class);
@@ -1029,8 +1032,7 @@ private function processStoreConfig(string $type, array $stores, ContainerBuilde
                         $store['table'],
                     ])
                     ->addTag('proxy', ['interface' => StoreInterface::class])
-                    ->addTag('ai.store')
-                ;
+                    ->addTag('ai.store');
 
                 $container->setDefinition('ai.store.'.$type.'.'.$name, $definition);
                 $container->registerAliasForArgument('ai.store.'.$type.'.'.$name, StoreInterface::class, $name);