Draft replies to refund requests with the order details

import base64
from email.mime.text import MIMEText
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, credentials, http, state, ai, report_fetched

STORE = "your-store.myshopify.com"

def gql(query, variables=None):
    resp = http.post(f"https://{STORE}/admin/api/2025-07/graphql.json", json={"query": query, "variables": variables or {}},
                     headers={"X-Shopify-Access-Token": credentials.connection("shopify", "store")})
    return resp.raise_for_status() or resp.json()["data"]

SCHEMA = {"type": "object", "properties": {"order": {"type": "string"}, "reason": {"type": "string"}}}

@on_trigger
def run(event):
    gmail = build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).users()
    seen = set(state.get("seen") or [])
    msgs = [m for m in gmail.messages().list(userId="me", q="refund newer_than:2d -from:me").execute().get("messages", []) if m["id"] not in seen]
    report_fetched(len(msgs))
    if not msgs:
        return {"drafts": 0}
    for m in msgs:
        msg = gmail.messages().get(userId="me", id=m["id"], format="metadata", metadataHeaders=["From", "Subject"]).execute()
        h = {x["name"]: x["value"] for x in msg["payload"]["headers"]}
        sender = h["From"].split("<")[-1].strip(">")
        ask = ai.generate(msg["snippet"], system="Extract the order number (or empty) and refund reason.", json_schema=SCHEMA, max_output_tokens=60).json
        q = f"name:{ask['order']}" if ask["order"] else f"email:{sender}"
        found = gql(f'{{ orders(first: 1, reverse: true, query: "{q}") {{ nodes {{ name createdAt }} }} }}')["orders"]["nodes"]
        facts = f"Order {found[0]['name']} placed {found[0]['createdAt'][:10]}, return window 30 days" if found else "No order found; ask for the order number"
        body = ai.generate(f"Customer wrote: {msg['snippet']}\nFacts: {facts}", system="Write a short, kind reply following the facts. Sign off as the Store team.", max_output_tokens=300).text
        raw = MIMEText(body)
        raw["To"], raw["Subject"] = sender, "Re: " + h.get("Subject", "")
        gmail.drafts().create(userId="me", body={"message": {"threadId": msg["threadId"], "raw": base64.urlsafe_b64encode(raw.as_bytes()).decode()}}).execute()
        seen.add(m["id"])
    state.set("seen", sorted(seen)[-500:])
    return {"drafts": len(msgs)}