A Telegram assistant that remembers what you tell it

from datetime import date
from clickruntime import on_event, credentials, http, ai

OWNER_CHAT_ID = 123456789

def embed(text):
    r = http.post("https://api.pinecone.io/embed", headers={"Api-Key": credentials.secret("PINECONE_API_KEY"), "X-Pinecone-API-Version": "2025-01"},
                  json={"model": "multilingual-e5-large", "inputs": [{"text": text}]})
    return r.json()["data"][0]["values"]

@on_event
def handle(event):
    msg = (event.json or {}).get("message") or {}
    if msg.get("chat", {}).get("id") != OWNER_CHAT_ID:
        return {"ignored": True}
    text, index = msg.get("text", ""), credentials.secret("PINECONE_HOST")
    pc = {"Api-Key": credentials.secret("PINECONE_API_KEY")}
    if text.lower().startswith("remember:"):
        fact = text[9:].strip()
        http.post(f"https://{index}/vectors/upsert", headers=pc, json={"vectors": [
            {"id": str(msg["message_id"]), "values": embed(fact), "metadata": {"text": fact, "date": date.today().isoformat()}}]})
        reply = "Saved."
    else:
        matches = http.post(f"https://{index}/query", headers=pc, json={"vector": embed(text), "topK": 8, "includeMetadata": True}).json()["matches"]
        memories = "\n".join(f"{m['metadata']['date']}: {m['metadata']['text']}" for m in matches)
        reply = ai.generate(f"Memories:\n{memories}\n\nQuestion: {text}", system="Answer from the memories only. Prefer the newest if they disagree. Say if nothing matches.", max_output_tokens=300).text
    http.post(f"https://api.telegram.org/bot{credentials.secret('TELEGRAM_BOT_TOKEN')}/sendMessage", json={"chat_id": OWNER_CHAT_ID, "text": reply})
    return {"saved": text.lower().startswith("remember:")}