From ab059311320d81b5cb010b18e9c976cd9e3878b7 Mon Sep 17 00:00:00 2001 From: Layne Penney Date: Thu, 7 May 2026 18:42:17 -0500 Subject: [PATCH] fix: export adapter dataclasses for docs imports --- src/synapt_eval/adapters/__init__.py | 6 ++++-- tests/unit/test_adapter_exports.py | 15 +++++++++++++++ 2 files changed, 19 insertions(+), 2 deletions(-) create mode 100644 tests/unit/test_adapter_exports.py 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"