Draft follow-ups for prospects who have gone quiet

import base64
from datetime import datetime, timedelta, timezone
from email.message import EmailMessage
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, credentials, ai, state
@on_trigger
def run(event):
    creds = Credentials(credentials.connection("google", "work"))
    gmail = build("gmail", "v1", credentials=creds, cache_discovery=False)
    done, drafted = set(state.get("done") or []), []
    for t in gmail.users().threads().list(userId="me", q="label:Prospects older_than:7d newer_than:60d").execute().get("threads", []):
        th = gmail.users().threads().get(userId="me", id=t["id"]).execute()
        last = th["messages"][-1]
        if "SENT" not in last["labelIds"] or last["id"] in done:
            continue
        h = {x["name"]: x["value"] for x in last["payload"]["headers"]}
        msg = EmailMessage()
        msg["To"], msg["Subject"] = h["To"], "Re: " + h["Subject"].removeprefix("Re: ")
        msg.set_content(ai.generate(f"Write a short, friendly follow-up to this email:\n{last['snippet']}", max_output_tokens=200).text)
        gmail.users().drafts().create(userId="me", body={"message": {"threadId": t["id"], "raw": base64.urlsafe_b64encode(msg.as_bytes()).decode()}}).execute()
        done.add(last["id"])
        drafted.append(h["To"])
    if drafted:
        start = datetime.now(timezone.utc) + timedelta(hours=1)
        build("calendar", "v3", credentials=creds, cache_discovery=False).events().insert(calendarId="primary", body={
            "summary": f"Send {len(drafted)} follow-ups", "description": "\n".join(drafted),
            "start": {"dateTime": start.isoformat()}, "end": {"dateTime": (start + timedelta(minutes=15)).isoformat()}}).execute()
    state.set("done", list(done)[-500:])
    return {"drafted": len(drafted)}