Chat with a Telegram assistant that remembers you

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

DOC_ID, MY_CHAT_ID = "your-doc-id", 123456789

@on_event
def handle(event):
    msg = (event.json or {}).get("message") or {}
    if msg.get("chat", {}).get("id") != MY_CHAT_ID or not msg.get("text"):
        return {"skipped": True}
    docs = build("docs", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).documents()
    doc = docs.get(documentId=DOC_ID).execute()
    memory = "".join(r["textRun"]["content"] for c in doc["body"]["content"] for r in c.get("paragraph", {}).get("elements", []) if "textRun" in r)
    out = ai.generate(f"Notes about me:\n{memory[-20000:]}\n\nMy message: {msg['text']}",
                      system="You are my personal assistant. Answer using the notes. If my message states a lasting fact, return it as new_fact, else empty.",
                      max_output_tokens=700, json_schema={"type": "object", "properties": {"reply": {"type": "string"}, "new_fact": {"type": "string"}},
                                                          "required": ["reply", "new_fact"]}).json
    if out["new_fact"]:
        end = doc["body"]["content"][-1]["endIndex"] - 1
        docs.batchUpdate(documentId=DOC_ID, body={"requests": [{"insertText": {"location": {"index": end},
                         "text": f"\n{date.today().isoformat()}: {out['new_fact']}"}}]}).execute()
    http.post(f"https://api.telegram.org/bot{credentials.secret('TELEGRAM_BOT_TOKEN')}/sendMessage",
              json={"chat_id": MY_CHAT_ID, "text": out["reply"]})
    return {"remembered": bool(out["new_fact"])}