from joblib import load

router = load("system-router.joblib")

queries = [
    "Use a human approval step before an agent can deploy code",
    "Route hard reasoning tasks to a stronger model",
    "Store long-term memories for a persistent agent"
]

for q in queries:
    label = router.predict([q])[0]
    scores = router.predict_proba([q])[0]
    confidence = scores.max()
    print(f"{q}\n  -> {label} ({confidence:.2f})")