"""
Tests for the public llm_connect package surface.
"""

import llm_connect


def test_wp_0004_primitives_are_exported_from_package_root():
    expected_names = [
        "AdaptiveRoutingPolicy",
        "BaselineGrader",
        "EmbeddingSimilarityJudge",
        "ExactMatchJudge",
        "GradingResult",
        "Judge",
        "LLMJudge",
        "PairedGrader",
        "QualityLedger",
        "QualityObservation",
        "ShadowingAdapter",
        "is_stale",
    ]

    for name in expected_names:
        assert hasattr(llm_connect, name)
        assert name in llm_connect.__all__