tegwick
e1bab67bd1
A hall visual that stays on the general branch uses the Efficient model from the 2026-10-08 intelligence-radar tree. render-hall writes only portraits whose files are still absent. Assistant: grok Assistant-Session: 01a11ab6-1534-7d63-8606-662b92a80ed6
120 lines
3.1 KiB
Python
120 lines
3.1 KiB
Python
"""
|
|
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__
|
|
|
|
|
|
def test_wp_0006_profile_primitives_are_exported_from_package_root():
|
|
expected_names = [
|
|
"CUSTODIAN_TRIAGE_BALANCED",
|
|
"RuntimeProfile",
|
|
"ProfiledLLMAdapter",
|
|
"default_runtime_profiles",
|
|
]
|
|
|
|
for name in expected_names:
|
|
assert hasattr(llm_connect, name)
|
|
assert name in llm_connect.__all__
|
|
|
|
|
|
def test_image_render_primitives_are_exported_from_package_root():
|
|
expected_names = [
|
|
"ImageRenderRequest",
|
|
"ImageRenderResult",
|
|
"ImageReceipt",
|
|
"ImageSpendPolicy",
|
|
"ImageSpendGate",
|
|
"OpenRouterImageClient",
|
|
"create_image_client",
|
|
"render_image",
|
|
"ImageSelection",
|
|
"choose_image_model",
|
|
"selection_for_prompt",
|
|
"HallPortraitJob",
|
|
"plan_missing_portraits",
|
|
]
|
|
|
|
for name in expected_names:
|
|
assert hasattr(llm_connect, name)
|
|
assert name in llm_connect.__all__
|
|
|
|
|
|
def test_wp_0007_spend_primitives_are_exported_from_package_root():
|
|
expected_names = [
|
|
"FxRate",
|
|
"resolve_fx_rate",
|
|
"UsageEvent",
|
|
"UsageLedger",
|
|
"UsageSummary",
|
|
"default_usage_ledger_path",
|
|
"event_from_response",
|
|
"maybe_record_usage",
|
|
"week_window",
|
|
]
|
|
|
|
for name in expected_names:
|
|
assert hasattr(llm_connect, name)
|
|
assert name in llm_connect.__all__
|
|
|
|
|
|
def test_wp_0008_balance_primitives_are_exported_from_package_root():
|
|
expected_names = [
|
|
"AccountBalance",
|
|
"BalanceClientRegistry",
|
|
"OpenRouterBalanceClient",
|
|
"ProviderBalanceClient",
|
|
"format_balance_human",
|
|
"get_account_balance",
|
|
"resolve_balance_provider",
|
|
"LLMBalanceUnsupportedError",
|
|
]
|
|
|
|
for name in expected_names:
|
|
assert hasattr(llm_connect, name)
|
|
assert name in llm_connect.__all__
|