A Telegram assistant that searches the web when it needs to

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

OWNER_CHAT_ID = 123456789

@on_event
def handle(event):
    msg = (event.json or {}).get("message") or {}
    text = msg.get("text", "").strip()
    if msg.get("chat", {}).get("id") != OWNER_CHAT_ID or not text:
        return {"ignored": True}
    if text == "/reset":
        state.set("history", [])
        reply = "Starting fresh."
    else:
        history = state.get("history") or []
        query = ai.generate(f"Message: {text}\nIf answering needs current web information, reply with a Google search query. Otherwise reply NONE.", max_output_tokens=30).text.strip()
        found = ""
        if query != "NONE":
            res = http.get("https://serpapi.com/search", params={"engine": "google", "q": query, "api_key": credentials.secret("SERPAPI_API_KEY")}).json()
            found = "\n".join(f"- {r['title']}: {r.get('snippet', '')}" for r in res.get("organic_results", [])[:5])
        chat = "\n".join(f"{h['who']}: {h['text']}" for h in history)
        reply = ai.generate(f"{chat}\nme: {text}\n\nSearch results:\n{found or 'none'}", system="You are a concise personal assistant.", max_output_tokens=400).text.strip()
        state.set("history", (history + [{"who": "me", "text": text}, {"who": "assistant", "text": reply}])[-10:])
    http.post(f"https://api.telegram.org/bot{credentials.secret('TELEGRAM_BOT_TOKEN')}/sendMessage", json={"chat_id": OWNER_CHAT_ID, "text": reply})
    return {"ok": True}