Zero Entropy

pip install zeroentropy

from mem0 import Memory

config = {
    "vector_store": {
        "provider": "chroma",
        "config": {
            "collection_name": "my_memories",
            "path": "./chroma_db"
        }
    },
    "llm": {
        "provider": "openai",
        "config": {
            "model": "gpt-4o-mini"
        }
    },
    "rerank": {
        "provider": "zero_entropy",
        "config": {
            "model": "zerank-1",  # or "zerank-1-small"
            "api_key": "your-zero-entropy-api-key",  # or set ZERO_ENTROPY_API_KEY
            "top_k": 5
        }
    }
}

memory = Memory.from_config(config)

export ZERO_ENTROPY_API_KEY="your-api-key"

import os
from mem0 import Memory

# Set API key
os.environ["ZERO_ENTROPY_API_KEY"] = "your-api-key"

# Initialize memory with Zero Entropy reranker
config = {
    "vector_store": {"provider": "chroma"},
    "llm": {"provider": "openai", "config": {"model": "gpt-4o-mini"}},
    "rerank": {"provider": "zero_entropy", "config": {"model": "zerank-1"}}
}

memory = Memory.from_config(config)

# Add memories
messages = [
    {"role": "user", "content": "I love Italian pasta, especially carbonara"},
    {"role": "user", "content": "Japanese sushi is also amazing"},
    {"role": "user", "content": "I enjoy cooking Mediterranean dishes"}
]

memory.add(messages, user_id="alice")

# Search with reranking
results = memory.search("What Italian food does the user like?", filters={"user_id": "alice"})

for result in results['results']:
    print(f"Memory: {result['memory']}")
    print(f"Vector Score: {result['score']:.3f}")
    print(f"Rerank Score: {result['rerank_score']:.3f}")
    print()