aiqtoolkit-zep-cloud 1.1.0rc5__py3-none-any.whl → 1.2.0__py3-none-any.whl

{aiqtoolkit_zep_cloud-1.1.0rc5.dist-info → aiqtoolkit_zep_cloud-1.2.0.dist-info}/METADATA

@@ -1,13 +1,10 @@
 Metadata-Version: 2.4
 Name: aiqtoolkit-zep-cloud
-Version: 1.1.0rc5
-Summary: Subpackage for Zep memory integration in AIQtoolkit
-Keywords: ai,agents,memory
+Version: 1.2.0
+Summary: Transitional package for nvidia-nat-zep-cloud, this package is deprecated and will be removed in the future.
 Classifier: Programming Language :: Python
-Requires-Python: <3.13,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: aiqtoolkit==v1.1.0-rc5
-Requires-Dist: zep-cloud~=2.2.0
+Requires-Dist: nvidia-nat-zep-cloud
 
 <!--
 SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
@@ -26,9 +23,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 -->
 
-![NVIDIA Agent Intelligence Toolkit](https://media.githubusercontent.com/media/NVIDIA/AIQToolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "AIQ toolkit banner image")
+# Transitional Package for `nvidia-nat-zep-cloud`
+This is a transitional package for `nvidia-nat-zep-cloud` to help ease the migration to `nvidia-nat-zep-cloud`, and will be removed in a future release. It is recommended to use `nvidia-nat-zep-cloud` directly for new projects.
 
-# NVIDIA Agent Intelligence Toolkit Subpackage
-This is a subpackage for Zep memory integration in AIQ toolkit.
-
-For more information about AIQ toolkit, please visit the [AIQ toolkit package](https://pypi.org/project/aiqtoolkit/).
+For more information about the NVIDIA NeMo Agent toolkit, please visit the [NeMo Agent toolkit package](https://pypi.org/project/nvidia-nat-zep-cloud/).

aiqtoolkit_zep_cloud-1.2.0.dist-info/RECORD

@@ -0,0 +1,4 @@
+aiqtoolkit_zep_cloud-1.2.0.dist-info/METADATA,sha256=GftdAr9q-vMUBxUXnPGI9PAVtrMN0huYWsfMLilpMIQ,1399
+aiqtoolkit_zep_cloud-1.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiqtoolkit_zep_cloud-1.2.0.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+aiqtoolkit_zep_cloud-1.2.0.dist-info/RECORD,,

{aiqtoolkit_zep_cloud-1.1.0rc5.dist-info → aiqtoolkit_zep_cloud-1.2.0.dist-info}/WHEEL

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

aiqtoolkit_zep_cloud-1.2.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+

aiq/meta/pypi.md

@@ -1,23 +0,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 Agent Intelligence Toolkit](https://media.githubusercontent.com/media/NVIDIA/AIQToolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "AIQ toolkit banner image")
-
-# NVIDIA Agent Intelligence Toolkit Subpackage
-This is a subpackage for Zep memory integration in AIQ toolkit.
-
-For more information about AIQ toolkit, please visit the [AIQ toolkit package](https://pypi.org/project/aiqtoolkit/).

aiq/plugins/zep_cloud/memory.py

@@ -1,46 +0,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.
-
-from aiq.builder.builder import Builder
-from aiq.cli.register_workflow import register_memory
-from aiq.data_models.memory import MemoryBaseConfig
-
-
-class ZepMemoryClientConfig(MemoryBaseConfig, 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 aiq.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)
-
-    yield memory_editor

aiq/plugins/zep_cloud/register.py

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

aiq/plugins/zep_cloud/zep_editor.py

@@ -1,105 +0,0 @@
-# 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 aiq.memory.interfaces import MemoryEditor
-from aiq.memory.models import MemoryItem
-
-
-class ZepEditor(MemoryEditor):
-    """
-    Wrapper class that implements AIQ Toolkit 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. ")

aiqtoolkit_zep_cloud-1.1.0rc5.dist-info/RECORD

@@ -1,10 +0,0 @@
-aiq/meta/pypi.md,sha256=OaXSseSQmZpGfu1gU1X-CA1agKyr9jjkimG2pFumUws,1079
-aiq/plugins/zep_cloud/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aiq/plugins/zep_cloud/memory.py,sha256=nCdjuqc88lbHgOzryX03lXsZvT26CLtIhaWjPyzYYBo,1728
-aiq/plugins/zep_cloud/register.py,sha256=_ffKNKnMfkB2HzX4Nk_9EW0pwebg3GuzAE-iB-CoC3E,839
-aiq/plugins/zep_cloud/zep_editor.py,sha256=n8CzRcN6FupQ4gJT6hB7o6-cGnAL3OP3mI2IiUfdvOA,3778
-aiqtoolkit_zep_cloud-1.1.0rc5.dist-info/METADATA,sha256=2Z8Cy29aW5fqrmxawXLaFaSPd9i64_b5quhT5auFpuc,1418
-aiqtoolkit_zep_cloud-1.1.0rc5.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-aiqtoolkit_zep_cloud-1.1.0rc5.dist-info/entry_points.txt,sha256=4YncaCW2G71ZTzahBnUuvTEhIbz9adjbPXC2waBqZ0Q,64
-aiqtoolkit_zep_cloud-1.1.0rc5.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-aiqtoolkit_zep_cloud-1.1.0rc5.dist-info/RECORD,,

aiqtoolkit_zep_cloud-1.1.0rc5.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[aiq.components]
-aiq_zep_cloud = aiq.plugins.zep_cloud.register

aiqtoolkit_zep_cloud-1.1.0rc5.dist-info/top_level.txt

@@ -1 +0,0 @@
-aiq