Use model FQCN for indexer config, to allow any Symfony\AI\Platform\Model child class

Author: welcoMattic

src/ai-bundle/config/options.php

@@ -196,8 +196,8 @@
                         ->end()
                         ->arrayNode('model')
                             ->children()
-                                ->scalarNode('name')->isRequired()->end()
-                                ->scalarNode('version')->defaultNull()->end()
+                                ->scalarNode('class')->isRequired()->end()
+                                ->scalarNode('name')->defaultNull()->end()
                                 ->arrayNode('options')
                                     ->variablePrototype()->end()
                                 ->end()

src/ai-bundle/doc/index.rst

@@ -90,11 +90,11 @@ Configuration
                     collection: 'my_collection'
         indexer:
             default:
-                # platform: 'symfony_ai.platform.anthropic'
+                # platform: 'symfony_ai.platform.mistral'
                 # store: 'symfony_ai.store.chroma_db.default'
                 model:
-                    name: 'Embeddings'
-                    version: 'text-embedding-ada-002'
+                    class: 'Symfony\AI\Platform\Bridge\Mistral\Embeddings'
+                    name: !php/const Symfony\AI\Platform\Bridge\Mistral\Embeddings::MISTRAL_EMBED
 
 Usage
 -----

src/ai-bundle/src/AIBundle.php

@@ -35,11 +35,9 @@
 use Symfony\AI\Platform\Bridge\Meta\Llama;
 use Symfony\AI\Platform\Bridge\Mistral\Mistral;
 use Symfony\AI\Platform\Bridge\Mistral\PlatformFactory as MistralPlatformFactory;
-use Symfony\AI\Platform\Bridge\OpenAI\Embeddings;
 use Symfony\AI\Platform\Bridge\OpenAI\GPT;
 use Symfony\AI\Platform\Bridge\OpenAI\PlatformFactory as OpenAIPlatformFactory;
 use Symfony\AI\Platform\Bridge\OpenRouter\PlatformFactory as OpenRouterPlatformFactory;
-use Symfony\AI\Platform\Bridge\Voyage\Voyage;
 use Symfony\AI\Platform\Model;
 use Symfony\AI\Platform\ModelClientInterface;
 use Symfony\AI\Platform\Platform;
@@ -473,20 +471,22 @@ private function processStoreConfig(string $type, array $stores, ContainerBuilde
      */
     private function processIndexerConfig(int|string $name, array $config, ContainerBuilder $container): void
     {
-        ['name' => $modelName, 'version' => $version, 'options' => $options] = $config['model'];
+        ['class' => $modelClass, 'name' => $name, 'options' => $options] = $config['model'];
 
-        $modelClass = match (strtolower((string) $modelName)) {
-            'embeddings' => Embeddings::class,
-            'voyage' => Voyage::class,
-            default => throw new \InvalidArgumentException(\sprintf('Model "%s" is not supported.', $modelName)),
-        };
-        $modelDefinition = (new Definition($modelClass));
-        if (null !== $version) {
-            $modelDefinition->setArgument('$name', $version);
+        if (!is_a($modelClass, Model::class, true)) {
+            throw new \InvalidArgumentException(\sprintf('"%s" class is not extending Symfony\AI\Platform\Model.', $modelClass));
         }
+
+        $modelDefinition = (new Definition((string) $modelClass));
+
+        if (null !== $name) {
+            $modelDefinition->setArgument('$name', $name);
+        }
+
         if ([] !== $options) {
             $modelDefinition->setArgument('$options', $options);
         }
+
         $modelDefinition->addTag('symfony_ai.model.embeddings_model');
         $container->setDefinition('symfony_ai.indexer.'.$name.'.model', $modelDefinition);
 

src/ai-bundle/tests/AIBundleTest.php

@@ -137,10 +137,10 @@ private function getFullConfig(): array
                 'indexer' => [
                     'my_text_indexer' => [
                         'store' => 'my_azure_search_store_service_id',
-                        'platform' => 'google_platform_service_id',
+                        'platform' => 'mistral_platform_service_id',
                         'model' => [
-                            'name' => 'embeddings',
-                            'version' => 'text-embedding-004',
+                            'class' => 'Symfony\AI\Platform\Bridge\Mistral\Embeddings',
+                            'name' => 'text-embedding-004',
                             'options' => ['dimension' => 768],
                         ],
                     ],