code/self_healing_finetuning/risk/tiers.py

from __future__ import annotations


RISK_TIERS = {
    "internal_dev": {"require_human_approval": False, "allow_mock_providers": True},
    "internal_analyst": {"require_human_approval": False, "require_prompt_replay": True},
    "regulated_internal": {"require_human_approval": True, "require_compliance_approval": True},
    "external_facing": {
        "require_human_approval": True,
        "require_compliance_approval": True,
        "require_product_owner_approval": True,
        "mandatory_prompt_replay_pass_rate": 1.0,
    },
}


def get_risk_policy(tier: str) -> dict[str, object]:
    if tier not in RISK_TIERS:
        raise ValueError(f"unknown risk tier: {tier}")
    return RISK_TIERS[tier]