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"