Memory for WhatsApp agents.

pip install 'niadra[whatsapp]'   # no framework dependency

"""A WhatsApp Cloud API webhook: each message is the customer's turn, each reply the agent's."""

import os

from fastapi import FastAPI, Request, Response

from niadra import Niadra
from niadra.integrations.whatsapp import parse_webhook, sent, subscribe

niadra = Niadra(channel="whatsapp")
app = FastAPI()


@app.get("/whatsapp")
def challenge(request: Request) -> Response:
    result = subscribe(request.query_params, os.environ["WHATSAPP_VERIFY_TOKEN"])
    return Response(result.text(), result.status, media_type=result.content_type)


@app.post("/whatsapp")
async def inbound(request: Request) -> Response:
    messages = parse_webhook(await request.body(), request.headers, os.environ["META_APP_SECRET"])
    if messages is None:
        return Response(status_code=401)
    for message in messages:
        with niadra.conversation(f"wa-{message.wa_id}", subject=message.subject) as chat:
            message.record(chat)
            niadra.flush()
            context = chat.context()
            reply = your_model(context.system_block, context.turn_block, message.text)
            sent(chat, reply, send_whatsapp(message.wa_id, reply))
    return Response(status_code=200)

npm install @niadra/sdk   # @niadra/sdk/whatsapp

// A WhatsApp Cloud API webhook on Hono: reads Meta's delivery, gives the agent the customer's
// context, sends the answer through the Graph API and records both turns.
import { Hono } from "hono";
import { Niadra } from "@niadra/sdk";
import { readWhatsApp, recordInbound, recordOutbound, whatsAppChallenge } from "@niadra/sdk/whatsapp";

const niadra = new Niadra();
const env = (name: string): string => process.env[name] ?? "";

export const app = new Hono();

app.get("/whatsapp", (c) => {
  const { status, body } = whatsAppChallenge(new URL(c.req.url).searchParams, env("META_VERIFY_TOKEN"));
  return c.text(body, status as 200);
});

app.post("/whatsapp", async (c) => {
  const { status, messages } = await readWhatsApp(await c.req.text(), c.req.raw.headers, { appSecret: env("META_APP_SECRET") });
  for (const inbound of messages) {
    const convo = niadra.conversation({ subject: inbound.subject, channel: "whatsapp", conversation_id: `wa:${inbound.waId}` });
    recordInbound(convo, inbound);
    const ctx = await convo.context();
    convo.markInjected(ctx);
    const reply = await answer(`You are Acme's WhatsApp agent.\n\n${ctx.text}`, ctx.suffix, inbound.text);
    const sent = await fetch(`https://graph.facebook.com/v23.0/${inbound.phoneNumberId}/messages`, {
      method: "POST",
      headers: { authorization: `Bearer ${env("META_ACCESS_TOKEN")}`, "content-type": "application/json" },
      body: JSON.stringify({ messaging_product: "whatsapp", to: inbound.waId, type: "text", text: { body: reply } }),
    });
    recordOutbound(convo, reply, await sent.json());
  }
  return c.body(null, status as 200);
});

export default app;