nvidia-nat-test 1.2.0rc5__py3-none-any.whl → 1.2.0rc6__py3-none-any.whl

{aiq → nat}/meta/pypi.md

@@ -15,7 +15,7 @@ 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](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 NeMo Agent toolkit test utilities.

{aiq → nat}/test/embedder.py

@@ -15,12 +15,12 @@
 
 from pydantic import ConfigDict
 
-from aiq.builder.builder import Builder
-from aiq.builder.embedder import EmbedderProviderInfo
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.cli.register_workflow import register_embedder_client
-from aiq.cli.register_workflow import register_embedder_provider
-from aiq.data_models.embedder import EmbedderBaseConfig
+from nat.builder.builder import Builder
+from nat.builder.embedder import EmbedderProviderInfo
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.cli.register_workflow import register_embedder_client
+from nat.cli.register_workflow import register_embedder_provider
+from nat.data_models.embedder import EmbedderBaseConfig
 
 
 class EmbedderTestConfig(EmbedderBaseConfig, name="test_embedder"):

{aiq → nat}/test/functions.py

@@ -15,13 +15,13 @@
 
 from collections.abc import AsyncGenerator
 
-from aiq.builder.builder import Builder
-from aiq.builder.function_info import FunctionInfo
-from aiq.cli.register_workflow import register_function
-from aiq.data_models.api_server import AIQChatRequest
-from aiq.data_models.api_server import AIQChatResponse
-from aiq.data_models.api_server import AIQChatResponseChunk
-from aiq.data_models.function import FunctionBaseConfig
+from nat.builder.builder import Builder
+from nat.builder.function_info import FunctionInfo
+from nat.cli.register_workflow import register_function
+from nat.data_models.api_server import ChatRequest
+from nat.data_models.api_server import ChatResponse
+from nat.data_models.api_server import ChatResponseChunk
+from nat.data_models.function import FunctionBaseConfig
 
 
 class EchoFunctionConfig(FunctionBaseConfig, name="test_echo"):
@@ -34,8 +34,8 @@ async def echo_function(config: EchoFunctionConfig, builder: Builder):
     async def inner(message: str) -> str:
         return message
 
-    async def inner_oai(message: AIQChatRequest) -> AIQChatResponse:
-        return AIQChatResponse.from_string(message.messages[0].content)
+    async def inner_oai(message: ChatRequest) -> ChatResponse:
+        return ChatResponse.from_string(message.messages[0].content)
 
     if (config.use_openai_api):
         yield inner_oai
@@ -50,16 +50,16 @@ class StreamingEchoFunctionConfig(FunctionBaseConfig, name="test_streaming_echo"
 @register_function(config_type=StreamingEchoFunctionConfig)
 async def streaming_function(config: StreamingEchoFunctionConfig, builder: Builder):
 
-    def oai_to_list(message: AIQChatRequest) -> list[str]:
+    def oai_to_list(message: ChatRequest) -> list[str]:
         return [m.content for m in message.messages]
 
     async def inner(message: list[str]) -> AsyncGenerator[str]:
         for value in message:
             yield value
 
-    async def inner_oai(message: AIQChatRequest) -> AsyncGenerator[AIQChatResponseChunk]:
+    async def inner_oai(message: ChatRequest) -> AsyncGenerator[ChatResponseChunk]:
         for value in oai_to_list(message):
-            yield AIQChatResponseChunk.from_string(value)
+            yield ChatResponseChunk.from_string(value)
 
     yield FunctionInfo.from_fn(inner_oai if config.use_openai_api else inner, converters=[oai_to_list])
 

{aiq → nat}/test/memory.py

@@ -13,11 +13,11 @@
 # 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
-from aiq.memory.interfaces import MemoryEditor
-from aiq.memory.models import MemoryItem
+from nat.builder.builder import Builder
+from nat.cli.register_workflow import register_memory
+from nat.data_models.memory import MemoryBaseConfig
+from nat.memory.interfaces import MemoryEditor
+from nat.memory.models import MemoryItem
 
 
 class DummyMemoryConfig(MemoryBaseConfig, name="test_dummy"):

{aiq → nat}/test/object_store_tests.py

@@ -20,10 +20,10 @@ from contextlib import asynccontextmanager
 import pytest
 import pytest_asyncio
 
-from aiq.data_models.object_store import KeyAlreadyExistsError
-from aiq.data_models.object_store import NoSuchKeyError
-from aiq.object_store.interfaces import ObjectStore
-from aiq.object_store.models import ObjectStoreItem
+from nat.data_models.object_store import KeyAlreadyExistsError
+from nat.data_models.object_store import NoSuchKeyError
+from nat.object_store.interfaces import ObjectStore
+from nat.object_store.models import ObjectStoreItem
 
 
 @pytest.mark.asyncio(loop_scope="class")

{aiq → nat}/test/plugin.py

@@ -59,16 +59,16 @@ def pytest_runtest_setup(item):
 
 @pytest.fixture(name="register_components", scope="session", autouse=True)
 def register_components_fixture():
-    from aiq.runtime.loader import PluginTypes
-    from aiq.runtime.loader import discover_and_register_plugins
+    from nat.runtime.loader import PluginTypes
+    from nat.runtime.loader import discover_and_register_plugins
 
     # Ensure that all components which need to be registered as part of an import are done so. This is necessary
     # because imports will not be reloaded between tests, so we need to ensure that all components are registered
     # before any tests are run.
     discover_and_register_plugins(PluginTypes.ALL)
 
-    # Also import the aiq.test.register module to register test-only components
-    import aiq.test.register  # pylint: disable=unused-import # noqa: F401
+    # Also import the nat.test.register module to register test-only components
+    import nat.test.register  # pylint: disable=unused-import # noqa: F401
 
 
 @pytest.fixture(name="module_registry", scope="module", autouse=True)
@@ -78,7 +78,7 @@ def module_registry_fixture():
 
     This gets automatically used at the module level to ensure no state is leaked between modules
     """
-    from aiq.cli.type_registry import GlobalTypeRegistry
+    from nat.cli.type_registry import GlobalTypeRegistry
 
     with GlobalTypeRegistry.push() as registry:
         yield registry
@@ -91,7 +91,7 @@ def function_registry_fixture():
 
     This gets automatically used at the function level to ensure no state is leaked between functions
     """
-    from aiq.cli.type_registry import GlobalTypeRegistry
+    from nat.cli.type_registry import GlobalTypeRegistry
 
     with GlobalTypeRegistry.push() as registry:
         yield registry

{aiq → nat}/test/tool_test_runner.py

@@ -20,15 +20,15 @@ from contextlib import asynccontextmanager
 from unittest.mock import AsyncMock
 from unittest.mock import MagicMock
 
-from aiq.builder.builder import Builder
-from aiq.builder.function import Function
-from aiq.builder.function_info import FunctionInfo
-from aiq.cli.type_registry import GlobalTypeRegistry
-from aiq.data_models.function import FunctionBaseConfig
-from aiq.data_models.object_store import ObjectStoreBaseConfig
-from aiq.object_store.interfaces import ObjectStore
-from aiq.runtime.loader import PluginTypes
-from aiq.runtime.loader import discover_and_register_plugins
+from nat.builder.builder import Builder
+from nat.builder.function import Function
+from nat.builder.function_info import FunctionInfo
+from nat.cli.type_registry import GlobalTypeRegistry
+from nat.data_models.function import FunctionBaseConfig
+from nat.data_models.object_store import ObjectStoreBaseConfig
+from nat.object_store.interfaces import ObjectStore
+from nat.runtime.loader import PluginTypes
+from nat.runtime.loader import discover_and_register_plugins
 
 logger = logging.getLogger(__name__)
 
@@ -231,7 +231,7 @@ class MockBuilder(Builder):
 
 class ToolTestRunner:
     """
-    A test runner that enables isolated testing of AIQ tools without requiring
+    A test runner that enables isolated testing of NAT tools without requiring
     full workflow setup, LLMs, or complex dependencies.
 
     Usage:

{nvidia_nat_test-1.2.0rc5.dist-info → nvidia_nat_test-1.2.0rc6.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-test
-Version: 1.2.0rc5
-Summary: Testing utilities for AIQtoolkit
+Version: 1.2.0rc6
+Summary: Testing utilities for NeMo Agent toolkit
 Keywords: ai,rag,agents
 Classifier: Programming Language :: Python
 Requires-Python: <3.13,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: nvidia-nat~=1.2
+Requires-Dist: nvidia-nat==v1.2.0-rc6
 Requires-Dist: langchain-community~=0.3
 Requires-Dist: pytest~=8.3
 
@@ -27,7 +27,7 @@ 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](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 NeMo Agent toolkit test utilities.

nvidia_nat_test-1.2.0rc6.dist-info/RECORD

@@ -0,0 +1,14 @@
+nat/meta/pypi.md,sha256=LLKJHg5oN1-M9Pqfk3Bmphkk4O2TFsyiixuK5T0Y-gw,1100
+nat/test/__init__.py,sha256=_RnTJnsUucHvla_nYKqD4O4g8Bz0tcuDRzWk1bEhcy0,875
+nat/test/embedder.py,sha256=ClDyK1kna4hCBSlz71gK1B-ZjlwcBHTDQRekoNM81Bs,1809
+nat/test/functions.py,sha256=0ScrdsjcxCsPRLnyb5gfwukmvZxFi_ptCswLSIG0DVY,3095
+nat/test/memory.py,sha256=xki_A2yiMhEZuQk60K7t04QRqf32nQqnfzD5Iv7fkvw,1456
+nat/test/object_store_tests.py,sha256=PyJioOtoSzILPq6LuD-sOZ_89PIcgXWZweoHBQpK2zQ,4281
+nat/test/plugin.py,sha256=fp39ib0W63vfqX6Ssvq4sCuSd8Lm6yQyknL3_qRijgI,3610
+nat/test/register.py,sha256=jU1pW5wf20ZmCOTgkaQshKZfvYh8_-sMJ4P3xXilfTY,891
+nat/test/tool_test_runner.py,sha256=ccErldob2VwBbVL0_pmLrOcKLc18qYjxeAEACYoKKGQ,17469
+nvidia_nat_test-1.2.0rc6.dist-info/METADATA,sha256=C1-6QU-qmIaRCq9qQI9ylmU3U2es_g4U26n1by4ojQw,1455
+nvidia_nat_test-1.2.0rc6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_test-1.2.0rc6.dist-info/entry_points.txt,sha256=7dOP9XB6iMDqvav3gYx9VWUwA8RrFzhbAa8nGeC8e4Y,99
+nvidia_nat_test-1.2.0rc6.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_test-1.2.0rc6.dist-info/RECORD,,

nvidia_nat_test-1.2.0rc6.dist-info/entry_points.txt

@@ -0,0 +1,5 @@
+[nat.components]
+nvidia-nat-test = nat.test.register
+
+[pytest11]
+nvidia-nat-test = nat.test.plugin

nvidia_nat_test-1.2.0rc6.dist-info/top_level.txt

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

nvidia_nat_test-1.2.0rc5.dist-info/RECORD

@@ -1,14 +0,0 @@
-aiq/meta/pypi.md,sha256=XfzdATwjeiVEfSdXmjS5mhVbrHIwHNyK2CR4KtB9Qaw,1111
-aiq/test/__init__.py,sha256=_RnTJnsUucHvla_nYKqD4O4g8Bz0tcuDRzWk1bEhcy0,875
-aiq/test/embedder.py,sha256=Sszon2lvuET3u3mMcbdWNDh5xkYLSOFxy1WMIhKC2Ig,1809
-aiq/test/functions.py,sha256=qlYAGip6cYh0hG7kySZzlqvdr2caZSMjQGyv9jP7rK4,3125
-aiq/test/memory.py,sha256=UnUJuL0foiOqh6emneIg5gHnK9n1MAnaD2CO7voFfok,1456
-aiq/test/object_store_tests.py,sha256=zPON1yYyhf07CpBbwN_ymIcBFxhogAS7tqcF-7vc9zk,4281
-aiq/test/plugin.py,sha256=Clt81eqpJimOn5ZeUlVBlgOUfogFmPNAkCc5SxS76qo,3610
-aiq/test/register.py,sha256=jU1pW5wf20ZmCOTgkaQshKZfvYh8_-sMJ4P3xXilfTY,891
-aiq/test/tool_test_runner.py,sha256=ime7S4KKQcRYKoh8q6RnYwdwdCWgdftrOV7AgUQ2DnI,17469
-nvidia_nat_test-1.2.0rc5.dist-info/METADATA,sha256=sc0Q-cUTtliaaQ-mUh8gGzFUCePoxMKP4Qr2rNfn7zE,1451
-nvidia_nat_test-1.2.0rc5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_test-1.2.0rc5.dist-info/entry_points.txt,sha256=whK6YfmQvlrRJtKNiEPuDNqv_O7ZGNA5WZ7ag57CPFU,99
-nvidia_nat_test-1.2.0rc5.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-nvidia_nat_test-1.2.0rc5.dist-info/RECORD,,

nvidia_nat_test-1.2.0rc5.dist-info/entry_points.txt

@@ -1,5 +0,0 @@
-[aiq.components]
-nvidia-nat-test = aiq.test.register
-
-[pytest11]
-nvidia-nat-test = aiq.test.plugin

nvidia_nat_test-1.2.0rc5.dist-info/top_level.txt

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