Add GEMINI_3.0_FLASH support and update docker-compose.yml (#4687)

Co-authored-by: Claude <noreply@anthropic.com>
2026-02-11 22:30:40 -05:00
parent dbca5f7094
commit fd241dba54
3 changed files with 20 additions and 7 deletions
--- a/skyvern/forge/sdk/api/llm/config_registry.py
+++ b/skyvern/forge/sdk/api/llm/config_registry.py
@@ -995,6 +995,19 @@ if settings.ENABLE_GEMINI:
            ),
        ),
    )
+    LLMConfigRegistry.register_config(
+        "GEMINI_3.0_FLASH",
+        LLMConfig(
+            "gemini/gemini-3-flash-preview",
+            ["GEMINI_API_KEY"],
+            supports_vision=True,
+            add_assistant_prefix=False,
+            max_completion_tokens=65536,
+            litellm_params=LiteLLMParams(
+                thinking_level="medium" if settings.GEMINI_INCLUDE_THOUGHT else "minimal",
+            ),
+        ),
+    )


 if settings.ENABLE_NOVITA: