Description

Add Apache AGE as a graph store provider for mem0's graph memory, providing an open-source PostgreSQL-based alternative to Neo4j. Apache AGE adds graph database capabilities (Cypher queries) to PostgreSQL, letting users run graph + relational + vector operations on a single Postgres instance.

Closes #3292

Type of change

• New feature (non-breaking change which adds functionality)

How Has This Been Tested?

62 tests total — 19 unit + 43 end-to-end integration tests, all passing.

Unit tests (tests/memory/test_apache_age_memory.py — 19 tests)

Run without any external dependencies (all mocked):

• TestCosineSimilarity (3) — identical, orthogonal, and zero vectors
• TestRetrieveNodesFromData (4) — normal extraction, malformed entity handling, missing keys, null tool_calls
• TestEstablishNodesRelationsFromData (3) — null LLM response, empty tool_calls, valid entities
• TestRemoveSpacesFromEntities (1) — space/case sanitization
• TestFindSimilarNode (3) — empty graph, threshold matching, agent_id filtering
• TestDeleteAll (1) — commit called
• TestGetAll (2) — result formatting, limit enforcement
• TestAdd (1) — orchestration returns correct shape
• TestSearch (1) — empty search output

E2E integration tests (tests/memory/test_apache_age_e2e.py — 43 tests)

Run against a real Apache AGE instance via Docker (docker run -d -p 15432:5432 apache/age):

• TestExecCypher (3) — vertex creation/return, scalar column mapping, empty results
• TestMergeNode (3) — first-create, idempotent merge with mentions increment, agent_id storage
• TestRelationships (2) — single and multiple relationship creation with type(r) queries
• TestEmbeddingSimilarity (5) — embedding roundtrip, exact match, threshold filtering, closest-pick, user_id isolation
• TestPublicAPICRUD (5) — get_all, user isolation, limit, delete_all, reset
• TestDeleteEntities (1) — selective relationship deletion preserving other relationships
• TestAddEntities (3) — new entity creation, merge to existing similar node, multiple relationships
• TestSearchGraphDB (2) — similarity-based relationship discovery, empty results
• TestAddSearchIntegration (2) — full add()→search()→delete_all() cycle, relationship update (delete old + add new)
• TestMultiTenantIsolation (1) — cross-user data isolation, delete doesn't affect other users
• TestAgentRunIdFiltering (4) — get_all/delete_all with agent_id filter, get_all/delete_all with run_id filter
• TestEdgeCases (7) — underscored names, apostrophe (known driver limitation), empty graph operations, idempotent merge, bidirectional relationships, self-referencing relationships
• TestSearchGraphDBFiltering (2) — search with agent_id, _find_similar_node with run_id
• TestFullLifecycleWithAgentId (1) — complete add→get_all→search→delete_all cycle scoped by agent_id
• TestMergeNodeTimestamp (1) — created timestamp preserved across merges

Test categories

• Unit Test
• Integration Test

Files Changed

Usage

from mem0 import Memory

config = {
    "graph_store": {
        "provider": "apache_age",
        "config": {
            "host": "localhost",
            "port": 5432,
            "database": "mem0_db",
            "username": "mem0_user",
            "password": "mem0_pass",
            "graph_name": "mem0_graph",
        },
    },
}

m = Memory.from_config(config_dict=config)

Checklist:

• My code follows the style guidelines of this project
• I have performed a self-review of my own code
• I have made corresponding changes to the documentation
• My changes generate no new warnings
• I have added tests that prove my fix is effective or that my feature works
• New and existing unit tests pass locally with my changes

Maintainer Checklist

• closes Add Apache AGE database support for Graph operations #3292
• Made sure Checks passed