nvidia-nat-zep-cloud 1.2.0__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/banner.png "NeMo Agent toolkit banner image")
+
+# NVIDIA NeMo Agent Toolkit Subpackage
+This is a subpackage for Zep 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/zep_cloud/memory.py

@@ -0,0 +1,54 @@
+# 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 ZepMemoryClientConfig(MemoryBaseConfig, RetryMixin, name="zep_memory"):
+    base_url: str | None = None
+    timeout: float | None = None
+    follow_redirects: bool | None = None
+
+
+@register_memory(config_type=ZepMemoryClientConfig)
+async def zep_memory_client(config: ZepMemoryClientConfig, builder: Builder):
+    import os
+
+    from zep_cloud.client import AsyncZep
+
+    from nat.plugins.zep_cloud.zep_editor import ZepEditor
+
+    zep_api_key = os.environ.get("ZEP_API_KEY")
+
+    if zep_api_key is None:
+        raise RuntimeError("Zep API key is not set. Please specify it in the environment variable 'ZEP_API_KEY'.")
+
+    zep_client = AsyncZep(api_key=zep_api_key,
+                          base_url=config.base_url,
+                          timeout=config.timeout,
+                          follow_redirects=config.follow_redirects)
+    memory_editor = ZepEditor(zep_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/zep_cloud/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

nat/plugins/zep_cloud/zep_editor.py

@@ -0,0 +1,105 @@
+# 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.
+
+from __future__ import annotations
+
+import asyncio
+
+from zep_cloud.client import AsyncZep
+from zep_cloud.types import Message
+
+from nat.memory.interfaces import MemoryEditor
+from nat.memory.models import MemoryItem
+
+
+class ZepEditor(MemoryEditor):
+    """
+    Wrapper class that implements NAT interfaces for Zep Integrations Async.
+    """
+
+    def __init__(self, zep_client: AsyncZep):
+        """
+        Initialize class with Predefined Mem0 Client.
+
+        Args:
+        zep_client (AsyncZep): Async client instance.
+        """
+        self._client = zep_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:
+            conversation = memory_item.conversation
+            session_id = memory_item.user_id
+            messages = []
+            for msg in conversation:
+                messages.append(Message(content=msg["content"], role_type=msg["role"]))
+
+            coroutines.append(self._client.memory.add(session_id=session_id, messages=messages))
+
+        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.
+        """
+
+        session_id = kwargs.pop("user_id")  # Ensure user ID is in keyword arguments
+        limit = top_k
+
+        search_result = await self._client.memory.search_sessions(session_ids=[session_id],
+                                                                  text=query,
+                                                                  limit=limit,
+                                                                  search_scope="messages",
+                                                                  **kwargs)
+
+        # Construct MemoryItem instances
+        memories = []
+
+        for res in search_result.results:
+            memories.append(
+                MemoryItem(conversation=[],
+                           user_id=session_id,
+                           memory=res.message.content,
+                           metadata={
+                               "relevance_score": res.score,
+                               "created_at": res.message.created_at,
+                               "updated_at": res.message.updated_at
+                           }))
+
+        return memories
+
+    async def remove_items(self, **kwargs):
+
+        if "session_id" in kwargs:
+            session_id = kwargs.pop("session_id")
+            await self._client.memory.delete(session_id)
+
+        else:
+            raise ValueError("session_id not provided as part of the tool call. ")

nvidia_nat_zep_cloud-1.2.0.dist-info/METADATA

@@ -0,0 +1,34 @@
+Metadata-Version: 2.4
+Name: nvidia-nat-zep-cloud
+Version: 1.2.0
+Summary: Subpackage for Zep 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.0
+Requires-Dist: zep-cloud~=2.2.0
+
+<!--
+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/banner.png "NeMo Agent toolkit banner image")
+
+# NVIDIA NeMo Agent Toolkit Subpackage
+This is a subpackage for Zep 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_zep_cloud-1.2.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+nat/meta/pypi.md,sha256=BMN_V2SVyWPMymcWAX8PypnNKJphAqckAbCEqGPaN6I,1111
+nat/plugins/zep_cloud/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nat/plugins/zep_cloud/memory.py,sha256=WrRbHBn_bnHdDVa6hThKR5BPl0wSPOJfKXbbR_XOroM,2198
+nat/plugins/zep_cloud/register.py,sha256=_ffKNKnMfkB2HzX4Nk_9EW0pwebg3GuzAE-iB-CoC3E,839
+nat/plugins/zep_cloud/zep_editor.py,sha256=JrLS6S956PzR2-Jaheaq2EvDCrlYKDEc5eD1HIdzsWA,3770
+nvidia_nat_zep_cloud-1.2.0.dist-info/METADATA,sha256=LoFzY1wZheIjgbiIs5xgPBfi7P49YG6FZgmTPoqRNys,1444
+nvidia_nat_zep_cloud-1.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_zep_cloud-1.2.0.dist-info/entry_points.txt,sha256=r7vO0ft8P0EBtuNO3cm0nYo4FScapvptPjHwtlfnvcA,64
+nvidia_nat_zep_cloud-1.2.0.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_zep_cloud-1.2.0.dist-info/RECORD,,

nvidia_nat_zep_cloud-1.2.0.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_zep_cloud-1.2.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[nat.components]
+nat_zep_cloud = nat.plugins.zep_cloud.register

nvidia_nat_zep_cloud-1.2.0.dist-info/top_level.txt

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