Build a User Memory Chatbot

import OpenAI from "openai";
import { RetainDB } from "@retaindb/sdk";

const openai = new OpenAI({ apiKey: process.env.OPENAI_API_KEY });
const db = new RetainDB({
  apiKey: process.env.RETAINDB_API_KEY,
  project: "production"
});

export async function answerUser(input: {
  userId: string;
  sessionId: string;
  message: string;
}) {
  const session = db.user(input.userId).session(input.sessionId);

  const { context } = await session.getContext(input.message);

  const completion = await openai.chat.completions.create({
    model: "gpt-4o-mini",
    messages: [
      {
        role: "system",
        content: `Use this memory only when relevant:\n${context}`
      },
      {
        role: "user",
        content: input.message
      }
    ]
  });

  const answer = completion.choices[0]?.message?.content ?? "";

  await session.remember([
    { role: "user", content: input.message },
    { role: "assistant", content: answer }
  ]);

  return { answer };
}