68 lines
3.2 KiB
Python
68 lines
3.2 KiB
Python
from __future__ import annotations
|
||
|
||
from router.query_classifier import classify_query_for_route
|
||
from router.route_selector import choose_route
|
||
from router.store_sufficiency import check_store_sufficiency
|
||
|
||
|
||
def _metadata() -> dict:
|
||
return {
|
||
"freshness_threshold_hours": 6.0,
|
||
"refresh_age_hours": 1.0,
|
||
"feature_age_hours": 1.0,
|
||
"risk_age_hours": 1.0,
|
||
"feature_ready": True,
|
||
"risk_ready": True,
|
||
"ranking_ready": False,
|
||
"aggregate_ready": True,
|
||
"precomputed_aggregates": [
|
||
"baseline_period_summary",
|
||
"period_trend_summary",
|
||
],
|
||
"canonical_semantic_coverage": 0.95,
|
||
"canonical_relation_types": 25,
|
||
"canonical_links_total": 2000,
|
||
"canonical_entities_total": 400,
|
||
}
|
||
|
||
|
||
def _route(question_class: str, question_text: str) -> str:
|
||
meta = _metadata()
|
||
flags = classify_query_for_route(question_text, {"question_class": question_class}, meta)
|
||
suff = check_store_sufficiency(flags, meta)
|
||
parsed_as_trend_or_risk = question_class in {"period_trend", "anomaly_control", "ambiguous_fuzzy"} or (
|
||
question_class == "heavy_analytical" and "baseline" in question_text.lower()
|
||
)
|
||
selection = choose_route(flags, suff, parsed_as_trend_or_risk=parsed_as_trend_or_risk)
|
||
return selection.chosen_route
|
||
|
||
|
||
def test_router_subset_q06_to_q12() -> None:
|
||
cases = [
|
||
("drilldown_explain", "Объясни сальдо через движения.", "hybrid_store_plus_live"),
|
||
("drilldown_explain", "Почему проводка на этот счет?", "live_mcp_drilldown"),
|
||
("drilldown_explain", "Цепочка документ -> проводки -> субконто.", "live_mcp_drilldown"),
|
||
("drilldown_explain", "Источник регистра для строки движения.", "live_mcp_drilldown"),
|
||
("drilldown_explain", "Почему выбрано это субконто3?", "live_mcp_drilldown"),
|
||
("cross_entity", "Свяжи документы покупателей и проводки.", "hybrid_store_plus_live"),
|
||
("cross_entity", "Свяжи контрагентов, договоры и проводки.", "hybrid_store_plus_live"),
|
||
]
|
||
for question_class, text, expected_route in cases:
|
||
assert _route(question_class, text) == expected_route
|
||
|
||
|
||
def test_router_subset_q26_to_q30() -> None:
|
||
cases = [
|
||
("heavy_analytical", "Полный риск-срез за июнь.", "batch_refresh_then_store"),
|
||
("heavy_analytical", "Рейтинг риск-счетов.", "batch_refresh_then_store"),
|
||
("heavy_analytical", "Рейтинг риск-контрагентов.", "batch_refresh_then_store"),
|
||
("heavy_analytical", "Baseline closed/open periods.", "store_feature_risk"),
|
||
("heavy_analytical", "Company anomaly summary.", "batch_refresh_then_store"),
|
||
]
|
||
for question_class, text, expected_route in cases:
|
||
assert _route(question_class, text) == expected_route
|
||
|
||
|
||
def test_router_period_trend_anomaly_stays_feature_store() -> None:
|
||
assert _route("period_trend", "Аномальный рост расходных операций?") == "store_feature_risk"
|