gemini reasoning effort medium by default (#3282)

Co-authored-by: lawyzheng <lawyzheng1106@gmail.com>
This commit is contained in:
Shuchang Zheng
2025-08-25 01:06:46 -07:00
committed by GitHub
parent 5fb0378434
commit 0a9b58956f
2 changed files with 12 additions and 0 deletions

View File

@@ -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

View File

@@ -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(