add gemini

HumanSignal · Jan 17, 2025 · b990ba4 · b990ba4
1 parent 5669420
commit b990ba4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/label_studio/ml_model_providers/models.py b/label_studio/ml_model_providers/models.py
@@ -30,6 +30,7 @@ class ModelProviderChoices(Enum):
     AZURE_OPENAI = ModelProviderConfig(litellm_slug='azure', display_name='AzureOpenAI')
     VERTEX_AI = ModelProviderConfig(litellm_slug='vertex_ai', display_name='VertexAI')
     CUSTOM = ModelProviderConfig(litellm_slug='openai', display_name='Custom')
+    GEMINI = ModelProviderConfig(litellm_slug='gemini', display_name='Gemini')
 
 
 class ModelProviderConnectionScopes(models.TextChoices):