def getRiskRegisterData(**context):
    ti = context.get("ti")
    token = ti.xcom_pull(key="token")

    response = requests.get(
        url=f"{dRoW_api_end_url}/api/module/document-export/airflow/workflow/68f72cb8ee99729ef014ca38?export_type=0",
        headers={
            "x-access-token": f"Bearer {token}",
            "ICWPxAccessKey": "5WSD21ICWP_[1AG:4UdI){n=b~"
        }
    )

    risk_register_data = response.json()
    print(f"check rr data: {risk_register_data}")

    # thresholds: 3 months ≈ 90 days, 6 months ≈ 180 days
    THRESHOLD_3M = 90
    THRESHOLD_6M = 180

    today = datetime.now(timezone.utc).date()

    kpi9_count = 0  # Open records with elapsed_days > 90
    kpi10_count = 0 # Open records with elapsed_days > 180

    for item in risk_register_data:
        rec = item.get("data", {}) or {}

        status = rec.get("Status")
        if status != "Open":
            continue  # only consider open records

        ew_str = rec.get("Date of Early Warning")
        if not ew_str:
            continue  # cannot compute elapsed time

        try:
            ew_date = parse_iso_date(ew_str).date()
        except ValueError:
            print(f"Unable to parse Date of Early Warning: {ew_str}")
            continue

        elapsed_days = (today - ew_date).days

        if elapsed_days > THRESHOLD_3M:
            kpi9_count += 1
        if elapsed_days > THRESHOLD_6M:
            kpi10_count += 1

    print(f"KPI 9 (Open > 3 months): {kpi9_count}")
    print(f"KPI 10 (Open > 6 months): {kpi10_count}")

    result = {
        "kpi9_open_gt_3m": kpi9_count,
        "kpi10_open_gt_6m": kpi10_count,
    }

    # If you want to use XCom:
    # ti.xcom_push(key="RISK_REGISTER_KPIS", value=result)

    return result