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 c3833696adc666a512b90ca64826f16bee0e412d..83899153ca06ee7eab10b7e1cc93155e0156ecd7 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", )