Introduce generic platform for openai embeddings- and completions-based platforms

Author: chr-hertel

examples/litellm/chat.php

@@ -9,13 +9,35 @@
  * file that was distributed with this source code.
  */
 
-use Symfony\AI\Platform\Bridge\LiteLlm\PlatformFactory;
+use Symfony\AI\Platform\Bridge\Generic\CompletionsModel;
+use Symfony\AI\Platform\Bridge\Generic\ModelCatalog;
+use Symfony\AI\Platform\Bridge\Generic\PlatformFactory;
+use Symfony\AI\Platform\Capability;
 use Symfony\AI\Platform\Message\Message;
 use Symfony\AI\Platform\Message\MessageBag;
 
 require_once dirname(__DIR__).'/bootstrap.php';
 
-$platform = PlatformFactory::create(env('LITELLM_HOST_URL'), env('LITELLM_API_KEY'), http_client());
+$modelCatalog = new ModelCatalog([
+    'mistral-small-latest' => [
+        'class' => CompletionsModel::class,
+        'capabilities' => [
+            Capability::INPUT_MESSAGES,
+            Capability::OUTPUT_TEXT,
+            Capability::OUTPUT_STREAMING,
+            Capability::OUTPUT_STRUCTURED,
+            Capability::INPUT_IMAGE,
+            Capability::TOOL_CALLING,
+        ],
+    ],
+]);
+
+$platform = PlatformFactory::create(
+    env('LITELLM_HOST_URL'),
+    env('LITELLM_API_KEY'),
+    http_client(),
+    $modelCatalog,
+);
 
 $messages = new MessageBag(
     Message::forSystem('You are a pirate and you write funny.'),

src/ai-bundle/config/options.php

@@ -111,6 +111,26 @@
                             ->end()
                         ->end()
                     ->end()
+                    ->arrayNode('generic')
+                        ->useAttributeAsKey('name')
+                        ->arrayPrototype()
+                            ->children()
+                                ->stringNode('base_url')->isRequired()->end()
+                                ->stringNode('api_key')->end()
+                                ->stringNode('http_client')
+                                    ->defaultValue('http_client')
+                                    ->info('Service ID of the HTTP client to use')
+                                ->end()
+                                ->stringNode('model_catalog')
+                                    ->info('Service ID of the model catalog to use')
+                                ->end()
+                                ->booleanNode('supports_completions')->defaultTrue()->end()
+                                ->booleanNode('supports_embeddings')->defaultTrue()->end()
+                                ->stringNode('completions_path')->defaultValue('/v1/chat/completions')->end()
+                                ->stringNode('embeddings_path')->defaultValue('/v1/embeddings')->end()
+                            ->end()
+                        ->end()
+                    ->end()
                     ->arrayNode('huggingface')
                         ->children()
                             ->stringNode('api_key')->isRequired()->end()

src/ai-bundle/src/AiBundle.php

@@ -58,6 +58,7 @@
 use Symfony\AI\Platform\Bridge\DockerModelRunner\PlatformFactory as DockerModelRunnerPlatformFactory;
 use Symfony\AI\Platform\Bridge\ElevenLabs\PlatformFactory as ElevenLabsPlatformFactory;
 use Symfony\AI\Platform\Bridge\Gemini\PlatformFactory as GeminiPlatformFactory;
+use Symfony\AI\Platform\Bridge\Generic\PlatformFactory as GenericPlatformFactory;
 use Symfony\AI\Platform\Bridge\HuggingFace\PlatformFactory as HuggingFacePlatformFactory;
 use Symfony\AI\Platform\Bridge\LmStudio\PlatformFactory as LmStudioPlatformFactory;
 use Symfony\AI\Platform\Bridge\Mistral\PlatformFactory as MistralPlatformFactory;
@@ -466,6 +467,33 @@ private function processPlatformConfig(string $type, array $platform, ContainerB
             return;
         }
 
+        if ('generic' === $type) {
+            foreach ($platform as $name => $config) {
+                $platformId = 'ai.platform.generic.'.$name;
+                $definition = (new Definition(Platform::class))
+                    ->setFactory(GenericPlatformFactory::class.'::create')
+                    ->setLazy(true)
+                    ->addTag('proxy', ['interface' => PlatformInterface::class])
+                    ->setArguments([
+                        $config['base_url'],
+                        $config['api_key'],
+                        new Reference($config['http_client'], ContainerInterface::NULL_ON_INVALID_REFERENCE),
+                        new Reference($config['model_catalog'], ContainerInterface::NULL_ON_INVALID_REFERENCE),
+                        null,
+                        new Reference('event_dispatcher'),
+                        $config['supports_completions'],
+                        $config['supports_embeddings'],
+                        $config['completions_path'],
+                        $config['embeddings_path'],
+                    ])
+                    ->addTag('ai.platform', ['name' => 'generic.'.$name]);
+
+                $container->setDefinition($platformId, $definition);
+            }
+
+            return;
+        }
+
         if ('huggingface' === $type) {
             $platformId = 'ai.platform.huggingface';
             $definition = (new Definition(Platform::class))