Draft replies to support emails from your Drive help docs

import base64
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, report_fetched, credentials, http, notify, ai
DRAFTED_LABEL_ID = "Label_789"
HOST, NS = "https://docs-abc123.svc.aped-1234.pinecone.io", "company"  # your index host and namespace
def search(text, k=5):
    r = http.post(f"{HOST}/records/namespaces/{NS}/search", json={"query": {"inputs": {"text": text}, "top_k": k}}, headers={
        "Api-Key": credentials.secret("PINECONE_API_KEY"), "X-Pinecone-API-Version": "2025-04"}).json()
    return [h["fields"] for h in r["result"]["hits"]]

@on_trigger
def run(event):
    gmail = build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).users()
    new = gmail.messages().list(userId="me", q="label:Support -label:Drafted newer_than:1d").execute().get("messages", [])
    report_fetched(len(new))
    if not new:
        return {"new": 0}
    for m in new:
        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"]}
        docs = "\n\n".join(f"[{d.get('file_name')}] {d['chunk_text']}" for d in search(f"{h.get('Subject')} {msg['snippet']}", 4))
        reply = ai.generate(f"Write a helpful support reply using only these help docs. If they don't cover it, reply exactly "
                            f"NO_ANSWER.\n{docs}\n\nCustomer email: {h.get('Subject')}\n{msg['snippet']}", max_output_tokens=400).text
        if "NO_ANSWER" in reply:
            notify.slack(f"Needs a person: {h.get('Subject')} from {h.get('From')}", channel="#support")
        else:
            gmail.drafts().create(userId="me", body={"message": {"threadId": msg["threadId"], "raw": base64.urlsafe_b64encode(
                f"To: {h.get('From')}\r\nSubject: Re: {h.get('Subject')}\r\n\r\n{reply}".encode()).decode()}}).execute()
        gmail.messages().modify(userId="me", id=m["id"], body={"addLabelIds": [DRAFTED_LABEL_ID]}).execute()