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

{aiq → nat}/tool/nvidia_rag.py

@@ -18,10 +18,10 @@ import logging
 
 from pydantic import Field
 
-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.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.function import FunctionBaseConfig
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/tool/retriever.py

@@ -18,21 +18,21 @@ import logging
 from pydantic import BaseModel
 from pydantic import Field
 
-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.component_ref import RetrieverRef
-from aiq.data_models.function import FunctionBaseConfig
-from aiq.retriever.interface import AIQRetriever
-from aiq.retriever.models import RetrieverError
-from aiq.retriever.models import RetrieverOutput
+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.component_ref import RetrieverRef
+from nat.data_models.function import FunctionBaseConfig
+from nat.retriever.interface import Retriever
+from nat.retriever.models import RetrieverError
+from nat.retriever.models import RetrieverOutput
 
 logger = logging.getLogger(__name__)
 
 
-class AIQRetrieverConfig(FunctionBaseConfig, name="aiq_retriever"):
+class RetrieverConfig(FunctionBaseConfig, name="nat_retriever"):
     """
-    AIQRetriever tool which provides a common interface for different vectorstores. Its
+    Retriever tool which provides a common interface for different vectorstores. Its
     configuration uses clients, which are the vectorstore-specific implementaiton of the retriever interface.
     """
     retriever: RetrieverRef = Field(description="The retriever instance name from the workflow configuration object.")
@@ -44,7 +44,7 @@ class AIQRetrieverConfig(FunctionBaseConfig, name="aiq_retriever"):
     description: str | None = Field(default=None, description="If present it will be used as the tool description")
 
 
-def _get_description_from_config(config: AIQRetrieverConfig) -> str:
+def _get_description_from_config(config: RetrieverConfig) -> str:
     """
     Generate a description of what the tool will do based on how it is configured.
     """
@@ -55,10 +55,10 @@ def _get_description_from_config(config: AIQRetrieverConfig) -> str:
     return description.format(topic=_topic) if not config.description else config.description
 
 
-@register_function(config_type=AIQRetrieverConfig)
-async def aiq_retriever_tool(config: AIQRetrieverConfig, builder: Builder):
+@register_function(config_type=RetrieverConfig)
+async def retriever_tool(config: RetrieverConfig, builder: Builder):
     """
-    Configure an AIQ Toolkit Retriever Tool which supports different clients such as Milvus and Nemo Retriever.
+    Configure a NAT Retriever Tool which supports different clients such as Milvus and Nemo Retriever.
 
     Args:
         config: A config object with required parameters 'client' and 'client_config'
@@ -68,7 +68,7 @@ async def aiq_retriever_tool(config: AIQRetrieverConfig, builder: Builder):
     class RetrieverInputSchema(BaseModel):
         query: str = Field(description="The query to be searched in the configured data store")
 
-    client: AIQRetriever = await builder.get_retriever(config.retriever)
+    client: Retriever = await builder.get_retriever(config.retriever)
 
     async def _retrieve(query: str) -> RetrieverOutput:
         try:
@@ -87,3 +87,8 @@ async def aiq_retriever_tool(config: AIQRetrieverConfig, builder: Builder):
         input_schema=RetrieverInputSchema,
         description=_get_description_from_config(config),
     )
+
+
+# Compatibility aliases with previous releases
+AIQRetrieverConfig = RetrieverConfig
+aiq_retriever_tool = retriever_tool

{aiq → nat}/tool/server_tools.py

@@ -13,10 +13,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-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.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.function import FunctionBaseConfig
 
 
 class RequestAttributesTool(FunctionBaseConfig, name="current_request_attributes"):
@@ -37,19 +37,19 @@ async def current_request_attributes(config: RequestAttributesTool, builder: Bui
 
     async def _get_request_attributes(unused: str) -> str:
 
-        from aiq.builder.context import AIQContext
-        aiq_context = AIQContext.get()
+        from nat.builder.context import Context
+        nat_context = Context.get()
 
-        method: str | None = aiq_context.metadata.method
-        url_path: str | None = aiq_context.metadata.url_path
-        url_scheme: str | None = aiq_context.metadata.url_scheme
-        headers: Headers | None = aiq_context.metadata.headers
-        query_params: QueryParams | None = aiq_context.metadata.query_params
-        path_params: dict[str, str] | None = aiq_context.metadata.path_params
-        client_host: str | None = aiq_context.metadata.client_host
-        client_port: int | None = aiq_context.metadata.client_port
-        cookies: dict[str, str] | None = aiq_context.metadata.cookies
-        conversation_id: str | None = aiq_context.conversation_id
+        method: str | None = nat_context.metadata.method
+        url_path: str | None = nat_context.metadata.url_path
+        url_scheme: str | None = nat_context.metadata.url_scheme
+        headers: Headers | None = nat_context.metadata.headers
+        query_params: QueryParams | None = nat_context.metadata.query_params
+        path_params: dict[str, str] | None = nat_context.metadata.path_params
+        client_host: str | None = nat_context.metadata.client_host
+        client_port: int | None = nat_context.metadata.client_port
+        cookies: dict[str, str] | None = nat_context.metadata.cookies
+        conversation_id: str | None = nat_context.conversation_id
 
         return (f"Method: {method}, "
                 f"URL Path: {url_path}, "

{aiq → nat}/utils/dump_distro_mapping.py

@@ -16,11 +16,11 @@
 import argparse
 import json
 
-from aiq.runtime.loader import get_all_aiq_entrypoints_distro_mapping
+from nat.runtime.loader import get_all_entrypoints_distro_mapping
 
 
 def dump_distro_mapping(path: str):
-    mapping = get_all_aiq_entrypoints_distro_mapping()
+    mapping = get_all_entrypoints_distro_mapping()
     with open(path, "w", encoding="utf-8") as f:
         json.dump(mapping, f, indent=4)
 

{aiq → nat}/utils/exception_handlers/mcp.py

@@ -22,14 +22,14 @@ from typing import Any
 
 import httpx
 
-from aiq.tool.mcp.exceptions import MCPAuthenticationError
-from aiq.tool.mcp.exceptions import MCPConnectionError
-from aiq.tool.mcp.exceptions import MCPError
-from aiq.tool.mcp.exceptions import MCPProtocolError
-from aiq.tool.mcp.exceptions import MCPRequestError
-from aiq.tool.mcp.exceptions import MCPSSLError
-from aiq.tool.mcp.exceptions import MCPTimeoutError
-from aiq.tool.mcp.exceptions import MCPToolNotFoundError
+from nat.tool.mcp.exceptions import MCPAuthenticationError
+from nat.tool.mcp.exceptions import MCPConnectionError
+from nat.tool.mcp.exceptions import MCPError
+from nat.tool.mcp.exceptions import MCPProtocolError
+from nat.tool.mcp.exceptions import MCPRequestError
+from nat.tool.mcp.exceptions import MCPSSLError
+from nat.tool.mcp.exceptions import MCPTimeoutError
+from nat.tool.mcp.exceptions import MCPToolNotFoundError
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/utils/io/yaml_tools.py

@@ -20,7 +20,7 @@ import typing
 import expandvars
 import yaml
 
-from aiq.utils.type_utils import StrPath
+from nat.utils.type_utils import StrPath
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/utils/metadata_utils.py

@@ -15,8 +15,8 @@
 
 from pydantic_core import PydanticUndefined
 
-from aiq.data_models.common import TypedBaseModelT
-from aiq.utils.type_utils import DecomposedType
+from nat.data_models.common import TypedBaseModelT
+from nat.utils.type_utils import DecomposedType
 
 
 def generate_config_type_docs(config_type: TypedBaseModelT) -> str:

{aiq → nat}/utils/reactive/base/observable_base.py

@@ -20,8 +20,8 @@ from collections.abc import Callable
 from typing import Generic
 from typing import TypeVar
 
-from aiq.utils.reactive.base.observer_base import ObserverBase
-from aiq.utils.reactive.subscription import Subscription
+from nat.utils.reactive.base.observer_base import ObserverBase
+from nat.utils.reactive.subscription import Subscription
 
 # Covariant type param: An Observable producing type X can also produce
 # a subtype of X.

{aiq → nat}/utils/reactive/base/subject_base.py

@@ -22,7 +22,7 @@ from .observable_base import ObservableBase
 from .observer_base import ObserverBase
 
 if typing.TYPE_CHECKING:
-    from aiq.utils.reactive.subscription import Subscription
+    from nat.utils.reactive.subscription import Subscription
 
 T = TypeVar("T")
 

{aiq → nat}/utils/reactive/observable.py

@@ -16,11 +16,11 @@
 from collections.abc import Callable
 from typing import TypeVar
 
-from aiq.utils.reactive.base.observable_base import ObservableBase
-from aiq.utils.reactive.base.observer_base import ObserverBase
-from aiq.utils.reactive.observer import Observer
-from aiq.utils.reactive.subscription import Subscription
-from aiq.utils.type_utils import override
+from nat.utils.reactive.base.observable_base import ObservableBase
+from nat.utils.reactive.base.observer_base import ObserverBase
+from nat.utils.reactive.observer import Observer
+from nat.utils.reactive.subscription import Subscription
+from nat.utils.type_utils import override
 
 # Covariant type param: An Observable producing type X can also produce
 # a subtype of X.

{aiq → nat}/utils/reactive/observer.py

@@ -17,7 +17,7 @@ import logging
 from collections.abc import Callable
 from typing import TypeVar
 
-from aiq.utils.reactive.base.observer_base import ObserverBase
+from nat.utils.reactive.base.observer_base import ObserverBase
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/utils/reactive/subject.py

@@ -17,10 +17,10 @@ import threading
 from collections.abc import Callable
 from typing import TypeVar
 
-from aiq.utils.reactive.base.subject_base import SubjectBase
-from aiq.utils.reactive.observable import Observable
-from aiq.utils.reactive.observer import Observer
-from aiq.utils.reactive.subscription import Subscription
+from nat.utils.reactive.base.subject_base import SubjectBase
+from nat.utils.reactive.observable import Observable
+from nat.utils.reactive.observer import Observer
+from nat.utils.reactive.subscription import Subscription
 
 T = TypeVar("T")
 

{aiq → nat}/utils/reactive/subscription.py

@@ -19,7 +19,7 @@ from typing import Generic
 from typing import TypeVar
 
 if typing.TYPE_CHECKING:
-    from aiq.utils.reactive.base.subject_base import SubjectBase
+    from nat.utils.reactive.base.subject_base import SubjectBase
 
 _T = TypeVar("_T")  # pylint: disable=invalid-name
 

{aiq → nat}/utils/settings/global_settings.py

@@ -17,9 +17,9 @@ import logging
 
 from pydantic import create_model
 
-from aiq.cli.type_registry import GlobalTypeRegistry
-from aiq.data_models.registry_handler import RegistryHandlerBaseConfig
-from aiq.settings.global_settings import GlobalSettings
+from nat.cli.type_registry import GlobalTypeRegistry
+from nat.data_models.registry_handler import RegistryHandlerBaseConfig
+from nat.settings.global_settings import GlobalSettings
 
 logger = logging.getLogger(__name__)
 
@@ -66,7 +66,7 @@ def configure_registry_channel(config_type: RegistryHandlerBaseConfig, channel_n
 
 
 def add_channel_interative(channel_type: str) -> None:
-    """Add a remote registry channel to publish/search/pull AIQ Toolkit plugin packages.
+    """Add a remote registry channel to publish/search/pull NAT plugin packages.
 
     Args:
         channel_type (str): They type of channel to configure.

{aiq → nat}/utils/type_converter.py

@@ -19,7 +19,7 @@ from collections import OrderedDict
 from collections.abc import Callable
 from io import TextIOWrapper
 
-from aiq.utils.type_utils import DecomposedType
+from nat.utils.type_utils import DecomposedType
 
 logger = logging.getLogger(__name__)
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat
-Version: 1.2.0rc5
+Version: 1.2.0rc6
 Summary: NVIDIA NeMo Agent toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation
@@ -273,28 +273,28 @@ Requires-Dist: nvidia-nat-weave; extra == "weave"
 Provides-Extra: zep-cloud
 Requires-Dist: nvidia-nat-zep-cloud; extra == "zep-cloud"
 Provides-Extra: examples
-Requires-Dist: aiq_agno_personal_finance; extra == "examples"
-Requires-Dist: aiq_alert_triage_agent; extra == "examples"
-Requires-Dist: aiq_automated_description_generation; extra == "examples"
-Requires-Dist: aiq_email_phishing_analyzer; extra == "examples"
-Requires-Dist: aiq_multi_frameworks; extra == "examples"
-Requires-Dist: aiq_plot_charts; extra == "examples"
-Requires-Dist: aiq_por_to_jiratickets; extra == "examples"
-Requires-Dist: aiq_profiler_agent; extra == "examples"
-Requires-Dist: aiq_redact_pii; extra == "examples"
-Requires-Dist: aiq_semantic_kernel_demo; extra == "examples"
-Requires-Dist: aiq_simple_auth; extra == "examples"
-Requires-Dist: aiq_simple_web_query; extra == "examples"
-Requires-Dist: aiq_simple_web_query_eval; extra == "examples"
-Requires-Dist: aiq_simple_calculator; extra == "examples"
-Requires-Dist: aiq_simple_calculator_custom_routes; extra == "examples"
-Requires-Dist: aiq_simple_calculator_eval; extra == "examples"
-Requires-Dist: aiq_simple_calculator_mcp; extra == "examples"
-Requires-Dist: aiq_simple_calculator_observability; extra == "examples"
-Requires-Dist: aiq_simple_calculator_hitl; extra == "examples"
-Requires-Dist: aiq_simple_rag; extra == "examples"
-Requires-Dist: aiq_swe_bench; extra == "examples"
-Requires-Dist: aiq_user_report; extra == "examples"
+Requires-Dist: nat_agno_personal_finance; extra == "examples"
+Requires-Dist: nat_alert_triage_agent; extra == "examples"
+Requires-Dist: nat_automated_description_generation; extra == "examples"
+Requires-Dist: nat_email_phishing_analyzer; extra == "examples"
+Requires-Dist: nat_multi_frameworks; extra == "examples"
+Requires-Dist: nat_plot_charts; extra == "examples"
+Requires-Dist: nat_por_to_jiratickets; extra == "examples"
+Requires-Dist: nat_profiler_agent; extra == "examples"
+Requires-Dist: nat_redact_pii; extra == "examples"
+Requires-Dist: nat_semantic_kernel_demo; extra == "examples"
+Requires-Dist: nat_simple_auth; extra == "examples"
+Requires-Dist: nat_simple_web_query; extra == "examples"
+Requires-Dist: nat_simple_web_query_eval; extra == "examples"
+Requires-Dist: nat_simple_calculator; extra == "examples"
+Requires-Dist: nat_simple_calculator_custom_routes; extra == "examples"
+Requires-Dist: nat_simple_calculator_eval; extra == "examples"
+Requires-Dist: nat_simple_calculator_mcp; extra == "examples"
+Requires-Dist: nat_simple_calculator_observability; extra == "examples"
+Requires-Dist: nat_simple_calculator_hitl; extra == "examples"
+Requires-Dist: nat_simple_rag; extra == "examples"
+Requires-Dist: nat_swe_bench; extra == "examples"
+Requires-Dist: nat_user_report; extra == "examples"
 Provides-Extra: profiling
 Requires-Dist: matplotlib~=3.9; extra == "profiling"
 Requires-Dist: prefixspan~=0.5.2; extra == "profiling"
@@ -320,30 +320,30 @@ 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
 
-AIQ toolkit is a flexible library designed to seamlessly integrate your enterprise agents—regardless of framework—with various data sources and tools. By treating agents, tools, and agentic workflows as simple function calls, AIQ toolkit enables true composability: build once and reuse anywhere.
+NeMo Agent toolkit is a flexible library designed to seamlessly integrate your enterprise agents—regardless of framework—with various data sources and tools. By treating agents, tools, and agentic workflows as simple function calls, NeMo Agent toolkit enables true composability: build once and reuse anywhere.
 
 ## Key Features
 
-- [**Framework Agnostic:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/extend/plugins.html) Works with any agentic framework, so you can use your current technology stack without replatforming.
-- [**Reusability:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/extend/sharing-components.html) Every agent, tool, or workflow can be combined and repurposed, allowing developers to leverage existing work in new scenarios.
-- [**Rapid Development:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/tutorials/index.html) Start with a pre-built agent, tool, or workflow, and customize it to your needs.
-- [**Profiling:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/workflows/profiler.html) Profile entire workflows down to the tool and agent level, track input/output tokens and timings, and identify bottlenecks.
-- [**Observability:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/workflows/observe/observe-workflow-with-phoenix.html) Monitor and debug your workflows with any OpenTelemetry-compatible observability tool, with examples using [Phoenix](https://docs.nvidia.com/aiqtoolkit/1.2.0/workflows/observe/observe-workflow-with-phoenix.html) and [W&B Weave](https://docs.nvidia.com/aiqtoolkit/1.2.0/workflows/observe/observe-workflow-with-weave.html).
-- [**Evaluation System:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/workflows/evaluate.html) Validate and maintain accuracy of agentic workflows with built-in evaluation tools.
-- [**User Interface:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/quick-start/launching-ui.html) Use the AIQ toolkit UI chat interface to interact with your agents, visualize output, and debug workflows.
-- [**MCP Compatibility**](https://docs.nvidia.com/aiqtoolkit/1.2.0/workflows/mcp/mcp-client.html) Compatible with Model Context Protocol (MCP), allowing tools served by MCP Servers to be used as AIQ toolkit functions.
+- [**Framework Agnostic:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/extend/plugins.html) Works with any agentic framework, so you can use your current technology stack without replatforming.
+- [**Reusability:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/extend/sharing-components.html) Every agent, tool, or workflow can be combined and repurposed, allowing developers to leverage existing work in new scenarios.
+- [**Rapid Development:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/tutorials/index.html) Start with a pre-built agent, tool, or workflow, and customize it to your needs.
+- [**Profiling:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/profiler.html) Profile entire workflows down to the tool and agent level, track input/output tokens and timings, and identify bottlenecks.
+- [**Observability:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/observe/observe-workflow-with-phoenix.html) Monitor and debug your workflows with any OpenTelemetry-compatible observability tool, with examples using [Phoenix](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/observe/observe-workflow-with-phoenix.html) and [W&B Weave](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/observe/observe-workflow-with-weave.html).
+- [**Evaluation System:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/evaluate.html) Validate and maintain accuracy of agentic workflows with built-in evaluation tools.
+- [**User Interface:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/quick-start/launching-ui.html) Use the NeMo Agent toolkit UI chat interface to interact with your agents, visualize output, and debug workflows.
+- [**MCP Compatibility**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/mcp/mcp-client.html) Compatible with Model Context Protocol (MCP), allowing tools served by MCP Servers to be used as NeMo Agent toolkit functions.
 
-With AIQ toolkit, you can move quickly, experiment freely, and ensure reliability across all your agent-driven projects.
+With NeMo Agent toolkit, you can move quickly, experiment freely, and ensure reliability across all your agent-driven projects.
 
 ## Links
- * [Documentation](https://docs.nvidia.com/aiqtoolkit/1.2.0/index.html): Explore the full documentation for AIQ toolkit.
+ * [Documentation](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/index.html): Explore the full documentation for NeMo Agent toolkit.
 
 ## First time user?
- If this is your first time using AIQ toolkit, it is recommended to install the latest version from the [source repository](https://github.com/NVIDIA/NeMo-Agent-Toolkit?tab=readme-ov-file#quick-start) on GitHub. This package is intended for users who are familiar with AIQ toolkit applications and need to add AIQ toolkit as a dependency to their project.
+ If this is your first time using NeMo Agent toolkit, it is recommended to install the latest version from the [source repository](https://github.com/NVIDIA/NeMo-Agent-Toolkit?tab=readme-ov-file#quick-start) on GitHub. This package is intended for users who are familiar with NeMo Agent toolkit applications and need to add NeMo Agent toolkit as a dependency to their project.
 
 ## Feedback
 
@@ -351,7 +351,7 @@ We would love to hear from you! Please file an issue on [GitHub](https://github.
 
 ## Acknowledgements
 
-We would like to thank the following open source projects that made AIQ toolkit possible:
+We would like to thank the following open source projects that made NeMo Agent toolkit possible:
 
 - [CrewAI](https://github.com/crewAIInc/crewAI)
 - [FastAPI](https://github.com/tiangolo/fastapi)