Answer chat questions and take bookings into a sheet

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_event, credentials, http, ai

SHEET_ID, REPLY_URL = "your-spreadsheet-id", "https://chat.example.com/api/reply"
SCHEMA = {"type": "object", "properties": {k: {"type": "string"} for k in ["intent", "name", "date", "time", "party", "missing"]}}
HOST, NS = "https://kb-abc123.svc.aped-1234.pinecone.io", "faq"  # 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_event
def handle(event):
    body = event.json or {}
    msg = str(body.get("message", ""))[:1000]
    b = ai.generate(f"Is this a 'question' or a 'booking'? For a booking, extract name, date (YYYY-MM-DD), time, party size, and "
                    f"list any missing fields in 'missing'.\n{msg}", json_schema=SCHEMA, max_output_tokens=150).json
    if b.get("intent") == "booking" and not b.get("missing"):
        build("sheets", "v4", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False) \
            .spreadsheets().values().append(spreadsheetId=SHEET_ID, range="Bookings!A:E", valueInputOption="RAW",
                                            body={"values": [[b["name"], b["date"], b["time"], b["party"], event.received_at[:16]]]}).execute()
        reply = f"Thanks {b['name']}, you're booked for {b['party']} on {b['date']} at {b['time']}."
    elif b.get("intent") == "booking":
        reply = f"Happy to book that. Could you tell me your {b['missing']}?"
    else:
        faq = "\n".join(h["chunk_text"] for h in search(msg, 3))
        reply = ai.generate(f"Answer briefly using only this FAQ, or say we'll get back to them:\n{faq}\nQ: {msg}", max_output_tokens=150).text
    http.post(REPLY_URL, json={"session_id": body.get("session_id"), "text": reply})
    return {"intent": b.get("intent")}