diff --git a/src/synapt_eval/adapters/__init__.py b/src/synapt_eval/adapters/__init__.py index 7f6082d..0e47cb6 100644 --- a/src/synapt_eval/adapters/__init__.py +++ b/src/synapt_eval/adapters/__init__.py @@ -1,15 +1,17 @@ """Adapter interfaces for customer integration.""" from synapt_eval.adapters.fixture_loader import FixtureLoader -from synapt_eval.adapters.generation_adapter import GenerationAdapter +from synapt_eval.adapters.generation_adapter import GenerationAdapter, GenerationOutput from synapt_eval.adapters.judge_adapter import JudgeAdapter, JudgeRequest, JudgeResponse -from synapt_eval.adapters.retrieval_adapter import RetrievalAdapter +from synapt_eval.adapters.retrieval_adapter import RetrievalAdapter, RetrievalCandidate __all__ = [ "FixtureLoader", "GenerationAdapter", + "GenerationOutput", "JudgeAdapter", "JudgeRequest", "JudgeResponse", "RetrievalAdapter", + "RetrievalCandidate", ] diff --git a/tests/unit/test_adapter_exports.py b/tests/unit/test_adapter_exports.py new file mode 100644 index 0000000..816a910 --- /dev/null +++ b/tests/unit/test_adapter_exports.py @@ -0,0 +1,15 @@ +"""Public adapter export contract used by README and docs.""" + +from synapt_eval.adapters import ( + GenerationAdapter, + GenerationOutput, + RetrievalAdapter, + RetrievalCandidate, +) + + +def test_documented_adapter_imports_are_exported() -> None: + assert RetrievalAdapter.__name__ == "RetrievalAdapter" + assert RetrievalCandidate(id="doc1", score=0.9).id == "doc1" + assert GenerationAdapter.__name__ == "GenerationAdapter" + assert GenerationOutput(text="ok", latency_ms=1.0).text == "ok"