From f544b78c0858596c5ca7aba90bddb0e292409ad4 Mon Sep 17 00:00:00 2001
From: Riddhimaan Senapati
<114703025+Riddhimaan-Senapati@users.noreply.github.com>
Date: Sun, 16 Feb 2025 16:12:27 -0500
Subject: [PATCH] Feat: add new gemini models (#17824)
---
.../llama-index-llms-gemini/llama_index/llms/gemini/base.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/llama-index-integrations/llms/llama-index-llms-gemini/llama_index/llms/gemini/base.py b/llama-index-integrations/llms/llama-index-llms-gemini/llama_index/llms/gemini/base.py
index c3833696ad..83899153ca 100644
--- a/llama-index-integrations/llms/llama-index-llms-gemini/llama_index/llms/gemini/base.py
+++ b/llama-index-integrations/llms/llama-index-llms-gemini/llama_index/llms/gemini/base.py
@@ -50,8 +50,12 @@ dispatcher = instrument.get_dispatcher(__name__)
GEMINI_MODELS = (
"models/gemini-2.0-flash",
"models/gemini-2.0-flash-thinking",
+ "models/gemini-2.0-flash-thinking-exp-01-21",
"models/gemini-2.0-flash-lite",
+ "models/gemini-2.0-flash-lite-preview-02-05",
+ "models/gemini-2.0-pro-exp-02-05",
"models/gemini-1.5-flash",
+ "models/gemini-1.5-flash-8b",
"models/gemini-1.0-pro",
)
--
GitLab