Add Gemini 1.5 to the list of chat options

jupyterlab · Aug 27, 2024 · 2d24a60 · 2d24a60
1 parent 41ccd10
commit 2d24a60
Showing 1 changed file with 2 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,6 +6,8 @@ class GeminiProvider(BaseProvider, GoogleGenerativeAI):
     id = "gemini"
     name = "Gemini"
     models = [
+        "gemini-1.5-pro",
+        "gemini-1.5-flash",
         "gemini-1.0-pro",
         "gemini-1.0-pro-001",
         "gemini-1.0-pro-latest",