app/knowledge/substitution_rules.py

"""Substitution maps."""

from __future__ import annotations

SUBSTITUTIONS: dict[str, list[str]] = {
    "nsaid_like": ["acetaminophen_like", "topical_nsaid_like"],
    "benzodiazepine_like": ["non_benzo_sleep_support"],
    "opioid_like": ["non_opioid_analgesic"],
}


def get_substitutions(drug: str) -> list[str]:
    return SUBSTITUTIONS.get(drug, [])