Implement-LLM-WP-0005-cost-model-estimators

2026-05-19 05:02:20 +02:00
parent 0054afe689
commit c11c6afa3f
16 changed files with 1525 additions and 10 deletions
--- a/tests/test_package_exports.py
+++ b/tests/test_package_exports.py
@@ -24,3 +24,27 @@ def test_wp_0004_primitives_are_exported_from_package_root():
    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__