Memory for agents on the OpenAI Agents SDK.

pip install 'niadra[openai-agents]'   # openai-agents 0.22.3 or newer, below 1

"""An OpenAI Agents SDK agent with the customer's memory."""

import asyncio

from agents import Agent, Runner

from niadra import AsyncNiadra, phone
from niadra.integrations.openai_agents import NiadraAgentsMemory

niadra = AsyncNiadra(channel="chat")


async def main() -> None:
    async with niadra.conversation("thread-81", subject=phone("+5511912345678")) as conversation:
        memory = NiadraAgentsMemory(conversation, agent_memory=True)
        agent = Agent(
            name="Support", instructions="You are Acme's agent.", model="gpt-4.1", tools=memory.tools
        )
        result = await Runner.run(
            agent, "Where is my replacement lid?", hooks=memory.hooks, run_config=memory.run_config()
        )
        print(result.final_output)
    await niadra.close()


asyncio.run(main())

npm install @niadra/sdk @openai/agents   # @openai/agents 0.18, as an optional peer dependency

import { Agent, Runner } from "@openai/agents";
import { Niadra, handles } from "@niadra/sdk";
import { NiadraSession, niadraInstructions, niadraRunHooks, niadraTools } from "@niadra/sdk/openai-agents";

const niadra = new Niadra();
const runner = new Runner();

/** One customer message in; `userId` comes from your session, never from the model. */
export async function reply(userId: string, chatId: string, text: string): Promise<string> {
  const convo = niadra.conversation({ subject: handles.appUserId(userId), channel: "web_chat", conversation_id: chatId });
  const billing = new Agent({ name: "Billing", instructions: niadraInstructions("You handle invoices and credits.", convo), tools: niadraTools(convo) });
  const support = new Agent({
    name: "Support",
    instructions: niadraInstructions("You are Acme's support agent. Hand billing questions to Billing.", convo),
    tools: niadraTools(convo),
    handoffs: [billing],
  });
  const stop = niadraRunHooks(runner, convo);
  try {
    const result = await runner.run(support, text, { session: new NiadraSession(convo) });
    return String(result.finalOutput ?? "");
  } finally {
    stop();
  }
}