feat: Add optional Neptune Analytics graph backend for mem0_memory tool (#230)

---------

Signed-off-by: Andy Kwok <[email protected]>

Author: andy-k-improving

README.md

@@ -828,11 +828,25 @@ The Mem0 Memory Tool supports three different backend configurations:
    - Uses FAISS as the local vector store backend
    - Requires faiss-cpu package for local vector storage
 
+4. **Neptune Analytics** (Optional Graph backend for search enhancement):
+   - Uses Neptune Analytics as the graph store backend to enhance memory recall.
+   - Requires AWS credentials and Neptune Analytics configuration
+   ```
+   # Configure your Neptune Analytics graph ID in the .env file:
+   export NEPTUNE_ANALYTICS_GRAPH_IDENTIFIER=sample-graph-id
+   
+   # Configure your Neptune Analytics graph ID in Python code:
+   import os
+   os.environ['NEPTUNE_ANALYTICS_GRAPH_IDENTIFIER'] = "g-sample-graph-id"
+   
+   ```
+
 | Environment Variable | Description | Default | Required For |
 |----------------------|-------------|---------|--------------|
 | MEM0_API_KEY | Mem0 Platform API key | None | Mem0 Platform |
 | OPENSEARCH_HOST | OpenSearch Host URL | None | OpenSearch |
 | AWS_REGION | AWS Region for OpenSearch | us-west-2 | OpenSearch |
+| NEPTUNE_ANALYTICS_GRAPH_IDENTIFIER | Neptune Analytics Graph Identifier | None | Neptune Analytics |
 | DEV | Enable development mode (bypasses confirmations) | false | All modes |
 | MEM0_LLM_PROVIDER | LLM provider for memory processing | aws_bedrock | All modes |
 | MEM0_LLM_MODEL | LLM model for memory processing | anthropic.claude-3-5-haiku-20241022-v1:0 | All modes |
@@ -846,6 +860,7 @@ The Mem0 Memory Tool supports three different backend configurations:
 - If `MEM0_API_KEY` is set, the tool will use the Mem0 Platform
 - If `OPENSEARCH_HOST` is set, the tool will use OpenSearch
 - If neither is set, the tool will default to FAISS (requires `faiss-cpu` package)
+- If `NEPTUNE_ANALYTICS_GRAPH_IDENTIFIER` is set, the tool will configure Neptune Analytics as graph store to enhance memory search
 - LLM configuration applies to all backend modes and allows customization of the language model used for memory processing
 
 #### Bright Data Tool

src/strands_tools/mem0_memory.py

@@ -204,13 +204,33 @@ def _initialize_client(self, config: Optional[Dict] = None) -> Any:
             logger.debug("Using Mem0 Platform backend (MemoryClient)")
             return MemoryClient()
 
+        if os.environ.get("NEPTUNE_ANALYTICS_GRAPH_IDENTIFIER"):
+            logger.debug("Using Neptune Analytics graph backend (Mem0Memory with Neptune Analytics)")
+            config = self._configure_neptune_analytics_backend(config)
+
         if os.environ.get("OPENSEARCH_HOST"):
             logger.debug("Using OpenSearch backend (Mem0Memory with OpenSearch)")
             return self._initialize_opensearch_client(config)
 
         logger.debug("Using FAISS backend (Mem0Memory with FAISS)")
         return self._initialize_faiss_client(config)
 
+    def _configure_neptune_analytics_backend(self, config: Optional[Dict] = None) -> Dict:
+        """Initialize a Mem0 client with Neptune Analytics graph backend.
+
+        Args:
+            config: Optional configuration dictionary to override defaults.
+
+        Returns:
+            An configuration dict with graph backend.
+        """
+        config = config or {}
+        config["graph_store"] = {
+            "provider": "neptune",
+            "config": {"endpoint": f"neptune-graph://{os.environ.get('NEPTUNE_ANALYTICS_GRAPH_IDENTIFIER')}"},
+        }
+        return config
+
     def _initialize_opensearch_client(self, config: Optional[Dict] = None) -> Mem0Memory:
         """Initialize a Mem0 client with OpenSearch backend.
 

tests/test_mem0.py

@@ -424,6 +424,15 @@ def test_mem0_service_client_init(mock_opensearch, mock_mem0_memory, mock_sessio
         client = Mem0ServiceClient()
         assert client.region == os.environ.get("AWS_REGION", "us-west-2")
 
+    # Test with optional Graph backend
+    with patch.dict(
+        os.environ,
+        {"OPENSEARCH_HOST": "test.opensearch.amazonaws.com", "NEPTUNE_ANALYTICS_GRAPH_IDENTIFIER": "g-5aaaaa1234"},
+    ):
+        client = Mem0ServiceClient()
+        assert client.region == os.environ.get("AWS_REGION", "us-west-2")
+        assert client.mem0 is not None
+
     # Test with custom config (OpenSearch)
     custom_config = {
         "embedder": {"provider": "custom", "config": {"model": "custom-model"}},