app/models/graph/severe_alert_head.py

"""Severe alert head."""

from __future__ import annotations

from app.knowledge.ddi_knowledge import top_risky_pairs


def severe_alert_probability(drugs: list[str]) -> float:
    return min(0.99, 0.1 + 0.3 * len(top_risky_pairs(drugs)))