"""
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__


def test_wp_0005_primitives_are_exported_from_package_root():
    expected_names = [
        "ModelRate",
        "ModelRateRegistry",
        "CostEstimate",
        "CostModel",
        "estimate_cost",
        "TokenEstimate",
        "Observation",
        "ProblemClass",
        "ProblemClassRegistry",
        "default_problem_class_registry",
        "ChunkSummarizationProblemClass",
        "EntityExtractionProblemClass",
        "RelationExtractionProblemClass",
        "JudgeEvalProblemClass",
        "ReportSynthesisProblemClass",
    ]

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