feat: support google gemini

adrienbrault · May 30, 2024 · f331713 · f331713
1 parent f3fe034
commit f331713
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/DependencyInjection/InstructriceExtension.php b/src/DependencyInjection/InstructriceExtension.php
@@ -12,6 +12,7 @@
 use AdrienBrault\Instructrice\LLM\Provider\Anyscale;
 use AdrienBrault\Instructrice\LLM\Provider\DeepInfra;
 use AdrienBrault\Instructrice\LLM\Provider\Fireworks;
+use AdrienBrault\Instructrice\LLM\Provider\Google;
 use AdrienBrault\Instructrice\LLM\Provider\Groq;
 use AdrienBrault\Instructrice\LLM\Provider\Mistral;
 use AdrienBrault\Instructrice\LLM\Provider\Ollama;
@@ -41,6 +42,9 @@ class InstructriceExtension extends ConfigurableExtension
      *     groq?: array{
      *         api_key?: string
      *     },
+     *     google?: array{
+     *         api_key?: string
+     *     },
      *     mistral?: array{
      *         api_key?: string
      *     },
@@ -66,6 +70,7 @@ protected function loadInternal(array $mergedConfig, ContainerBuilder $container
             DeepInfra::class => $mergedConfig['deepinfra']['api_key'] ?? null,
             Fireworks::class => $mergedConfig['fireworks']['api_key'] ?? null,
             Groq::class => $mergedConfig['groq']['api_key'] ?? null,
+            Google::class => $mergedConfig['google']['api_key'] ?? null,
             Mistral::class => $mergedConfig['mistral']['api_key'] ?? null,
             Ollama::class => $mergedConfig['ollama']['api_key'] ?? null,
             OpenAi::class => $mergedConfig['openai']['api_key'] ?? null,