From 27d08b61d3182d99c2daee838f9cc7937a22b1ad Mon Sep 17 00:00:00 2001
From: Mathieu Santostefano <msantostefano@proton.me>
Date: Mon, 16 Jun 2025 17:23:38 +0200
Subject: [PATCH] feat(config): Add support for Mistral in configuration

---
 src/DependencyInjection/Configuration.php     |  5 +++++
 src/DependencyInjection/LlmChainExtension.php | 18 ++++++++++++++++++
 2 files changed, 23 insertions(+)

diff --git a/src/DependencyInjection/Configuration.php b/src/DependencyInjection/Configuration.php
index 340a337..2612045 100644
--- a/src/DependencyInjection/Configuration.php
+++ b/src/DependencyInjection/Configuration.php
@@ -48,6 +48,11 @@ public function getConfigTreeBuilder(): TreeBuilder
                                 ->scalarNode('api_key')->isRequired()->end()
                             ->end()
                         ->end()
+                        ->arrayNode('mistral')
+                            ->children()
+                                ->scalarNode('api_key')->isRequired()->end()
+                            ->end()
+                        ->end()
                     ->end()
                 ->end()
                 ->arrayNode('chain')
diff --git a/src/DependencyInjection/LlmChainExtension.php b/src/DependencyInjection/LlmChainExtension.php
index 0263855..731b5ea 100644
--- a/src/DependencyInjection/LlmChainExtension.php
+++ b/src/DependencyInjection/LlmChainExtension.php
@@ -23,6 +23,8 @@
 use PhpLlm\LlmChain\Platform\Bridge\Google\Gemini;
 use PhpLlm\LlmChain\Platform\Bridge\Google\PlatformFactory as GooglePlatformFactory;
 use PhpLlm\LlmChain\Platform\Bridge\Meta\Llama;
+use PhpLlm\LlmChain\Platform\Bridge\Mistral\Mistral;
+use PhpLlm\LlmChain\Platform\Bridge\Mistral\PlatformFactory as MistralPlatformFactory;
 use PhpLlm\LlmChain\Platform\Bridge\OpenAI\Embeddings;
 use PhpLlm\LlmChain\Platform\Bridge\OpenAI\GPT;
 use PhpLlm\LlmChain\Platform\Bridge\OpenAI\PlatformFactory as OpenAIPlatformFactory;
@@ -202,6 +204,21 @@ private function processPlatformConfig(string $type, array $platform, ContainerB
             return;
         }
 
+        if ('mistral' === $type) {
+            $platformId = 'llm_chain.platform.mistral';
+            $definition = (new Definition(Platform::class))
+                ->setFactory(MistralPlatformFactory::class.'::create')
+                ->setAutowired(true)
+                ->setLazy(true)
+                ->addTag('proxy', ['interface' => PlatformInterface::class])
+                ->setArguments(['$apiKey' => $platform['api_key']])
+                ->addTag('llm_chain.platform');
+
+            $container->setDefinition($platformId, $definition);
+
+            return;
+        }
+
         throw new \InvalidArgumentException(sprintf('Platform "%s" is not supported for configuration via bundle at this point.', $type));
     }
 
@@ -218,6 +235,7 @@ private function processChainConfig(string $name, array $config, ContainerBuilde
             'claude' => Claude::class,
             'llama' => Llama::class,
             'gemini' => Gemini::class,
+            'mistral' => Mistral::class,
             default => throw new \InvalidArgumentException(sprintf('Model "%s" is not supported.', $modelName)),
         };
         $modelDefinition = new Definition($modelClass);