Sort outreach replies in Gmail into interested and not

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, credentials, ai, notify, report_fetched

LABELS = {"interested": "Label_Interested", "not_interested": "Label_NotInterested", "out_of_office": "Label_OOO"}

@on_trigger
def run(event):
    gmail = build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)
    query = 'subject:"Re: Quick question about hiring" -label:Interested -label:NotInterested -label:OOO newer_than:7d'
    msgs = gmail.users().messages().list(userId="me", q=query).execute().get("messages", [])
    report_fetched(len(msgs))
    if not msgs:
        return {"summary": "No new replies"}
    for m in msgs:
        full = gmail.users().messages().get(userId="me", id=m["id"], format="metadata", metadataHeaders=["From"]).execute()
        sender = full["payload"]["headers"][0]["value"]
        label = ai.generate(f"Classify this reply as interested, not_interested or out_of_office: {full['snippet']}",
                            max_output_tokens=10).text.strip().lower()
        label = label if label in LABELS else "not_interested"
        gmail.users().messages().modify(userId="me", id=m["id"], body={"addLabelIds": [LABELS[label]]}).execute()
        if label == "interested":
            notify.slack(f":tada: {sender} is interested: \"{full['snippet'][:140]}\"", channel="#sales")
    return {"sorted": len(msgs)}