[AiBundle] Add more alias for argument: Indexer and Vectorizer

Author: lyrixx

src/ai-bundle/src/AiBundle.php

@@ -91,6 +91,7 @@
 use Symfony\AI\Store\Bridge\Typesense\Store as TypesenseStore;
 use Symfony\AI\Store\Bridge\Weaviate\Store as WeaviateStore;
 use Symfony\AI\Store\Document\Vectorizer;
+use Symfony\AI\Store\Document\VectorizerInterface;
 use Symfony\AI\Store\Indexer;
 use Symfony\AI\Store\IndexerInterface;
 use Symfony\AI\Store\StoreInterface;
@@ -1627,7 +1628,9 @@ private function processVectorizerConfig(string $name, array $config, ContainerB
             new Reference('logger', ContainerInterface::IGNORE_ON_INVALID_REFERENCE),
         ]);
         $vectorizerDefinition->addTag('ai.vectorizer', ['name' => $name]);
-        $container->setDefinition('ai.vectorizer.'.$name, $vectorizerDefinition);
+        $serviceId = 'ai.vectorizer.'.$name;
+        $container->setDefinition($serviceId, $vectorizerDefinition);
+        $container->registerAliasForArgument($serviceId, VectorizerInterface::class, (new Target((string) $name))->getParsedName());
     }
 
     /**
@@ -1656,7 +1659,9 @@ private function processIndexerConfig(int|string $name, array $config, Container
         ]);
         $definition->addTag('ai.indexer', ['name' => $name]);
 
-        $container->setDefinition('ai.indexer.'.$name, $definition);
+        $serviceId = 'ai.indexer.'.$name;
+        $container->setDefinition($serviceId, $definition);
+        $container->registerAliasForArgument($serviceId, IndexerInterface::class, (new Target((string) $name))->getParsedName());
     }
 
     /**

src/ai-bundle/tests/DependencyInjection/AiBundleTest.php

@@ -27,6 +27,8 @@
 use Symfony\AI\Store\Document\Loader\InMemoryLoader;
 use Symfony\AI\Store\Document\Transformer\TextTrimTransformer;
 use Symfony\AI\Store\Document\Vectorizer;
+use Symfony\AI\Store\Document\VectorizerInterface;
+use Symfony\AI\Store\IndexerInterface;
 use Symfony\AI\Store\StoreInterface;
 use Symfony\Component\Clock\ClockInterface;
 use Symfony\Component\Config\Definition\Exception\InvalidConfigurationException;
@@ -1784,6 +1786,25 @@ public function testVectorizerWithLoggerInjection()
         $this->assertSame(ContainerInterface::IGNORE_ON_INVALID_REFERENCE, $arguments[2]->getInvalidBehavior());
     }
 
+    public function testInjectionVectorizerAliasIsRegistered()
+    {
+        $container = $this->buildContainer([
+            'ai' => [
+                'vectorizer' => [
+                    'test' => [
+                        'model' => 'text-embedding-3-small?dimensions=512',
+                    ],
+                    'another' => [
+                        'model' => 'text-embedding-3-small?dimensions=512',
+                    ],
+                ],
+            ],
+        ]);
+
+        $this->assertTrue($container->hasAlias(VectorizerInterface::class.' $test'));
+        $this->assertTrue($container->hasAlias(VectorizerInterface::class.' $another'));
+    }
+
     public function testIndexerWithConfiguredVectorizer()
     {
         $container = $this->buildContainer([
@@ -2262,6 +2283,36 @@ public function testIndexerWithSourceFiltersAndTransformers()
         $this->assertSame('logger', (string) $arguments[6]);
     }
 
+    public function testInjectionIndexerAliasIsRegistered()
+    {
+        $container = $this->buildContainer([
+            'ai' => [
+                'store' => [
+                    'memory' => [
+                        'my_store' => [],
+                    ],
+                ],
+                'indexer' => [
+                    'my_indexer' => [
+                        'loader' => InMemoryLoader::class,
+                        'transformers' => [],
+                        'vectorizer' => 'my_vectorizer_service',
+                        'store' => 'ai.store.memory.my_store',
+                    ],
+                    'another' => [
+                        'loader' => InMemoryLoader::class,
+                        'transformers' => [],
+                        'vectorizer' => 'my_vectorizer_service',
+                        'store' => 'ai.store.memory.my_store',
+                    ],
+                ],
+            ],
+        ]);
+
+        $this->assertTrue($container->hasAlias(IndexerInterface::class.' $myIndexer'));
+        $this->assertTrue($container->hasAlias(IndexerInterface::class.' $another'));
+    }
+
     public function testValidMultiAgentConfiguration()
     {
         $container = $this->buildContainer([