import os
from mem0 import Memory

os.environ["OPENAI_API_KEY"] = "sk-xx"

config = {
    "vector_store": {
        "provider": "mongodb",
        "config": {
            "db_name": "mem0-db",
            "collection_name": "mem0-collection",
            "mongo_uri": "mongodb://username:password@localhost:27017"
        }
    }
}

m = Memory.from_config(config)
messages = [
    {
        "role": "user",
        "content": "I'm planning to watch a movie tonight. Any recommendations?",
    },
    {
        "role": "assistant",
        "content": "How about thriller movies? They can be quite engaging.",
    },
    {
        "role": "user",
        "content": "I’m not a big fan of thriller movies but I love sci-fi movies.",
    },
    {
        "role": "assistant",
        "content": "Got it! I'll avoid thriller recommendations and suggest sci-fi movies in the future.",
    },
]

m.add(messages, user_id="alice", metadata={"category": "movies"})

import { Memory } from "mem0ai/oss";

const config = {
  vectorStore: {
    provider: "mongodb",
    config: {
      dbName: "mem0-db",
      collectionName: "mem0-collection",
      url: "mongodb://username:password@localhost:27017",
    },
  },
};

const memory = new Memory(config);

const messages = [
  {
    role: "user",
    content: "I'm planning to watch a movie tonight. Any recommendations?",
  },
  {
    role: "assistant",
    content: "How about thriller movies? They can be quite engaging.",
  },
  {
    role: "user",
    content: "I’m not a big fan of thriller movies but I love sci-fi movies.",
  },
  {
    role: "assistant",
    content:
      "Got it! I'll avoid thriller recommendations and suggest sci-fi movies in the future.",
  },
];

await memory.add(messages, {
  userId: "alice",
  metadata: {
    category: "movies",
  },
});