Add Gemini 1.5 to the list of chat options (#964)

* Add Gemini 1.5 to the list of chat options * Include URL listing available Gemini models
jupyterlab · Aug 27, 2024 · 29e2a47 · 29e2a47
1 parent 41ccd10
commit 29e2a47
Showing 1 changed file with 3 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
@@ -2,10 +2,13 @@
 from langchain_google_genai import GoogleGenerativeAI
 
 
+# See list of model ids here: https://ai.google.dev/gemini-api/docs/models/gemini
 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",