diff --git a/lib/sycamore/sycamore/llms/gemini.py b/lib/sycamore/sycamore/llms/gemini.py index 2796ed4a9..9583deda0 100644 --- a/lib/sycamore/sycamore/llms/gemini.py +++ b/lib/sycamore/sycamore/llms/gemini.py @@ -26,6 +26,7 @@ class GeminiModels(Enum): GEMINI_2_FLASH = GeminiModel(name="gemini-2.0-flash", is_chat=True) GEMINI_2_FLASH_LITE = GeminiModel(name="gemini-2.0-flash-lite", is_chat=True) GEMINI_2_FLASH_THINKING = GeminiModel(name="gemini-2.0-flash-thinking-exp", is_chat=True) + GEMINI_2_PRO = GeminiModel(name="gemini-2.0-pro-exp-02-05", is_chat=True) GEMINI_1_5_PRO = GeminiModel(name="gemini-1.5-pro", is_chat=True) @classmethod