def getCEImplementationData(**context):
    ti = context["ti"]
    engine = get_dw_engine()

    q_kpi7 = """
    SELECT COALESCE(SUM(total_events), 0) AS v
    FROM c5_health_check_report_ce_merged_metrics
    WHERE metric = 'ce_not_yet_implemented'
      AND label IN (
        '3 - 6 months',
        '6 - 9 months',
        '9 - 12 months',
        '12 - 24 months',
        '> 24 months'
      );
    """

    q_kpi8 = """
    SELECT COALESCE(SUM(total_events), 0) AS v
    FROM c5_health_check_report_ce_merged_metrics
    WHERE metric = 'ce_not_yet_implemented'
      AND label IN (
        '6 - 9 months',
        '9 - 12 months',
        '12 - 24 months',
        '> 24 months'
      );
    """

    with engine.begin() as conn:
        kpi7 = conn.execute(text(q_kpi7)).scalar() or 0
        kpi8 = conn.execute(text(q_kpi8)).scalar() or 0


    kpi7 = int(kpi7)
    kpi8 = int(kpi8)

    print(f"KPI 7 > 3 months: {kpi7}")
    print(f"KPI 7 > 6 months: {kpi8}")

    ti.xcom_push(key="kpi7", value=kpi7)
    ti.xcom_push(key="kpi8", value=kpi8)

    return {"kpi7": kpi7, "kpi8": kpi8}