diff --git a/skyvern/forge/sdk/api/llm/config_registry.py b/skyvern/forge/sdk/api/llm/config_registry.py index f9e31a80..64ef22e1 100644 --- a/skyvern/forge/sdk/api/llm/config_registry.py +++ b/skyvern/forge/sdk/api/llm/config_registry.py @@ -305,6 +305,26 @@ if settings.ENABLE_AZURE_O3_MINI: ) if settings.ENABLE_GEMINI: + LLMConfigRegistry.register_config( + "GEMINI_FLASH_2_0", + LLMConfig( + "gemini/gemini-2.0-flash-001", + ["GEMINI_API_KEY"], + supports_vision=True, + add_assistant_prefix=False, + max_completion_tokens=8192, + ), + ) + LLMConfigRegistry.register_config( + "GEMINI_FLASH_2_0_LITE", + LLMConfig( + "gemini/gemini-2.0-flash-lite-preview-02-05", + ["GEMINI_API_KEY"], + supports_vision=True, + add_assistant_prefix=False, + max_completion_tokens=8192, + ), + ) LLMConfigRegistry.register_config( "GEMINI_PRO", LLMConfig(