Draft email replies from your knowledge base

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

LABEL_ID = "Label_321"  # "Processed"
HOST, NS = "https://kb-abc123.svc.aped-1234.pinecone.io", "help"  # 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()
    q = "in:inbox -label:Processed -category:promotions -category:updates -from:noreply newer_than:1d"
    new = gmail.messages().list(userId="me", q=q).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(d["chunk_text"] for d in search(f"{h.get('Subject')} {msg['snippet']}", 4))
        reply = ai.generate(f"Write a friendly reply using only these docs. If they don't answer it, say you'll check and ask "
                            f"for details.\nDocs:\n{docs}\n\nEmail: {h.get('Subject')}\n{msg['snippet']}", max_output_tokens=400).text
        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": [LABEL_ID]}).execute()
    return {"drafted": len(new)}