From b394bd8d3bc317c6a0b77c0fee46eb1f3398d0e4 Mon Sep 17 00:00:00 2001 From: Qiusheng Wu Date: Tue, 5 Mar 2024 15:30:16 -0500 Subject: [PATCH] Add more gemini models --- .../jupyter_ai_magics/partner_providers/gemini.py | 5 +++++ 1 file changed, 5 insertions(+) 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")