File size: 1,430 Bytes
638c806 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | from __future__ import annotations
import pytest
from conftest import call_with_supported_kwargs, resolve_callable, text_blob
ADAPTER_MODULES = (
"promptboost.adapters",
"promptboost.adapter",
"promptboost.core",
)
ADAPTER_SELECTOR_FUNCS = (
"select_adapter",
"get_adapter",
"resolve_adapter",
)
def _select_adapter(target_harness: str, target_model: str):
selector = resolve_callable(ADAPTER_MODULES, ADAPTER_SELECTOR_FUNCS)
return call_with_supported_kwargs(
selector,
target_harness=target_harness,
harness=target_harness,
target_model=target_model,
model=target_model,
)
def test_select_adapter_uses_harness_and_model():
adapter = _select_adapter("codex", "gpt-5.4")
blob = text_blob(adapter)
assert "codex" in blob
assert any(token in blob for token in ("gpt", "5.4", "openai"))
def test_select_adapter_distinguishes_supported_targets():
codex = _select_adapter("codex", "gpt-5.4")
opencode = _select_adapter("opencode", "minimax-m2.7")
assert text_blob(codex) != text_blob(opencode)
assert "opencode" in text_blob(opencode)
assert any(token in text_blob(opencode) for token in ("minimax", "m2.7", "m2"))
def test_select_adapter_rejects_unknown_targets():
with pytest.raises((KeyError, LookupError, ValueError, NotImplementedError)):
_select_adapter("unknown-harness", "unknown-model")
|