Skip to content

Commit 02a823d

Browse files
authored
Refresh the list of supported Gemini models. (#1381)
1 parent 019bab6 commit 02a823d

File tree

1 file changed

+3
-6
lines changed
  • packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers

1 file changed

+3
-6
lines changed

packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/gemini.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7,14 +7,11 @@ class GeminiProvider(BaseProvider, GoogleGenerativeAI):
77
id = "gemini"
88
name = "Gemini"
99
models = [
10+
"gemini-2.5-pro",
11+
"gemini-2.5-flash",
12+
"gemini-2.0-flash-lite",
1013
"gemini-1.5-pro",
1114
"gemini-1.5-flash",
12-
"gemini-1.0-pro",
13-
"gemini-1.0-pro-001",
14-
"gemini-1.0-pro-latest",
15-
"gemini-1.0-pro-vision-latest",
16-
"gemini-pro",
17-
"gemini-pro-vision",
1815
]
1916
model_id_key = "model"
2017
auth_strategy = EnvAuthStrategy(name="GOOGLE_API_KEY")

0 commit comments

Comments
 (0)