wehub-resource-sync
e4dcfc49aa
Tests / Import Check (Python 3.13) (push) Has been cancelled
Tests / Import Check (Python 3.14) (push) Has been cancelled
Tests / Python Tests (Python 3.11) (push) Has been cancelled
Tests / Python Tests (Python 3.12) (push) Has been cancelled
Tests / Python Tests (Python 3.14) (push) Has been cancelled
Tests / Test Summary (push) Has been cancelled
Tests / Lint and Format (push) Has been cancelled
Tests / Web Node Tests (push) Has been cancelled
Tests / Import Check (Python 3.11) (push) Has been cancelled
Tests / Import Check (Python 3.12) (push) Has been cancelled
Tests / Python Tests (Python 3.13) (push) Has been cancelled
93 lines
3.3 KiB
Python
93 lines
3.3 KiB
Python
"""Tests for the centralized reasoning-effort registry."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from deeptutor.services.llm.reasoning_params import (
|
|
build_openai_compatible_reasoning_kwargs,
|
|
default_reasoning_effort_for,
|
|
)
|
|
|
|
|
|
class TestDefaultReasoningEffortFor:
|
|
"""Single source of truth for the implicit per-provider/model effort."""
|
|
|
|
@pytest.mark.parametrize(
|
|
"model",
|
|
[
|
|
"gemini-2.5-flash",
|
|
"gemini-2.5-pro",
|
|
"gemini-2.5-flash-lite",
|
|
"GEMINI-2.5-FLASH",
|
|
"models/gemini-2.5-flash",
|
|
"gemini-3.0-pro",
|
|
],
|
|
)
|
|
def test_gemini_thinking_models_default_to_none(self, model: str) -> None:
|
|
assert default_reasoning_effort_for("gemini", model) == "none"
|
|
|
|
@pytest.mark.parametrize(
|
|
"model",
|
|
["gemini-1.5-flash", "gemini-1.5-pro", "gemini-2.0-flash"],
|
|
)
|
|
def test_gemini_legacy_models_unaffected(self, model: str) -> None:
|
|
assert default_reasoning_effort_for("gemini", model) is None
|
|
|
|
def test_other_providers_unaffected(self) -> None:
|
|
assert default_reasoning_effort_for("openai", "gpt-5") is None
|
|
assert default_reasoning_effort_for("deepseek", "deepseek-v4") is None
|
|
assert default_reasoning_effort_for("dashscope", "qwen3-max") is None
|
|
|
|
def test_missing_provider_or_model(self) -> None:
|
|
assert default_reasoning_effort_for(None, "gemini-2.5-flash") is None
|
|
assert default_reasoning_effort_for("gemini", None) is None
|
|
assert default_reasoning_effort_for("", "") is None
|
|
|
|
def test_provider_name_case_insensitive(self) -> None:
|
|
assert default_reasoning_effort_for("Gemini", "gemini-2.5-flash") == "none"
|
|
assert default_reasoning_effort_for("GEMINI", "gemini-2.5-flash") == "none"
|
|
|
|
|
|
class TestBuildOpenAICompatibleReasoningKwargsForGemini:
|
|
"""The OpenAI-compat helper consults the same registry."""
|
|
|
|
def test_gemini_25_defaults_off_when_unspecified(self) -> None:
|
|
kwargs = build_openai_compatible_reasoning_kwargs(
|
|
spec=None, binding="gemini", model="gemini-2.5-flash", reasoning_effort=None
|
|
)
|
|
assert kwargs == {"reasoning_effort": "none"}
|
|
|
|
def test_models_prefix_still_matches(self) -> None:
|
|
kwargs = build_openai_compatible_reasoning_kwargs(
|
|
spec=None,
|
|
binding="gemini",
|
|
model="models/gemini-2.5-flash",
|
|
reasoning_effort=None,
|
|
)
|
|
assert kwargs == {"reasoning_effort": "none"}
|
|
|
|
def test_explicit_effort_takes_precedence(self) -> None:
|
|
kwargs = build_openai_compatible_reasoning_kwargs(
|
|
spec=None,
|
|
binding="gemini",
|
|
model="gemini-2.5-flash",
|
|
reasoning_effort="high",
|
|
)
|
|
assert kwargs == {"reasoning_effort": "high"}
|
|
|
|
def test_gemini_15_left_untouched(self) -> None:
|
|
kwargs = build_openai_compatible_reasoning_kwargs(
|
|
spec=None,
|
|
binding="gemini",
|
|
model="gemini-1.5-flash",
|
|
reasoning_effort=None,
|
|
)
|
|
assert kwargs == {}
|
|
|
|
def test_openai_left_untouched(self) -> None:
|
|
kwargs = build_openai_compatible_reasoning_kwargs(
|
|
spec=None, binding="openai", model="gpt-4o", reasoning_effort=None
|
|
)
|
|
assert kwargs == {}
|