Spot customers at risk of leaving from their tickets

from datetime import date, timedelta
from clickruntime import on_trigger, http, credentials, ai, notify

ZD, AGENT = "https://yourcompany.zendesk.com/api/v2", "you@yourcompany.com"

def zd(method, path, **kw):
    r = http.request(method, ZD + path, auth=(f"{AGENT}/token", credentials.secret("ZENDESK_API_TOKEN")), **kw)
    return r.raise_for_status() or (r.json() if r.content else None)


@on_trigger
def run(event):
    since = (date.today() - timedelta(days=1)).isoformat()
    tickets = zd("GET", "/search.json", params={"query": f"type:ticket updated>={since}"})["results"][:100]
    if not tickets:
        return {"flagged": 0}
    numbered = "\n".join(f"{i}. {t['subject']} | {t['description'][:300]}" for i, t in enumerate(tickets))
    res = ai.generate(numbered, system="Return the numbers of tickets that suggest the customer may cancel or switch provider, with a 6-word reason each.",
                      json_schema={"type": "object", "properties": {"risks": {"type": "array", "items": {"type": "object",
                      "properties": {"n": {"type": "integer"}, "reason": {"type": "string"}}}}}}, max_output_tokens=600).json
    by_org = {}
    for r in res["risks"]:
        t = tickets[r["n"]]
        by_org.setdefault(t.get("organization_id") or "No organisation", []).append(f"  #{t['id']} {r['reason']}")
    if by_org:
        notify.slack("Possible churn risk:\n" + "\n".join(f"• Org {o}\n" + "\n".join(v) for o, v in by_org.items()), channel="#customer-success")
    return {"flagged": len(res["risks"])}