mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 20:53:15 +00:00
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
from agent.dashboard.options import SUPPORTED_MODELS, provider_fallback_pair
|
|
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("minimal") == "minimal"
|
|
assert google_thinking_level_for("none") == "minimal"
|
|
assert google_thinking_level_for("medium") == "medium"
|
|
assert google_thinking_level_for("high") == "high"
|
|
assert google_thinking_level_for("unknown") is None
|
|
|
|
|
|
def test_gemini_35_flash_is_supported_with_documented_efforts() -> None:
|
|
gemini = next(m for m in SUPPORTED_MODELS if m["id"] == "google_genai:gemini-3.5-flash")
|
|
assert gemini["label"] == "Gemini 3.5 Flash"
|
|
assert gemini["efforts"] == ["minimal", "low", "medium", "high"]
|
|
assert gemini["default_effort"] == "medium"
|
|
|
|
|
|
def test_google_provider_fallback_uses_gemini_35_flash() -> None:
|
|
assert provider_fallback_pair("google_genai:gemini-3-flash-preview", "high") == (
|
|
"google_genai:gemini-3.5-flash",
|
|
"high",
|
|
)
|
|
|
|
|
|
def test_google_provider_fallback_maps_legacy_none_to_minimal() -> None:
|
|
assert provider_fallback_pair("google_genai:gemini-3-flash-preview", "none") == (
|
|
"google_genai:gemini-3.5-flash",
|
|
"minimal",
|
|
)
|
|
|
|
|
|
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"
|