pnpm add @axiomkit/mongodb mongodb

import { createAgent } from "@axiomkit/core";
import { createMongoMemory } from "@axiomkit/mongodb";
import { groq } from "@ai-sdk/groq";

const memory = createMongoMemory({
  connectionString: process.env.MONGODB_URI,
  database: "axiomkit",
  collection: "memories",
});

const agent = createAgent({
  model: groq("gemma2-9b-it"),
  memory,
});

await agent.start();

interface MongoMemoryConfig {
  connectionString: string;     // MongoDB connection string
  database: string;             // Database name
  collection: string;           // Collection name
  encryption?: {                // Encryption settings
    enabled: boolean;
    key: string;
  };
  options?: {                   // Memory options
    maxSize?: number;           // Maximum memories to store
    ttl?: number;               // Time to live in milliseconds
  };
}

const memory = createMongoMemory({
  connectionString: process.env.MONGODB_URI,
  database: "axiomkit",
  collection: "memories",
  encryption: {
    enabled: true,
    key: process.env.ENCRYPTION_KEY,
  },
  options: {
    maxSize: 10000,
    ttl: 30 * 24 * 60 * 60 * 1000, // 30 days
  },
});

// Store memory
await agent.memory.store("episodic", {
  userId: "user123",
  content: "User asked about weather",
  timestamp: Date.now(),
});

// Retrieve memories
const memories = await agent.memory.retrieve("episodic", {
  userId: "user123",
  limit: 10,
});

// Search memories
const results = await agent.memory.search("weather", {
  type: "episodic",
  limit: 5,
});