Neo4j Graph Database

src/memos/graph_dbs/
├── base.py            # Abstract interface: BaseGraphDB
├── factory.py         # Factory to instantiate GraphDB from config
├── neo4j.py           # Neo4jGraphDB: production implementation

from memos.graph_dbs.factory import GraphStoreFactory
from memos.configs.graph_db import GraphDBConfigFactory

# Step 1: Build factory config
config = GraphDBConfigFactory(
    backend="neo4j",
    config={
        "uri": "bolt://localhost:7687",
        "user": "your_neo4j_user_name",
        "password": "your_password",
        "db_name": "memory_user1",
        "auto_create": True,
        "embedding_dimension": 768
    }
)

# Step 2: Instantiate the graph store
graph = GraphStoreFactory.from_config(config)

# Step 3: Add memory
graph.add_node(
    id="node-001",
    memory="Today I learned about retrieval-augmented generation.",
    metadata={"type": "WorkingMemory", "tags": ["RAG", "AI"], "timestamp": "2025-06-05", "sources": []}
)

Function Introduction:
1. Node Operations:
   Insert: add_node (Adds a single node)
           add_nodes_batch (Adds multiple nodes in batch)
   Query: get_node (Retrieves a single node)
          get_nodes (Retrieves multiple nodes)
          get_memory_count (Retrieves the count of nodes)
          node_not_exist (Checks if a node exists)
          search_by_embedding (Vector search supports adding filter conditions for filtering. For usage of the filter, refer to the function neo4j_example.example_complex_shared_db_search_filter for the complete method documentation.)
   Update: update_node (Updates a single node)
   Delete: delete_node (Deletes a single node)
           clear(deletes all associated nodes by the user_name attribute.)
           See neo4j_example.example_complex_shared_db_delete_memory for full method docs

2. Edge Operations:
   Insert: add_edge (Adds a triple/relation as a memory element)
   Query: get_edges (Retrieves multiple relations/edges)
          edge_exists (Checks if a relation/edge exists)
          get_children_with_embeddings (Retrieves a list of child nodes for the PARENT relation type)
          get_subgraph (Queries multi-hop nodes/retrieves a subgraph)
   Delete: delete_edge (Deletes a relation/edge)

3. Import/Export Operations:
   import_graph (Imports an entire graph from a serialized dictionary. Parameters: A dictionary containing all nodes and edges to load, format: {'nodes': [], 'edges': []})
   export_graph (Exports all graph nodes and edges in a structured format, with pagination support)

See src/memos/graph_dbs/base.py for full method docs.

config = GraphDBConfigFactory(
    backend="neo4j",
    config={
        "uri": "bolt://localhost:7687",
        "user": "neo4j",
        "password": "your_password",
        "db_name": "shared-graph",
        "user_name": "alice",
        "use_multi_db": False,
        "embedding_dimension": 768,
    },
)

config = GraphDBConfigFactory(
    backend="neo4j-community",
    config={
        "uri": "bolt://localhost:7687",
        "user": "neo4j",
        "password": "12345678",
        "db_name": "paper",
        "user_name": "bob",
        "auto_create": False,
        "embedding_dimension": 768,
        "use_multi_db": False,
        "vec_config": {
            "backend": "qdrant",
            "config": {
                "host": "localhost",
                "port": 6333,
                "collection_name": "neo4j_vec_db",
                "vector_dimension": 768,
                "distance_metric": "cosine"
            },
        },
    },
)