Add more gemini models

jupyterlab · Mar 6, 2024 · 7290ca8 · 7290ca8
1 parent 3bd207d
commit 7290ca8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/gemini.py b/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/gemini.py
@@ -6,7 +6,12 @@ class GeminiProvider(BaseProvider, GoogleGenerativeAI):
     id = "gemini"
     name = "Gemini"
     models = [
+        "gemini-1.0-pro",
+        "gemini-1.0-pro-001",
+        "gemini-1.0-pro-latest",
+        "gemini-1.0-pro-vision-latest",
         "gemini-pro",
+        "gemini-pro-vision",
     ]
     model_id_key = "model"
     auth_strategy = EnvAuthStrategy(name="GOOGLE_API_KEY")