Answer questions from your Drive documents, with the source named

from clickruntime import on_event, credentials, http, notify, ai

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_event
def handle(event):
    question = str((event.json or {}).get("question", "")).strip()[:500]
    if not question:
        return {"skipped": "no question"}
    hits = search(question)
    context = "\n\n".join(f"[{h.get('file_name')}] {h['chunk_text']}" for h in hits)
    answer = ai.generate(f"Answer using only these passages. Name the file in brackets and quote the key line. If they "
                         f"don't answer it, say so.\n\n{context}\n\nQuestion: {question}", max_output_tokens=500).text
    notify.slack(f"*Q:* {question}\n{answer}", channel="#ask-docs")
    return {"answer": answer, "sources": sorted({h.get("file_name") for h in hits})}