FEAT: Provide Neptune DB support for Mem0 Graph backend

src/strands_tools/mem0_memory.py

@@ -209,10 +209,22 @@ def _initialize_client(self, config: Optional[Dict] = None) -> Any:
             merged_config = self._append_faiss_config(config)
 
         # Graph backend providers
+
+        # Graph backend providers
+        if os.environ.get("NEPTUNE_ANALYTICS_GRAPH_IDENTIFIER") and os.environ.get("NEPTUNE_DATABASE_ENDPOINT"):
+            raise RuntimeError("""Conflicting backend configurations:
+                Both NEPTUNE_ANALYTICS_GRAPH_IDENTIFIER and NEPTUNE_DATABASE_ENDPOINT environment variables are set.
+                Please specify only one graph backend.""")
+
         if os.environ.get("NEPTUNE_ANALYTICS_GRAPH_IDENTIFIER"):
             logger.debug("Using Neptune Analytics graph backend (Mem0Memory with Neptune Analytics)")
             merged_config = self._append_neptune_analytics_graph_config(merged_config)
 
+        elif os.environ.get("NEPTUNE_DATABASE_ENDPOINT"):
+            logger.debug("Using Neptune Database graph backend (Mem0Memory with Neptune Database)")
+            merged_config = self._append_neptune_database_backend(merged_config)
+
+
         return Mem0Memory.from_config(config_dict=merged_config)
 
     def _append_neptune_analytics_vector_config(self, config: Optional[Dict] = None) -> Dict:
@@ -234,6 +246,26 @@ def _append_neptune_analytics_vector_config(self, config: Optional[Dict] = None)
         }
         return self._merge_config(config)
 
+    def _append_neptune_database_backend(self, config: Optional[Dict] = None) -> Dict:
+        """Update incoming configuration dictionary to include the configuration of Neptune Database graph backend.
+
+        Args:
+            config: Optional configuration dictionary to override defaults.
+
+        Returns:
+            An configuration dict with graph backend.
+        """
+        config = config or {}
+        config["graph_store"] = {
+            "provider": "neptunedb",
+            "config": {"endpoint": f"neptune-db://{os.environ.get('NEPTUNE_DATABASE_ENDPOINT')}"},
+        }
+        # To retrieve cosine similarity score instead for Faiss.
+        if "faiss" == config.get("vector_store", {}).get("provider"):
+            config["vector_store"]["config"]["distance_strategy"] = "cosine"
+
+        return config
+
     def _append_opensearch_config(self, config: Optional[Dict] = None) -> Dict:
         """Update incoming configuration dictionary to include the configuration of OpenSearch vector backend.
 

tests/test_mem0.py

@@ -445,6 +445,16 @@ def test_mem0_service_client_init(mock_opensearch, mock_mem0_memory, mock_sessio
         client = Mem0ServiceClient()
         assert client.mem0 is not None
 
+    # Test with Neptune Database with OpenSearch
+    with patch.dict(
+            os.environ,
+            {"OPENSEARCH_HOST": "test.opensearch.amazonaws.com",
+             "NEPTUNE_DATABASE_ENDPOINT": "xxx.us-west-2.neptune.amazonaws.com"},
+    ):
+        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"}},