feat: update SelfMemory memory provider with improved encryption handling and async item addition

Author: shrijayan

packages/nvidia_nat_selfmemory/pyproject.toml

@@ -56,7 +56,7 @@ dependencies = [
   # version when adding a new package. If unsure, default to using `~=` instead of `==`. Does not apply to nvidia-nat packages.
   # Keep sorted!!!
   "nvidia-nat-core == {version}",
-  "selfmemory>=0.9.4",
+  "selfmemory>=0.9.4,<2.0.0",
 ]
 
 [tool.setuptools_dynamic_dependencies.optional-dependencies]

packages/nvidia_nat_selfmemory/src/nat/meta/pypi.md

@@ -1,3 +1,22 @@
+<!--
+SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+-->
+
+![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/banner.png "NeMo Agent Toolkit banner image")
+
 # nvidia-nat-selfmemory
 
 SelfMemory memory provider plugin for the NVIDIA NeMo Agent Toolkit.

packages/nvidia_nat_selfmemory/src/nat/plugins/selfmemory/memory.py

@@ -53,11 +53,14 @@ async def selfmemory_provider(config: SelfMemoryProviderConfig, builder: Builder
             "config": config.llm_config,
         }
 
-    if config.encryption_key:
-        os.environ.setdefault("MASTER_ENCRYPTION_KEY", config.encryption_key)
+    encryption_key = config.encryption_key or os.environ.get("MASTER_ENCRYPTION_KEY")
 
-    memory = SelfMemory(config=config_dict)
+    if encryption_key:
+        os.environ["MASTER_ENCRYPTION_KEY"] = encryption_key
 
-    yield SelfMemoryEditor(memory)
+    memory = SelfMemory(config=config_dict)
 
-    memory.close()
+    try:
+        yield SelfMemoryEditor(memory)
+    finally:
+        memory.close()

packages/nvidia_nat_selfmemory/src/nat/plugins/selfmemory/selfmemory_editor.py

@@ -36,9 +36,11 @@ def __init__(self, backend: SelfMemory):
 
     async def add_items(self, items: list[MemoryItem]) -> None:
         """Insert multiple MemoryItems into SelfMemory."""
-        for item in items:
-            kwargs = memory_item_to_add_kwargs(item)
-            await asyncio.to_thread(self._backend.add, **kwargs)
+        coroutines = [
+            asyncio.to_thread(self._backend.add, **memory_item_to_add_kwargs(item))
+            for item in items
+        ]
+        await asyncio.gather(*coroutines)
 
     async def search(self, query: str, top_k: int = 5, **kwargs) -> list[MemoryItem]:
         """Retrieve items relevant to the given query."""

pyproject.toml

@@ -67,6 +67,7 @@ openpipe-art = ["nvidia-nat-openpipe-art == {version}"]
 opentelemetry = ["nvidia-nat-opentelemetry == {version}"]
 phoenix = ["nvidia-nat-phoenix == {version}"]
 profiler = ["nvidia-nat-profiler == {version}"]
+selfmemory = ["nvidia-nat-selfmemory == {version}"]
 rag = ["nvidia-nat-rag == {version}"]
 ragas = ["nvidia-nat-ragas == {version}"]
 ragaai = ["nvidia-nat-ragaai == {version}"]