Skip to content

Commit

Permalink
feat: support google gemini
Browse files Browse the repository at this point in the history
  • Loading branch information
adrienbrault committed May 30, 2024
1 parent f3fe034 commit f331713
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions src/DependencyInjection/InstructriceExtension.php
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down Expand Up @@ -41,6 +42,9 @@ class InstructriceExtension extends ConfigurableExtension
* groq?: array{
* api_key?: string
* },
* google?: array{
* api_key?: string
* },
* mistral?: array{
* api_key?: string
* },
Expand All @@ -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,
Expand Down

0 comments on commit f331713

Please sign in to comment.