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
index d56d30e91..c52e7ebdb 100644
--- 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")