From 0a9b58956f3eb8f06c7c2ead41d00a9bc747cd4d Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Mon, 25 Aug 2025 01:06:46 -0700 Subject: [PATCH] gemini reasoning effort medium by default (#3282) Co-authored-by: lawyzheng --- skyvern/config.py | 1 + skyvern/forge/sdk/api/llm/config_registry.py | 11 +++++++++++ 2 files changed, 12 insertions(+) diff --git a/skyvern/config.py b/skyvern/config.py index 2259bca3..7176b67c 100644 --- a/skyvern/config.py +++ b/skyvern/config.py @@ -245,6 +245,7 @@ class Settings(BaseSettings): # GEMINI GEMINI_API_KEY: str | None = None + GEMINI_REASONING_EFFORT: str | None = None # VERTEX_AI VERTEX_CREDENTIALS: str | None = None diff --git a/skyvern/forge/sdk/api/llm/config_registry.py b/skyvern/forge/sdk/api/llm/config_registry.py index 508bfb2b..8b3ce2f3 100644 --- a/skyvern/forge/sdk/api/llm/config_registry.py +++ b/skyvern/forge/sdk/api/llm/config_registry.py @@ -787,6 +787,7 @@ if settings.ENABLE_GEMINI: supports_vision=True, add_assistant_prefix=False, max_completion_tokens=65536, + reasoning_effort=settings.GEMINI_REASONING_EFFORT, ), ) LLMConfigRegistry.register_config( @@ -797,6 +798,7 @@ if settings.ENABLE_GEMINI: supports_vision=True, add_assistant_prefix=False, max_completion_tokens=65536, + reasoning_effort=settings.GEMINI_REASONING_EFFORT, ), ) LLMConfigRegistry.register_config( @@ -807,6 +809,7 @@ if settings.ENABLE_GEMINI: supports_vision=True, add_assistant_prefix=False, max_completion_tokens=65536, + reasoning_effort=settings.GEMINI_REASONING_EFFORT, ), ) LLMConfigRegistry.register_config( @@ -817,6 +820,7 @@ if settings.ENABLE_GEMINI: supports_vision=True, add_assistant_prefix=False, max_completion_tokens=65536, + reasoning_effort=settings.GEMINI_REASONING_EFFORT, ), ) LLMConfigRegistry.register_config( @@ -827,6 +831,7 @@ if settings.ENABLE_GEMINI: supports_vision=True, add_assistant_prefix=False, max_completion_tokens=65536, + reasoning_effort=settings.GEMINI_REASONING_EFFORT, ), ) @@ -1023,6 +1028,7 @@ if settings.ENABLE_VERTEX_AI and settings.VERTEX_CREDENTIALS: api_base=f"{api_base}/gemini-2.5-pro" if api_base else None, vertex_location=settings.VERTEX_LOCATION, ), + reasoning_effort=settings.GEMINI_REASONING_EFFORT, ), ) LLMConfigRegistry.register_config( @@ -1038,6 +1044,7 @@ if settings.ENABLE_VERTEX_AI and settings.VERTEX_CREDENTIALS: api_base=f"{api_base}/gemini-2.5-pro-preview-05-06" if api_base else None, vertex_location=settings.VERTEX_LOCATION, ), + reasoning_effort=settings.GEMINI_REASONING_EFFORT, ), ) LLMConfigRegistry.register_config( @@ -1053,6 +1060,7 @@ if settings.ENABLE_VERTEX_AI and settings.VERTEX_CREDENTIALS: api_base=f"{api_base}/gemini-2.5-flash" if api_base else None, vertex_location=settings.VERTEX_LOCATION, ), + reasoning_effort=settings.GEMINI_REASONING_EFFORT, ), ) LLMConfigRegistry.register_config( @@ -1068,6 +1076,7 @@ if settings.ENABLE_VERTEX_AI and settings.VERTEX_CREDENTIALS: api_base=f"{api_base}/gemini-2.5-flash-preview-05-20" if api_base else None, vertex_location=settings.VERTEX_LOCATION, ), + reasoning_effort=settings.GEMINI_REASONING_EFFORT, ), ) LLMConfigRegistry.register_config( @@ -1083,6 +1092,7 @@ if settings.ENABLE_VERTEX_AI and settings.VERTEX_CREDENTIALS: api_base=f"{api_base}/gemini-2.5-flash-preview-04-17" if api_base else None, vertex_location=settings.VERTEX_LOCATION, ), + reasoning_effort=settings.GEMINI_REASONING_EFFORT, ), ) LLMConfigRegistry.register_config( @@ -1098,6 +1108,7 @@ if settings.ENABLE_VERTEX_AI and settings.VERTEX_CREDENTIALS: api_base=f"{api_base}/gemini-2.5-flash-preview-05-20" if api_base else None, vertex_location=settings.VERTEX_LOCATION, ), + reasoning_effort=settings.GEMINI_REASONING_EFFORT, ), ) LLMConfigRegistry.register_config(