From 29e2a479e928416e490c4d84b48fac3f421e112a Mon Sep 17 00:00:00 2001 From: trducng Date: Tue, 27 Aug 2024 22:24:11 +0700 Subject: [PATCH] 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 --- .../jupyter_ai_magics/partner_providers/gemini.py | 3 +++ 1 file changed, 3 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 c52e7ebdb..5e72abac0 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 @@ -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",