Weaviate - Mem0

Weaviate

Weaviate is an open-source vector search engine. It allows efficient storage and retrieval of high-dimensional vector embeddings, enabling powerful search and retrieval capabilities.

Installation

Python

pip install weaviate-client

TypeScript

npm install weaviate-client

Usage

Python

import os
from mem0 import Memory

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

config = {
    "vector_store": {
        "provider": "weaviate",
        "config": {
            "collection_name": "test",
            "cluster_url": "http://localhost:8080",
            "auth_client_secret": None,
        }
    }
}

m = Memory.from_config(config)
messages = [\
    {"role": "user", "content": "I'm planning to watch a movie tonight. Any recommendations?"},\
    {"role": "assistant", "content": "How about a thriller movie? 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"})

TypeScript

import { Memory } from "mem0ai/oss";

const config = {
  vectorStore: {
    provider: "weaviate",
    config: {
      collectionName: "test",
      embeddingModelDims: 1536,
      clusterUrl: "http://localhost:8080",
    },
  },
};

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 a thriller movie? 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",
  },
});

The TypeScript SDK picks the connection mode from the config you pass:

• clusterUrl pointing at localhost connects to a local instance.
• clusterUrl plus apiKey connects to a Weaviate Cloud cluster (for example https://my-cluster.weaviate.cloud).
• Any other clusterUrl without an apiKey connects to a custom deployment, using the host and port from the URL.

You can also pass a pre-configured client (a WeaviateClient instance) to reuse an existing connection.

Config

Here are the parameters available for configuring Weaviate: