wehub-resource-sync
81 lines
3.3 KiB
Python
81 lines
3.3 KiB
Python
"""Worker-side pragma resolution: model + thinking overrides."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from robomp.config import Settings, reset_settings_cache
|
|
from robomp.worker import DirectiveInfo, _resolve_pragma_overrides
|
|
|
|
|
|
@pytest.fixture
|
|
def settings_with_pool(monkeypatch: pytest.MonkeyPatch, env: dict[str, str]) -> Settings:
|
|
monkeypatch.setenv(
|
|
"ROBOMP_MODEL",
|
|
"anthropic/claude-sonnet-4-6,openai/gpt-5.5,openai/gpt-5.5-mini",
|
|
)
|
|
reset_settings_cache()
|
|
return Settings() # type: ignore[call-arg]
|
|
|
|
|
|
def test_no_directive_means_no_override(settings_with_pool: Settings) -> None:
|
|
assert _resolve_pragma_overrides(None, settings_with_pool) == (None, None)
|
|
|
|
|
|
def test_directive_without_pragmas_means_no_override(settings_with_pool: Settings) -> None:
|
|
directive = DirectiveInfo(body="run it", author="can1357")
|
|
assert _resolve_pragma_overrides(directive, settings_with_pool) == (None, None)
|
|
|
|
|
|
def test_model_pragma_resolves_to_pool_entry(settings_with_pool: Settings) -> None:
|
|
directive = DirectiveInfo(body="run", author="can1357", pragmas=(("model", "gpt"),))
|
|
model_override, thinking_override = _resolve_pragma_overrides(directive, settings_with_pool)
|
|
assert model_override == "openai/gpt-5.5"
|
|
assert thinking_override is None
|
|
|
|
|
|
def test_model_alias_exact_short_name(settings_with_pool: Settings) -> None:
|
|
directive = DirectiveInfo(body="run", author="can1357", pragmas=(("model", "gpt-5.5-mini"),))
|
|
model_override, _ = _resolve_pragma_overrides(directive, settings_with_pool)
|
|
assert model_override == "openai/gpt-5.5-mini"
|
|
|
|
|
|
def test_unmatched_model_alias_falls_back_to_random_pick(settings_with_pool: Settings) -> None:
|
|
directive = DirectiveInfo(body="run", author="can1357", pragmas=(("model", "qwen"),))
|
|
model_override, _ = _resolve_pragma_overrides(directive, settings_with_pool)
|
|
assert model_override is None
|
|
|
|
|
|
def test_thinking_pragma_normalized(settings_with_pool: Settings) -> None:
|
|
directive = DirectiveInfo(body="run", author="can1357", pragmas=(("thinking", "LOW"),))
|
|
model_override, thinking_override = _resolve_pragma_overrides(directive, settings_with_pool)
|
|
assert model_override is None
|
|
assert thinking_override == "low"
|
|
|
|
|
|
def test_unknown_thinking_level_dropped(settings_with_pool: Settings) -> None:
|
|
directive = DirectiveInfo(body="run", author="can1357", pragmas=(("thinking", "ultra"),))
|
|
_, thinking_override = _resolve_pragma_overrides(directive, settings_with_pool)
|
|
assert thinking_override is None
|
|
|
|
|
|
def test_both_pragmas_resolved_together(settings_with_pool: Settings) -> None:
|
|
directive = DirectiveInfo(
|
|
body="run",
|
|
author="can1357",
|
|
pragmas=(("model", "claude"), ("thinking", "medium")),
|
|
)
|
|
model_override, thinking_override = _resolve_pragma_overrides(directive, settings_with_pool)
|
|
assert model_override == "anthropic/claude-sonnet-4-6"
|
|
assert thinking_override == "medium"
|
|
|
|
|
|
def test_last_value_wins_for_duplicate_keys(settings_with_pool: Settings) -> None:
|
|
directive = DirectiveInfo(
|
|
body="run",
|
|
author="can1357",
|
|
pragmas=(("model", "claude"), ("model", "gpt")),
|
|
)
|
|
model_override, _ = _resolve_pragma_overrides(directive, settings_with_pool)
|
|
assert model_override == "openai/gpt-5.5"
|