Draft email replies from your FAQ Google Doc

import base64
from datetime import date
from email.message import EmailMessage
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, credentials, http, state, ai, report_fetched

DOC_ID, INDEX = "your-doc-id", "https://faq-xxxx.svc.pinecone.io"

def pc(path, body):
    return http.post(INDEX + path, json=body, headers={"Api-Key": credentials.secret("PINECONE_API_KEY")}).json()

@on_trigger
def run(event):
    creds = Credentials(credentials.connection("google", "work"))
    if state.get("indexed") != date.today().isoformat():
        doc = build("docs", "v1", credentials=creds, cache_discovery=False).documents().get(documentId=DOC_ID).execute()
        paras = [p for p in ("".join(r["textRun"]["content"] for r in c.get("paragraph", {}).get("elements", []) if "textRun" in r)
                 for c in doc["body"]["content"]) if p.strip()]
        pc("/records/namespaces/faq/upsert", [{"_id": f"p{i}", "text": p} for i, p in enumerate(paras)])
        state.set("indexed", date.today().isoformat())
    gmail = build("gmail", "v1", credentials=creds, cache_discovery=False).users()
    since = state.get("since")
    msgs = gmail.messages().list(userId="me", q=f"in:inbox after:{since}" if since else "in:inbox newer_than:10m").execute().get("messages", [])
    report_fetched(len(msgs))
    if not msgs:
        return {"summary": "Nothing new"}
    for m in msgs:
        full = gmail.messages().get(userId="me", id=m["id"], format="full").execute()
        hdr = {h["name"]: h["value"] for h in full["payload"]["headers"]}
        hits = pc("/records/namespaces/faq/search", {"query": {"inputs": {"text": full["snippet"]}, "top_k": 4}})
        ctx = "\n".join(h["fields"]["text"] for h in hits["result"]["hits"])
        reply = ai.generate(f"FAQ:\n{ctx}\n\nCustomer email:\n{full['snippet']}\n\nWrite a short reply using only the FAQ.", max_output_tokens=500).text
        msg = EmailMessage()
        msg["To"], msg["Subject"] = hdr.get("From"), "Re: " + hdr.get("Subject", "")
        msg.set_content(reply)
        gmail.drafts().create(userId="me", body={"message": {"threadId": full["threadId"], "raw": base64.urlsafe_b64encode(msg.as_bytes()).decode()}}).execute()
    state.set("since", int(gmail.messages().get(userId="me", id=msgs[0]["id"], format="minimal").execute()["internalDate"]) // 1000 + 1)
    return {"drafts": len(msgs)}