From 4b318babd997a3ff152d5916289f0c45d33d24c7 Mon Sep 17 00:00:00 2001 From: trducng Date: Tue, 27 Aug 2024 09:15:44 +0700 Subject: [PATCH] Include URL listing available Gemini models --- .../jupyter_ai_magics/partner_providers/gemini.py | 1 + 1 file changed, 1 insertion(+) 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 a96927957..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,6 +2,7 @@ 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"