nvidia-nat-mem0ai 1.2.0a20250813__py3-none-any.whl

nat/meta/pypi.md

@@ -0,0 +1,23 @@
+<!--
+SPDX-FileCopyrightText: Copyright (c) 2025, 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/aiqtoolkit_banner.png "NeMo Agent toolkit banner image")
+
+# NVIDIA NeMo Agent Toolkit Subpackage
+This is a subpackage for Mem0 memory integration in NeMo Agent toolkit.
+
+For more information about the NVIDIA NeMo Agent toolkit, please visit the [NeMo Agent toolkit GitHub Repo](https://github.com/NVIDIA/NeMo-Agent-Toolkit).

nat/plugins/mem0ai/mem0_editor.py

@@ -0,0 +1,112 @@
+# SPDX-FileCopyrightText: Copyright (c) 2024-2025, 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.
+
+import asyncio
+
+from mem0 import AsyncMemoryClient
+
+from nat.memory.interfaces import MemoryEditor
+from nat.memory.models import MemoryItem
+
+
+class Mem0Editor(MemoryEditor):
+    """
+    Wrapper class that implements NAT interfaces for Mem0 Integrations Async.
+    """
+
+    def __init__(self, mem0_client: AsyncMemoryClient):
+        """
+        Initialize class with Predefined Mem0 Client.
+
+        Args:
+        mem0_client (AsyncMemoryClient): Preinstantiated
+        AsyncMemoryClient object for Mem0.
+        """
+        self._client = mem0_client
+
+    async def add_items(self, items: list[MemoryItem]) -> None:
+        """
+        Insert Multiple MemoryItems into the memory.
+        Each MemoryItem is translated and uploaded.
+        """
+
+        coroutines = []
+
+        # Iteratively insert memories into Mem0
+        for memory_item in items:
+            item_meta = memory_item.metadata
+            content = memory_item.conversation
+
+            user_id = memory_item.user_id  # This must be specified
+            run_id = item_meta.pop("run_id", None)
+            tags = memory_item.tags
+
+            coroutines.append(
+                self._client.add(content,
+                                 user_id=user_id,
+                                 run_id=run_id,
+                                 tags=tags,
+                                 metadata=item_meta,
+                                 output_format="v1.1"))
+
+        await asyncio.gather(*coroutines)
+
+    async def search(self, query: str, top_k: int = 5, **kwargs) \
+            -> list[MemoryItem]:
+        """
+        Retrieve items relevant to the given query.
+
+        Args:
+            query (str): The query string to match.
+            top_k (int): Maximum number of items to return.
+            **kwargs: Other keyword arguments for search.
+
+        Returns:
+            list[MemoryItem]: The most relevant
+            MemoryItems for the given query.
+        """
+
+        user_id = kwargs.pop("user_id")  # Ensure user ID is in keyword arguments
+
+        search_result = await self._client.search(query, user_id=user_id, top_k=top_k, output_format="v1.1", **kwargs)
+
+        # Construct MemoryItem instances
+        memories = []
+
+        for res in search_result["results"]:
+            item_meta = res.pop("metadata", {})
+
+            memories.append(
+                MemoryItem(conversation=res.pop("input", []),
+                           user_id=user_id,
+                           memory=res["memory"],
+                           tags=res.pop("categories", []) or [],
+                           metadata=item_meta))
+
+        return memories
+
+    async def remove_items(self, **kwargs):
+
+        if "memory_id" in kwargs:
+
+            memory_id = kwargs.pop("memory_id")
+            await self._client.delete(memory_id)
+
+        elif "user_id" in kwargs:
+
+            user_id = kwargs.pop("user_id")
+            await self._client.delete_all(user_id=user_id)
+
+        return

nat/plugins/mem0ai/memory.py

@@ -0,0 +1,57 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025, 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.
+
+from nat.builder.builder import Builder
+from nat.cli.register_workflow import register_memory
+from nat.data_models.memory import MemoryBaseConfig
+from nat.data_models.retry_mixin import RetryMixin
+from nat.utils.exception_handlers.automatic_retries import patch_with_retry
+
+
+class Mem0MemoryClientConfig(MemoryBaseConfig, RetryMixin, name="mem0_memory"):
+    host: str | None = None
+    organization: str | None = None
+    project: str | None = None
+    org_id: str | None = None
+    project_id: str | None = None
+
+
+@register_memory(config_type=Mem0MemoryClientConfig)
+async def mem0_memory_client(config: Mem0MemoryClientConfig, builder: Builder):
+    import os
+
+    from mem0 import AsyncMemoryClient
+
+    from nat.plugins.mem0ai.mem0_editor import Mem0Editor
+
+    mem0_api_key = os.environ.get("MEM0_API_KEY")
+
+    if mem0_api_key is None:
+        raise RuntimeError("Mem0 API key is not set. Please specify it in the environment variable 'MEM0_API_KEY'.")
+
+    mem0_client = AsyncMemoryClient(api_key=mem0_api_key,
+                                    host=config.host,
+                                    org_id=config.org_id,
+                                    project_id=config.project_id)
+
+    memory_editor = Mem0Editor(mem0_client=mem0_client)
+
+    if isinstance(config, RetryMixin):
+        memory_editor = patch_with_retry(memory_editor,
+                                         retries=config.num_retries,
+                                         retry_codes=config.retry_on_status_codes,
+                                         retry_on_messages=config.retry_on_errors)
+
+    yield memory_editor

nat/plugins/mem0ai/register.py

@@ -0,0 +1,22 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025, 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.
+
+# pylint: disable=unused-import
+# flake8: noqa
+# isort:skip_file
+
+# Import any providers which need to be automatically registered here
+
+from . import memory

nvidia_nat_mem0ai-1.2.0a20250813.dist-info/METADATA

@@ -0,0 +1,34 @@
+Metadata-Version: 2.4
+Name: nvidia-nat-mem0ai
+Version: 1.2.0a20250813
+Summary: Subpackage for Mem0 integration in NeMo Agent toolkit
+Keywords: ai,agents,memory
+Classifier: Programming Language :: Python
+Requires-Python: <3.13,>=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: nvidia-nat==v1.2.0a20250813
+Requires-Dist: mem0ai~=0.1.30
+
+<!--
+SPDX-FileCopyrightText: Copyright (c) 2025, 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/aiqtoolkit_banner.png "NeMo Agent toolkit banner image")
+
+# NVIDIA NeMo Agent Toolkit Subpackage
+This is a subpackage for Mem0 memory integration in NeMo Agent toolkit.
+
+For more information about the NVIDIA NeMo Agent toolkit, please visit the [NeMo Agent toolkit GitHub Repo](https://github.com/NVIDIA/NeMo-Agent-Toolkit).

nvidia_nat_mem0ai-1.2.0a20250813.dist-info/RECORD

@@ -0,0 +1,10 @@
+nat/meta/pypi.md,sha256=M3fUU11Pxni0woH-GJUeaKcy2lQE3OLNhjDqU-lO244,1123
+nat/plugins/mem0ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nat/plugins/mem0ai/mem0_editor.py,sha256=8BKaUia8VLW_NRYHQgFKyqX5EAoDTwmlBxck0JhIv7I,3647
+nat/plugins/mem0ai/memory.py,sha256=-uO0kcuCpAXrRZnzU2dfM5yHm87go4nOhBUZ03i9Qtk,2291
+nat/plugins/mem0ai/register.py,sha256=_ffKNKnMfkB2HzX4Nk_9EW0pwebg3GuzAE-iB-CoC3E,839
+nvidia_nat_mem0ai-1.2.0a20250813.dist-info/METADATA,sha256=qF1RtPdEI8HNR2gp1F73HXptQZcI5xRrbDw7pnpjVGY,1470
+nvidia_nat_mem0ai-1.2.0a20250813.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_mem0ai-1.2.0a20250813.dist-info/entry_points.txt,sha256=3eEklM2ToRyptUsWn00bINNml1FWTMaEZlNq29G4SjQ,58
+nvidia_nat_mem0ai-1.2.0a20250813.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_mem0ai-1.2.0a20250813.dist-info/RECORD,,

nvidia_nat_mem0ai-1.2.0a20250813.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

nvidia_nat_mem0ai-1.2.0a20250813.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[nat.components]
+nat_mem0ai = nat.plugins.mem0ai.register

nvidia_nat_mem0ai-1.2.0a20250813.dist-info/top_level.txt

@@ -0,0 +1 @@
+nat