mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 13:53:15 +00:00
27 lines
802 B
Python
27 lines
802 B
Python
|
from agent.utils.model import (
|
||
|
|
google_thinking_level_for,
|
||
|
|
is_gemini_3_family,
|
||
|
|
provider_model_kwargs,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_gemini_3_family_detection() -> None:
|
||
|
|
assert is_gemini_3_family("google_genai:gemini-3.5-flash") is True
|
||
|
|
assert is_gemini_3_family("google_genai:gemini-2.5-flash") is False
|
||
|
|
|
||
|
|
|
||
|
|
def test_google_thinking_level_maps_effort() -> None:
|
||
|
|
assert google_thinking_level_for("medium") == "medium"
|
||
|
|
assert google_thinking_level_for("high") == "high"
|
||
|
|
assert google_thinking_level_for("unknown") is None
|
||
|
|
|
||
|
|
|
||
|
|
def test_provider_model_kwargs_for_google() -> None:
|
||
|
|
kwargs = provider_model_kwargs(
|
||
|
|
"google_genai:gemini-3.5-flash",
|
||
|
|
"high",
|
||
|
|
max_tokens=16_000,
|
||
|
|
)
|
||
|
|
assert kwargs["max_tokens"] == 16_000
|
||
|
|
assert kwargs["thinking_level"] == "high"
|