gemini reasoning effort medium by default (#3282)
Co-authored-by: lawyzheng <lawyzheng1106@gmail.com>
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user