shu/gemini 2.5 pro support (#2109)

2025-04-06 00:03:24 -04:00
parent a72fcadd9a
commit 7e5e684f53
3 changed files with 37 additions and 2 deletions
--- a/skyvern/forge/sdk/api/llm/config_registry.py
+++ b/skyvern/forge/sdk/api/llm/config_registry.py
@@ -348,6 +348,27 @@ if settings.ENABLE_GEMINI:
            max_completion_tokens=8192,
        ),
    )
+    LLMConfigRegistry.register_config(
+        "GEMINI_2.5_PRO_PREVIEW_03_25",
+        LLMConfig(
+            "gemini/gemini-2.5-pro-preview-03-25",
+            ["GEMINI_API_KEY"],
+            supports_vision=True,
+            add_assistant_prefix=False,
+            max_completion_tokens=1048576,
+        ),
+    )
+    LLMConfigRegistry.register_config(
+        "GEMINI_2.5_PRO_EXP_03_25",
+        LLMConfig(
+            "gemini/gemini-2.5-pro-exp-03-25",
+            ["GEMINI_API_KEY"],
+            supports_vision=True,
+            add_assistant_prefix=False,
+            max_completion_tokens=1048576,
+        ),
+    )
+

 if settings.ENABLE_NOVITA:
    LLMConfigRegistry.register_config(