Memory for voice agents on LiveKit.

pip install 'niadra[livekit]'   # livekit-agents 1.8.3 or newer, below 2

"""A LiveKit voice agent that starts every call knowing the caller. Run: python livekit_agent.py dev"""

from livekit.agents import AgentServer, AgentSession, JobContext, cli, inference

from niadra import AsyncNiadra
from niadra.integrations.livekit import NiadraAgent, conversation_for

niadra = AsyncNiadra(channel="voice")
server = AgentServer()


@server.rtc_session()
async def entrypoint(ctx: JobContext) -> None:
    await ctx.connect()
    caller = await ctx.wait_for_participant()
    conversation = conversation_for(niadra, caller, room=ctx.room)  # the SIP number and call id
    session = AgentSession(
        stt=inference.STT("deepgram/nova-3"),
        llm=inference.LLM("openai/gpt-4.1-mini"),
        tts=inference.TTS("cartesia/sonic-2"),
    )
    agent = NiadraAgent(
        conversation, instructions="You are Acme's support agent. Be brief.", agent_memory=True
    )
    await session.start(agent, room=ctx.room)


if __name__ == "__main__":
    cli.run_app(server)

npm install @niadra/sdk @livekit/agents   # @livekit/agents 1.9, as an optional peer dependency

import { type JobContext, ServerOptions, cli, defineAgent, voice } from "@livekit/agents";
import { fileURLToPath } from "node:url";
import { Niadra } from "@niadra/sdk";
import { NiadraAgent, NiadraMemory, attestationProof, sipConversationId, sipSubject } from "@niadra/sdk/livekit";

const niadra = new Niadra();

export default defineAgent({
  entry: async (ctx: JobContext) => {
    await ctx.connect();
    const caller = await ctx.waitForParticipant();
    const conversation = niadra.conversation({
      subject: sipSubject(caller),
      channel: "voice",
      conversation_id: sipConversationId(caller, ctx.room.name ?? "room"),
    });
    const memory = new NiadraMemory({
      conversation,
      // Map the carrier's STIR/SHAKEN header to this attribute in your SIP trunk's header settings.
      verify: attestationProof(caller.attributes["sip.h.x-stir-verstat"]),
    });
    const session = new voice.AgentSession({
      stt: "deepgram/nova-3",
      llm: "openai/gpt-4.1-mini",
      tts: "cartesia/sonic-3",
    });
    memory.attach(session);
    await session.start({
      agent: new NiadraAgent({ instructions: "You answer the phone for Acme Energy. Be brief.", memory }),
      room: ctx.room,
    });
  },
});

if (process.argv[1] === fileURLToPath(import.meta.url)) {
  cli.runApp(new ServerOptions({ agent: fileURLToPath(import.meta.url) }));
}